def do_GET(self):
try:
path, body, session, element = self.process_request()
logger.debug("%s - %s - %s - %s" % (path, body, session, element))
if path.startswith("/attribute/"):
logger.info("Getting Attribute %s - %s" % (element, session))
assert session
name = path[len("/attribute/") :]
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session, value=marionette_element.get_attribute(name))
elif path == "/displayed":
logger.info("Displayed %s - %s" % (element, session))
assert session
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session, value=marionette_element.is_displayed())
elif path == "/enabled":
logger.info("Enabled %s - %s" % (element, session))
assert session
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session, value=marionette_element.enabled())
elif path.startswith("/equals/"):
assert session
other = path[len("/equals") :]
marionette_element = HTMLElement(self.server.marionette, element)
other_element = HTMLElement(self.server.marionette, other)
self.send_JSON(session=session, value=marionette_element.equals(other_element))
elif path == "/selected":
logger.info("Selected %s - %s" % (element, session))
assert session
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session, value=marionette_element.selected())
elif path == "/status":
self.send_JSON(data=self.server.marionette.status())
elif path == "/text":
logger.info("Getting text %s - %s" % (element, session))
assert session
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session, value=marionette_element.text)
elif path == "/url":
logger.info("Getting url - %s" % session)
assert session
self.send_JSON(value=self.server.marionette.get_url(), session=session)
elif path == "/window_handle":
logger.info("Getting window handle - %s" % session)
assert session
self.send_JSON(session=session, value=self.server.marionette.current_window_handle)
elif path == "/window_handles":
logger.info("Getting window handles - %s" % session)
assert session
self.send_JSON(session=session, value=self.server.marionette.window_handles)
elif path == "/title":
logger.info("Getting Window Title")
assert session
self.send_JSON(session=session, value=self.server.marionette.title)
elif path == "/name":
logger.info("Getting element tag name")
assert session
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session, value=marionette_element.tag_name)
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")
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_GET(self):
try:
path, body, session, element = self.process_request()
if path.startswith('/attribute/'):
assert(session)
name = path[len('/attribute/'):]
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.get_attribute(name))
elif path == '/displayed':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.displayed())
elif path == '/enabled':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.enabled())
elif path.startswith('/equals/'):
assert(session)
other = path[len('/equals'):]
marionette_element = HTMLElement(self.server.marionette, element)
other_element = HTMLElement(self.server.marionette, other)
self.send_JSON(session=session,
value=marionette_element.equals(other))
elif path == '/selected':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.selected())
elif path == '/status':
self.send_JSON(data=self.server.marionette.status())
elif path == '/text':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.text())
elif path == '/url':
assert(session)
self.send_JSON(value=self.server.marionette.get_url(),
session=session)
elif path == '/value':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
send.send_JSON(session=session,
value=marionette_element.value())
elif path == '/window_handle':
assert(session)
self.send_JSON(session=session,
value=self.server.marionette.get_window())
elif path == '/window_handles':
assert(session)
self.send_JSON(session=session,
value=self.server.marionette.get_windows())
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()
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()
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_GET(self):
try:
path, body, session, element = self.process_request()
if path.startswith('/attribute/'):
assert(session)
name = path[len('/attribute/'):]
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.get_attribute(name))
elif path == '/displayed':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.displayed())
elif path == '/enabled':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.enabled())
elif path.startswith('/equals/'):
assert(session)
other = path[len('/equals'):]
marionette_element = HTMLElement(self.server.marionette, element)
other_element = HTMLElement(self.server.marionette, other)
self.send_JSON(session=session,
value=marionette_element.equals(other))
elif path == '/selected':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.selected())
elif path == '/status':
self.send_JSON(data=self.server.marionette.status())
elif path == '/text':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
self.send_JSON(session=session,
value=marionette_element.text())
elif path == '/url':
assert(session)
self.send_JSON(value=self.server.marionette.get_url(),
session=session)
elif path == '/value':
assert(session)
marionette_element = HTMLElement(self.server.marionette, element)
send.send_JSON(session=session,
value=marionette_element.value())
elif path == '/window_handle':
assert(session)
self.send_JSON(session=session,
value=self.server.marionette.get_window())
elif path == '/window_handles':
assert(session)
self.send_JSON(session=session,
value=self.server.marionette.get_windows())
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_)
