From c856b5e7fcb51dfeb0dc2f070400ef720261fd07 Mon Sep 17 00:00:00 2001 From: Salad Dais Date: Mon, 31 May 2021 10:25:31 +0000 Subject: [PATCH] Add test for post-filtering logged messages --- tests/proxy/integration/test_lludp.py | 24 +++++++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) diff --git a/tests/proxy/integration/test_lludp.py b/tests/proxy/integration/test_lludp.py index 9f038c4..0f65287 100644 --- a/tests/proxy/integration/test_lludp.py +++ b/tests/proxy/integration/test_lludp.py @@ -208,16 +208,34 @@ class LLUDPIntegrationTests(BaseProxyTest): self.assertEqual(len(entries), 1) self.assertEqual(entries[0].name, "ObjectUpdateCompressed") - async def test_logging_taken_message(self): + async def test_filtering_logged_messages(self): message_logger = SimpleMessageLogger() self.session_manager.message_logger = message_logger self._setup_default_circuit() - obj_update = self.serializer.serialize(ProxiedMessage( + obj_update = self._make_objectupdate_compressed(1234) + self.protocol.datagram_received(obj_update, self.region_addr) + msg = self.serializer.serialize(ProxiedMessage( "UndoLand", Block("AgentData", AgentID=UUID(), SessionID=UUID()), direction=Direction.IN, )) - self.protocol.datagram_received(obj_update, self.region_addr) + self.protocol.datagram_received(msg, self.region_addr) + await self._wait_drained() + message_logger.set_filter("ObjectUpdateCompressed") + entries = message_logger.entries + self.assertEqual(len(entries), 1) + self.assertEqual(entries[0].name, "ObjectUpdateCompressed") + + async def test_logging_taken_message(self): + message_logger = SimpleMessageLogger() + self.session_manager.message_logger = message_logger + self._setup_default_circuit() + msg = self.serializer.serialize(ProxiedMessage( + "UndoLand", + Block("AgentData", AgentID=UUID(), SessionID=UUID()), + direction=Direction.IN, + )) + self.protocol.datagram_received(msg, self.region_addr) await self._wait_drained() entries = message_logger.entries self.assertEqual(len(entries), 1)