test stacktrace with decorated method

Author: hjgraca

libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/Handlers/ExceptionFunctionHandler.cs

@@ -22,11 +22,22 @@ public Task<string> HandleDecoratorOutsideHandler(string input)
 
         return Task.FromResult(input.ToUpper(CultureInfo.InvariantCulture));
     }
+    
+    [Metrics(Namespace = "ns", Service = "svc")]
+    public Task<string> HandleDecoratorOutsideHandlerException(string input)
+    {
+        MethodDecorated();
+        
+        Metrics.AddMetric($"Metric Name", 1, MetricUnit.Count);
+        
+        ThisThrowsDecorated();
+        
+        return Task.FromResult(input.ToUpper(CultureInfo.InvariantCulture));
+    }
 
     [Metrics(Namespace = "ns", Service = "svc")]
     private void MethodDecorated()
     {
-        // NOOP
         Metrics.AddMetric($"Metric Name", 1, MetricUnit.Count);
         Metrics.AddMetric($"Metric Name Decorated", 1, MetricUnit.Count);
     }
@@ -35,4 +46,10 @@ private void ThisThrows()
     {
         throw new NullReferenceException();
     }
+    
+    [Metrics(Namespace = "ns", Service = "svc")]
+    private void ThisThrowsDecorated()
+    {
+        throw new NullReferenceException();
+    }
 }

libraries/tests/AWS.Lambda.Powertools.Metrics.Tests/Handlers/ExceptionFunctionHandlerTests.cs

@@ -20,7 +20,21 @@ public async Task Stack_Trace_Included_When_Decorator_Present()
         // Assert
         var tracedException = await Assert.ThrowsAsync<NullReferenceException>(Handle);
         Assert.StartsWith("at AWS.Lambda.Powertools.Metrics.Tests.Handlers.ExceptionFunctionHandler.ThisThrows()", tracedException.StackTrace?.TrimStart());
+    }
+    
+    [Fact]
+    public async Task Stack_Trace_Included_When_Decorator_Present_In_Method()
+    {
+        // Arrange
+        Metrics.ResetForTest();
+        var handler = new ExceptionFunctionHandler();
 
+        // Act
+        Task Handle() => handler.HandleDecoratorOutsideHandlerException("whatever");
+        
+        // Assert
+        var tracedException = await Assert.ThrowsAsync<NullReferenceException>(Handle);
+        Assert.StartsWith("at AWS.Lambda.Powertools.Metrics.Tests.Handlers.ExceptionFunctionHandler.__a$_around_ThisThrows", tracedException.StackTrace?.TrimStart());
     }
 
     [Fact]