From 178d820541e874fe9913080c5125876b9c0b82ca Mon Sep 17 00:00:00 2001
From: Vladimir Gorej
Date: Wed, 20 Jun 2012 20:49:43 +0200
Subject: Issue #5; Configure protocol of local backend via settings

---
 brevisurl/tests/backends/test_local.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

(limited to 'brevisurl/tests/backends/test_local.py')

diff --git a/brevisurl/tests/backends/test_local.py b/brevisurl/tests/backends/test_local.py
index 620b9a4..ebf27e8 100644
--- a/brevisurl/tests/backends/test_local.py
+++ b/brevisurl/tests/backends/test_local.py
@@ -14,6 +14,7 @@ class TestLocalBrevisUrlBackend(TestCase):
         self.connection = get_connection('brevisurl.backends.local.BrevisUrlBackend')
 
     def test_shorten_url_use_site_framework(self):
+        _original_domain = brevisurl.settings.LOCAL_BACKEND_DOMAIN
         brevisurl.settings.LOCAL_BACKEND_DOMAIN = None
         self.assertEqual(ShortUrl.objects.all().count(), 0)
         original_url = 'http://www.codescale.net/'
@@ -21,15 +22,18 @@ class TestLocalBrevisUrlBackend(TestCase):
         self.assertEqual(ShortUrl.objects.all().count(), 1)
         self.assertEqual(short_url.original_url, original_url)
         self.assertRegexpMatches(short_url.shortened_url, URLValidator.regex)
+        brevisurl.settings.LOCAL_BACKEND_DOMAIN = _original_domain
 
     def test_shorten_url_domain_from_settings(self):
         self.assertEqual(ShortUrl.objects.all().count(), 0)
+        _original_domain = brevisurl.settings.LOCAL_BACKEND_DOMAIN
         brevisurl.settings.LOCAL_BACKEND_DOMAIN = 'http://brevisurl.net/'
         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, r'^http://brevisurl\.net/[a-zA-Z0-9]{5}$')
+        brevisurl.settings.LOCAL_BACKEND_DOMAIN = _original_domain
 
     def test_shorten_url_reuse_old(self):
         original_url = 'http://www.codescale.net/'
@@ -95,4 +99,21 @@ class TestLocalBrevisUrlBackend(TestCase):
         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
+        self.assertRegexpMatches(short_url.shortened_url, r'^http://test\.com/.{5}')
+
+    def test_configurable_protocol(self):
+        _original_domain = brevisurl.settings.LOCAL_BACKEND_DOMAIN
+        brevisurl.settings.LOCAL_BACKEND_DOMAIN = None
+        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.assertRegexpMatches(short_url.shortened_url, '^http://')
+        original_url = 'http://www.codescale.net/another'
+        _default_protocol = brevisurl.settings.LOCAL_BACKEND_DOMAIN_PROTOCOL
+        brevisurl.settings.LOCAL_BACKEND_DOMAIN_PROTOCOL = 'https'
+        short_url = self.connection.shorten_url(original_url)
+        self.assertEqual(ShortUrl.objects.all().count(), 2)
+        self.assertRegexpMatches(short_url.shortened_url, '^https://')
+        brevisurl.settings.LOCAL_BACKEND_DOMAIN_PROTOCOL = _default_protocol
+        brevisurl.settings.LOCAL_BACKEND_DOMAIN = None
\ No newline at end of file
-- 
cgit v1.2.3