diff options
Diffstat (limited to 'test/test_fetcher.py')
-rw-r--r-- | test/test_fetcher.py | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/test/test_fetcher.py b/test/test_fetcher.py index bf4a3a9..7e529bc 100644 --- a/test/test_fetcher.py +++ b/test/test_fetcher.py @@ -15,7 +15,7 @@ from kafka.structs import TopicPartition, OffsetAndMetadata @pytest.fixture def client(mocker): - return mocker.Mock(spec=KafkaClient) + return mocker.Mock(spec=KafkaClient(bootstrap_servers=[])) @pytest.fixture @@ -71,6 +71,19 @@ def test_init_fetches(fetcher, mocker): assert len(ret) == len(fetch_requests) +@pytest.mark.parametrize(("api_version", "fetch_version"), [ + ((0, 10), 2), + ((0, 9), 1), + ((0, 8), 0) +]) +def test_create_fetch_requests(fetcher, mocker, api_version, fetch_version): + fetcher._client.in_flight_request_count.return_value = 0 + fetcher.config['api_version'] = api_version + by_node = fetcher._create_fetch_requests() + requests = by_node.values() + assert all([isinstance(r, FetchRequest[fetch_version]) for r in requests]) + + def test_update_fetch_positions(fetcher, mocker): mocker.patch.object(fetcher, '_reset_offset') partition = TopicPartition('foobar', 0) |