Test that HE effects url navigation correctly.

cowlicks · cowlicks · commit e3a796b115dd · 2017-10-06T11:48:06.000-07:00
diff --git a/test/selenium/test_navigation.py b/test/selenium/test_navigation.py
@@ -0,0 +1,30 @@
+import unittest
+
+from util import ExtensionTestCase
+
+kittens_url = 'http://freerangekitten.com/'
+
+http_url = 'http://http.badssl.com/'
+
+
+class TestNavigation(ExtensionTestCase):
+    def test_redirect(self):
+        self.driver.get(kittens_url)
+        self.assertTrue(self.driver.current_url.startswith('https'))
+
+    def test_no_redirect_when_disabled(self):
+        self.toggle_disabled()
+        self.driver.get(kittens_url)
+        self.assertEqual(self.driver.current_url, kittens_url)  # not https
+
+    def test_httpnowhere_blocks(self):
+        if self.shim.browser_type == 'firefox':
+            raise unittest.SkipTest('broken on firefox')
+        href_script = 'return window.location.href;'
+        self.toggle_http_nowhere()
+        self.driver.get(http_url)
+        self.assertFalse(http_url == self.driver.execute_script(href_script))
+
+    def test_http_site_not_blocked(self):
+        self.driver.get(http_url)
+        self.assertTrue(self.driver.current_url == http_url)
