encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-05 13:00:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add test for UserRateThrottle

Browse Source
This commit is contained in:
Artem Muterko 2017-01-12 17:55:46 +02:00
parent fa64ead4d4
commit 52963ce29d
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/test_throttling.py
View File

@ -424,6 +424,9 @@ class AnonRateThrottleTests(TestCase):
def setUp(self): def setUp(self):
self.throttle = AnonRateThrottle() self.throttle = AnonRateThrottle()
def tearDown(self):
User.objects.all().delete()
def test_authenticated_user_not_affected(self): def test_authenticated_user_not_affected(self):
request = Request(HttpRequest()) request = Request(HttpRequest())
user = User.objects.create(username='test') user = User.objects.create(username='test')
@ -435,3 +438,20 @@ class AnonRateThrottleTests(TestCase):
request = Request(HttpRequest()) request = Request(HttpRequest())
cache_key = self.throttle.get_cache_key(request, view={}) cache_key = self.throttle.get_cache_key(request, view={})
assert cache_key == 'throttle_anon_None' 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