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()])