def test_blob(self): msg = Msg("hello", blob.StringBlob("world")) actual = rpc.serialize(msg) actual = actual.read_all() actual = twistit.extract(actual) pickleout = pickle.dumps(msg, pickle.HIGHEST_PROTOCOL) header = rpc.HEADER.pack(len(pickleout), 5) expected = header + pickleout + "world" self.assertEqual(expected, actual)
def test_noblob(self): msg = Msg("hello", None) actual = rpc.serialize(msg) actual = actual.read_all() actual = twistit.extract(actual) pickleout = pickle.dumps(msg, pickle.HIGHEST_PROTOCOL) header = rpc.HEADER.pack(len(pickleout), 0) expected = header + pickleout self.assertEqual(expected, actual)
def test_blob_stream(self): b = blob.PassthroughBlob(6) msg = Msg("hello", b) actual = rpc.serialize(msg) actual = actual.read_all() b.write("world!") b.done() actual = twistit.extract(actual) pickleout = pickle.dumps(msg, pickle.HIGHEST_PROTOCOL) header = rpc.HEADER.pack(len(pickleout), 6) expected = header + pickleout + "world!" self.assertEqual(expected, actual)