15
15
16
16
import pytest
17
17
18
- from aws_lambda_powertools import Logger , Tracer , set_package_logger_handler
18
+ from aws_lambda_powertools import Logger , Metrics , set_package_logger_handler
19
19
from aws_lambda_powertools .logging import correlation_paths
20
20
from aws_lambda_powertools .logging .exceptions import InvalidLoggerSamplingRateError
21
21
from aws_lambda_powertools .logging .formatter import (
@@ -219,13 +219,14 @@ def test_package_logger_stream(stdout):
219
219
# GIVEN package logger "aws_lambda_powertools" is explicitly set with no params
220
220
set_package_logger (stream = stdout )
221
221
222
- # WHEN Tracer is initialized in disabled mode
223
- Tracer (disabled = True )
222
+ # WHEN we add a dimension in Metrics feature
223
+ my_metrics = Metrics (namespace = "powertools" )
224
+ my_metrics .add_dimension (name = "dimension" , value = "test" )
224
225
225
- # THEN Tracer debug log statement should be logged
226
+ # THEN Metrics debug log statement should be logged
226
227
output = stdout .getvalue ()
227
228
logger = logging .getLogger ("aws_lambda_powertools" )
228
- assert "Tracing has been disabled " in output
229
+ assert "Adding dimension: " in output
229
230
assert logger .level == logging .DEBUG
230
231
231
232
@@ -235,10 +236,11 @@ def test_package_logger_format(capsys):
235
236
formatter = logging .Formatter ("message=%(message)s" )
236
237
set_package_logger (formatter = formatter )
237
238
238
- # WHEN Tracer is initialized in disabled mode
239
- Tracer (disabled = True )
239
+ # WHEN we add a dimension in Metrics feature
240
+ my_metrics = Metrics (namespace = "powertools" )
241
+ my_metrics .add_dimension (name = "dimension" , value = "test" )
240
242
241
- # THEN Tracer debug log statement should be logged using `message=` format
243
+ # THEN Metrics debug log statement should be logged using `message=` format
242
244
output = capsys .readouterr ().out
243
245
logger = logging .getLogger ("aws_lambda_powertools" )
244
246
assert "message=" in output
@@ -977,13 +979,15 @@ def test_set_package_logger_handler_with_powertools_debug_env_var(stdout, monkey
977
979
logger = logging .getLogger ("aws_lambda_powertools" )
978
980
979
981
# WHEN set_package_logger is used at initialization
980
- # and any Powertools for AWS Lambda (Python) operation is used (e.g., Tracer )
982
+ # and any Powertools for AWS Lambda (Python) operation is used (e.g., Metrics add_dimension )
981
983
set_package_logger_handler (stream = stdout )
982
- Tracer (disabled = True )
983
984
984
- # THEN Tracer debug log statement should be logged
985
+ my_metrics = Metrics (namespace = "powertools" )
986
+ my_metrics .add_dimension (name = "dimension" , value = "test" )
987
+
988
+ # THEN Metrics debug log statement should be logged
985
989
output = stdout .getvalue ()
986
- assert "Tracing has been disabled " in output
990
+ assert "Adding dimension: " in output
987
991
assert logger .level == logging .DEBUG
988
992
989
993
0 commit comments