summaryrefslogtreecommitdiffstats
path: root/tests/test_ssh_tunnel.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-18 13:21:43 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-18 13:21:43 +0000
commit98038631ba9672eafb28f518ff30db76d5954d81 (patch)
tree01c930bf54ae05bd6fb271c7161086f0eaf4da47 /tests/test_ssh_tunnel.py
parentAdding upstream version 4.0.1. (diff)
downloadpgcli-upstream/4.1.0.tar.xz
pgcli-upstream/4.1.0.zip
Adding upstream version 4.1.0.upstream/4.1.0upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tests/test_ssh_tunnel.py')
-rw-r--r--tests/test_ssh_tunnel.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/test_ssh_tunnel.py b/tests/test_ssh_tunnel.py
index ae865f4..983212b 100644
--- a/tests/test_ssh_tunnel.py
+++ b/tests/test_ssh_tunnel.py
@@ -6,7 +6,7 @@ from configobj import ConfigObj
from click.testing import CliRunner
from sshtunnel import SSHTunnelForwarder
-from pgcli.main import cli, PGCli
+from pgcli.main import cli, notify_callback, PGCli
from pgcli.pgexecute import PGExecute
@@ -61,6 +61,7 @@ def test_ssh_tunnel(
"127.0.0.1",
pgcli.ssh_tunnel.local_bind_ports[0],
"",
+ notify_callback,
)
mock_ssh_tunnel_forwarder.reset_mock()
mock_pgexecute.reset_mock()
@@ -96,6 +97,7 @@ def test_ssh_tunnel(
"127.0.0.1",
pgcli.ssh_tunnel.local_bind_ports[0],
"",
+ notify_callback,
)
mock_ssh_tunnel_forwarder.reset_mock()
mock_pgexecute.reset_mock()