try: from unittest import mock except ImportError: import mock import mitogen.core import testlib class ReceiveOneTest(testlib.TestCase): klass = mitogen.core.MitogenProtocol def test_corruption(self): broker = mock.Mock() router = mock.Mock() stream = mock.Mock() protocol = self.klass(router, 1) protocol.stream = stream junk = mitogen.core.b('x') * mitogen.core.Message.HEADER_LEN capture = testlib.LogCapturer() capture.start() protocol.on_receive(broker, junk) capture.stop() self.assertEqual(1, stream.on_disconnect.call_count) expect = self.klass.corrupt_msg % (stream.name, junk) self.assertIn(expect, capture.raw())