def test_noblob(self): expected = Msg("hello", None) pickleout = pickle.dumps(expected, pickle.HIGHEST_PROTOCOL) header = rpc.HEADER.pack(len(pickleout), 0) data = header + pickleout src = blob.StringBlob(data) d = rpc.deserialize(src) actual = twistit.extract(d) self.assertEqual(expected, actual)
def test_blob(self): expected = Msg("hello", None) pickleout = pickle.dumps(expected, pickle.HIGHEST_PROTOCOL) header = rpc.HEADER.pack(len(pickleout), 6) data = header + pickleout + "World!" src = blob.StringBlob(data) d = rpc.deserialize(src) actual = twistit.extract(d) payload = twistit.extract(actual.blob.read_all()) self.assertEqual(expected, actual) self.assertEqual("World!", payload)
def test_blob_stream(self): expected = Msg("hello", None) pickleout = pickle.dumps(expected, pickle.HIGHEST_PROTOCOL) header = rpc.HEADER.pack(len(pickleout), 6) data = header + pickleout src = blob.PassthroughBlob(len(data)+6) d = rpc.deserialize(src) src.write(data) actual = twistit.extract(d) self.assertEqual(expected, actual) src.write("World!") src.done() payload = twistit.extract(actual.blob.read_all()) self.assertEqual("World!", payload)