diff options
| -rw-r--r-- | tests/functional/features/environment.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/tests/functional/features/environment.py b/tests/functional/features/environment.py index cc3179e..b88ddd7 100644 --- a/tests/functional/features/environment.py +++ b/tests/functional/features/environment.py @@ -22,9 +22,13 @@ def before_all(context): context.host = 'https://{}'.format(context.host) context.hostname = urlparse(context.host).hostname - context.username = os.environ['TEST_USERNAME'] - context.password = os.environ['TEST_PASSWORD'] - context.user_email = '{}@{}'.format(context.username, context.hostname) + try: + context.username = os.environ['TEST_USERNAME'] + context.password = os.environ['TEST_PASSWORD'] + context.user_email = '{}@{}'.format(context.username, context.hostname) + except KeyError: + print('TEST_USERNAME or TEST_PASSWORD not set') + sys.exit(0) def _setup_webdriver(context): |