diff options
| -rw-r--r-- | selenium/test.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/selenium/test.py b/selenium/test.py index 7de65d294..587de893f 100644 --- a/selenium/test.py +++ b/selenium/test.py @@ -18,6 +18,7 @@ import logging import sys import os import re +import json logging.basicConfig(format='%(levelname)s:%(message)s', level=logging.INFO) logger = logging.getLogger(__name__) @@ -27,9 +28,13 @@ def client_setup(args): """Return a dict containing the driver and the extension's id""" co = webdriver.ChromeOptions() co.add_argument("load-extension=" + args.extdir) - cap = webdriver.DesiredCapabilities.CHROME.copy() + cap = co.to_capabilities() cap['loggingPrefs'] = {'driver': 'INFO', 'browser': 'INFO'} - client = webdriver.Chrome(chrome_options=co, desired_capabilities=cap) + if args.remote: + client = webdriver.Remote(desired_capabilities=cap, command_executor=args.remote) + else: + client = webdriver.Chrome(desired_capabilities=cap) + # client = webdriver.Chrome(chrome_options=co, desired_capabilities=cap) client.get('https://taler.net') listener = """\ document.addEventListener('taler-id', function(evt){ @@ -216,6 +221,7 @@ def withdraw(client, amount_value=None): parser = argparse.ArgumentParser() parser.add_argument('--extdir', help="Folder containing the unpacked extension", metavar="EXTDIR", type=str, dest="extdir", required=True) +parser.add_argument('--remote', help="Whether the test is to be run against URI, or locally", metavar="URI", type=str, dest="remote") args = parser.parse_args() logger.info("Getting extension's ID..") ret = client_setup(args) |