mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-11 04:07:39 +03:00
994e1ba927
* Added TEMPLATES setting to tests
* Remove deprecated view-string in URL conf
* Replace 'urls = ...' in test classes with override_settings('ROOT_URLCONF=...')
* Refactor UsingURLPatterns to use override_settings(ROOT_URLCONF=...) style
* Get model managers and names in a version-compatible manner.
* Apply override_settings to a TestCase, not a mixin class
* Use '.callback' property instead of private attributes when inspecting urlpatterns
* Pass 'user' to template explicitly
* Correct sorting of import statements.
* Remove unused TEMPLATE_LOADERS setting, in favor of TEMPLATES.
* Remove code style issue
* BaseFilter test requires a concrete model
* Resolve tox.ini issues
* Resolve isort differences between local and tox environments
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
from django.conf.urls import url
|
|
from django.contrib.auth.models import User
|
|
from django.test import override_settings
|
|
|
|
from rest_framework.authentication import TokenAuthentication
|
|
from rest_framework.authtoken.models import Token
|
|
from rest_framework.test import APITestCase
|
|
from rest_framework.views import APIView
|
|
|
|
urlpatterns = [
|
|
url(r'^$', APIView.as_view(authentication_classes=(TokenAuthentication,))),
|
|
]
|
|
|
|
|
|
class MyMiddleware(object):
|
|
|
|
def process_response(self, request, response):
|
|
assert hasattr(request, 'user'), '`user` is not set on request'
|
|
assert request.user.is_authenticated(), '`user` is not authenticated'
|
|
return response
|
|
|
|
|
|
@override_settings(ROOT_URLCONF='tests.test_middleware')
|
|
class TestMiddleware(APITestCase):
|
|
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)
|