#414: reload_user function

misago/users/tests/test_testutils.py

@@ -68,6 +68,13 @@ class AuthenticatedUserTestCaseTests(AuthenticatedUserTestCase):
         response = self.client.get(reverse('misago:index'))
         self.assertIn(self.user.username, response.content)
 
+    def test_reload_user(self):
+        """reload_user reloads user"""
+        user_pk = self.user.pk
+
+        self.reload_user()
+        self.assertEqual(user_pk, self.user.pk)
+
 
 class SuperUserTestCaseTests(SuperUserTestCase):
     def test_setup(self):

misago/users/testutils.py

@@ -41,8 +41,12 @@ class AuthenticatedUserTestCase(UserTestCase):
         self.user = self.get_authenticated_user()
         self.login_user(self.user)
 
+    def reload_user(self):
+        User = get_user_model()
+        self.user = User.objects.get(id=self.user.id)
+
 
-class SuperUserTestCase(UserTestCase):
+class SuperUserTestCase(AuthenticatedUserTestCase):
     def setUp(self):
         self.user = self.get_superuser()
         self.login_user(self.user)