Skip to content

Commit e85a2c8

Browse files
committed
add callback attr name test
1 parent 68f9783 commit e85a2c8

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

Diff for: instrumentation/opentelemetry-instrumentation-pika/tests/test_pika_instrumentation.py

+13
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@
1818
from wrapt import BoundFunctionWrapper
1919

2020
from opentelemetry.instrumentation.pika import PikaInstrumentor
21+
from opentelemetry.instrumentation.pika.pika_instrumentor import (
22+
_consumer_callback_attribute_name,
23+
)
2124
from opentelemetry.instrumentation.pika.utils import dummy_callback
2225
from opentelemetry.trace import Tracer
2326

@@ -113,3 +116,13 @@ def test_uninstrument_channel_functions(self) -> None:
113116
self.channel.basic_publish._original_function = original_function
114117
PikaInstrumentor._uninstrument_channel_functions(self.channel)
115118
self.assertEqual(self.channel.basic_publish, original_function)
119+
120+
def test_consumer_callback_attribute_name(self) -> None:
121+
with mock.patch("pika.__version__", "1.0.0"):
122+
self.assertEqual(
123+
_consumer_callback_attribute_name(), "on_message_callback"
124+
)
125+
with mock.patch("pika.__version__", "0.12.0"):
126+
self.assertEqual(
127+
_consumer_callback_attribute_name(), "consumer_cb"
128+
)

0 commit comments

Comments
 (0)