From d661f63eec393359600ff1277faeb2ccede66f51 Mon Sep 17 00:00:00 2001 From: Irina Truong Date: Tue, 21 May 2019 20:28:05 -0700 Subject: pep8. --- tests/features/environment.py | 3 ++- tests/features/steps/basic_commands.py | 6 ++++-- 2 files changed, 6 insertions(+), 3 deletions(-) (limited to 'tests') diff --git a/tests/features/environment.py b/tests/features/environment.py index b21b9010..6b4d4241 100644 --- a/tests/features/environment.py +++ b/tests/features/environment.py @@ -94,7 +94,8 @@ def before_all(context): os.environ['PGUSER'] = context.conf['user'] os.environ['PGHOST'] = context.conf['host'] os.environ['PGPORT'] = context.conf['port'] - os.environ['PGSERVICEFILE'] = os.path.join(fixture_dir, 'mock_pg_service.conf') + os.environ['PGSERVICEFILE'] = os.path.join( + fixture_dir, 'mock_pg_service.conf') if context.conf['pass']: os.environ['PGPASSWORD'] = context.conf['pass'] diff --git a/tests/features/steps/basic_commands.py b/tests/features/steps/basic_commands.py index fcb250c7..71b591a0 100644 --- a/tests/features/steps/basic_commands.py +++ b/tests/features/steps/basic_commands.py @@ -46,11 +46,13 @@ def step_run_cli_using_arg(context, arg): if arg == '--password': arg = '--password' prompt_check = False - if arg == 'dsn_password': # This uses the mock_pg_service.conf file in fixtures folder. + # This uses the mock_pg_service.conf file in fixtures folder. + if arg == 'dsn_password': arg = 'service=mock_postgres --password' prompt_check = False currentdb = "postgres" - wrappers.run_cli(context, run_args=[arg], prompt_check=prompt_check, currentdb=currentdb) + wrappers.run_cli(context, run_args=[ + arg], prompt_check=prompt_check, currentdb=currentdb) @when('we wait for prompt') -- cgit v1.2.3