@@ -96,16 +96,17 @@ def test_robust_publish(self):
96
96
def _test_publish_works_with_not_recording_span (self , exchange_type ):
97
97
exchange = exchange_type (CONNECTION_7 , CHANNEL_7 , EXCHANGE_NAME )
98
98
with mock .patch .object (
99
- PublishDecorator , "_get_publish_span"
99
+ PublishDecorator , "_get_publish_span"
100
100
) as mock_get_publish_span :
101
101
mocked_not_recording_span = MagicMock ()
102
102
mocked_not_recording_span .is_recording .return_value = False
103
103
mock_get_publish_span .return_value = mocked_not_recording_span
104
104
with mock .patch .object (
105
- Exchange , "publish" , return_value = asyncio .sleep (0 )
105
+ Exchange , "publish" , return_value = asyncio .sleep (0 )
106
106
) as mock_publish :
107
107
with mock .patch (
108
- "opentelemetry.instrumentation.aio_pika.publish_decorator.propagate.inject" ) as mock_inject :
108
+ "opentelemetry.instrumentation.aio_pika.publish_decorator.propagate.inject"
109
+ ) as mock_inject :
109
110
decorated_publish = PublishDecorator (
110
111
self .tracer , exchange
111
112
).decorate (mock_publish )
@@ -178,16 +179,17 @@ def test_robust_publish(self):
178
179
def _test_publish_works_with_not_recording_span (self , exchange_type ):
179
180
exchange = exchange_type (CONNECTION_7 , CHANNEL_7 , EXCHANGE_NAME )
180
181
with mock .patch .object (
181
- PublishDecorator , "_get_publish_span"
182
+ PublishDecorator , "_get_publish_span"
182
183
) as mock_get_publish_span :
183
184
mocked_not_recording_span = MagicMock ()
184
185
mocked_not_recording_span .is_recording .return_value = False
185
186
mock_get_publish_span .return_value = mocked_not_recording_span
186
187
with mock .patch .object (
187
- Exchange , "publish" , return_value = asyncio .sleep (0 )
188
+ Exchange , "publish" , return_value = asyncio .sleep (0 )
188
189
) as mock_publish :
189
190
with mock .patch (
190
- "opentelemetry.instrumentation.aio_pika.publish_decorator.propagate.inject" ) as mock_inject :
191
+ "opentelemetry.instrumentation.aio_pika.publish_decorator.propagate.inject"
192
+ ) as mock_inject :
191
193
decorated_publish = PublishDecorator (
192
194
self .tracer , exchange
193
195
).decorate (mock_publish )
0 commit comments