def __init__(self, remote_server_addr, browser_name, platform):
self._conn = RemoteConnection(
remote_server_addr, browser_name, platform)
class WebDriver(object):
def __init__(self, remote_server_addr, browser_name, platform):
self._conn = RemoteConnection(
remote_server_addr, browser_name, platform)
def get(self, url):
"""Loads a web page in the current browser."""
self._post("url", url)
def get_title(self):
"""Gets the title of the current page."""
resp = self._get("title")
return resp
def find_element_by_id(self, id_):
"""Finds element by id."""
return self._find_element_by("id", id_)
def find_elements_by_xpath(self, xpath):
"""Finds multiple elements by xpath."""
resp = self._post("elements", "xpath", xpath)
elems = []
for token in resp:
elems.append(self._get_elem(token))
return elems
def find_element_by_xpath(self, xpath):
"""Finds an element by xpath."""
return self._find_element_by("xpath", xpath)
def find_element_by_link_text(self, link_text):
"""Finds an element by its link text.
Returns None if the element is not a link.
"""
return self._find_element_by("link text", link_text)
def find_element_by_name(self, name):
"""Finds and element by its name."""
return self._find_element_by("name", name)
def execute_script(self, script, *args):
converted_args = []
for arg in args:
if type(arg) == WebElement:
converted_args.append({"type": "ELEMENT", "value": arg.id})
else:
converted_args.append({"type": "STRING", "value": arg})
resp = self._post("execute", script, converted_args)
if "NULL" == resp["type"]:
pass
elif "ELEMENT" == resp["type"]:
return self._get_elem(resp["value"])
else:
return resp["value"]
def get_current_url(self):
"""Gets the current url."""
return self._get("url")
def get_page_source(self):
"""Gets the page source."""
return self._get("source")
def close(self):
"""Closes the current window.
Quit the browser if it's the last window open.
"""
self._delete("window")
def quit(self):
"""Quits the driver and close every associated window."""
self._conn.quit()
def switch_to_window(self, window_name):
"""Switches focus to a window."""
resp = self._post("window/%s" % window_name)
if resp and "No window found" in resp:
raise InvalidSwitchToTargetException(
"Window %s not found" % window_name)
def switch_to_frame(self, index_or_name):
"""Switches focus to a frame by index or name."""
self._post("frame/%s" % str(index_or_name))
def back(self):
"""Goes back in browser history."""
self._post("back")
def forward(self):
"""Goes forward in browser history."""
self._post("forward")
# Options
def get_cookie(self):
"""Gets all the cookies. Return a set of dicts."""
return self._get("cookie")
def delete_cookie(self, name):
"""Delete a cookie with the given name."""
self._delete("cookie/%s" % name)
def delete_all_cookies(self):
"""Delete all the cookies."""
self._delete("cookie")
def add_cookie(self, cookie_dict):
self._post("addCookie", cookie_dict)
@property
def conn(self):
return self._conn
def _get_elem(self, resp_value):
"""Creates a WebElement from a response token."""
return WebElement(self, resp_value.split("/")[1])
def _find_element_by(self, by, value):
try:
resp = self._post("element", by, value)
if not resp:
raise NoSuchElementException(resp)
return self._get_elem(resp[0])
except ErrorInResponseException, ex:
utils.handle_find_element_exception(ex)
