diff options
author | Taras Voinarovskyi <voyn1991@gmail.com> | 2017-10-14 23:06:27 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-10-14 23:06:27 +0300 |
commit | fbbd6ca5d999a8520d483ecfe0ad6f805eb8833f (patch) | |
tree | 52e5860b1f8738b15e7c757c205961b761badd2b /test/test_buffer.py | |
parent | dd8e33654f2270097d6c1373dc272153670e48f8 (diff) | |
parent | 365cae02da59721df77923bb5f5a2d94a84b2e83 (diff) | |
download | kafka-python-fbbd6ca5d999a8520d483ecfe0ad6f805eb8833f.tar.gz |
Merge pull request #1252 from dpkp/legacy_records_refactor
Refactor MessageSet and Message into LegacyRecordBatch
Diffstat (limited to 'test/test_buffer.py')
-rw-r--r-- | test/test_buffer.py | 72 |
1 files changed, 0 insertions, 72 deletions
diff --git a/test/test_buffer.py b/test/test_buffer.py deleted file mode 100644 index db6cbb3..0000000 --- a/test/test_buffer.py +++ /dev/null @@ -1,72 +0,0 @@ -# pylint: skip-file -from __future__ import absolute_import - -import io -import platform - -import pytest - -from kafka.producer.buffer import MessageSetBuffer -from kafka.protocol.message import Message, MessageSet - - -def test_buffer_close(): - records = MessageSetBuffer(io.BytesIO(), 100000) - orig_msg = Message(b'foobar') - records.append(1234, orig_msg) - records.close() - - msgset = MessageSet.decode(records.buffer()) - assert len(msgset) == 1 - (offset, size, msg) = msgset[0] - assert offset == 1234 - assert msg == orig_msg - - # Closing again should work fine - records.close() - - msgset = MessageSet.decode(records.buffer()) - assert len(msgset) == 1 - (offset, size, msg) = msgset[0] - assert offset == 1234 - assert msg == orig_msg - - -@pytest.mark.parametrize('compression', [ - 'gzip', - 'snappy', - pytest.mark.skipif(platform.python_implementation() == 'PyPy', - reason='python-lz4 crashes on older versions of pypy')('lz4'), -]) -def test_compressed_buffer_close(compression): - records = MessageSetBuffer(io.BytesIO(), 100000, compression_type=compression) - orig_msg = Message(b'foobar') - records.append(1234, orig_msg) - records.close() - - msgset = MessageSet.decode(records.buffer()) - assert len(msgset) == 1 - (offset, size, msg) = msgset[0] - assert offset == 0 - assert msg.is_compressed() - - msgset = msg.decompress() - (offset, size, msg) = msgset[0] - assert not msg.is_compressed() - assert offset == 1234 - assert msg == orig_msg - - # Closing again should work fine - records.close() - - msgset = MessageSet.decode(records.buffer()) - assert len(msgset) == 1 - (offset, size, msg) = msgset[0] - assert offset == 0 - assert msg.is_compressed() - - msgset = msg.decompress() - (offset, size, msg) = msgset[0] - assert not msg.is_compressed() - assert offset == 1234 - assert msg == orig_msg |