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)
