Initial commit

1 files changed, 38 insertions, 0 deletions

diff --git a/brevisurl/tests/test_models.py b/brevisurl/tests/test_models.py
new file mode 100644
index 0000000..6942124
--- /dev/null
+++ b/brevisurl/tests/test_models.py
@@ -0,0 +1,38 @@
+from django.core.exceptions import ValidationError
+from django.test import TestCase
+from django.contrib.sites.models import Site
+
+from brevisurl.models import ShortUrl
+from brevisurl import get_connection
+
+
+class TestModels(TestCase):
+
+    def test_model_save(self):
+        site = Site.objects.get_current()
+        connection = get_connection('brevisurl.backends.local.BrevisUrlBackend')
+        short_url = ShortUrl()
+        short_url.original_url = 'http://www.codescale.net/'
+        short_url.shortened_url = '{0}://{1}/12345'.format(connection.PROTOCOL, site.domain)
+        short_url.backend = connection.class_path
+        short_url.save()
+        self.assertIsNotNone(short_url.pk)
+
+    def test_model_save_invalid_original_url(self):
+        with self.assertRaises(ValidationError):
+            self.site = Site.objects.get_current()
+            self.connection = get_connection('brevisurl.backends.local.BrevisUrlBackend')
+            self.short_url = ShortUrl()
+            self.short_url.original_url = 'www.codescale.'
+            self.short_url.shortened_url = '{0}://{1}/12345'.format(self.connection.PROTOCOL, self.site.domain)
+            self.short_url.backend = self.connection.class_path
+            self.short_url.save()
+
+    def test_model_save_invalid_shortened_url(self):
+        with self.assertRaises(ValidationError):
+            connection = get_connection('brevisurl.backends.local.BrevisUrlBackend')
+            short_url = ShortUrl()
+            short_url.original_url = 'http://www.codescale.net/'
+            short_url.shortened_url = 'www.codescale.'
+            short_url.backend = connection.class_path
+            short_url.save()
\ No newline at end of file