blob: 0dee3127d297ef1aeebfbbe7841a75538785dc89 [file] [log] [blame]
"""WebDriver alert handling."""
class Alert(object):
"""Class that provides access to the WebDriver alert handling functions."""
def __init__(self, driver):
self._driver = driver
def _execute(self, method, path, name, body=None):
return self._driver.execute(method, path, name, body)
def dismiss(self):
"""Dismiss the alert."""
self._execute('POST', '/dismiss_alert', 'dismiss')
def accept(self):
"""Accept the alert."""
self._execute('POST', '/accept_alert', 'accept')
def get_text(self):
"""Get the text displayed in the alert."""
return self._execute('GET', '/alert_text', 'getText')
def send_keys(self, keys):
"""Type into the text input of the alert if available."""
self._execute('POST', '/alert_text', 'sendKeys', { 'text': keys })