diff options
| -rw-r--r-- | src/shorturls/tests/test_templatetag.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/src/shorturls/tests/test_templatetag.py b/src/shorturls/tests/test_templatetag.py index 2feef91..2127817 100644 --- a/src/shorturls/tests/test_templatetag.py +++ b/src/shorturls/tests/test_templatetag.py @@ -1,4 +1,4 @@ -from django import template +from django.template import Template, Context from django.conf import settings from django.test import TestCase from shorturls.tests.models import Animal, Vegetable, Mineral @@ -15,7 +15,7 @@ class RedirectViewTestCase(TestCase): 'A': 'shorturls.animal', 'V': 'shorturls.vegetable', } - + def tearDown(self): if self.old_shorten is not None: settings.SHORTEN_MODELS = self.old_shorten @@ -23,12 +23,12 @@ class RedirectViewTestCase(TestCase): settings.SHORT_BASE_URL = self.old_base def render(self, t, **c): - return template.Template('{% load shorturl %}'+t).render(c) - + return Template('{% load shorturl %}'+t).render(Context(c)) + def test_shorturl(self): r = self.render('{% shorturl a %}', a=Animal.objects.get(id=12345)) self.assertEqual(r, '/ADNH') - + def test_bad_context(self): r = self.render('{% shorturl a %}') self.assertEqual(r, '') @@ -36,13 +36,12 @@ class RedirectViewTestCase(TestCase): def test_no_prefix(self): r = self.render('{% shorturl m %}', m=Mineral.objects.all()[0]) self.assertEqual(r, '') - + def test_short_base_url(self): settings.SHORT_BASE_URL = 'http://example.com/' r = self.render('{% shorturl a %}', a=Animal.objects.get(id=12345)) self.assertEqual(r, 'http://example.com/ADNH') - + def test_revcanonical(self): r = self.render('{% revcanonical a %}', a=Animal.objects.get(id=12345)) self.assertEqual(r, '<link rev="canonical" href="/ADNH">') -
\ No newline at end of file |