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)
