From 9ee77dfdbc4aeb5723ce7ebdae76f8b7141962af Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Wed, 28 Sep 2016 12:39:34 -0700 Subject: KAFKA-3007: KafkaConsumer max_poll_records (#831) --- test/test_fetcher.py | 16 ++-------------- 1 file changed, 2 insertions(+), 14 deletions(-) (limited to 'test/test_fetcher.py') diff --git a/test/test_fetcher.py b/test/test_fetcher.py index 6afd547..fea3f7d 100644 --- a/test/test_fetcher.py +++ b/test/test_fetcher.py @@ -33,7 +33,7 @@ def fetcher(client, subscription_state): return Fetcher(client, subscription_state, Metrics()) -def test_init_fetches(fetcher, mocker): +def test_send_fetches(fetcher, mocker): fetch_requests = [ FetchRequest[0]( -1, fetcher.config['fetch_max_wait_ms'], @@ -53,19 +53,7 @@ def test_init_fetches(fetcher, mocker): mocker.patch.object(fetcher, '_create_fetch_requests', return_value = dict(enumerate(fetch_requests))) - fetcher._records.append('foobar') - ret = fetcher.init_fetches() - assert fetcher._create_fetch_requests.call_count == 0 - assert ret == [] - fetcher._records.clear() - - fetcher._iterator = 'foo' - ret = fetcher.init_fetches() - assert fetcher._create_fetch_requests.call_count == 0 - assert ret == [] - fetcher._iterator = None - - ret = fetcher.init_fetches() + ret = fetcher.send_fetches() for node, request in enumerate(fetch_requests): fetcher._client.send.assert_any_call(node, request) assert len(ret) == len(fetch_requests) -- cgit v1.2.1