57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.conf.urls import url
|
|
from django.core.urlresolvers import NoReverseMatch
|
|
from django.test import TestCase
|
|
|
|
from rest_framework.reverse import reverse
|
|
from rest_framework.test import APIRequestFactory
|
|
|
|
factory = APIRequestFactory()
|
|
|
|
|
|
def null_view(request):
|
|
pass
|
|
|
|
urlpatterns = [
|
|
url(r'^view$', null_view, name='view'),
|
|
]
|
|
|
|
|
|
class MockVersioningScheme(object):
|
|
|
|
def __init__(self, raise_error=False):
|
|
self.raise_error = raise_error
|
|
|
|
def reverse(self, *args, **kwargs):
|
|
if self.raise_error:
|
|
raise NoReverseMatch()
|
|
|
|
return 'http://scheme-reversed/view'
|
|
|
|
|
|
class ReverseTests(TestCase):
|
|
"""
|
|
Tests for fully qualified URLs when using `reverse`.
|
|
"""
|
|
urls = 'tests.test_reverse'
|
|
|
|
def test_reversed_urls_are_fully_qualified(self):
|
|
request = factory.get('/view')
|
|
url = reverse('view', request=request)
|
|
self.assertEqual(url, 'http://testserver/view')
|
|
|
|
def test_reverse_with_versioning_scheme(self):
|
|
request = factory.get('/view')
|
|
request.versioning_scheme = MockVersioningScheme()
|
|
|
|
url = reverse('view', request=request)
|
|
self.assertEqual(url, 'http://scheme-reversed/view')
|
|
|
|
def test_reverse_with_versioning_scheme_fallback_to_default_on_error(self):
|
|
request = factory.get('/view')
|
|
request.versioning_scheme = MockVersioningScheme(raise_error=True)
|
|
|
|
url = reverse('view', request=request)
|
|
self.assertEqual(url, 'http://testserver/view')
|