def test__acquire_has_no_consumer(self, mock_consumer, cluster,
example_partitions):
config = KafkaConsumerConfig(self.group, cluster)
consumer = KafkaConsumerGroup([], config)
consumer._acquire(example_partitions)
mock_consumer.assert_called_once_with(example_partitions,
**consumer.config)
def test__acquire_has_consumer(self, cluster, example_partitions,
mock_post_rebalance_cb):
config = KafkaConsumerConfig(
self.group,
cluster,
post_rebalance_callback=mock_post_rebalance_cb)
consumer = KafkaConsumerGroup([], config)
consumer.consumer = mock.Mock()
consumer._acquire(example_partitions)
consumer.consumer.set_topic_partitions.assert_called_once_with(
example_partitions)
mock_post_rebalance_cb.assert_called_once_with(example_partitions)