如何確保在循環中調用所有處理程序?我想出的最好方法是模擬fake_handler函數并檢查該函數是否被調用了一定次數,但我認為可能有更好的解決方案。class MessageHandler: def __init__(self, handlers=None): self.handlers = handlers or [] def handle(self, event, body): for handler in self.handlers: handler(event, body)測試:def fake_handler(*args, **kwargs): passclass TestMessageHandler(TestCase): @patch('tests.test_handlers.fake_handler') def test_handle(self, fake_handler_mock): messages = MessageHandler([fake_handler_mock, fake_handler_mock]) messages.handle(None, None) self.assertEqual(fake_handler_mock.call_count, 2)
添加回答
舉報
0/150
提交
取消