From 1a75b1e0b8c1f62b90f533243466f54e6ce42ee8 Mon Sep 17 00:00:00 2001 From: Vladan Popovic Date: Sun, 6 Sep 2020 01:58:38 +0200 Subject: [PATCH] Use the producer and collector properly --- tests/test_producer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_producer.py b/tests/test_producer.py index 6f61bec..e71f585 100644 --- a/tests/test_producer.py +++ b/tests/test_producer.py @@ -21,7 +21,7 @@ async def test_producer_called(config, event_loop): producer.producer = Mock() - task = event_loop.create_task(producer.produce()) + task = event_loop.create_task(producer()) await asyncio.sleep(0) producer.producer.send_and_wait.assert_called_with( config.kafka.topic, bytes(check.json().encode('utf-8'))) @@ -41,7 +41,7 @@ async def test_producer_called_invalid(config, event_loop): producer.producer = Mock() - task = event_loop.create_task(producer.produce()) + task = event_loop.create_task(producer()) await asyncio.sleep(0) producer.logger.error.assert_called() assert task.done()