def test_data_received(loop): proto = parsers.StreamProtocol(loop=loop) proto.connection_made(mock.Mock()) proto.reader = mock.Mock() proto.data_received(b'data') proto.reader.feed_data.assert_called_with(b'data')
def test_connection_lost(loop): proto = parsers.StreamProtocol(loop=loop) proto.connection_made(mock.Mock()) proto.connection_lost(None) assert proto.transport is None assert proto.writer is None assert proto.reader._eof
def test_connection_lost_exc(loop): proto = parsers.StreamProtocol(loop=loop) proto.connection_made(mock.Mock()) exc = ValueError() proto.connection_lost(exc) assert proto.reader.exception() is exc
def test_connection_lost_exc(self): proto = parsers.StreamProtocol(loop=self.loop) proto.connection_made(unittest.mock.Mock()) exc = ValueError() proto.connection_lost(exc) self.assertIs(proto.reader.exception(), exc)
def test_connection_lost(self): proto = parsers.StreamProtocol(loop=self.loop) proto.connection_made(unittest.mock.Mock()) proto.connection_lost(None) self.assertIsNone(proto.transport) self.assertIsNone(proto.writer) self.assertTrue(proto.reader._eof)
def test_data_received(self): proto = parsers.StreamProtocol() proto.connection_made(unittest.mock.Mock()) proto.reader = unittest.mock.Mock() proto.data_received(b'data') proto.reader.feed_data.assert_called_with(b'data')
def test_connection_made(loop): tr = mock.Mock() proto = parsers.StreamProtocol(loop=loop) assert proto.transport is None proto.connection_made(tr) assert proto.transport is tr
def test_connection_made(self): tr = unittest.mock.Mock() proto = parsers.StreamProtocol(loop=self.loop) self.assertIsNone(proto.transport) proto.connection_made(tr) self.assertIs(proto.transport, tr)
def test_drain_waiter(self): proto = parsers.StreamProtocol(loop=unittest.mock.Mock()) proto._paused = False self.assertEqual(proto._make_drain_waiter(), ()) proto._paused = True fut = proto._make_drain_waiter() self.assertIsInstance(fut, asyncio.Future) fut2 = proto._make_drain_waiter() self.assertIs(fut, fut2)