add ability to change port in test

server/test_all.py

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