We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 27f7c2a commit 82340b3Copy full SHA for 82340b3
1 file changed
instrumentation/opentelemetry-instrumentation-aiokafka/tests/test_utils.py
@@ -256,7 +256,7 @@ async def test_wrap_getmany(
256
extract_bootstrap_servers: mock.MagicMock,
257
_enrich_getmany_poll_span: mock.MagicMock,
258
_enrich_getmany_topic_span: mock.MagicMock,
259
- _create_consumer_span: mock.MagicMock,
+ _create_consumer_span: mock.AsyncMock,
260
extract: mock.MagicMock,
261
) -> None:
262
tracer = mock.MagicMock()
@@ -270,6 +270,7 @@ async def test_wrap_getmany(
270
}
271
)
272
kafka_consumer = mock.MagicMock()
273
+ _create_consumer_span.return_value = mock.MagicMock()
274
275
wrapped_getmany = _wrap_getmany(tracer, consume_hook)
276
records = await wrapped_getmany(
0 commit comments