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)
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 _read_message(chan):
nets = Netstring(chan)
return nets.recv()
def _read_message(self):
nets = Netstring(self._chan)
return nets.recv()
