Untie ConsumerRebalanceListener

aiokafka/abc.py

@@ -1,8 +1,7 @@
 import abc
-from kafka import ConsumerRebalanceListener as BaseConsumerRebalanceListener
 
 
-class ConsumerRebalanceListener(BaseConsumerRebalanceListener):
+class ConsumerRebalanceListener(abc.ABC):
     """
     A callback interface that the user can implement to trigger custom actions
     when the set of partitions assigned to the consumer changes.

aiokafka/consumer/subscription_state.py

@@ -5,7 +5,7 @@
 from asyncio import shield, Event, Future
 from enum import Enum
 
-from typing import Dict, FrozenSet, Iterable, List, Pattern, Set
+from typing import Dict, Iterable, List, Pattern, Set
 
 from aiokafka.errors import IllegalStateError
 from aiokafka.structs import OffsetAndMetadata, TopicPartition

kafka/__init__.py

@@ -20,7 +20,6 @@ def emit(self, record):
 
 from kafka.admin import KafkaAdminClient
 from kafka.client_async import KafkaClient
-from kafka.consumer.subscription_state import ConsumerRebalanceListener
 from kafka.conn import BrokerConnection
 from kafka.serializer import Serializer, Deserializer
 from kafka.structs import TopicPartition, OffsetAndMetadata

tests/kafka/conftest.py

@@ -42,14 +42,6 @@ def factory(**broker_params):
         broker.close()
 
 
-@pytest.fixture
-def kafka_client(kafka_broker, request):
-    """Return a KafkaClient fixture"""
-    (client,) = kafka_broker.get_clients(cnt=1, client_id='%s_client' % (request.node.name,))
-    yield client
-    client.close()
-
-
 @pytest.fixture
 def kafka_consumer(kafka_consumer_factory):
     """Return a KafkaConsumer fixture"""