From 37bb5c99e085364cd71bbb44b225f0eb37a7aac7 Mon Sep 17 00:00:00 2001 From: Tomasz Pietruszkiewicz Date: Mon, 12 Apr 2021 14:33:41 +0200 Subject: Fix Kafka consumer Change-Id: I8b52e2d1859b964a582289e2f63d70272d60f39a Issue-ID: INT-1869 Signed-off-by: Tomasz Pietruszkiewicz --- src/python/tests/unit/test_netconf_kafka_client.py | 21 ++++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) (limited to 'src/python/tests') diff --git a/src/python/tests/unit/test_netconf_kafka_client.py b/src/python/tests/unit/test_netconf_kafka_client.py index 9eff761..b3f45c1 100644 --- a/src/python/tests/unit/test_netconf_kafka_client.py +++ b/src/python/tests/unit/test_netconf_kafka_client.py @@ -32,8 +32,10 @@ class TestNetconfKafkaClient(TestCase): def setUp(self): self.producer = MagicMock() + self.kafkaConsumerResponse = KafkaConsumerResponse(MagicMock(value=MESSAGE_1), MagicMock(value=MESSAGE_2)) + self.kafkaConsumerResponse.close = MagicMock() self.kafka_customer_func = MagicMock( - return_value=[MagicMock(value=MESSAGE_1), MagicMock(value=MESSAGE_2)] + return_value=self.kafkaConsumerResponse ) self.test_obj = NetconfKafkaClient( producer=self.producer, @@ -58,3 +60,20 @@ class TestNetconfKafkaClient(TestCase): self.assertTrue(len(messages) == 2) self.assertTrue(MESSAGE_1 in messages) self.assertTrue(MESSAGE_2 in messages) + + +class KafkaConsumerResponse(list): + + def __new__(self, *args, **kwargs): + return super(KafkaConsumerResponse, self).__new__(self, args, kwargs) + + def __init__(self, *args, **kwargs): + if len(args) == 1 and hasattr(args[0], '__iter__'): + list.__init__(self, args[0]) + else: + list.__init__(self, args) + self.__dict__.update(kwargs) + + def __call__(self, **kwargs): + self.__dict__.update(kwargs) + return self -- cgit 1.2.3-korg