1 files changed, 159 insertions, 0 deletions
diff --git a/testing/marionette/harness/marionette_harness/tests/unit/test_proxy.py b/testing/marionette/harness/marionette_harness/tests/unit/test_proxy.py
new file mode 100644
index 0000000000..21a37639d9
--- /dev/null
+++ b/testing/marionette/harness/marionette_harness/tests/unit/test_proxy.py
@@ -0,0 +1,159 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+from marionette_driver import errors
+
+from marionette_harness import MarionetteTestCase
+
+
+class TestProxyCapabilities(MarionetteTestCase):
+    def setUp(self):
+        super(TestProxyCapabilities, self).setUp()
+
+        self.marionette.delete_session()
+
+    def tearDown(self):
+        if not self.marionette.session:
+            self.marionette.start_session()
+
+        with self.marionette.using_context("chrome"):
+            self.marionette.execute_script(
+                """
+                const { Preferences } = ChromeUtils.importESModule(
+                  "resource://gre/modules/Preferences.sys.mjs"
+                );
+                Preferences.resetBranch("network.proxy");
+            """
+            )
+
+        super(TestProxyCapabilities, self).tearDown()
+
+    def test_proxy_object_in_returned_capabilities(self):
+        capabilities = {"proxy": {"proxyType": "system"}}
+
+        self.marionette.start_session(capabilities)
+        self.assertEqual(
+            self.marionette.session_capabilities["proxy"], capabilities["proxy"]
+        )
+
+    def test_proxy_type_autodetect(self):
+        capabilities = {"proxy": {"proxyType": "autodetect"}}
+
+        self.marionette.start_session(capabilities)
+        self.assertEqual(
+            self.marionette.session_capabilities["proxy"], capabilities["proxy"]
+        )
+
+    def test_proxy_type_direct(self):
+        capabilities = {"proxy": {"proxyType": "direct"}}
+
+        self.marionette.start_session(capabilities)
+        self.assertEqual(
+            self.marionette.session_capabilities["proxy"], capabilities["proxy"]
+        )
+
+    def test_proxy_type_manual(self):
+        proxy_hostname = "marionette.test"
+        capabilities = {
+            "proxy": {
+                "proxyType": "manual",
+                "httpProxy": "{}:80".format(proxy_hostname),
+                "sslProxy": "{}:443".format(proxy_hostname),
+                "socksProxy": proxy_hostname,
+                "socksVersion": 4,
+            }
+        }
+
+        self.marionette.start_session(capabilities)
+        self.assertEqual(
+            self.marionette.session_capabilities["proxy"], capabilities["proxy"]
+        )
+
+    def test_proxy_type_manual_socks_requires_version(self):
+        proxy_port = 4444
+        proxy_hostname = "marionette.test"
+        proxy_host = "{}:{}".format(proxy_hostname, proxy_port)
+        capabilities = {
+            "proxy": {
+                "proxyType": "manual",
+                "socksProxy": proxy_host,
+            }
+        }
+
+        with self.assertRaises(errors.SessionNotCreatedException):
+            self.marionette.start_session(capabilities)
+
+    def test_proxy_type_manual_no_proxy_on(self):
+        capabilities = {
+            "proxy": {
+                "proxyType": "manual",
+                "noProxy": ["foo", "bar"],
+            }
+        }
+
+        self.marionette.start_session(capabilities)
+        self.assertEqual(
+            self.marionette.session_capabilities["proxy"], capabilities["proxy"]
+        )
+
+    def test_proxy_type_manual_invalid_no_proxy_on(self):
+        capabilities = {
+            "proxy": {
+                "proxyType": "manual",
+                "noProxy": "foo, bar",
+            }
+        }
+
+        with self.assertRaises(errors.SessionNotCreatedException):
+            self.marionette.start_session(capabilities)
+
+    def test_proxy_type_pac(self):
+        pac_url = "http://marionette.test"
+        capabilities = {"proxy": {"proxyType": "pac", "proxyAutoconfigUrl": pac_url}}
+
+        self.marionette.start_session(capabilities)
+        self.assertEqual(
+            self.marionette.session_capabilities["proxy"], capabilities["proxy"]
+        )
+
+    def test_proxy_type_system(self):
+        capabilities = {"proxy": {"proxyType": "system"}}
+
+        self.marionette.start_session(capabilities)
+        self.assertEqual(
+            self.marionette.session_capabilities["proxy"], capabilities["proxy"]
+        )
+
+    def test_invalid_proxy_object(self):
+        capabilities = {"proxy": "I really should be a dictionary"}
+
+        with self.assertRaises(errors.SessionNotCreatedException):
+            self.marionette.start_session(capabilities)
+
+    def test_missing_proxy_type(self):
+        with self.assertRaises(errors.SessionNotCreatedException):
+            self.marionette.start_session({"proxy": {"proxyAutoconfigUrl": "foobar"}})
+
+    def test_invalid_proxy_type(self):
+        capabilities = {"proxy": {"proxyType": "NOPROXY"}}
+
+        with self.assertRaises(errors.SessionNotCreatedException):
+            self.marionette.start_session(capabilities)
+
+    def test_invalid_autoconfig_url_for_pac(self):
+        with self.assertRaises(errors.SessionNotCreatedException):
+            self.marionette.start_session({"proxy": {"proxyType": "pac"}})
+
+        with self.assertRaises(errors.SessionNotCreatedException):
+            self.marionette.start_session(
+                {"proxy": {"proxyType": "pac", "proxyAutoconfigUrl": None}}
+            )
+
+    def test_missing_socks_version_for_manual(self):
+        capabilities = {
+            "proxy": {"proxyType": "manual", "socksProxy": "marionette.test"}
+        }
+
+        with self.assertRaises(errors.SessionNotCreatedException):
+            self.marionette.start_session(capabilities)