Issue #13; using absurl util whenrever possible

1 files changed, 13 insertions, 2 deletions

diff --git a/brevisurl/tests/test_utils.py b/brevisurl/tests/test_utils.py
index 3fedecd..a6a8fd2 100644
--- a/brevisurl/tests/test_utils.py
+++ b/brevisurl/tests/test_utils.py
@@ -1,7 +1,7 @@
 from django.utils import unittest
 from django.contrib.sites.models import Site
 
-from brevisurl.utils import load_object, absurl
+from brevisurl.utils import load_object, absurl, url_has_protocol
 from brevisurl.backends.local import BrevisUrlBackend
 
 
@@ -51,4 +51,15 @@ class TestAbsUrl(unittest.TestCase):
     def test_site_and_domain(self):
         site = Site(domain='www.codescale.net', name='CodeScale.net')
         abs_url = absurl(protocol='http', domain='sub.codescale.net', site=site, path='/')
-        self.assertEqual(abs_url, 'http://sub.codescale.net/')
\ No newline at end of file
+        self.assertEqual(abs_url, 'http://sub.codescale.net/')
+
+
+class TestUrlHasProtocol(unittest.TestCase):
+
+    def test_url_with_protocol(self):
+        url = 'http://test.sk'
+        self.assertTrue(url_has_protocol(url))
+
+    def test_url_without_protocol(self):
+        url = 'test.sk/path/'
+        self.assertFalse(url_has_protocol(url))
\ No newline at end of file