add callback attr name test

open-telemetry · Dec 20, 2021 · 29fbd93 · 29fbd93
1 parent 68f9783
commit 29fbd93
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/instrumentation/opentelemetry-instrumentation-pika/tests/test_pika_instrumentation.py b/instrumentation/opentelemetry-instrumentation-pika/tests/test_pika_instrumentation.py
@@ -18,6 +18,9 @@
 from wrapt import BoundFunctionWrapper
 
 from opentelemetry.instrumentation.pika import PikaInstrumentor
+from opentelemetry.instrumentation.pika.pika_instrumentor import (
+    _consumer_callback_attribute_name,
+)
 from opentelemetry.instrumentation.pika.utils import dummy_callback
 from opentelemetry.trace import Tracer
 
@@ -113,3 +116,9 @@ def test_uninstrument_channel_functions(self) -> None:
         self.channel.basic_publish._original_function = original_function
         PikaInstrumentor._uninstrument_channel_functions(self.channel)
         self.assertEqual(self.channel.basic_publish, original_function)
+
+    def test_consumer_callback_attribute_name(self) -> None:
+        with mock.patch("pika.__version__", "1.0.0"):
+            self.assertEqual(_consumer_callback_attribute_name(), "on_message_callback")
+        with mock.patch("pika.__version__", "0.12.0"):
+            self.assertEqual(_consumer_callback_attribute_name(), "consumer_cb")