diff --git a/axes/tests/test_access_attempt.py b/axes/tests/test_access_attempt.py index 87ee91c..7b3324a 100644 --- a/axes/tests/test_access_attempt.py +++ b/axes/tests/test_access_attempt.py @@ -310,8 +310,8 @@ class AccessAttemptTest(TestCase): [], ) - cache.get.assert_called_once() - cache.set.assert_called_once() + self.assertTrue(cache.get.call_count) + self.assertTrue(cache.set.call_count) @patch('axes.utils.get_client_ip', return_value='127.0.0.1') def test_get_cache_key(self, _): @@ -661,7 +661,7 @@ class AttemptUtilsTestCase(TestCase): get_cache.return_value = cache self.assertTrue(is_already_locked(self.request, {})) - cache.get.assert_called_once() + self.assertTrue(cache.get.call_count) @override_settings(AXES_NEVER_LOCKOUT_GET=True) def test_is_already_locked_never_lockout_get(self): diff --git a/axes/tests/test_handlers.py b/axes/tests/test_handlers.py index 50f5972..ebd6fdf 100644 --- a/axes/tests/test_handlers.py +++ b/axes/tests/test_handlers.py @@ -100,8 +100,8 @@ class AxesHandlerTestCase(TestCase): self.handler.post_save_access_attempt(self.attempt) - cache.get.assert_called_once() - cache.set.assert_called_once() + self.assertTrue(cache.get.call_count) + self.assertTrue(cache.set.call_count) @patch('axes.handlers.get_axes_cache') def test_user_login_failed_utilizes_cache(self, get_cache): @@ -118,4 +118,4 @@ class AxesHandlerTestCase(TestCase): self.handler.user_login_failed(sender, credentials, request) - self.assertGreater(cache.get.call_count, 0) + self.assertTrue(cache.get.call_count)