fix: correct in_subsegment assertion

Author: heitorlessa

tests/unit/test_tracing.py

@@ -179,10 +179,9 @@ def test_tracer_no_autopatch(patch_mock):
     assert patch_mock.call_count == 0
 
 
-def test_tracer_lambda_handler_does_not_add_empty_response_as_metadata(mocker, provider_stub):
+def test_tracer_lambda_handler_does_not_add_empty_response_as_metadata(mocker, provider_stub, in_subsegment_mock):
     # GIVEN tracer is initialized
-    put_metadata_mock = mocker.MagicMock()
-    provider = provider_stub(put_metadata_mock=put_metadata_mock)
+    provider = provider_stub(in_subsegment=in_subsegment_mock.in_subsegment)
     tracer = Tracer(provider=provider)
 
     # WHEN capture_lambda_handler decorator is used
@@ -194,13 +193,12 @@ def handler(event, context):
     handler({}, mocker.MagicMock())
 
     # THEN we should not add empty metadata
-    assert put_metadata_mock.call_count == 0
+    assert in_subsegment_mock.put_metadata.call_count == 0
 
 
-def test_tracer_method_does_not_add_empty_response_as_metadata(mocker, provider_stub):
+def test_tracer_method_does_not_add_empty_response_as_metadata(mocker, provider_stub, in_subsegment_mock):
     # GIVEN tracer is initialized
-    put_metadata_mock = mocker.MagicMock()
-    provider = provider_stub(put_metadata_mock=put_metadata_mock)
+    provider = provider_stub(in_subsegment=in_subsegment_mock.in_subsegment)
     tracer = Tracer(provider=provider)
 
     # WHEN capture_method decorator is used
@@ -212,7 +210,7 @@ def greeting(name, message):
     greeting(name="Foo", message="Bar")
 
     # THEN we should not add empty metadata
-    assert put_metadata_mock.call_count == 0
+    assert in_subsegment_mock.put_metadata.call_count == 0
 
 
 @mock.patch("aws_lambda_powertools.tracing.tracer.aws_xray_sdk.core.patch")