def _write_message(chan, data): nets = Netstring.compose(data) remaining = len(nets) while remaining > 0: n = chan.send(nets) if n <= 0: raise BastioEOFError("channel closed") remaining -= n
def test_netstring(self): data = 'hello world' enc = Netstring.compose(data) res = Netstring.parse(enc) self.assertEqual(data, res) buf = StringIO.StringIO(enc) buf.recv = buf.read net = Netstring(buf) res = net.recv() self.assertEqual(data, res)