Tag enhanced metrics with datadog lambda library version (#94)

* tag enhanced metrics witht he datadog_lambda:${version} string

* remove ddtrace

* format

* add datadog_lambda tag to enhanced metrics

* black format

Author: Czechh

datadog_lambda/tags.py

@@ -78,13 +78,20 @@ 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
     """
     return parse_lambda_tags_from_arn(lambda_context) + [
         get_cold_start_tag(),
         "memorysize:{}".format(lambda_context.memory_limit_in_mb),
         get_runtime_tag(),
+        get_library_version_tag(),
     ]
 
 

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,