Add test for additional WithArgs overload

dotnet · Sep 9, 2024 · 8132c74 · 8132c74
1 parent d85371b
commit 8132c74
Showing 1 changed file with 34 additions and 6 deletions.
diff --git a/tests/Aspire.Hosting.Tests/ExecutableResourceTests.cs b/tests/Aspire.Hosting.Tests/ExecutableResourceTests.cs
@@ -24,12 +24,7 @@ public async Task AddExecutableWithArgs()
             });
 
         var exe2 = appBuilder.AddExecutable("e2", "python", ".", "app.py")
-             .WithArgs(context =>
-             {
-                 context.Args.Add("arg1");
-                 context.Args.Add(exe1.GetEndpoint("ep"));
-                 context.Args.Add(testResource);
-             });
+             .WithArgs("arg1", exe1.GetEndpoint("ep"), testResource);
 
         using var app = appBuilder.Build();
 
@@ -61,6 +56,39 @@ public async Task AddExecutableWithArgs()
         Assert.Equal(expectedManifest, manifest.ToString());
     }
 
+    [Fact]
+    public async Task AddExecutableWithArgsCallback()
+    {
+        var appBuilder = DistributedApplication.CreateBuilder();
+
+        var testResource = new TestResource("test", "connectionString");
+
+        var exe1 = appBuilder.AddExecutable("e1", "ruby", ".", "app.rb")
+            .WithEndpoint("ep", e =>
+            {
+                e.UriScheme = "http";
+                e.AllocatedEndpoint = new(e, "localhost", 1234);
+            });
+
+        var exe2 = appBuilder.AddExecutable("e2", "python", ".", "app.py")
+             .WithArgs(context =>
+             {
+                 context.Args.Add("arg1");
+                 context.Args.Add(exe1.GetEndpoint("ep"));
+                 context.Args.Add(testResource);
+             });
+
+        using var app = appBuilder.Build();
+
+        var args = await ArgumentEvaluator.GetArgumentListAsync(exe2.Resource);
+
+        Assert.Collection(args,
+            arg => Assert.Equal("app.py", arg),
+            arg => Assert.Equal("arg1", arg),
+            arg => Assert.Equal("http://localhost:1234", arg),
+            arg => Assert.Equal("connectionString", arg));
+    }    
+
     private sealed class TestResource(string name, string connectionString) : Resource(name), IResourceWithConnectionString
     {
         public ReferenceExpression ConnectionStringExpression =>