Skip to content

Commit 82340b3

Browse files
committed
fix(aiokafka): use AsyncMock to fix unawaited coroutine warning
1 parent 27f7c2a commit 82340b3

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

  • instrumentation/opentelemetry-instrumentation-aiokafka/tests

instrumentation/opentelemetry-instrumentation-aiokafka/tests/test_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -256,7 +256,7 @@ async def test_wrap_getmany(
256256
extract_bootstrap_servers: mock.MagicMock,
257257
_enrich_getmany_poll_span: mock.MagicMock,
258258
_enrich_getmany_topic_span: mock.MagicMock,
259-
_create_consumer_span: mock.MagicMock,
259+
_create_consumer_span: mock.AsyncMock,
260260
extract: mock.MagicMock,
261261
) -> None:
262262
tracer = mock.MagicMock()
@@ -270,6 +270,7 @@ async def test_wrap_getmany(
270270
}
271271
)
272272
kafka_consumer = mock.MagicMock()
273+
_create_consumer_span.return_value = mock.MagicMock()
273274

274275
wrapped_getmany = _wrap_getmany(tracer, consume_hook)
275276
records = await wrapped_getmany(

0 commit comments

Comments
 (0)