diff options
Diffstat (limited to 'brevisurl/tests/backends/test_local.py')
| -rw-r--r-- | brevisurl/tests/backends/test_local.py | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/brevisurl/tests/backends/test_local.py b/brevisurl/tests/backends/test_local.py new file mode 100644 index 0000000..8939314 --- /dev/null +++ b/brevisurl/tests/backends/test_local.py @@ -0,0 +1,55 @@ +from django.core.exceptions import ValidationError +from django.test import TestCase +from django.core.validators import URLValidator + +from brevisurl import get_connection +from brevisurl.models import ShortUrl + + +class TestLocalBrevisUrlBackend(TestCase): + + def setUp(self): + self.connection = get_connection('brevisurl.backends.local.BrevisUrlBackend') + + def test_shorten_url(self): + self.assertEqual(ShortUrl.objects.all().count(), 0) + 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, URLValidator.regex) + + def test_shorten_url_reuse_old(self): + original_url = 'http://www.codescale.net/' + self.assertEqual(ShortUrl.objects.all().count(), 0) + 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, URLValidator.regex) + 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, URLValidator.regex) + + def test_shorten_url_create_new(self): + original_url = 'http://www.codescale.net/' + self.assertEqual(ShortUrl.objects.all().count(), 0) + 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, URLValidator.regex) + original_url = 'http://www.codescale.net/en/company/' + short_url = self.connection.shorten_url(original_url) + self.assertEqual(ShortUrl.objects.all().count(), 2) + self.assertEqual(short_url.original_url, original_url) + self.assertRegexpMatches(short_url.shortened_url, URLValidator.regex) + + def test_shorten_url_invalid_original_url(self): + with self.assertRaises(ValidationError): + self.connection.shorten_url('www.codescale.') + self.assertEqual(ShortUrl.objects.all().count(), 0) + + def test_shorten_url_invalid_original_url_fail_silently(self): + self.connection.fail_silently = True + shorl_url = self.connection.shorten_url('www.codescale.') + self.assertIsNone(shorl_url)
\ No newline at end of file |