Fix Tests

Author: oxeye-nikolay

instrumentation/opentelemetry-instrumentation-pika/src/opentelemetry/instrumentation/pika/pika_instrumentor.py

@@ -122,12 +122,12 @@ def wrapper(wrapped, instance, args, kwargs):
 
     def _instrument(self, **kwargs: Dict[str, Any]) -> None:
         tracer_provider: TracerProvider = kwargs.get("tracer_provider", None)
-        channel.__setattr__("__opentelemetry_tracer", tracer)
+        self.__setattr__("__opentelemetry_tracer", tracer_provider)
         self._decorate_channel_function(tracer_provider)
 
     def _uninstrument(self, **kwargs: Dict[str, Any]) -> None:
-        if hasattr(channel, "__opentelemetry_tracer"):
-            delattr(channel, "__opentelemetry_tracer")
+        if hasattr(self, "__opentelemetry_tracer"):
+            delattr(self, "__opentelemetry_tracer")
         unwrap(BlockingConnection, "channel")
 
     def instrumentation_dependencies(self) -> Collection[str]:

instrumentation/opentelemetry-instrumentation-pika/tests/test_pika_instrumentation.py

@@ -34,6 +34,9 @@ def test_instrument_api(self) -> None:
         self.assertTrue(
             isinstance(BlockingConnection.channel, BoundFunctionWrapper)
         )
+        assert hasattr(
+            instrumentation, "__opentelemetry_tracer"
+        ), "Tracer not stored for the object!"
         instrumentation.uninstrument(channel=self.channel)
         self.assertFalse(
             isinstance(BlockingConnection.channel, BoundFunctionWrapper)
@@ -52,8 +55,8 @@ def test_instrument(
     ):
         PikaInstrumentor.instrument_channel(channel=self.channel)
         assert hasattr(
-            self.channel, "__opentelemetry_tracer"
-        ), "Tracer not set for the channel!"
+            self.channel, "_is_instrumented_by_opentelemetry"
+        ), "channel is not marked as instrumented!"
         instrument_consumers.assert_called_once()
         instrument_channel_functions.assert_called_once()