summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorDana Powers <dana.powers@gmail.com>2016-07-17 11:48:19 -0700
committerDana Powers <dana.powers@gmail.com>2016-07-17 12:55:42 -0700
commitd8bb5d2efa9f2df09421e77040edd0caa2643cec (patch)
treeb3530170dada8b1b0cccb3d217dee00da8cf519f /test
parentcd9aa73c8463fa0214d6ee6109c63396e623dbaa (diff)
downloadkafka-python-message_set_encode_size.tar.gz
Always encode size with MessageSetmessage_set_encode_size
Diffstat (limited to 'test')
-rw-r--r--test/test_protocol.py4
1 files changed, 2 insertions, 2 deletions
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