black formting

open-telemetry · Nov 27, 2023 · 589ea7c · 589ea7c
1 parent addc029
commit 589ea7c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/instrumentation/opentelemetry-instrumentation-aio-pika/tests/test_publish_decorator.py b/instrumentation/opentelemetry-instrumentation-aio-pika/tests/test_publish_decorator.py
@@ -96,16 +96,17 @@ def test_robust_publish(self):
     def _test_publish_works_with_not_recording_span(self, exchange_type):
         exchange = exchange_type(CONNECTION_7, CHANNEL_7, EXCHANGE_NAME)
         with mock.patch.object(
-                PublishDecorator, "_get_publish_span"
+            PublishDecorator, "_get_publish_span"
         ) as mock_get_publish_span:
             mocked_not_recording_span = MagicMock()
             mocked_not_recording_span.is_recording.return_value = False
             mock_get_publish_span.return_value = mocked_not_recording_span
             with mock.patch.object(
-                    Exchange, "publish", return_value=asyncio.sleep(0)
+                Exchange, "publish", return_value=asyncio.sleep(0)
             ) as mock_publish:
                 with mock.patch(
-                        "opentelemetry.instrumentation.aio_pika.publish_decorator.propagate.inject") as mock_inject:
+                    "opentelemetry.instrumentation.aio_pika.publish_decorator.propagate.inject"
+                ) as mock_inject:
                     decorated_publish = PublishDecorator(
                         self.tracer, exchange
                     ).decorate(mock_publish)
@@ -178,16 +179,17 @@ def test_robust_publish(self):
     def _test_publish_works_with_not_recording_span(self, exchange_type):
         exchange = exchange_type(CONNECTION_7, CHANNEL_7, EXCHANGE_NAME)
         with mock.patch.object(
-                PublishDecorator, "_get_publish_span"
+            PublishDecorator, "_get_publish_span"
         ) as mock_get_publish_span:
             mocked_not_recording_span = MagicMock()
             mocked_not_recording_span.is_recording.return_value = False
             mock_get_publish_span.return_value = mocked_not_recording_span
             with mock.patch.object(
-                    Exchange, "publish", return_value=asyncio.sleep(0)
+                Exchange, "publish", return_value=asyncio.sleep(0)
             ) as mock_publish:
                 with mock.patch(
-                        "opentelemetry.instrumentation.aio_pika.publish_decorator.propagate.inject") as mock_inject:
+                    "opentelemetry.instrumentation.aio_pika.publish_decorator.propagate.inject"
+                ) as mock_inject:
                     decorated_publish = PublishDecorator(
                         self.tracer, exchange
                     ).decorate(mock_publish)

diff --git a/instrumentation/opentelemetry-instrumentation-pika/tests/test_utils.py b/instrumentation/opentelemetry-instrumentation-pika/tests/test_utils.py
@@ -447,4 +447,4 @@ def test_decorate_basic_publish_when_span_is_not_recording(
         callback.assert_called_once_with(
             exchange_name, routing_key, mock_body, properties, False
         )
-        self.assertEqual(retval, callback.return_value)
+        self.assertEqual(retval, callback.return_value)