Add Firefox specific command to switch context between Browser content and Browser chrome

Author: AutomatedTester

py/selenium/webdriver/firefox/remote_connection.py

@@ -0,0 +1,25 @@
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+from selenium.webdriver.remote.remote_connection import RemoteConnection
+
+class FirefoxRemoteConnection(RemoteConnection):
+
+ def __init__(self, remote_server_addr, keep_alive=True):
+ RemoteConnection.__init__(self, remote_server_addr, keep_alive)
+ self._commands["SET_CONTEXT"] = ('POST',
+ '/session/$sessionId/moz/context')

py/selenium/webdriver/firefox/webdriver.py

@@ -28,6 +28,7 @@
 from .service import Service
 from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
 from selenium.webdriver.firefox.extension_connection import ExtensionConnection
+from .remote_connection import FirefoxRemoteConnection
 from selenium.webdriver.firefox.firefox_profile import FirefoxProfile
 from selenium.webdriver.remote.webdriver import WebDriver as RemoteWebDriver
 
@@ -60,7 +61,8 @@ def __init__(self, firefox_profile=None, firefox_binary=None, timeout=30,
  self.service.start()
 
  RemoteWebDriver.__init__(self,
- command_executor=self.service.service_url,
+ command_executor=FirefoxRemoteConnection(
+ remote_server_addr=self.service.service_url),
  desired_capabilities=capabilities,
  keep_alive=True)
 
@@ -103,3 +105,6 @@ def quit(self):
  @property
  def firefox_profile(self):
  return self.profile
+
+ def set_context(self, context):
+ self.execute("SET_CONTEXT", {"context": context})

py/test/selenium/webdriver/firefox/mn_set_context_tests.py

@@ -0,0 +1,45 @@
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+
+from selenium import webdriver
+from selenium.test.selenium.webdriver.common.webserver import SimpleWebServer
+
+class TestMarionetteSpecific:
+
+ def setup_method(self, method):
+ self.driver = webdriver.Firefox()
+ self.CHROME = 'chrome'
+ self.CONTENT = 'content'
+
+ def test_we_can_switch_context_to_chrome(self):
+ self.driver.set_context(self.CHROME)
+ self.driver.execute_script("var c = Components.classes; return 1;");
+
+ def teardown_method(self, method):
+ try:
+ self.driver.set_context(self.CONTENT)
+ self.driver.quit()
+ except:
+ pass # Don't care since we may have killed the browser above
+
+
+def teardown_module(module):
+ try:
+ TestMarionetteSpecific.driver.quit()
+ except:
+ pass # Don't Care since we may have killed the browser above