32 lines
754 B
Python
32 lines
754 B
Python
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())
|