From c48f5c07fde2ea6ca8d058e9a65a32b82fcebec9 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Thu, 9 Mar 2017 12:22:22 -0800 Subject: Add tests for Struct kwarg behavior --- test/test_protocol.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/test/test_protocol.py b/test/test_protocol.py index 1c9f0f9..aa3dd17 100644 --- a/test/test_protocol.py +++ b/test/test_protocol.py @@ -7,8 +7,9 @@ import six from kafka.protocol.api import RequestHeader from kafka.protocol.commit import GroupCoordinatorRequest -from kafka.protocol.fetch import FetchResponse +from kafka.protocol.fetch import FetchRequest, FetchResponse from kafka.protocol.message import Message, MessageSet, PartialMessage +from kafka.protocol.metadata import MetadataRequest from kafka.protocol.types import Int16, Int32, Int64, String @@ -244,3 +245,16 @@ def test_decode_fetch_response_partial(): m1 = partitions[0][3] assert len(m1) == 2 assert m1[1] == (None, None, PartialMessage()) + + +def test_struct_unrecognized_kwargs(): + try: + mr = MetadataRequest[0](topicz='foo') + assert False, 'Structs should not allow unrecognized kwargs' + except ValueError: + pass + + +def test_struct_missing_kwargs(): + fr = FetchRequest[0](max_wait_time=100) + assert fr.min_bytes is None -- cgit v1.2.1