diff --git a/datadog_lambda/tags.py b/datadog_lambda/tags.py index acef9691..f21c8f92 100644 --- a/datadog_lambda/tags.py +++ b/datadog_lambda/tags.py @@ -78,6 +78,12 @@ def get_runtime_tag(): ) +def get_library_version_tag(): + """Get datadog lambda library tag + """ + return "datadog_lambda:v{}".format(__version__) + + def get_enhanced_metrics_tags(lambda_context): """Get the list of tags to apply to enhanced metrics """ @@ -85,6 +91,7 @@ def get_enhanced_metrics_tags(lambda_context): get_cold_start_tag(), "memorysize:{}".format(lambda_context.memory_limit_in_mb), get_runtime_tag(), + get_library_version_tag(), ] diff --git a/tests/test_wrapper.py b/tests/test_wrapper.py index a79297ce..6cfce48a 100644 --- a/tests/test_wrapper.py +++ b/tests/test_wrapper.py @@ -68,6 +68,11 @@ def setUp(self): self.mock_write_metric_point_to_stdout = patcher.start() self.addCleanup(patcher.stop) + patcher = patch("datadog_lambda.tags.get_library_version_tag") + self.mock_format_dd_lambda_layer_tag = patcher.start() + # Mock the layer version so we don't have to update tests on every version bump + self.mock_format_dd_lambda_layer_tag.return_value = "datadog_lambda:v6.6.6" + patcher = patch("datadog_lambda.tags._format_dd_lambda_layer_tag") self.mock_format_dd_lambda_layer_tag = patcher.start() # Mock the layer version so we don't have to update tests on every version bump @@ -146,6 +151,7 @@ def lambda_handler(event, context): "cold_start:true", "memorysize:256", "runtime:python2.7", + "datadog_lambda:v6.6.6", "dd_lambda_layer:datadog-python27_0.1.0", ], timestamp=None, @@ -176,6 +182,7 @@ def lambda_handler(event, context): "cold_start:true", "memorysize:256", "runtime:python2.7", + "datadog_lambda:v6.6.6", "dd_lambda_layer:datadog-python27_0.1.0", ], timestamp=None, @@ -191,6 +198,7 @@ def lambda_handler(event, context): "cold_start:true", "memorysize:256", "runtime:python2.7", + "datadog_lambda:v6.6.6", "dd_lambda_layer:datadog-python27_0.1.0", ], timestamp=None, @@ -226,6 +234,7 @@ def lambda_handler(event, context): "cold_start:true", "memorysize:256", "runtime:python2.7", + "datadog_lambda:v6.6.6", "dd_lambda_layer:datadog-python27_0.1.0", ], timestamp=None, @@ -241,6 +250,7 @@ def lambda_handler(event, context): "cold_start:false", "memorysize:256", "runtime:python2.7", + "datadog_lambda:v6.6.6", "dd_lambda_layer:datadog-python27_0.1.0", ], timestamp=None, @@ -274,6 +284,7 @@ def lambda_handler(event, context): "cold_start:true", "memorysize:256", "runtime:python2.7", + "datadog_lambda:v6.6.6", "dd_lambda_layer:datadog-python27_0.1.0", ], timestamp=None, @@ -307,6 +318,7 @@ def lambda_handler(event, context): "cold_start:true", "memorysize:256", "runtime:python2.7", + "datadog_lambda:v6.6.6", "dd_lambda_layer:datadog-python27_0.1.0", ], timestamp=None,