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