Skip to content

Commit 7536bcb

Browse files
committed
Fixup py27 fetcher test failure
1 parent 590ef93 commit 7536bcb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/test_fetcher.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,8 @@ def test__reset_offsets_async(fetcher, mocker):
140140
fetcher._subscriptions.assign_from_subscribed([tp0, tp1])
141141
fetcher._subscriptions.request_offset_reset(tp0)
142142
fetcher._subscriptions.request_offset_reset(tp1)
143-
mocker.patch.object(fetcher._client.cluster, "leader_for_partition", side_effect=[0, 1])
143+
leaders = {tp0: 0, tp1: 1}
144+
mocker.patch.object(fetcher._client.cluster, "leader_for_partition", side_effect=lambda tp: leaders[tp])
144145
mocker.patch.object(fetcher._client, 'ready', return_value=True)
145146
future1 = Future()
146147
future2 = Future()

0 commit comments

Comments
 (0)