diff options
Diffstat (limited to 'test/haketilo_test/profiles.py')
-rwxr-xr-x | test/haketilo_test/profiles.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/test/haketilo_test/profiles.py b/test/haketilo_test/profiles.py index e1fc368..0487064 100755 --- a/test/haketilo_test/profiles.py +++ b/test/haketilo_test/profiles.py @@ -81,7 +81,8 @@ def set_webextension_uuid(profile, extension_id, uuid=default_extension_uuid): json.dumps({extension_id: uuid})) def firefox_safe_mode(proxy_port, proxy_host=default_proxy_host, - firefox_binary=conf_settings['BROWSER_BINARY']): + firefox_binary=conf_settings['BROWSER_BINARY'], + geckodriver_binary=conf_settings['DRIVER']): """ Initialize a Firefox instance controlled by selenium. The instance is started in safe mode. @@ -94,11 +95,13 @@ def firefox_safe_mode(proxy_port, proxy_host=default_proxy_host, options.add_argument('--safe-mode') return HaketiloFirefox(options=options, firefox_profile=profile, - firefox_binary=firefox_binary) + firefox_binary=firefox_binary, + executable_path=geckodriver_binary) def firefox_with_profile(proxy_port, proxy_host=default_proxy_host, firefox_binary=conf_settings['BROWSER_BINARY'], - profile_dir=conf_settings['CLEAN_PROFILE']): + profile_dir=conf_settings['CLEAN_PROFILE'], + geckodriver_binary=conf_settings['DRIVER']): """ Initialize a Firefox instance controlled by selenium. The instance is started using an empty profile (either the default one or the one passed to @@ -111,4 +114,5 @@ def firefox_with_profile(proxy_port, proxy_host=default_proxy_host, set_webextension_uuid(profile, default_haketilo_id) return HaketiloFirefox(firefox_profile=profile, - firefox_binary=firefox_binary) + firefox_binary=firefox_binary, + executable_path=geckodriver_binary) |