diff options
| author | Taras Voinarovskiy <voyn1991@gmail.com> | 2017-08-25 09:38:27 +0000 |
|---|---|---|
| committer | Taras Voinarovskiy <voyn1991@gmail.com> | 2017-10-24 15:12:11 +0000 |
| commit | 3af66bc542efff3f7010bec18b72d844e09488c4 (patch) | |
| tree | a20623631b36230c9425b08ee95b85afbc9a9455 /test/test_producer.py | |
| parent | e06af5343a55cf8d03e32a645ee970d872cb9ba0 (diff) | |
| download | kafka-python-v2_records.tar.gz | |
Add DefaultRecordBatch implementation aka V2 message format parser/builder.v2_records
Added bytecode optimization for varint and append/read_msg functions. Mostly based on avoiding LOAD_GLOBAL calls.
Diffstat (limited to 'test/test_producer.py')
| -rw-r--r-- | test/test_producer.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/test/test_producer.py b/test/test_producer.py index 41bd52e..20dffc2 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -88,10 +88,7 @@ def test_kafka_producer_proper_record_metadata(kafka_broker, compression): retries=5, max_block_ms=10000, compression_type=compression) - if producer.config['api_version'] >= (0, 10): - magic = 1 - else: - magic = 0 + magic = producer._max_usable_produce_magic() topic = random_string(5) future = producer.send( @@ -109,7 +106,9 @@ def test_kafka_producer_proper_record_metadata(kafka_broker, compression): else: assert record.timestamp == -1 # NO_TIMESTAMP - if magic == 1: + if magic >= 2: + assert record.checksum is None + elif magic == 1: assert record.checksum == 1370034956 else: assert record.checksum == 3296137851 |