diff options
| -rw-r--r-- | tests/functional/features/environment.py | 11 | ||||
| -rw-r--r-- | tests/functional/features/steps/vpn.py | 10 |
2 files changed, 15 insertions, 6 deletions
diff --git a/tests/functional/features/environment.py b/tests/functional/features/environment.py index 5b179cd..c92a39d 100644 --- a/tests/functional/features/environment.py +++ b/tests/functional/features/environment.py @@ -1,12 +1,14 @@ +import commands import os +import sys import shutil import re -import time + from urlparse import urlparse -import commands from selenium import webdriver from selenium.webdriver.chrome.options import Options + from leap.common.config import get_path_prefix DEFAULT_IMPLICIT_WAIT_TIMEOUT_IN_S = 10 @@ -102,7 +104,10 @@ def _save_screenshot(context, step): def _save_config(context, step): filepath = _artifact_path(step, 'config') - shutil.copytree(get_path_prefix(), filepath) + try: + shutil.copytree(get_path_prefix(), filepath) + except OSError: + pass print('copied config to: file://%s' % filepath) diff --git a/tests/functional/features/steps/vpn.py b/tests/functional/features/steps/vpn.py index 05ed2b5..4367359 100644 --- a/tests/functional/features/steps/vpn.py +++ b/tests/functional/features/steps/vpn.py @@ -15,12 +15,16 @@ def record_ip(context): def _current_ip(): - url = 'https://wtfismyip.com/json' + url = 'https://ipapi.co/json' r = requests.get(url) - data = r.json() + try: + data = r.json() + except Exception: + print("ERROR: data received was %s" % r.content) + raise - current_ip = data['YourFuckingIPAddress'] + current_ip = data.get('ip') print("Current IP: %s\n\n" % current_ip) return current_ip |