From d8bb5d2efa9f2df09421e77040edd0caa2643cec Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sun, 17 Jul 2016 11:48:19 -0700 Subject: Always encode size with MessageSet --- test/test_protocol.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'test') diff --git a/test/test_protocol.py b/test/test_protocol.py index 2b52f48..1c9f0f9 100644 --- a/test/test_protocol.py +++ b/test/test_protocol.py @@ -72,8 +72,7 @@ def test_encode_message_set(): Message(b'v2', key=b'k2') ] encoded = MessageSet.encode([(0, msg.encode()) - for msg in messages], - size=False) + for msg in messages]) expect = b''.join([ struct.pack('>q', 0), # MsgSet Offset struct.pack('>i', 18), # Msg Size @@ -93,6 +92,7 @@ def test_encode_message_set(): struct.pack('>i', 2), # Length of value b'v2', # Value ]) + expect = struct.pack('>i', len(expect)) + expect assert encoded == expect -- cgit v1.2.1