Added test for parsed_url

jazzband · Jun 5, 2015 · 7e4b72a · 7e4b72a
1 parent 41fce2a
commit 7e4b72a
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/test_views.py b/tests/test_views.py
@@ -4,6 +4,7 @@
 import os
 
 from django.test import TestCase, RequestFactory
+from django.utils.six.moves.urllib.parse import ParseResult
 
 from revproxy.exceptions import InvalidUpstream
 from revproxy.views import ProxyView, DiazoProxyView
@@ -25,6 +26,21 @@ def test_upstream_not_implemented(self):
         with self.assertRaises(NotImplementedError):
             upstream = proxy_view.upstream
 
+    def test_upstream_parsed_url_cache(self):
+        class CustomProxyView(ProxyView):
+            upstream = 'http://www.example.com'
+
+        proxy_view = CustomProxyView()
+        with self.assertRaises(AttributeError):
+            proxy_view._parsed_url
+
+        # Test for parsed URL
+        proxy_view.get_upstream()
+        self.assertIsInstance(proxy_view._parsed_url, ParseResult)
+        # Get parsed URL from cache
+        proxy_view.get_upstream()
+        self.assertIsInstance(proxy_view._parsed_url, ParseResult)
+
     def test_upstream_without_scheme(self):
         class BrokenProxyView(ProxyView):
             upstream = 'www.example.com'