niklas/tracetools_analysis
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add test for get_handler_by_type()

Browse source
This commit is contained in:
Christophe Bedard 2019-12-31 20:17:26 -05:00
parent 088b555ae0
commit 19652f165e
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tracetools_analysis/test/tracetools_analysis/test_processor.py
View file

@ -159,6 +159,13 @@ class TestProcessor(unittest.TestCase):
# Passes check
Processor(EventHandlerWithRequiredEvent()).process([required_mock_event, mock_event])
def test_get_handler_by_type(self) -> None:
handler1 = StubHandler1()
handler2 = StubHandler2()
processor = Processor(handler1, handler2)
result = processor.get_handler_by_type(StubHandler1)
self.assertTrue(result is handler1)
if __name__ == '__main__':
unittest.main()