add ability to change port in test

This commit is contained in:
Dima Gerasimov 2019-06-29 18:40:33 +01:00 committed by Dmitrii Gerasimov
parent e8260f2f47
commit 2fa4f81968

View file

@ -20,6 +20,11 @@ def get_webdriver():
driver = webdriver.Firefox(profile, firefox_binary='/L/soft/firefox-dev/firefox/firefox') driver = webdriver.Firefox(profile, firefox_binary='/L/soft/firefox-dev/firefox/firefox')
driver.install_addon(str(addon_path), temporary=True) driver.install_addon(str(addon_path), temporary=True)
yield driver
# TODO close driver?
def open_options_page(driver):
# necessary to trigger prefs.js initialisation.. # necessary to trigger prefs.js initialisation..
driver.get('http://example.com') driver.get('http://example.com')
sleep(1) sleep(1)
@ -27,26 +32,30 @@ def get_webdriver():
moz_profile = Path(driver.capabilities['moz:profile']) moz_profile = Path(driver.capabilities['moz:profile'])
prefs_file = moz_profile / 'prefs.js' prefs_file = moz_profile / 'prefs.js'
addon_id = None addon_id = None
print(prefs_file)
for line in prefs_file.read_text().splitlines(): for line in prefs_file.read_text().splitlines():
print(line)
# temporary-addon\":\"53104c22-acd0-4d44-904c-22d11d31559a\"}") # temporary-addon\":\"53104c22-acd0-4d44-904c-22d11d31559a\"}")
m = re.search(r'temporary-addon.....([0-9a-z-]+)."', line) m = re.search(r'temporary-addon.....([0-9a-z-]+)."', line)
if m is None: if m is None:
continue continue
addon_id = m.group(1) addon_id = m.group(1)
print("ADDON " + addon_id)
assert addon_id is not None assert addon_id is not None
options = f'moz-extension://{addon_id}/options.html' options = f'moz-extension://{addon_id}/options.html'
print(options)
driver.get(options) driver.get(options)
import ipdb; ipdb.set_trace()
yield driver # TODO could also check for errors
# TODO close driver?
# TODO need to fix hotkey! def change_port(driver, port: str):
open_options_page(driver)
ep = driver.find_element_by_id('endpoint_id')
ep.clear()
ep.send_keys(f'http://localhost:{port}')
se = driver.find_element_by_id('save_id')
se.click()
driver.switch_to.alert.accept()
def trigger_grasp(): def trigger_grasp():
@ -65,12 +74,13 @@ def trigger_grasp():
# pyautogui.locateOnScreen('/L/soft/browser-extensions/grasp/unicorn.png') # pyautogui.locateOnScreen('/L/soft/browser-extensions/grasp/unicorn.png')
print("sending hotkey!") print("sending hotkey!")
import ipdb; ipdb.set_trace()
pyautogui.hotkey('ctrl', 'alt', 'c') pyautogui.hotkey('ctrl', 'alt', 'c')
def test(): def test():
with get_webdriver() as driver: with get_webdriver() as driver:
change_port(driver, port='17777')
driver.get('https://en.wikipedia.org/wiki/Automation') driver.get('https://en.wikipedia.org/wiki/Automation')
sleep(2) # just in case.. sleep(2) # just in case..