From 09fe33095779a4bb9cb4a40394e89767a8ccfd31 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aleksi=20Ha=CC=88kli?= Date: Wed, 13 Feb 2019 19:02:02 +0200 Subject: [PATCH] Fix test compatibility on Python 3.5 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Aleksi Häkli --- axes/tests/test_access_attempt.py | 6 +++--- axes/tests/test_handlers.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) 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)