def test_reset_should_remove_added_prefs(self): """Check that when we call reset the items we expect are updated""" profile = Profile() prefs_file = os.path.join(profile.profile, 'user.js') # we shouldn't have any initial preferences initial_prefs = Preferences.read_prefs(prefs_file) self.assertFalse(initial_prefs) initial_prefs = file(prefs_file).read().strip() self.assertFalse(initial_prefs) # add some preferences prefs1 = [("mr.t.quotes", "i aint getting on no plane!")] profile.set_preferences(prefs1) self.assertEqual(prefs1, Preferences.read_prefs(prefs_file)) lines = file(prefs_file).read().strip().splitlines() self.assertTrue( bool([ line for line in lines if line.startswith('#MozRunner Prefs Start') ])) self.assertTrue( bool([ line for line in lines if line.startswith('#MozRunner Prefs End') ])) profile.reset() self.assertNotEqual( prefs1, Preferences.read_prefs(os.path.join(profile.profile, 'user.js')), "I pity the fool who left my pref")
def test_reset_should_remove_added_prefs(self): """Check that when we call reset the items we expect are updated""" profile = Profile() prefs_file = os.path.join(profile.profile, 'user.js') # we shouldn't have any initial preferences initial_prefs = Preferences.read_prefs(prefs_file) self.assertFalse(initial_prefs) initial_prefs = file(prefs_file).read().strip() self.assertFalse(initial_prefs) # add some preferences prefs1 = [("mr.t.quotes", "i aint getting on no plane!")] profile.set_preferences(prefs1) self.assertEqual(prefs1, Preferences.read_prefs(prefs_file)) lines = file(prefs_file).read().strip().splitlines() self.assertTrue(bool([line for line in lines if line.startswith('#MozRunner Prefs Start')])) self.assertTrue(bool([line for line in lines if line.startswith('#MozRunner Prefs End')])) profile.reset() self.assertNotEqual(prefs1, Preferences.read_prefs(os.path.join(profile.profile, 'user.js')), "I pity the fool who left my pref")
def test_magic_markers(self): """ensure our magic markers are working""" profile = Profile() prefs_file = os.path.join(profile.profile, 'user.js') # we shouldn't have any initial preferences initial_prefs = Preferences.read_prefs(prefs_file) self.assertFalse(initial_prefs) initial_prefs = file(prefs_file).read().strip() self.assertFalse(initial_prefs) # add some preferences prefs1 = [("browser.startup.homepage", "http://planet.mozilla.org/"), ("zoom.minPercent", 30)] profile.set_preferences(prefs1) self.assertEqual(prefs1, Preferences.read_prefs(prefs_file)) lines = file(prefs_file).read().strip().splitlines() self.assertTrue( bool([ line for line in lines if line.startswith('#MozRunner Prefs Start') ])) self.assertTrue( bool([ line for line in lines if line.startswith('#MozRunner Prefs End') ])) # add some more preferences prefs2 = [("zoom.maxPercent", 300), ("webgl.verbose", 'false')] profile.set_preferences(prefs2) self.assertEqual(prefs1 + prefs2, Preferences.read_prefs(prefs_file)) lines = file(prefs_file).read().strip().splitlines() self.assertTrue( len([ line for line in lines if line.startswith('#MozRunner Prefs Start') ]) == 2) self.assertTrue( len([ line for line in lines if line.startswith('#MozRunner Prefs End') ]) == 2) # now clean it up profile.clean_preferences() final_prefs = Preferences.read_prefs(prefs_file) self.assertFalse(final_prefs) lines = file(prefs_file).read().strip().splitlines() self.assertTrue('#MozRunner Prefs Start' not in lines) self.assertTrue('#MozRunner Prefs End' not in lines)
def test_magic_markers(): """ensure our magic markers are working""" profile = Profile() prefs_file = os.path.join(profile.profile, "user.js") # we shouldn't have any initial preferences initial_prefs = Preferences.read_prefs(prefs_file) assert not initial_prefs initial_prefs = open(prefs_file).read().strip() assert not initial_prefs # add some preferences prefs1 = [ ("browser.startup.homepage", "http://planet.mozilla.org/"), ("zoom.minPercent", 30), ] profile.set_preferences(prefs1) assert prefs1 == Preferences.read_prefs(prefs_file) lines = open(prefs_file).read().strip().splitlines() assert bool( [line for line in lines if line.startswith("#MozRunner Prefs Start")]) assert bool( [line for line in lines if line.startswith("#MozRunner Prefs End")]) # add some more preferences prefs2 = [("zoom.maxPercent", 300), ("webgl.verbose", "false")] profile.set_preferences(prefs2) assert prefs1 + prefs2 == Preferences.read_prefs(prefs_file) lines = open(prefs_file).read().strip().splitlines() assert (len([ line for line in lines if line.startswith("#MozRunner Prefs Start") ]) == 2) assert len([ line for line in lines if line.startswith("#MozRunner Prefs End") ]) == 2 # now clean it up profile.clean_preferences() final_prefs = Preferences.read_prefs(prefs_file) assert not final_prefs lines = open(prefs_file).read().strip().splitlines() assert "#MozRunner Prefs Start" not in lines assert "#MozRunner Prefs End" not in lines
def start_firefox(self): env = os.environ.copy() env['MOZ_CRASHREPORTER_NO_REPORT'] = '1' profile = Profile() profile.set_preferences({"marionette.defaultPrefs.enabled": True, "marionette.defaultPrefs.port": self.marionette_port, "dom.disable_open_during_load": False, "dom.max_script_run_time": 0}) self.firefox_runner = FirefoxRunner(profile, self.firefox_binary, cmdargs=["--marionette"], env=env, kp_kwargs = {"processOutputLine":[self.on_output]}, process_class=self.process_cls) self.logger.debug("Starting Firefox") self.firefox_runner.start() self.logger.debug("Firefox Started")
def test_magic_markers(self): """ensure our magic markers are working""" profile = Profile() prefs_file = os.path.join(profile.profile, 'user.js') # we shouldn't have any initial preferences initial_prefs = Preferences.read_prefs(prefs_file) self.assertFalse(initial_prefs) initial_prefs = file(prefs_file).read().strip() self.assertFalse(initial_prefs) # add some preferences prefs1 = [("browser.startup.homepage", "http://planet.mozilla.org/"), ("zoom.minPercent", 30)] profile.set_preferences(prefs1) self.assertEqual(prefs1, Preferences.read_prefs(prefs_file)) lines = file(prefs_file).read().strip().splitlines() self.assertTrue(bool([line for line in lines if line.startswith('#MozRunner Prefs Start')])) self.assertTrue(bool([line for line in lines if line.startswith('#MozRunner Prefs End')])) # add some more preferences prefs2 = [("zoom.maxPercent", 300), ("webgl.verbose", 'false')] profile.set_preferences(prefs2) self.assertEqual(prefs1 + prefs2, Preferences.read_prefs(prefs_file)) lines = file(prefs_file).read().strip().splitlines() self.assertTrue(len([line for line in lines if line.startswith('#MozRunner Prefs Start')]) == 2) self.assertTrue(len([line for line in lines if line.startswith('#MozRunner Prefs End')]) == 2) # now clean it up profile.clean_preferences() final_prefs = Preferences.read_prefs(prefs_file) self.assertFalse(final_prefs) lines = file(prefs_file).read().strip().splitlines() self.assertTrue('#MozRunner Prefs Start' not in lines) self.assertTrue('#MozRunner Prefs End' not in lines)
def test_reset_should_remove_added_prefs(): """Check that when we call reset the items we expect are updated""" profile = Profile() prefs_file = os.path.join(profile.profile, 'user.js') # we shouldn't have any initial preferences initial_prefs = Preferences.read_prefs(prefs_file) assert not initial_prefs initial_prefs = open(prefs_file).read().strip() assert not initial_prefs # add some preferences prefs1 = [("mr.t.quotes", "i aint getting on no plane!")] profile.set_preferences(prefs1) assert prefs1 == Preferences.read_prefs(prefs_file) lines = open(prefs_file).read().strip().splitlines() assert any(line.startswith('#MozRunner Prefs Start') for line in lines) assert any(line.startswith('#MozRunner Prefs End') for line in lines) profile.reset() assert prefs1 != Preferences.read_prefs(os.path.join(profile.profile, 'user.js'))
def do_POST(self): try: path, body, session, element = self.process_request() logger.debug("%s - %s - %s - %s" % (path, body, session, element)) if path == "/back": logger.info("Navigating back - %s" % session) assert session assert self.server.marionette.go_back() self.send_JSON(session=session) elif path == "/clear": logger.info("Clearing %s - %s" % (element, session)) assert session marionette_element = HTMLElement(self.server.marionette, element) marionette_element.clear() self.send_JSON(session=session) elif path == "/click": logger.info("Clicking %s - %s" % (element, session)) assert session marionette_element = HTMLElement(self.server.marionette, element) marionette_element.click() self.send_JSON(session=session) elif path == "/element": logger.info("Find Element using - %s, value - %s - %s" % (body["using"], body["value"], session)) # find element variants assert session self.send_JSON( session=session, value={"ELEMENT": self.server.marionette.find_element(body["using"], body["value"], id=element).id}, ) elif path == "/elements": logger.info("Find Elements using - %s, value - %s - %s" % (body["using"], body["value"], session)) # find elements variants assert session self.send_JSON( session=session, value=[ {"ELEMENT": x.id} for x in self.server.marionette.find_elements(body["using"], body["value"]) ], ) elif path == "/execute": logger.info("Executing Script - %s" % session) assert session if body["args"]: result = self.server.marionette.execute_script( body["script"], script_args=body["args"], new_sandbox=False ) else: result = self.server.marionette.execute_script(body["script"], new_sandbox=False) self.send_JSON(session=session, value=result) elif path == "/execute_async": logger.info("Executing Async Script - %s" % session) assert session result = None if body["args"]: result = self.server.marionette.execute_async_script( body["script"], script_args=body["args"], new_sandbox=False ) else: result = self.server.marionette.execute_async_script(body["script"], new_sandbox=False) self.send_JSON(session=session, value=result) elif path == "/forward": logger.info("Forwarding - %s" % session) assert session assert self.server.marionette.go_forward() self.send_JSON(session=session) elif path == "/frame": logger.info("Switch to Frame %s - %s" % (body["id"], session)) assert session frame = body["id"] if isinstance(frame, dict) and "ELEMENT" in frame: frame = HTMLElement(self.server.marionette, frame["ELEMENT"]) assert self.server.marionette.switch_to_frame(frame) self.send_JSON(session=session) elif path == "/refresh": logger.info("Refreshing the page - %s" % session) assert session assert self.server.marionette.refresh() self.send_JSON(session=session) elif path == "/session": logger.info("Creating new session") logger.debug("loading webdriver prefs") with open("webdriver.json") as webpref: read_prefs = webpref.read() prefs = json.loads(read_prefs) port = free_port() logger.debug("Creating Profile") profile = Profile() profile.set_preferences(prefs["frozen"]) profile.set_preferences(prefs["mutable"]) profile.set_preferences({"marionette.defaultPrefs.enabled": True, "marionette.defaultPrefs.port": port}) logger.debug("Profile created at %s" % profile.profile) logger.debug("Creating runner") firefox_binary = ( body["desiredCapabilities"]["firefoxBinary"] if body["desiredCapabilities"]["firefoxBinary"] else firefox_binary_path() ) self.server.runner = FirefoxRunner(profile, firefox_binary) self.server.runner.start() logger.debug("Browser has been started") logger.info("Creating Marionette instance on %s:%s" % ("localhost", port)) self.server.marionette = Marionette("localhost", port) self.server.marionette.wait_for_port() session = self.server.marionette.start_session() self.send_JSON(session=session, value={}) # 'value' is the browser capabilities, which we're ignoring for now elif path == "/timeouts/async_script": logger.info("Script timeout %s ms - %s" % (body["ms"], session)) assert session assert self.server.marionette.set_script_timeout(body["ms"]) self.send_JSON(session=session) elif path == "/timeouts/implicit_wait": logger.info("Implicit timeout %s ms - %s" % (body["ms"], session)) assert session assert self.server.marionette.set_search_timeout(body["ms"]) self.send_JSON(session=session) elif path == "/url": logger.info("Navigating to %s - %s" % (body["url"], session)) assert session assert self.server.marionette.navigate(body["url"]) self.send_JSON(session=session) elif path == "/value": logger.info("Send Keys %s - %s" % ("".join(body["value"]), session)) assert session keys = "".join(body["value"]) marionette_element = HTMLElement(self.server.marionette, element) assert marionette_element.send_keys(keys) self.send_JSON(session=session) elif path == "/window": logger.info("Switch to Window %s - %s" % (body["name"], session)) assert session assert self.server.marionette.switch_to_window(body["name"]) self.send_JSON(session=session) else: logger.error("Unknown path - %s" % session) self.file_not_found() except MarionetteException as e: logger.error("Status: %s - Message: %s" % (e.status, e.message)) self.send_JSON(data={"status": e.status}, value={"message": e.message}) except: trace_ = traceback.format_exc() logger.critical("Server Exception: %s" % trace_) self.server_error(trace_)
def do_POST(self): try: path, body, session, element = self.process_request() logger.debug("%s - %s - %s - %s" % (path, body, session, element)) if path == '/back': logger.info("Navigating back - %s" % session) assert(session) assert(self.server.marionette.go_back()) self.send_JSON(session=session) elif path == '/clear': logger.info("Clearing %s - %s" % (element, session)) assert(session) marionette_element = HTMLElement(self.server.marionette, element) marionette_element.clear() self.send_JSON(session=session) elif path == '/click': logger.info("Clicking %s - %s" % (element, session)) assert(session) marionette_element = HTMLElement(self.server.marionette, element) marionette_element.click() self.send_JSON(session=session) elif path == '/element': logger.info("Find Element using - %s, value - %s - %s" \ % (body['using'], body['value'], session)) # find element variants assert(session) self.send_JSON(session=session, value={'ELEMENT': self.server.marionette.find_element(body['using'], body['value'], id=element).id}) elif path == '/elements': logger.info("Find Elements using - %s, value - %s - %s" \ % (body['using'], body['value'], session)) # find elements variants assert(session) self.send_JSON(session=session, value=[{'ELEMENT': x.id} for x in self.server.marionette.find_elements(body['using'], body['value'])]) elif path == '/execute': logger.info("Executing Script - %s" % session) assert(session) if body['args']: result = self.server.marionette.execute_script(body['script'], script_args=body['args'], new_sandbox=False) else: result = self.server.marionette.execute_script(body['script'],new_sandbox=False) self.send_JSON(session=session, value=result) elif path == '/execute_async': logger.info("Executing Async Script - %s" % session) assert(session) result = None if body['args']: result = self.server.marionette.execute_async_script(body['script'], script_args=body['args'], new_sandbox=False) else: result = self.server.marionette.execute_async_script(body['script'], new_sandbox=False) self.send_JSON(session=session, value=result) elif path == '/forward': logger.info("Forwarding - %s" % session) assert(session) assert(self.server.marionette.go_forward()) self.send_JSON(session=session) elif path == '/frame': logger.info("Switch to Frame %s - %s" % (body['id'], session)) assert(session) frame = body['id'] if isinstance(frame, dict) and 'ELEMENT' in frame: frame = HTMLElement(self.server.marionette, frame['ELEMENT']) assert(self.server.marionette.switch_to_frame(frame)) self.send_JSON(session=session) elif path == '/refresh': logger.info("Refreshing the page - %s" % session) assert(session) assert(self.server.marionette.refresh()) self.send_JSON(session=session) elif path == '/session': logger.info("Creating new session") logger.debug("loading webdriver prefs") with open('webdriver.json') as webpref: read_prefs = webpref.read() prefs = json.loads(read_prefs) port = free_port() logger.debug("Creating Profile") profile = Profile() profile.set_preferences(prefs['frozen']) profile.set_preferences(prefs['mutable']) profile.set_preferences({"marionette.defaultPrefs.enabled": True, "marionette.defaultPrefs.port": port}) logger.debug("Profile created at %s" % profile.profile) logger.debug("Creating runner") firefox_binary = body["desiredCapabilities"]['firefoxBinary'] if body["desiredCapabilities"]['firefoxBinary'] else firefox_binary_path() self.server.runner = FirefoxRunner(profile, firefox_binary) self.server.runner.start() logger.debug("Browser has been started") import time time.sleep(10) logger.info("Creating Marionette instance on %s:%s" % ("localhost", port)) self.server.marionette = Marionette("localhost", port) session = self.server.marionette.start_session() self.send_JSON(session=session, value={}) # 'value' is the browser capabilities, which we're ignoring for now elif path == '/timeouts/async_script': logger.info("Script timeout %s ms - %s" % (body['ms'], session)) assert(session) assert(self.server.marionette.set_script_timeout(body['ms'])) self.send_JSON(session=session) elif path == '/timeouts/implicit_wait': logger.info("Implicit timeout %s ms - %s" % (body['ms'], session)) assert(session) assert(self.server.marionette.set_search_timeout(body['ms'])) self.send_JSON(session=session) elif path == '/url': logger.info("Navigating to %s - %s" % (body['url'], session)) assert(session) assert(self.server.marionette.navigate(body['url'])) self.send_JSON(session=session) elif path == '/value': logger.info("Send Keys %s - %s" % (''.join(body['value']), session)) assert(session) keys = ''.join(body['value']) marionette_element = HTMLElement(self.server.marionette, element) assert(marionette_element.send_keys(keys)) self.send_JSON(session=session) elif path == '/window': logger.info("Switch to Window %s - %s" % (body['name'], session)) assert(session) assert(self.server.marionette.switch_to_window(body['name'])) self.send_JSON(session=session) else: logger.error("Unknown path - %s" % session) self.file_not_found() except MarionetteException as e: logger.error("Status: %s - Message: %s" % (e.status, e.message)) self.send_JSON(data={'status': e.status}, value={'message': e.message}) except: trace_ = traceback.format_exc() logger.critical("Server Exception: %s" % trace_) self.server_error(trace_)