diff options
Diffstat (limited to 'tests/test_versioning.py')
| -rw-r--r-- | tests/test_versioning.py | 62 | 
1 files changed, 49 insertions, 13 deletions
| diff --git a/tests/test_versioning.py b/tests/test_versioning.py index c44f727d..553463d1 100644 --- a/tests/test_versioning.py +++ b/tests/test_versioning.py @@ -1,9 +1,13 @@ +from .utils import UsingURLPatterns  from django.conf.urls import include, url +from rest_framework import serializers  from rest_framework import status, versioning  from rest_framework.decorators import APIView  from rest_framework.response import Response  from rest_framework.reverse import reverse  from rest_framework.test import APIRequestFactory, APITestCase +from rest_framework.versioning import NamespaceVersioning +import pytest  class RequestVersionView(APIView): @@ -28,17 +32,8 @@ class RequestInvalidVersionView(APIView):  factory = APIRequestFactory() -mock_view = lambda request: None - -included_patterns = [ -    url(r'^namespaced/$', mock_view, name='another'), -] - -urlpatterns = [ -    url(r'^v1/', include(included_patterns, namespace='v1')), -    url(r'^another/$', mock_view, name='another'), -    url(r'^(?P<version>[^/]+)/another/$', mock_view, name='another') -] +dummy_view = lambda request: None +dummy_pk_view = lambda request, pk: None  class TestRequestVersion: @@ -114,8 +109,17 @@ class TestRequestVersion:          assert response.data == {'version': None} -class TestURLReversing(APITestCase): -    urls = 'tests.test_versioning' +class TestURLReversing(UsingURLPatterns, APITestCase): +    included = [ +        url(r'^namespaced/$', dummy_view, name='another'), +        url(r'^example/(?P<pk>\d+)/$', dummy_pk_view, name='example-detail') +    ] + +    urlpatterns = [ +        url(r'^v1/', include(included, namespace='v1')), +        url(r'^another/$', dummy_view, name='another'), +        url(r'^(?P<version>[^/]+)/another/$', dummy_view, name='another'), +    ]      def test_reverse_unversioned(self):          view = ReverseView.as_view() @@ -221,3 +225,35 @@ class TestInvalidVersion:          request.resolver_match = FakeResolverMatch          response = view(request, version='v3')          assert response.status_code == status.HTTP_404_NOT_FOUND + + +class TestHyperlinkedRelatedField(UsingURLPatterns, APITestCase): +    included = [ +        url(r'^namespaced/(?P<pk>\d+)/$', dummy_view, name='namespaced'), +    ] + +    urlpatterns = [ +        url(r'^v1/', include(included, namespace='v1')), +        url(r'^v2/', include(included, namespace='v2')) +    ] + +    def setUp(self): +        super(TestHyperlinkedRelatedField, self).setUp() + +        class MockQueryset(object): +            def get(self, pk): +                return 'object %s' % pk + +        self.field = serializers.HyperlinkedRelatedField( +            view_name='namespaced', +            queryset=MockQueryset() +        ) +        request = factory.get('/') +        request.versioning_scheme = NamespaceVersioning() +        request.version = 'v1' +        self.field._context = {'request': request} + +    def test_bug_2489(self): +        assert self.field.to_internal_value('/v1/namespaced/3/') == 'object 3' +        with pytest.raises(serializers.ValidationError): +            self.field.to_internal_value('/v2/namespaced/3/') | 
