@@ -81,7 +81,8 @@ def lambda_handler(evt, context, additional_arg, additional_kw_arg="default_valu
8181
8282
8383def test_log_metrics_with_durable_context (capsys , metric , durable_context ):
84- metrics = Metrics (namespace = "test" )
84+ provider = FakeMetricsProvider ()
85+ metrics = Metrics (provider = provider )
8586
8687 @metrics .log_metrics
8788 def lambda_handler (evt , ctx ):
@@ -90,18 +91,20 @@ def lambda_handler(evt, ctx):
9091 lambda_handler ({}, durable_context )
9192 output = capture_metrics_output (capsys )
9293
93- assert output ["single_metric" ] == [1.0 ]
94+ assert output [0 ]["name" ] == metric ["name" ]
95+ assert output [0 ]["value" ] == metric ["value" ]
9496
9597
9698def test_log_metrics_cold_start_with_durable_context (capsys , durable_context ):
97- metrics = Metrics (namespace = "test" )
99+ provider = FakeMetricsProvider ()
100+ metrics = Metrics (provider = provider )
98101
99102 @metrics .log_metrics (capture_cold_start_metric = True )
100103 def lambda_handler (evt , ctx ):
101- pass
104+ return True
102105
103106 lambda_handler ({}, durable_context )
104107 output = capture_metrics_output (capsys )
105108
106- assert output ["ColdStart " ] == [ 1.0 ]
107- assert output ["function_name " ] == "example_fn"
109+ assert output [0 ][ "name " ] == "ColdStart"
110+ assert output [0 ][ "value " ] == 1
0 commit comments