Issue #9; absurl util for constructing absolute urls directly in code

1 files changed, 34 insertions, 2 deletions

diff --git a/brevisurl/tests/test_utils.py b/brevisurl/tests/test_utils.py
index 6d31e77..3fedecd 100644
--- a/brevisurl/tests/test_utils.py
+++ b/brevisurl/tests/test_utils.py
@@ -1,8 +1,10 @@
 from django.utils import unittest
+from django.contrib.sites.models import Site
 
-from brevisurl.utils import load_object
+from brevisurl.utils import load_object, absurl
 from brevisurl.backends.local import BrevisUrlBackend
 
+
 class TestLoadObject(unittest.TestCase):
 
     def test_load_valid_object(self):
@@ -19,4 +21,34 @@ class TestLoadObject(unittest.TestCase):
 
     def test_load_value_error(self):
         with self.assertRaises(ValueError):
-            load_object('brevisurl')
\ No newline at end of file
+            load_object('brevisurl')
+
+
+class TestAbsUrl(unittest.TestCase):
+
+    def test_protocol(self):
+        abs_url = absurl(protocol='http', domain='www.codescale.net', path='/')
+        self.assertEqual(abs_url, 'http://www.codescale.net/')
+        abs_url = absurl(protocol='https', domain='www.codescale.net', path='/')
+        self.assertEqual(abs_url, 'https://www.codescale.net/')
+
+    def test_domain(self):
+        abs_url = absurl(protocol='http', domain='www.codescale.net', path='/')
+        self.assertEqual(abs_url, 'http://www.codescale.net/')
+        abs_url = absurl(protocol='http', domain='sub.codescale.net', path='/')
+        self.assertEqual(abs_url, 'http://sub.codescale.net/')
+
+    def test_site(self):
+        site = Site(domain='www.codescale.net', name='CodeScale.net')
+        abs_url = absurl(protocol='http', site=site, path='/')
+        self.assertEqual(abs_url, 'http://www.codescale.net/')
+
+    def test_no_site_no_domain(self):
+        abs_url = absurl(protocol='http', path='/')
+        current_site = Site.objects.get_current()
+        self.assertEqual(abs_url, 'http://' + current_site.domain + '/')
+
+    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