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

Adjust test

Browse Source
This commit is contained in:
Artem Muterko 2017-01-12 18:26:17 +02:00
parent 52963ce29d
commit 3103961f03
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_throttling.py
View File

@ -424,9 +424,6 @@ 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')
@ -445,13 +442,11 @@ class UserRateThrottleTests(TestCase):
def setUp(self): def setUp(self):
self.throttle = UserRateThrottle() self.throttle = UserRateThrottle()
def tearDown(self):
User.objects.all().delete()
def test_get_cache_key_returns_correct_key_if_user_is_authenticated(self): def test_get_cache_key_returns_correct_key_if_user_is_authenticated(self):
request = Request(HttpRequest()) request = Request(HttpRequest())
user = User.objects.create(username='test') user = User.objects.create(username='test')
force_authenticate(request, user) force_authenticate(request, user)
request.user = user request.user = user
cache_key = self.throttle.get_cache_key(request, view={}) 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