mirror of
https://github.com/django/django.git
synced 2025-03-27 01:30:46 +00:00
a8ded528b3
Thanks jukvalim for the report and initial patch, and Preston Timmons for review. Backport of c32bc1a7a7 from master
55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
from django.db import models
|
|
from django.test import TestCase
|
|
|
|
|
|
class AbsoluteUrlOverrideTests(TestCase):
|
|
|
|
def test_get_absolute_url(self):
|
|
"""
|
|
get_absolute_url() functions as a normal method.
|
|
"""
|
|
get_absolute_url = lambda o: '/test-a/%s/' % o.pk
|
|
TestA = self._create_model_class('TestA', get_absolute_url)
|
|
|
|
self.assertTrue(hasattr(TestA, 'get_absolute_url'))
|
|
obj = TestA(pk=1, name='Foo')
|
|
self.assertEqual('/test-a/%s/' % obj.pk, obj.get_absolute_url())
|
|
|
|
def test_override_get_absolute_url(self):
|
|
"""
|
|
ABSOLUTE_URL_OVERRIDES should override get_absolute_url().
|
|
"""
|
|
get_absolute_url = lambda o: '/test-b/%s/' % o.pk
|
|
with self.settings(
|
|
ABSOLUTE_URL_OVERRIDES={
|
|
'absolute_url_overrides.testb': lambda o: '/overridden-test-b/%s/' % o.pk,
|
|
},
|
|
):
|
|
TestB = self._create_model_class('TestB', get_absolute_url)
|
|
obj = TestB(pk=1, name='Foo')
|
|
self.assertEqual('/overridden-test-b/%s/' % obj.pk, obj.get_absolute_url())
|
|
|
|
def test_insert_get_absolute_url(self):
|
|
"""
|
|
ABSOLUTE_URL_OVERRIDES should work even if the model doesn't have a
|
|
get_absolute_url() method.
|
|
"""
|
|
with self.settings(
|
|
ABSOLUTE_URL_OVERRIDES={
|
|
'absolute_url_overrides.testc': lambda o: '/test-c/%s/' % o.pk,
|
|
},
|
|
):
|
|
TestC = self._create_model_class('TestC')
|
|
obj = TestC(pk=1, name='Foo')
|
|
self.assertEqual('/test-c/%s/' % obj.pk, obj.get_absolute_url())
|
|
|
|
def _create_model_class(self, class_name, get_absolute_url_method=None):
|
|
attrs = {
|
|
'name': models.CharField(max_length=50),
|
|
'__module__': 'absolute_url_overrides',
|
|
}
|
|
if get_absolute_url_method:
|
|
attrs['get_absolute_url'] = get_absolute_url_method
|
|
|
|
return type(class_name, (models.Model,), attrs)
|