refs #12836 - Added a test to assure permalink wraps method attributes

This commit is contained in:
Caleb Smith 2012-09-08 11:58:48 -04:00
parent 617d077f1f
commit 210fd7c658
2 changed files with 22 additions and 0 deletions

View File

@ -1,6 +1,13 @@
from django.db import models
def set_attr(name, value):
def wrapper(function):
setattr(function, name, value)
return function
return wrapper
class Guitarist(models.Model):
name = models.CharField(max_length=50)
slug = models.CharField(max_length=50)
@ -9,3 +16,9 @@ class Guitarist(models.Model):
def url(self):
"Returns the URL for this guitarist."
return ('guitarist_detail', [self.slug])
@models.permalink
@set_attr('attribute', 'value')
def url_with_attribute(self):
"Returns the URL for this guitarist and holds an attribute"
return ('guitarist_detail', [self.slug])

View File

@ -16,3 +16,12 @@ class PermalinkTests(TestCase):
"Methods using the @permalink decorator retain their docstring."
g = Guitarist(name='Adrien Moignard', slug='adrienmoignard')
self.assertEqual(g.url.__doc__, "Returns the URL for this guitarist.")
def test_wrapped_attribute(self):
"""
Methods using the @permalink decorator can have attached attributes
from other decorators
"""
g = Guitarist(name='Adrien Moignard', slug='adrienmoignard')
self.assertTrue(hasattr(g.url_with_attribute, 'attribute'))
self.assertEqual(g.url_with_attribute.attribute, 'value')