Issue #10; Local backend short url domain

1 files changed, 9 insertions, 1 deletions

diff --git a/brevisurl/tests/backends/test_local.py b/brevisurl/tests/backends/test_local.py
index 39ab8c1..620b9a4 100644
--- a/brevisurl/tests/backends/test_local.py
+++ b/brevisurl/tests/backends/test_local.py
@@ -87,4 +87,12 @@ class TestLocalBrevisUrlBackend(TestCase):
         with self.assertRaises(TokensExhaustedError):
             original_url = 'http://www.codescale.net/another/'
             short_url = self.connection.shorten_url(original_url)
-        brevisurl.settings.LOCAL_BACKEND_TOKEN_CHARS = _default_chars
\ No newline at end of file
+        brevisurl.settings.LOCAL_BACKEND_TOKEN_CHARS = _default_chars
+
+    def test_custom_domain(self):
+        original_url = 'http://www.codescale.net/'
+        connection = get_connection('brevisurl.backends.local.BrevisUrlBackend', domain='http://test.com/')
+        self.assertEqual(ShortUrl.objects.all().count(), 0)
+        short_url = connection.shorten_url(original_url)
+        self.assertEqual(ShortUrl.objects.all().count(), 1)
+        self.assertRegexpMatches(short_url.shortened_url, r'^http://test\.com/.{5}')
\ No newline at end of file