Browse Source

Clean removeoldips test output (#1020)

Rafał Pitoń 7 years ago
parent
commit
98cb6202b3
1 changed files with 7 additions and 3 deletions
  1. 7 3
      misago/users/tests/test_removeoldips.py

+ 7 - 3
misago/users/tests/test_removeoldips.py

@@ -17,9 +17,11 @@ class RemoveOldIpsTests(TestCase):
     def test_removeoldips_recent_user(self):
     def test_removeoldips_recent_user(self):
         """command is not removing user's IP if its recent"""
         """command is not removing user's IP if its recent"""
         user = UserModel.objects.create_user('Bob', 'bob@bob.com')
         user = UserModel.objects.create_user('Bob', 'bob@bob.com')
-        call_command(removeoldips.Command())
-        user_joined_from_ip = UserModel.objects.get(pk=user.pk).joined_from_ip
         
         
+        out = StringIO()
+        call_command(removeoldips.Command(), stdout=out)
+
+        user_joined_from_ip = UserModel.objects.get(pk=user.pk).joined_from_ip
         self.assertNotEqual(user_joined_from_ip, ANONYMOUS_IP)
         self.assertNotEqual(user_joined_from_ip, ANONYMOUS_IP)
     
     
     def test_removeoldips_old_user(self):
     def test_removeoldips_old_user(self):
@@ -29,7 +31,9 @@ class RemoveOldIpsTests(TestCase):
         user.joined_on = joined_on_past
         user.joined_on = joined_on_past
         user.save()
         user.save()
 
 
-        call_command(removeoldips.Command())
+        out = StringIO()
+        call_command(removeoldips.Command(), stdout=out)
+
         user_joined_from_ip = UserModel.objects.get(pk=user.pk).joined_from_ip
         user_joined_from_ip = UserModel.objects.get(pk=user.pk).joined_from_ip
         self.assertEqual(user_joined_from_ip, ANONYMOUS_IP)
         self.assertEqual(user_joined_from_ip, ANONYMOUS_IP)