diff options
Diffstat (limited to 'brevisurl/tests/test_views.py')
| -rw-r--r-- | brevisurl/tests/test_views.py | 6 | 
1 files changed, 6 insertions, 0 deletions
| diff --git a/brevisurl/tests/test_views.py b/brevisurl/tests/test_views.py index ca20c43..ec46911 100644 --- a/brevisurl/tests/test_views.py +++ b/brevisurl/tests/test_views.py @@ -3,6 +3,7 @@ from django.test.client import Client  from django.core.urlresolvers import reverse  from django.contrib.sites.models import Site +import brevisurl.settings  from brevisurl import get_connection  from brevisurl.models import ShortUrl @@ -10,6 +11,8 @@ from brevisurl.models import ShortUrl  class TestBrevisUrlRedirectView(TestCase):      def setUp(self): +        self.LOCAL_BACKEND_DOMAIN = brevisurl.settings.LOCAL_BACKEND_DOMAIN +        brevisurl.settings.LOCAL_BACKEND_DOMAIN = None          self.site = Site.objects.get_current()          self.connection = get_connection('brevisurl.backends.local.BrevisUrlBackend')          self.short_url = ShortUrl() @@ -19,6 +22,9 @@ class TestBrevisUrlRedirectView(TestCase):          self.short_url.save()          self.client = Client() +    def tearDown(self): +        brevisurl.settings.LOCAL_BACKEND_DOMAIN = self.LOCAL_BACKEND_DOMAIN +      def test_redirect(self):          response = self.client.get(reverse('brevisurl_redirect', kwargs={'token': 12345}))          self.assertEqual(response.status_code, 301) | 
