diff options
Diffstat (limited to 'tests/test_main.py')
| -rw-r--r-- | tests/test_main.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_main.py b/tests/test_main.py index cc7693a3..fc4de7d9 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -182,9 +182,9 @@ def pset_pager_mocks(): cli.watch_command = None with mock.patch('pgcli.main.click.echo') as mock_echo, \ mock.patch('pgcli.main.click.echo_via_pager') as mock_echo_via_pager, \ - mock.patch.object(cli, 'cli') as mock_cli: + mock.patch.object(cli, 'prompt_app') as mock_app: - yield cli, mock_echo, mock_echo_via_pager, mock_cli + yield cli, mock_echo, mock_echo_via_pager, mock_app @pytest.mark.parametrize('term_height,term_width,text', test_data, ids=test_ids) |