Skip to content

Commit

Permalink
Minor test fix
Browse files Browse the repository at this point in the history
  • Loading branch information
dkmstr committed Mar 6, 2024
1 parent 6f8bf4f commit cc36366
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/test_tunnel.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,7 +121,9 @@ def test_forward_fnc(self) -> None:
"""
with mock.patch('uds.tunnel._run') as run:
with mock.patch('uds.tunnel.ForwardServer') as ForwardServer:
fs = tunnel.forward(('localhost', 1234), fixtures.TICKET, 1, 1222, check_certificate=False, use_ipv6=False)
fs = tunnel.forward(
('localhost', 1234), fixtures.TICKET, 1, 1222, check_certificate=False, use_ipv6=False
)
# Ensure that thread is invoked with _run as target, and fs as argument
run.assert_called_once_with(fs)
# And that ForwardServer is called with the correct parameters
Expand All @@ -135,8 +137,6 @@ def test_forward_fnc(self) -> None:
keep_listening=True,
)

# Ensure server is stopped, we have not used it.. send some data to make it fail an stop

def test_forward_stoppable(self) -> None:
# Patch fs._set_stoppable to check if it is called
with mock.patch('uds.tunnel.ForwardServer._set_stoppable') as _set_stoppable:
Expand Down

0 comments on commit cc36366

Please sign in to comment.