From 52963ce29d22c96d4a181d84072f01466830ac19 Mon Sep 17 00:00:00 2001 From: Artem Muterko Date: Thu, 12 Jan 2017 17:55:46 +0200 Subject: [PATCH] Add test for UserRateThrottle --- tests/test_throttling.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/tests/test_throttling.py b/tests/test_throttling.py index 456467e5d..00b31ad7e 100644 --- a/tests/test_throttling.py +++ b/tests/test_throttling.py @@ -424,6 +424,9 @@ 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') @@ -435,3 +438,20 @@ class AnonRateThrottleTests(TestCase): request = Request(HttpRequest()) cache_key = self.throttle.get_cache_key(request, view={}) assert cache_key == 'throttle_anon_None' + + +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