diff --git a/oslo_messaging/_drivers/impl_kafka.py b/oslo_messaging/_drivers/impl_kafka.py index b448fcdbb..aa060eedc 100644 --- a/oslo_messaging/_drivers/impl_kafka.py +++ b/oslo_messaging/_drivers/impl_kafka.py @@ -347,7 +347,7 @@ class KafkaDriver(base.BaseDriver): with self._get_connection(purpose=PURPOSE_SEND) as conn: conn.notify_send(target_to_topic(target), ctxt, message, retry) - def listen(self, target): + def listen(self, target, batch_size, batch_timeout): raise NotImplementedError( 'The RPC implementation for Kafka is not implemented') diff --git a/oslo_messaging/tests/drivers/test_impl_kafka.py b/oslo_messaging/tests/drivers/test_impl_kafka.py index 4579453a7..33c45d9bd 100644 --- a/oslo_messaging/tests/drivers/test_impl_kafka.py +++ b/oslo_messaging/tests/drivers/test_impl_kafka.py @@ -108,7 +108,8 @@ class TestKafkaDriver(test_utils.BaseTestCase): def test_listen(self): target = oslo_messaging.Target(topic="topic_test") - self.assertRaises(NotImplementedError, self.driver.listen, target) + self.assertRaises(NotImplementedError, self.driver.listen, target, + None, None) class TestKafkaConnection(test_utils.BaseTestCase):