add test functions. refactor and add more tests. not aot yet.

Author: hjgraca

libraries/tests/e2e/functions/idempotency/Function/src/Function/Function.cs

@@ -24,21 +24,50 @@ public APIGatewayProxyResponse FunctionHandler(APIGatewayProxyRequest apigwProxy
     }
 }
 
+namespace IdempotencyAttributeTest
+{
+    public class Function
+    {
+        public Function()
+        {
+            var tableName = Environment.GetEnvironmentVariable("IDEMPOTENCY_TABLE_NAME");
+            Idempotency.Configure(builder => builder.UseDynamoDb(tableName));
+        }
+    
+        public APIGatewayProxyResponse FunctionHandler(APIGatewayProxyRequest apigwProxyEvent, ILambdaContext context)
+        {
+            return new APIGatewayProxyResponse
+            {
+                Body = MyInternalMethod("dummy", apigwProxyEvent.RequestContext.RequestId),
+                StatusCode = 200
+            };
+        }
+        
+        [Idempotent]
+        private string MyInternalMethod(string argOne, [IdempotencyKey] string argTwo) {
+            return Guid.NewGuid().ToString();
+        }
+    }
+}
 
-namespace Function2
+namespace IdempotencyPayloadSubsetTest
 {
     public class Function
     {
         public Function()
         {
-            // var tableName = Environment.GetEnvironmentVariable("IDEMPOTENCY_TABLE_NAME");
-            // Idempotency.Configure(builder => builder.UseDynamoDb(tableName));
+            var tableName = Environment.GetEnvironmentVariable("IDEMPOTENCY_TABLE_NAME");
+            Idempotency.Configure(builder =>
+                builder
+                    .WithOptions(optionsBuilder =>
+                        optionsBuilder.WithEventKeyJmesPath("powertools_json(Body).[\"user_id\", \"product_id\"]"))
+                    .UseDynamoDb(tableName));
         }
 
-        // [Idempotent]
-        public string FunctionHandler(APIGatewayProxyRequest apigwProxyEvent, ILambdaContext context)
+        [Idempotent]
+        public APIGatewayProxyResponse FunctionHandler(APIGatewayProxyRequest apigwProxyEvent, ILambdaContext context)
         {
-            return "Hello, World!";
+            return TestHelper.TestMethod(apigwProxyEvent);
         }
     }
 }

libraries/tests/e2e/functions/idempotency/Function/src/Function/TestHelper.cs

@@ -7,13 +7,10 @@ public static class TestHelper
 {
     public static APIGatewayProxyResponse TestMethod(APIGatewayProxyRequest apigwProxyEvent)
     {
-        var requestContextRequestId = apigwProxyEvent.RequestContext.RequestId;
         var response = new
         {
-            RequestId = requestContextRequestId,
             Greeting = "Hello Powertools for AWS Lambda (.NET)",
-            MethodGuid = GenerateGuid(), // Guid generated by the GenerateGuid method. used to compare Method output
-            HandlerGuid = Guid.NewGuid().ToString() // Guid generated in the Handler. used to compare Handler output
+            Guid = Guid.NewGuid().ToString() // Guid generated in the Handler. used to compare Handler output
         };
 
         try
@@ -35,13 +32,4 @@ public static APIGatewayProxyResponse TestMethod(APIGatewayProxyRequest apigwPro
             };
         }
     }
-    
-    /// <summary>
-    /// Generates a new Guid to check if value is the same between calls (should be when idempotency enabled)
-    /// </summary>
-    /// <returns>GUID</returns>
-    private static string GenerateGuid()
-    {
-        return Guid.NewGuid().ToString();
-    }
 }