@@ -179,10 +179,9 @@ def test_tracer_no_autopatch(patch_mock):
179
179
assert patch_mock .call_count == 0
180
180
181
181
182
- def test_tracer_lambda_handler_does_not_add_empty_response_as_metadata (mocker , provider_stub ):
182
+ def test_tracer_lambda_handler_does_not_add_empty_response_as_metadata (mocker , provider_stub , in_subsegment_mock ):
183
183
# GIVEN tracer is initialized
184
- put_metadata_mock = mocker .MagicMock ()
185
- provider = provider_stub (put_metadata_mock = put_metadata_mock )
184
+ provider = provider_stub (in_subsegment = in_subsegment_mock .in_subsegment )
186
185
tracer = Tracer (provider = provider )
187
186
188
187
# WHEN capture_lambda_handler decorator is used
@@ -194,13 +193,12 @@ def handler(event, context):
194
193
handler ({}, mocker .MagicMock ())
195
194
196
195
# THEN we should not add empty metadata
197
- assert put_metadata_mock .call_count == 0
196
+ assert in_subsegment_mock . put_metadata .call_count == 0
198
197
199
198
200
- def test_tracer_method_does_not_add_empty_response_as_metadata (mocker , provider_stub ):
199
+ def test_tracer_method_does_not_add_empty_response_as_metadata (mocker , provider_stub , in_subsegment_mock ):
201
200
# GIVEN tracer is initialized
202
- put_metadata_mock = mocker .MagicMock ()
203
- provider = provider_stub (put_metadata_mock = put_metadata_mock )
201
+ provider = provider_stub (in_subsegment = in_subsegment_mock .in_subsegment )
204
202
tracer = Tracer (provider = provider )
205
203
206
204
# WHEN capture_method decorator is used
@@ -212,7 +210,7 @@ def greeting(name, message):
212
210
greeting (name = "Foo" , message = "Bar" )
213
211
214
212
# THEN we should not add empty metadata
215
- assert put_metadata_mock .call_count == 0
213
+ assert in_subsegment_mock . put_metadata .call_count == 0
216
214
217
215
218
216
@mock .patch ("aws_lambda_powertools.tracing.tracer.aws_xray_sdk.core.patch" )
0 commit comments