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,