Fix style

znick · Apr 4, 2024 · 8ae6fcd · 8ae6fcd
1 parent b6c07a4
commit 8ae6fcd
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/anytask/users/tests.py b/anytask/users/tests.py
@@ -107,7 +107,7 @@ def test_login_user__bad_email(self):
 
     def test_register_user(self):
         client = self.client
-        
+
         form_data = {"username": u"test_user1",
                      "email": u"[email protected]",
                      "first_name": u"Иван",
@@ -130,13 +130,11 @@ def test_register_user(self):
         self.assertEqual(user.last_name, "Иванов")
 
         self.assertTrue(client.login(username="test_user1", password="qwer1"))
-
-        user_profile = UserProfile.objects.get(user=user)
-
-        self.assertEqual(user_profile.show_email, True)
 
+        user_profile = user.profile
+        self.assertEqual(user_profile.show_email, True)
         self.assertEqual(user_profile.is_active(), True)
-        
+
     def test_register_user__login_already_exists(self):
         client = self.client