[tbb-commits] [tor-browser-bundle-testsuite/master] Bug 40005: Add test for 'Replace security slider with security level UI' TB patch

[gk at torproject.org]

commit d97948001366242e3baab4d49dd0706cf42aa8ab
Author: Alex Catarineu <acat at torproject.org>
Date:   Mon Jun 29 19:56:14 2020 +0200

    Bug 40005: Add test for 'Replace security slider with security level UI' TB patch
---
 TBBTestSuite/TestSuite/BrowserBundleTests.pm       |  6 ++
 .../tor_browser_tests/test_security_level_ui.py    | 72 ++++++++++++++++++++++
 2 files changed, 78 insertions(+)

diff --git a/TBBTestSuite/TestSuite/BrowserBundleTests.pm b/TBBTestSuite/TestSuite/BrowserBundleTests.pm
index 931ab70..37efae6 100644
--- a/TBBTestSuite/TestSuite/BrowserBundleTests.pm
+++ b/TBBTestSuite/TestSuite/BrowserBundleTests.pm
@@ -494,6 +494,12 @@ our @tests = (
         use_net         => 1,
         descr           => 'Check that onion alias urlbar rewrites work properly',
     },
+    {
+        name            => 'security_level_ui',
+        type            => 'marionette',
+        use_net         => 1,
+        descr           => 'Check that the security level UI is working properly',
+    },
 );
 
 sub set_test_prefs {
diff --git a/marionette/tor_browser_tests/test_security_level_ui.py b/marionette/tor_browser_tests/test_security_level_ui.py
new file mode 100644
index 0000000..f24e7ea
--- /dev/null
+++ b/marionette/tor_browser_tests/test_security_level_ui.py
@@ -0,0 +1,72 @@
+from marionette_driver import By, Wait
+from marionette_driver.errors import MarionetteException
+
+from marionette_harness import MarionetteTestCase
+
+
+class Test(MarionetteTestCase):
+    def test_security_level_ui(self):
+        custom_pref = 'javascript.options.wasm'
+        m = self.marionette
+        m.timeout.implicit = 5
+        with m.using_context('chrome'):
+            m.find_element('id', 'security-level-button').click()
+            m.find_element(
+                'id', 'securityLevel-advancedSecuritySettings').click()
+            with m.using_context('content'):
+                # Test whether changing the security level value works
+                self.assertEqual(
+                    m.get_url(), "about:preferences#privacy-securitylevel")
+                self.assertEqual(4, m.get_pref(
+                    'extensions.torbutton.security_slider'))
+                m.find_element(
+                    'css selector', '#securityLevel-vbox-safer radio').click()
+                self.assertEqual(2, m.get_pref(
+                    'extensions.torbutton.security_slider'))
+                m.find_element(
+                    'css selector', '#securityLevel-vbox-safest radio').click()
+                self.assertEqual(1, m.get_pref(
+                    'extensions.torbutton.security_slider'))
+                m.find_element(
+                    'css selector', '#securityLevel-vbox-standard radio').click()
+                self.assertEqual(4, m.get_pref(
+                    'extensions.torbutton.security_slider'))
+
+                # Test custom security settings
+                elem = m.find_element('id', 'securityLevel-restoreDefaults')
+                self.assertEqual(elem.is_displayed(), False)
+                m.set_pref(custom_pref, False)
+                self.assertEqual(elem.is_displayed(), True)
+                elem.click()
+                self.assertEqual(True, m.get_pref(custom_pref))
+
+                # Test Learn More link
+                m.find_element('id', 'securityLevel-learnMore').click()
+                Wait(m, timeout=m.timeout.page_load).until(
+                    lambda _: len(m.window_handles) > 1)
+                m.switch_to_window(m.window_handles[1])
+                Wait(m, timeout=m.timeout.page_load).until(
+                    lambda _: m.get_url() != "about:blank")
+                self.assertTrue(
+                    m.get_url() in ["https://tb-manual.torproject.org/en-US/security-settings/", "https://tb-manual.torproject.org/security-settings/"])
+
+            # Test Learn More link from panel
+            m.find_element('id', 'security-level-button').click()
+            m.find_element('id', 'securityLevel-learnMore').click()
+            Wait(m, timeout=m.timeout.page_load).until(
+                lambda _: len(m.window_handles) > 2)
+            with m.using_context('content'):
+                m.switch_to_window(m.window_handles[2])
+                Wait(m, timeout=m.timeout.page_load).until(
+                    lambda _: m.get_url() != "about:blank")
+                self.assertTrue(
+                    m.get_url() in ["https://tb-manual.torproject.org/en-US/security-settings/", "https://tb-manual.torproject.org/security-settings/"])
+
+            # Test custom settings from panel
+            m.set_pref(custom_pref, False)
+            elem = m.find_element('id', 'securityLevel-restoreDefaults')
+            self.assertEqual(elem.is_displayed(), False)
+            m.find_element('id', 'security-level-button').click()
+            self.assertEqual(elem.is_displayed(), True)
+            elem.click()
+            self.assertEqual(True, m.get_pref(custom_pref))