1 files changed, 98 insertions, 0 deletions
diff --git a/testing/marionette/harness/marionette_harness/tests/unit/test_cli_arguments.py b/testing/marionette/harness/marionette_harness/tests/unit/test_cli_arguments.py
new file mode 100644
index 0000000000..c4bbbfad1b
--- /dev/null
+++ b/testing/marionette/harness/marionette_harness/tests/unit/test_cli_arguments.py
@@ -0,0 +1,98 @@
+# 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/.
+
+import copy
+
+import requests
+
+from marionette_harness import MarionetteTestCase
+
+
+class TestCommandLineArguments(MarionetteTestCase):
+    def setUp(self):
+        super(TestCommandLineArguments, self).setUp()
+
+        self.orig_arguments = copy.copy(self.marionette.instance.app_args)
+
+    def tearDown(self):
+        self.marionette.instance.app_args = self.orig_arguments
+        self.marionette.quit(in_app=False, clean=True)
+
+        super(TestCommandLineArguments, self).tearDown()
+
+    def test_debugger_address_cdp_status(self):
+        # By default Remote Agent is not enabled
+        debugger_address = self.marionette.session_capabilities.get(
+            "moz:debuggerAddress"
+        )
+        self.assertIsNone(debugger_address)
+
+        # With BiDi only enabled the capability shouldn't be returned
+        self.marionette.set_pref("remote.active-protocols", 1)
+        self.marionette.quit()
+
+        self.marionette.instance.app_args.append("-remote-debugging-port")
+        self.marionette.start_session()
+
+        debugger_address = self.marionette.session_capabilities.get(
+            "moz:debuggerAddress"
+        )
+        self.assertIsNone(debugger_address)
+
+        # Clean the profile so that the preference is definetely reset.
+        self.marionette.quit(in_app=False, clean=True)
+
+        # With all protocols enabled again the capability has to be returned
+        self.marionette.start_session()
+        debugger_address = self.marionette.session_capabilities.get(
+            "moz:debuggerAddress"
+        )
+
+        self.assertEqual(debugger_address, "127.0.0.1:9222")
+        result = requests.get(url="http://{}/json/version".format(debugger_address))
+        self.assertTrue(result.ok)
+
+    def test_websocket_url(self):
+        # By default Remote Agent is not enabled
+        self.assertNotIn("webSocketUrl", self.marionette.session_capabilities)
+
+        # With CDP only enabled the capability is still not returned
+        self.marionette.set_pref("remote.active-protocols", 2)
+
+        self.marionette.quit()
+        self.marionette.instance.app_args.append("-remote-debugging-port")
+        self.marionette.start_session({"webSocketUrl": True})
+
+        self.assertNotIn("webSocketUrl", self.marionette.session_capabilities)
+
+        # Clean the profile so that the preference is definetely reset.
+        self.marionette.quit(in_app=False, clean=True)
+
+        # With all protocols enabled again the capability has to be returned
+        self.marionette.start_session({"webSocketUrl": True})
+
+        session_id = self.marionette.session_id
+        websocket_url = self.marionette.session_capabilities.get("webSocketUrl")
+
+        self.assertEqual(
+            websocket_url, "ws://127.0.0.1:9222/session/{}".format(session_id)
+        )
+
+    # An issue in the command line argument handling lead to open Firefox on
+    # random URLs when remote-debugging-port is set to an explicit value, on macos.
+    # See Bug 1724251.
+    def test_start_page_about_blank(self):
+        self.marionette.quit()
+        self.marionette.instance.app_args.append("-remote-debugging-port=0")
+        self.marionette.start_session({"webSocketUrl": True})
+        self.assertEqual(self.marionette.get_url(), "about:blank")
+
+    def test_startup_timeout(self):
+        try:
+            self.marionette.quit()
+            with self.assertRaisesRegexp(IOError, "Process killed after 0s"):
+                # Use a small enough timeout which should always cause an IOError
+                self.marionette.start_session(timeout=0)
+        finally:
+            self.marionette.start_session()