Skip to content

Commit

Permalink
Merge branch 'issue5308' of github.com:spacewasp/mitmproxy into issue…
Browse files Browse the repository at this point in the history
…5308
  • Loading branch information
spacewasp committed Apr 12, 2024
2 parents ec581e5 + 438ed4e commit 3b32845
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions test/mitmproxy/proxy/test_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,27 +16,30 @@ def initialize_command() -> commands.OpenConnection:
)
return c


def server_connect_error_cb(hook_data: server_hooks.ServerConnectionHookData):
global hook_triggered
hook_triggered = True


def mock_simple_connection_handler_init(self):
self.client = Client(peername=("client", 1234), sockname=("server", 5678))
self.hook_handlers = {
server_hooks.ServerConnectErrorHook.name: server_connect_error_cb
}
}
self.max_conns = collections.defaultdict(lambda: asyncio.Semaphore(5))
self.timeout_watchdog = server.TimeoutWatchdog(0, lambda x: None)


def test_server_connect_error(monkeypatch):
global hook_triggered
hook_triggered = False

monkeypatch.setattr(
server.SimpleConnectionHandler, "__init__", mock_simple_connection_handler_init
)

s = server.SimpleConnectionHandler()
asyncio.run(s.open_connection(initialize_command()))

assert hook_triggered

0 comments on commit 3b32845

Please sign in to comment.