diff options
Diffstat (limited to 'brevisurl/tests')
| -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) |