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

Modernize middleware tests

Browse Source
This commit is contained in:
Ryan P Kilby 2017-11-13 15:04:02 -05:00
parent 265375c104
commit 2e5d4e9a09
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_middleware.py
View File

@ -13,22 +13,24 @@ urlpatterns = [
class MyMiddleware(object):
def __init__(self, get_response):
self.get_response = get_response
def process_response(self, request, response):
def __call__(self, request):
response = self.get_response(request)
assert hasattr(request, 'user'), '`user` is not set on request'
assert request.user.is_authenticated(), '`user` is not authenticated'
assert request.user.is_authenticated, '`user` is not authenticated'
return response
@override_settings(ROOT_URLCONF='tests.test_middleware')
class TestMiddleware(APITestCase):
@override_settings(MIDDLEWARE=('tests.test_middleware.MyMiddleware',))
def test_middleware_can_access_user_when_processing_response(self):
user = User.objects.create_user('john', 'john@example.com', 'password')
key = 'abcd1234'
Token.objects.create(key=key, user=user)
with self.settings(
MIDDLEWARE_CLASSES=('tests.test_middleware.MyMiddleware',)
):
auth = 'Token ' + key
self.client.get('/', HTTP_AUTHORIZATION=auth)
self.client.get('/auth', HTTP_AUTHORIZATION='Token %s' % key)