def _report_status(self, status: List[Tuple[bytes, bytes]]) -> None:
if self.has_capability(CAPABILITY_SIDE_BAND_64K):
writer = BufferedPktLineWriter(
lambda d: self.proto.write_sideband(SIDE_BAND_CHANNEL_DATA, d))
write = writer.write
def flush():
writer.flush()
self.proto.write_pkt_line(None)
else:
write = self.proto.write_pkt_line
def flush():
pass
for name, msg in status:
if name == b"unpack":
write(b"unpack " + msg + b"\n")
elif msg == b"ok":
write(b"ok " + name + b"\n")
else:
write(b"ng " + name + b" " + msg + b"\n")
write(None)
flush()
def _report_status(self, status):
if self.has_capability(CAPABILITY_SIDE_BAND_64K):
writer = BufferedPktLineWriter(
lambda d: self.proto.write_sideband(SIDE_BAND_CHANNEL_DATA, d))
write = writer.write
def flush():
writer.flush()
self.proto.write_pkt_line(None)
else:
write = self.proto.write_pkt_line
flush = lambda: None
for name, msg in status:
if name == b'unpack':
write(b'unpack ' + msg + b'\n')
elif msg == b'ok':
write(b'ok ' + name + b'\n')
else:
write(b'ng ' + name + b' ' + msg + b'\n')
write(None)
flush()
def _report_status(self, status):
if self.has_capability('side-band-64k'):
writer = BufferedPktLineWriter(
lambda d: self.proto.write_sideband(1, d))
write = writer.write
def flush():
writer.flush()
self.proto.write_pkt_line(None)
else:
write = self.proto.write_pkt_line
flush = lambda: None
for name, msg in status:
if name == 'unpack':
write('unpack %s\n' % msg)
elif msg == 'ok':
write('ok %s\n' % name)
else:
write('ng %s %s\n' % (name, msg))
write(None)
flush()
def setUp(self):
TestCase.setUp(self)
self._output = BytesIO()
self._writer = BufferedPktLineWriter(self._output.write, bufsize=16)
class BufferedPktLineWriterTests(TestCase):
def setUp(self):
TestCase.setUp(self)
self._output = BytesIO()
self._writer = BufferedPktLineWriter(self._output.write, bufsize=16)
def assertOutputEquals(self, expected):
self.assertEqual(expected, self._output.getvalue())
def _truncate(self):
self._output.seek(0)
self._output.truncate()
def test_write(self):
self._writer.write(b'foo')
self.assertOutputEquals(b'')
self._writer.flush()
self.assertOutputEquals(b'0007foo')
def test_write_none(self):
self._writer.write(None)
self.assertOutputEquals(b'')
self._writer.flush()
self.assertOutputEquals(b'0000')
def test_flush_empty(self):
self._writer.flush()
self.assertOutputEquals(b'')
def test_write_multiple(self):
self._writer.write(b'foo')
self._writer.write(b'bar')
self.assertOutputEquals(b'')
self._writer.flush()
self.assertOutputEquals(b'0007foo0007bar')
def test_write_across_boundary(self):
self._writer.write(b'foo')
self._writer.write(b'barbaz')
self.assertOutputEquals(b'0007foo000abarba')
self._truncate()
self._writer.flush()
self.assertOutputEquals(b'z')
def test_write_to_boundary(self):
self._writer.write(b'foo')
self._writer.write(b'barba')
self.assertOutputEquals(b'0007foo0009barba')
self._truncate()
self._writer.write(b'z')
self._writer.flush()
self.assertOutputEquals(b'0005z')
def setUp(self):
TestCase.setUp(self)
self._output = StringIO()
self._writer = BufferedPktLineWriter(self._output.write, bufsize=16)
class BufferedPktLineWriterTests(TestCase):
def setUp(self):
TestCase.setUp(self)
self._output = StringIO()
self._writer = BufferedPktLineWriter(self._output.write, bufsize=16)
def assertOutputEquals(self, expected):
self.assertEquals(expected, self._output.getvalue())
def _truncate(self):
self._output.seek(0)
self._output.truncate()
def test_write(self):
self._writer.write('foo')
self.assertOutputEquals('')
self._writer.flush()
self.assertOutputEquals('0007foo')
def test_write_none(self):
self._writer.write(None)
self.assertOutputEquals('')
self._writer.flush()
self.assertOutputEquals('0000')
def test_flush_empty(self):
self._writer.flush()
self.assertOutputEquals('')
def test_write_multiple(self):
self._writer.write('foo')
self._writer.write('bar')
self.assertOutputEquals('')
self._writer.flush()
self.assertOutputEquals('0007foo0007bar')
def test_write_across_boundary(self):
self._writer.write('foo')
self._writer.write('barbaz')
self.assertOutputEquals('0007foo000abarba')
self._truncate()
self._writer.flush()
self.assertOutputEquals('z')
def test_write_to_boundary(self):
self._writer.write('foo')
self._writer.write('barba')
self.assertOutputEquals('0007foo0009barba')
self._truncate()
self._writer.write('z')
self._writer.flush()
self.assertOutputEquals('0005z')
class BufferedPktLineWriterTests(TestCase):
def setUp(self):
TestCase.setUp(self)
self._output = BytesIO()
self._writer = BufferedPktLineWriter(self._output.write, bufsize=16)
def assertOutputEquals(self, expected):
self.assertEqual(expected, self._output.getvalue())
def _truncate(self):
self._output.seek(0)
self._output.truncate()
def test_write(self):
self._writer.write(b"foo")
self.assertOutputEquals(b"")
self._writer.flush()
self.assertOutputEquals(b"0007foo")
def test_write_none(self):
self._writer.write(None)
self.assertOutputEquals(b"")
self._writer.flush()
self.assertOutputEquals(b"0000")
def test_flush_empty(self):
self._writer.flush()
self.assertOutputEquals(b"")
def test_write_multiple(self):
self._writer.write(b"foo")
self._writer.write(b"bar")
self.assertOutputEquals(b"")
self._writer.flush()
self.assertOutputEquals(b"0007foo0007bar")
def test_write_across_boundary(self):
self._writer.write(b"foo")
self._writer.write(b"barbaz")
self.assertOutputEquals(b"0007foo000abarba")
self._truncate()
self._writer.flush()
self.assertOutputEquals(b"z")
def test_write_to_boundary(self):
self._writer.write(b"foo")
self._writer.write(b"barba")
self.assertOutputEquals(b"0007foo0009barba")
self._truncate()
self._writer.write(b"z")
self._writer.flush()
self.assertOutputEquals(b"0005z")