def test_write_eof(self):
def connection_made(transport):
transport.write_eof()
self.protocol.connection_made = connection_made
self._run_test(self.t, [TransportMock.WriteEof()])
def test_partial(self):
def connection_made(transport):
transport.write(b"foo")
self.protocol.connection_made = connection_made
self._run_test(
self.t,
[
TransportMock.Write(
b"foo",
),
],
partial=True
)
self.t.write_eof()
self.t.close()
self._run_test(
self.t,
[
TransportMock.WriteEof(),
TransportMock.Close()
]
)
def test_catch_unexpected_abort(self):
def connection_made(transport):
transport.abort()
self.protocol.connection_made = connection_made
with self.assertRaisesRegex(AssertionError, "unexpected abort"):
self._run_test(self.t, [TransportMock.WriteEof()])