Fix getEnhancedMetricTags on local env (#87)

* Fix getEnhancedMetricTags on local env * Update enhanced-metrics.spec.ts Co-authored-by: Harvinder Ghotra <[email protected]>
DataDog · Aug 19, 2020 · 69adaa9 · 69adaa9
1 parent 317796f
commit 69adaa9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/metrics/enhanced-metrics.spec.ts b/src/metrics/enhanced-metrics.spec.ts
@@ -14,6 +14,7 @@ const mockContext = ({
 } as any) as Context;
 const mockContextLocal = ({
   functionName: "my-test-lambda",
+  functionVersion: "1.0.0",
   memoryLimitInMB: "128",
 } as any) as Context;
 

diff --git a/src/metrics/enhanced-metrics.ts b/src/metrics/enhanced-metrics.ts
@@ -45,10 +45,8 @@ export function getRuntimeTag(): string | null {
 
 export function getEnhancedMetricTags(context: Context): string[] {
   let arnTags = [`functionname:${context.functionName}`];
-  if (context.functionVersion) {
+  if (context.invokedFunctionArn) {
     arnTags = parseTagsFromARN(context.invokedFunctionArn, context.functionVersion);
-  } else if (context.invokedFunctionArn) {
-    arnTags = parseTagsFromARN(context.invokedFunctionArn);
   }
   const tags = [...arnTags, getColdStartTag(), `memorysize:${context.memoryLimitInMB}`];