diff options
| author | Irina Truong <i.chernyavska@gmail.com> | 2019-05-21 20:28:05 -0700 |
|---|---|---|
| committer | Irina Truong <i.chernyavska@gmail.com> | 2019-05-21 20:28:05 -0700 |
| commit | d661f63eec393359600ff1277faeb2ccede66f51 (patch) | |
| tree | b1e53aa2f08ac45489859422efafad07c9246c45 /tests | |
| parent | b4fbe28286410f6b61bd8b9d4a06c2dbfc295bb9 (diff) | |
pep8.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/features/environment.py | 3 | ||||
| -rw-r--r-- | tests/features/steps/basic_commands.py | 6 |
2 files changed, 6 insertions, 3 deletions
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') |