From 3103961f034e061e1a5e112dba6dfe6d928aaaf6 Mon Sep 17 00:00:00 2001 From: Artem Muterko Date: Thu, 12 Jan 2017 18:26:17 +0200 Subject: [PATCH] Adjust test --- tests/test_throttling.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/tests/test_throttling.py b/tests/test_throttling.py index 00b31ad7e..6e0904bd7 100644 --- a/tests/test_throttling.py +++ b/tests/test_throttling.py @@ -424,9 +424,6 @@ class AnonRateThrottleTests(TestCase): def setUp(self): self.throttle = AnonRateThrottle() - def tearDown(self): - User.objects.all().delete() - def test_authenticated_user_not_affected(self): request = Request(HttpRequest()) user = User.objects.create(username='test') @@ -445,13 +442,11 @@ class UserRateThrottleTests(TestCase): def setUp(self): self.throttle = UserRateThrottle() - def tearDown(self): - User.objects.all().delete() - def test_get_cache_key_returns_correct_key_if_user_is_authenticated(self): request = Request(HttpRequest()) user = User.objects.create(username='test') force_authenticate(request, user) request.user = user + cache_key = self.throttle.get_cache_key(request, view={}) - assert cache_key == 'throttle_user_2' # user_1 is AnonymousUser + assert cache_key == 'throttle_user_%s' % user.pk