From 48b6c81b36c04c9c3e6944e2a405a26c56a05d6e Mon Sep 17 00:00:00 2001 From: shortcutme Date: Sat, 16 Mar 2019 00:56:25 +0100 Subject: [PATCH] Test msgpack streaming with binary data --- src/Test/TestMsgpack.py | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/src/Test/TestMsgpack.py b/src/Test/TestMsgpack.py index abbdaade..59a7d11d 100644 --- a/src/Test/TestMsgpack.py +++ b/src/Test/TestMsgpack.py @@ -36,16 +36,25 @@ class TestMsgpack: assert messages[0] == messages[1] def testStreaming(self): - f = StreamingMsgpack.FilePart("%s/users.json" % config.data_dir) - f.read_bytes = 10 + bin_data = os.urandom(20) + f = Msgpack.FilePart("%s/users.json" % config.data_dir, "rb") + f.read_bytes = 30 - data = {"cmd": "response", "params": f} + data = {"cmd": "response", "body": f, "bin": bin_data} out_buff = io.BytesIO() Msgpack.stream(data, out_buff.write) out_buff.seek(0) - data_packb = {"cmd": "response", "params": open("%s/users.json" % config.data_dir).read(10)} + data_packb = { + "cmd": "response", + "body": open("%s/users.json" % config.data_dir, "rb").read(30), + "bin": bin_data + } out_buff.seek(0) - assert msgpack.unpackb(out_buff.read()) == data_packb + data_unpacked = Msgpack.unpack(out_buff.read()) + assert data_unpacked == data_packb + assert data_unpacked["cmd"] == "response" + assert type(data_unpacked["body"]) == bytes +