Test base provider

Author: ConnorKirk

tests/functional/metrics/required_dependencies/test_metrics_provider.py

@@ -78,3 +78,30 @@ def lambda_handler(evt, context, additional_arg, additional_kw_arg="default_valu
     # the wrapped function is passed additional arguments
     assert lambda_handler({}, {}, "arg_value", additional_kw_arg="kw_arg_value") == ("arg_value", "kw_arg_value")
     assert lambda_handler({}, {}, "arg_value") == ("arg_value", "default_value")
+
+
+def test_log_metrics_with_durable_context(capsys, metric, durable_context):
+    metrics = Metrics(namespace="test")
+
+    @metrics.log_metrics
+    def lambda_handler(evt, ctx):
+        metrics.add_metric(**metric)
+
+    lambda_handler({}, durable_context)
+    output = capture_metrics_output(capsys)
+
+    assert output["single_metric"] == [1.0]
+
+
+def test_log_metrics_cold_start_with_durable_context(capsys, durable_context):
+    metrics = Metrics(namespace="test")
+
+    @metrics.log_metrics(capture_cold_start_metric=True)
+    def lambda_handler(evt, ctx):
+        pass
+
+    lambda_handler({}, durable_context)
+    output = capture_metrics_output(capsys)
+
+    assert output["ColdStart"] == [1.0]
+    assert output["function_name"] == "example_fn"