openstack/deb-python-kafka
Code Issues Proposed changes

Add SimpleBufferPool test to verify reallocated buffers are empty

Browse Source
This commit is contained in:
Dana Powers
2016-03-13 13:40:41 -07:00
parent bd5bd62b09
commit a03c7ace89
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
test/test_producer.py
View File

@@ -3,10 +3,23 @@ import sys
import pytest import pytest
from kafka import KafkaConsumer, KafkaProducer from kafka import KafkaConsumer, KafkaProducer
from kafka.producer.buffer import SimpleBufferPool
from test.conftest import version from test.conftest import version
from test.testutil import random_string from test.testutil import random_string
def test_buffer_pool():
pool = SimpleBufferPool(1000, 1000)
buf1 = pool.allocate(1000, 1000)
message = ''.join(map(str, range(100)))
buf1.write(message.encode('utf-8'))
pool.deallocate(buf1)
buf2 = pool.allocate(1000, 1000)
assert buf2.read() == b''
@pytest.mark.skipif(not version(), reason="No KAFKA_VERSION set") @pytest.mark.skipif(not version(), reason="No KAFKA_VERSION set")
@pytest.mark.parametrize("compression", [None, 'gzip', 'snappy', 'lz4']) @pytest.mark.parametrize("compression", [None, 'gzip', 'snappy', 'lz4'])
def test_end_to_end(kafka_broker, compression): def test_end_to_end(kafka_broker, compression):