From 5cbd4d75e5ad271e5981bda79e46e08c1f59d8a3 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Sat, 12 Jul 2025 21:57:25 +0100 Subject: [PATCH] Rev gateway tests --- .../test_rev_gateway/test_rev_gateway_service.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/tests/unit/test_rev_gateway/test_rev_gateway_service.py b/tests/unit/test_rev_gateway/test_rev_gateway_service.py index 6d87a2e0..26cab94a 100644 --- a/tests/unit/test_rev_gateway/test_rev_gateway_service.py +++ b/tests/unit/test_rev_gateway/test_rev_gateway_service.py @@ -308,11 +308,11 @@ class TestReverseGateway: mock_msg.type = WSMsgType.TEXT mock_msg.data = '{"test": "message"}' - # Mock receive to return message once, then close + # Mock receive to return message once, then raise exception to stop loop mock_ws.receive.side_effect = [mock_msg, Exception("Test stop")] - with pytest.raises(Exception, match="Test stop"): - await gateway.listen() + # listen() catches exceptions and breaks, so no exception should be raised + await gateway.listen() gateway.handle_message.assert_called_once_with('{"test": "message"}') @@ -341,11 +341,11 @@ class TestReverseGateway: mock_msg.type = WSMsgType.BINARY mock_msg.data = b'{"test": "binary"}' - # Mock receive to return message once, then close + # Mock receive to return message once, then raise exception to stop loop mock_ws.receive.side_effect = [mock_msg, Exception("Test stop")] - with pytest.raises(Exception, match="Test stop"): - await gateway.listen() + # listen() catches exceptions and breaks, so no exception should be raised + await gateway.listen() gateway.handle_message.assert_called_once_with('{"test": "binary"}') @@ -462,7 +462,8 @@ class TestReverseGatewayRun: mock_config_receiver_instance.start.assert_called_once() gateway.listen.assert_called_once() - gateway.disconnect.assert_called_once() + # disconnect is called twice: once in the main loop, once in shutdown + assert gateway.disconnect.call_count == 2 gateway.shutdown.assert_called_once()