fix: greedy variable can be any value

aws · Dec 19, 2024 · 87a8dcf · 87a8dcf
1 parent 8b90fa5
commit 87a8dcf
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Services/ApiGatewayRouteConfigService.cs b/Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Services/ApiGatewayRouteConfigService.cs
@@ -142,15 +142,20 @@ private bool IsRouteConfigValid(ApiGatewayRouteConfig routeConfig)
             return false;
         }
 
-        var occurrences = routeConfig.Path.Split("{proxy+}").Length - 1;
-        if (occurrences > 1)
+        var occurrences = routeConfig.Path
+            .Split('/')
+            .Where(
+                x => x.StartsWith("{") &&
+                     x.EndsWith("+}"))
+            .ToList();
+        if (occurrences.Count > 1)
         {
             _logger.LogError("The route config {Method} {Path} cannot have multiple greedy variables {{proxy+}}.",
                 routeConfig.HttpMethod, routeConfig.Path);
             return false;
         }
 
-        if (occurrences == 1 && !routeConfig.Path.EndsWith("/{proxy+}"))
+        if (occurrences.Count == 1 && !routeConfig.Path.EndsWith($"/{occurrences.Last()}"))
         {
             _logger.LogError("The route config {Method} {Path} uses a greedy variable {{proxy+}} but does not end with it.",
                 routeConfig.HttpMethod, routeConfig.Path);
@@ -412,7 +417,7 @@ private bool IsVariableSegment(string segment)
     /// <returns><c>true</c> if the segment is a greedy variable segment; <c>false</c> otherwise.</returns>
     private bool IsGreedyVariable(string segment)
     {
-        return segment.Equals("{proxy+}", StringComparison.InvariantCultureIgnoreCase);
+        return segment.StartsWith("{") && segment.EndsWith("+}");
     }
 
     /// <summary>

diff --git a/...l-v2/tests/Amazon.Lambda.TestTool.UnitTests/Services/ApiGatewayRouteConfigServiceTests.cs b/...l-v2/tests/Amazon.Lambda.TestTool.UnitTests/Services/ApiGatewayRouteConfigServiceTests.cs
@@ -159,8 +159,10 @@ public void Constructor_LoadsAndParsesListOfConfigs()
         Assert.Equal("Function2", result2.LambdaResourceName);
     }
 
-    [Fact]
-    public void CatchAllRouteConfig()
+    [Theory]
+    [InlineData("proxy")]
+    [InlineData("anyvalue")]
+    public void CatchAllRouteConfig(string variableName)
     {
         // Arange
         var routeConfigs = new List<ApiGatewayRouteConfig>
@@ -169,7 +171,7 @@ public void CatchAllRouteConfig()
             {
                 LambdaResourceName = "F1",
                 HttpMethod = "ANY",
-                Path = "/{proxy+}"
+                Path = $"/{{{variableName}+}}"
             }
         };