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() if path == '/back': assert(session) assert(self.server.marionette.go_back()) self.send_JSON(session=session) elif path == '/clear': assert(session) marionette_element = HTMLElement(self.server.marionette, element) marionette_element.clear() self.send_JSON(session=session) elif path == '/click': assert(session) marionette_element = HTMLElement(self.server.marionette, element) marionette_element.click() self.send_JSON(session=session) elif path == '/element': # find element variants assert(session) self.send_JSON(session=session, value={'ELEMENT': str(self.server.marionette.find_element(body['using'], body['value'], id=element))}) elif path == '/elements': # find elements variants assert(session) self.send_JSON(session=session, value=[{'ELEMENT': str(x)} for x in self.server.marionette.find_elements(body['using'], body['value'])]) elif path == '/execute': assert(session) if body['args']: result = self.server.marionette.execute_script(body['script'], script_args=body['args']) else: result = self.server.marionette.execute_script(body['script']) self.send_JSON(session=session, value=result) elif path == '/execute_async': assert(session) if body['args']: result = self.server.marionette.execute_async_script(body['script'], script_args=body['args']) else: result = self.server.marionette.execute_async_script(body['script']) self.send_JSON(session=session, value=result) elif path == '/forward': assert(session) assert(self.server.marionette.go_forward()) self.send_JSON(session=session) elif path == '/frame': 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': assert(session) assert(self.server.marionette.refresh()) self.send_JSON(session=session) elif path == '/session': session = self.server.marionette.start_session() # 'value' is the browser capabilities, which we're ignoring for now self.send_JSON(session=session, value={}) elif path == '/timeouts/async_script': assert(session) assert(self.server.marionette.set_script_timeout(body['ms'])) self.send_JSON(session=session) elif path == '/timeouts/implicit_wait': assert(session) assert(self.server.marionette.set_search_timeout(body['ms'])) self.send_JSON(session=session) elif path == '/url': assert(session) assert(self.server.marionette.navigate(body['url'])) self.send_JSON(session=session) elif path == '/value': 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': assert(session) assert(self.server.marionette.switch_to_window(body['name'])) self.send_JSON(session=session) else: self.file_not_found() except MarionetteException, e: self.send_JSON(data={'status': e.status}, value={'message': e.message})
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_)