diff options
| author | Tom Christie | 2015-02-05 00:58:16 +0000 | 
|---|---|---|
| committer | Tom Christie | 2015-02-05 00:58:16 +0000 | 
| commit | 65bbc4a1598350f3dd7847d3f2f5d98c5d258b92 (patch) | |
| tree | bbf4850dfb19781f137a9029b5cada3057501fe9 /tests | |
| parent | 41b213414df57d7e39f1bbf3aaa35a1b033e89a3 (diff) | |
| parent | e1c45133126e0c47b8470b4cf7a43c6a7f4fca43 (diff) | |
| download | django-rest-framework-65bbc4a1598350f3dd7847d3f2f5d98c5d258b92.tar.bz2 | |
Merge branch 'brandoncazander-version-3.1-2489' into version-3.1
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_relations_hyperlink.py | 7 | ||||
| -rw-r--r-- | tests/test_versioning.py | 69 | ||||
| -rw-r--r-- | tests/utils.py | 24 | 
3 files changed, 84 insertions, 16 deletions
| diff --git a/tests/test_relations_hyperlink.py b/tests/test_relations_hyperlink.py index f1b882ed..aede61d2 100644 --- a/tests/test_relations_hyperlink.py +++ b/tests/test_relations_hyperlink.py @@ -1,5 +1,5 @@  from __future__ import unicode_literals -from django.conf.urls import patterns, url +from django.conf.urls import url  from django.test import TestCase  from rest_framework import serializers  from rest_framework.test import APIRequestFactory @@ -14,8 +14,7 @@ request = factory.get('/')  # Just to ensure we have a request in the serializer  dummy_view = lambda request, pk: None -urlpatterns = patterns( -    '', +urlpatterns = [      url(r'^dummyurl/(?P<pk>[0-9]+)/$', dummy_view, name='dummy-url'),      url(r'^manytomanysource/(?P<pk>[0-9]+)/$', dummy_view, name='manytomanysource-detail'),      url(r'^manytomanytarget/(?P<pk>[0-9]+)/$', dummy_view, name='manytomanytarget-detail'), @@ -24,7 +23,7 @@ urlpatterns = patterns(      url(r'^nullableforeignkeysource/(?P<pk>[0-9]+)/$', dummy_view, name='nullableforeignkeysource-detail'),      url(r'^onetoonetarget/(?P<pk>[0-9]+)/$', dummy_view, name='onetoonetarget-detail'),      url(r'^nullableonetoonesource/(?P<pk>[0-9]+)/$', dummy_view, name='nullableonetoonesource-detail'), -) +]  # ManyToMany diff --git a/tests/test_versioning.py b/tests/test_versioning.py index c44f727d..cdd10065 100644 --- a/tests/test_versioning.py +++ b/tests/test_versioning.py @@ -1,9 +1,14 @@ +from .utils import MockObject, MockQueryset, UsingURLPatterns  from django.conf.urls import include, url +from django.core.exceptions import ObjectDoesNotExist +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): @@ -29,16 +34,7 @@ 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, pk: None  class TestRequestVersion: @@ -114,8 +110,17 @@ class TestRequestVersion:          assert response.data == {'version': None} -class TestURLReversing(APITestCase): -    urls = 'tests.test_versioning' +class TestURLReversing(UsingURLPatterns, APITestCase): +    included = [ +        url(r'^namespaced/$', mock_view, name='another'), +        url(r'^example/(?P<pk>\d+)/$', dummy_view, name='example-detail') +    ] + +    urlpatterns = [ +        url(r'^v1/', include(included, namespace='v1')), +        url(r'^another/$', mock_view, name='another'), +        url(r'^(?P<version>[^/]+)/another/$', mock_view, name='another'), +    ]      def test_reverse_unversioned(self):          view = ReverseView.as_view() @@ -221,3 +226,43 @@ 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+)/$', mock_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 HyperlinkedMockQueryset(MockQueryset): +            def get(self, **lookup): +                for item in self.items: +                    if item.pk == int(lookup.get('pk', -1)): +                        return item +                raise ObjectDoesNotExist() + +        self.queryset = HyperlinkedMockQueryset([ +            MockObject(pk=1, name='foo'), +            MockObject(pk=2, name='bar'), +            MockObject(pk=3, name='baz') +        ]) +        self.field = serializers.HyperlinkedRelatedField( +            view_name='namespaced', +            queryset=self.queryset +        ) +        request = factory.post('/', urlconf='tests.test_versioning') +        request.versioning_scheme = NamespaceVersioning() +        request.version = 'v1' +        self.field._context = {'request': request} + +    def test_bug_2489(self): +        self.field.to_internal_value('/v1/namespaced/3/') +        with pytest.raises(serializers.ValidationError): +            self.field.to_internal_value('/v2/namespaced/3/') diff --git a/tests/utils.py b/tests/utils.py index 5b2d7586..b9034996 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -2,6 +2,30 @@ from django.core.exceptions import ObjectDoesNotExist  from django.core.urlresolvers import NoReverseMatch +class UsingURLPatterns(object): +    """ +    Isolates URL patterns used during testing on the test class itself. +    For example: + +    class MyTestCase(UsingURLPatterns, TestCase): +        urlpatterns = [ +            ... +        ] + +        def test_something(self): +            ... +    """ +    urls = __name__ + +    def setUp(self): +        global urlpatterns +        urlpatterns = self.urlpatterns + +    def tearDown(self): +        global urlpatterns +        urlpatterns = [] + +  class MockObject(object):      def __init__(self, **kwargs):          self._kwargs = kwargs | 
