1 files changed, 12 insertions, 1 deletions
diff --git a/brevisurl/tests/backends/test_local.py b/brevisurl/tests/backends/test_local.py
index 8939314..86565d2 100644
--- a/brevisurl/tests/backends/test_local.py
+++ b/brevisurl/tests/backends/test_local.py
@@ -2,6 +2,7 @@ from django.core.exceptions import ValidationError
 from django.test import TestCase
 from django.core.validators import URLValidator
 
+import brevisurl.settings
 from brevisurl import get_connection
 from brevisurl.models import ShortUrl
 
@@ -11,7 +12,8 @@ class TestLocalBrevisUrlBackend(TestCase):
     def setUp(self):
         self.connection = get_connection('brevisurl.backends.local.BrevisUrlBackend')
 
-    def test_shorten_url(self):
+    def test_shorten_url_use_site_framework(self):
+        brevisurl.settings.LOCAL_BACKEND_DOMAIN = None
         self.assertEqual(ShortUrl.objects.all().count(), 0)
         original_url = 'http://www.codescale.net/'
         short_url = self.connection.shorten_url(original_url)
@@ -19,6 +21,15 @@ class TestLocalBrevisUrlBackend(TestCase):
         self.assertEqual(short_url.original_url, original_url)
         self.assertRegexpMatches(short_url.shortened_url, URLValidator.regex)
 
+    def test_shorten_url_domain_from_settings(self):
+        self.assertEqual(ShortUrl.objects.all().count(), 0)
+        brevisurl.settings.LOCAL_BACKEND_DOMAIN = 'http://brevisurl.net/'
+        original_url = 'http://www.codescale.net/'
+        short_url = self.connection.shorten_url(original_url)
+        self.assertEqual(ShortUrl.objects.all().count(), 1)
+        self.assertEqual(short_url.original_url, original_url)
+        self.assertRegexpMatches(short_url.shortened_url, r'^http://brevisurl\.net/[a-zA-Z0-9]{5}$')
+
     def test_shorten_url_reuse_old(self):
         original_url = 'http://www.codescale.net/'
         self.assertEqual(ShortUrl.objects.all().count(), 0)