diff options
Diffstat (limited to 'brevisurl/tests')
| -rw-r--r-- | brevisurl/tests/backends/test_local.py | 10 |
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 |