Skip to content

Commit 3cea4b0

Browse files
committed
Fix test
1 parent 53acf5b commit 3cea4b0

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

tests/functional/metrics/required_dependencies/test_metrics_provider.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,8 @@ def lambda_handler(evt, context, additional_arg, additional_kw_arg="default_valu
8181

8282

8383
def 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

9698
def 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

Comments
 (0)