diff options
| author | Dana Powers <dana.powers@gmail.com> | 2016-07-17 14:16:27 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-07-17 14:16:27 -0700 |
| commit | deebefa6abb0c4e2bce47431b592a523fd49732e (patch) | |
| tree | b3530170dada8b1b0cccb3d217dee00da8cf519f /test/test_protocol.py | |
| parent | cd9aa73c8463fa0214d6ee6109c63396e623dbaa (diff) | |
| download | kafka-python-deebefa6abb0c4e2bce47431b592a523fd49732e.tar.gz | |
Always encode size with MessageSet (#771)
Diffstat (limited to 'test/test_protocol.py')
| -rw-r--r-- | test/test_protocol.py | 4 |
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 |