[ft #145186309] Remove querries in test_login.py

andela · May 19, 2017 · dd65380 · dd65380
1 parent 2e8b128
commit dd65380
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/hc/accounts/tests/test_login.py b/hc/accounts/tests/test_login.py
@@ -27,8 +27,9 @@ def test_it_sends_link(self):
         ### Assert contents of the email body
         self.assertTrue(mail.outbox[0].body)
         ### Assert that check is associated with the new user
-        self.assertEqual(session["welcome_code"].replace('-', ''),
-                         Check.objects.filter(user_id=1)[0].code.hex)
+        check2 = Check.objects.get(code=check.code)
+        self.assertTrue(check2.user)
+
 
     def test_it_pops_bad_link_from_session(self):
         self.client.session["bad_link"] = True