Correct type signatures in MessageHandler

2023-12-15 19:07:17 +00:00
parent c768aeaf40
commit 4cc1513e58
3 changed files with 6 additions and 5 deletions
--- a/tests/client/test_hippo_client.py
+++ b/tests/client/test_hippo_client.py
@@ -105,7 +105,7 @@ class TestHippoClient(unittest.IsolatedAsyncioTestCase):
    }

    async def asyncSetUp(self):
-        self.server_handler = MessageHandler()
+        self.server_handler: MessageHandler[Message, str] = MessageHandler()
        self.server_transport = PacketForwardingTransport()
        self.server_circuit = Circuit(("127.0.0.1", 2), ("127.0.0.1", 99), self.server_transport)
        self.server = MockServer(self.server_circuit, self.server_handler)