def test_response_eof_received(self):
def connection_made(transport):
transport.close()
self.protocol.connection_made = connection_made
self._run_test(
self.t, [TransportMock.Close(response=TransportMock.ReceiveEof())])
self.assertSequenceEqual(self.protocol.mock_calls, [
unittest.mock.call.eof_received(),
unittest.mock.call.connection_lost(None)
])
def test_response_sequence(self):
def connection_made(transport):
transport.write(b"foo")
self.protocol.connection_made = connection_made
self._run_test(self.t, [
TransportMock.Write(b"foo",
response=[
TransportMock.Receive(b"foo"),
TransportMock.ReceiveEof()
])
])
self.assertSequenceEqual(self.protocol.mock_calls, [
unittest.mock.call.data_received(b"foo"),
unittest.mock.call.eof_received(),
unittest.mock.call.connection_lost(None),
])
