diff options
Diffstat (limited to 'brevisurl/tests')
| -rw-r--r-- | brevisurl/tests/templatetags/test_brevisurltags.py | 15 | 
1 files changed, 12 insertions, 3 deletions
| diff --git a/brevisurl/tests/templatetags/test_brevisurltags.py b/brevisurl/tests/templatetags/test_brevisurltags.py index de1ed00..38ccfcc 100644 --- a/brevisurl/tests/templatetags/test_brevisurltags.py +++ b/brevisurl/tests/templatetags/test_brevisurltags.py @@ -7,7 +7,7 @@ from brevisurl.models import ShortUrl  class TestShortenUrlTag(TestCase): -    def test_shorten_url_tag(self): +    def test_shorten_url_filter(self):          original_url = 'http://www.codescale.net/'          url = Template("""          {% load brevisurltags %} @@ -17,11 +17,20 @@ class TestShortenUrlTag(TestCase):          self.assertEqual(ShortUrl.objects.all()[0].original_url, original_url)          self.assertRegexpMatches(url, URLValidator.regex) -    def test_shorten_url_tag_invalid_url(self): +    def test_shorten_url_filter_invalid_url(self):          original_url = 'www.codescale.'          url = Template("""          {% load brevisurltags %}          {{ url|shorten_url }}          """).render(Context({'url': original_url})).strip()          self.assertEqual(ShortUrl.objects.all().count(), 0) -        self.assertEqual(url, original_url)
\ No newline at end of file +        self.assertEqual(url, original_url) + +    def test_absurl_tag(self): +        url = Template(""" +        {% load brevisurltags %} +        {% absurl brevisurl_redirect token='12345' as brevis_url %} +        {{ brevis_url|shorten_url }} +        """).render(Context()).strip() +        self.assertEqual(ShortUrl.objects.all().count(), 1) +        self.assertRegexpMatches(url, URLValidator.regex)
\ No newline at end of file | 
