use snapshots for tests

Author: philasmar

test/AWS.AgentCore.SourceGenerator.UnitTests/AWS.AgentCore.SourceGenerator.UnitTests.csproj

@@ -19,4 +19,11 @@
         <ProjectReference Include="..\..\src\AWS.AgentCore.SourceGenerator\AWS.AgentCore.SourceGenerator.csproj" />
     </ItemGroup>
 
+    <ItemGroup>
+        <Content Include="Snapshots\**\*.g.cs">
+            <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+        </Content>
+        <Compile Remove="Snapshots\**" />
+    </ItemGroup>
+
 </Project>

test/AWS.AgentCore.SourceGenerator.UnitTests/AgentCoreStartupGeneratorTests.cs

@@ -6,7 +6,7 @@ namespace AWS.AgentCore.SourceGenerator.UnitTests;
 public class AgentCoreStartupGeneratorTests
 {
     [Fact]
-    public void Generator_WithStartupAndAgent_EmitsProgram()
+    public async Task Generator_WithStartupAndAgent_EmitsProgram()
     {
         var source = @"
 using AWS.AgentCore;
@@ -34,18 +34,14 @@ public Task<string> Handle(PromptRequest request, AgentCoreRuntimeContext contex
 }";
 
         var result = GeneratorTestHelper.RunGenerator(source);
+        var expected = await ReadSnapshot("StartupAndAgent.g.cs");
 
         Assert.NotNull(result.GeneratedSource);
-        Assert.Contains("var builder = WebApplication.CreateBuilder(args);", result.GeneratedSource);
-        Assert.Contains("var startup = new TestApp.Startup();", result.GeneratedSource);
-        Assert.Contains("startup.ConfigureServices(builder);", result.GeneratedSource);
-        Assert.Contains("builder.Services.AddTransient<TestApp.MyAgent>();", result.GeneratedSource);
-        Assert.Contains("app.MapAgentCore<TestApp.PromptRequest>(", result.GeneratedSource);
-        Assert.Contains("agent.Handle(request, context, ct)", result.GeneratedSource);
+        Assert.Equal(expected, result.GeneratedSource);
     }
 
     [Fact]
-    public void Generator_WithPingHandler_EmitsPingDelegate()
+    public async Task Generator_WithPingHandler_EmitsPingDelegate()
     {
         var source = @"
 using AWS.AgentCore;
@@ -70,14 +66,14 @@ public Task<string> Handle(PromptRequest request, CancellationToken ct)
 }";
 
         var result = GeneratorTestHelper.RunGenerator(source);
+        var expected = await ReadSnapshot("WithPingHandler.g.cs");
 
         Assert.NotNull(result.GeneratedSource);
-        Assert.Contains("pingHandler:", result.GeneratedSource);
-        Assert.Contains("agent.Ping()", result.GeneratedSource);
+        Assert.Equal(expected, result.GeneratedSource);
     }
 
     [Fact]
-    public void Generator_WithoutStartup_SkipsConfigureServices()
+    public async Task Generator_WithoutStartup_SkipsConfigureServices()
     {
         var source = @"
 using AWS.AgentCore;
@@ -99,14 +95,14 @@ public Task<string> Handle(PromptRequest request, CancellationToken ct)
 }";
 
         var result = GeneratorTestHelper.RunGenerator(source);
+        var expected = await ReadSnapshot("WithoutStartup.g.cs");
 
         Assert.NotNull(result.GeneratedSource);
-        Assert.DoesNotContain("ConfigureServices", result.GeneratedSource);
-        Assert.Contains("builder.Services.AddTransient<TestApp.MyAgent>();", result.GeneratedSource);
+        Assert.Equal(expected, result.GeneratedSource);
     }
 
     [Fact]
-    public void Generator_WithoutInvocation_EmitsNothing()
+    public void Generator_WithoutHandler_EmitsNothing()
     {
         var source = @"
 using AWS.AgentCore;
@@ -126,7 +122,7 @@ public void ConfigureServices(Microsoft.AspNetCore.Builder.WebApplicationBuilder
     }
 
     [Fact]
-    public void Generator_StreamingHandler_DetectsReturnType()
+    public async Task Generator_StreamingHandler_DetectsReturnType()
     {
         var source = @"
 using AWS.AgentCore;
@@ -148,14 +144,14 @@ public IAsyncEnumerable<string> Handle(PromptRequest request, CancellationToken
 }";
 
         var result = GeneratorTestHelper.RunGenerator(source);
+        var expected = await ReadSnapshot("StreamingHandler.g.cs");
 
         Assert.NotNull(result.GeneratedSource);
-        Assert.Contains("app.MapAgentCore<TestApp.PromptRequest>(", result.GeneratedSource);
-        Assert.Contains("agent.Handle(request, ct)", result.GeneratedSource);
+        Assert.Equal(expected, result.GeneratedSource);
     }
 
     [Fact]
-    public void Generator_RequestOnlyParameter_IdentifiesRequestType()
+    public async Task Generator_RequestOnlyParameter_IdentifiesRequestType()
     {
         var source = @"
 using AWS.AgentCore;
@@ -176,13 +172,14 @@ public Task<string> Handle(MyRequest request)
 }";
 
         var result = GeneratorTestHelper.RunGenerator(source);
+        var expected = await ReadSnapshot("RequestOnly.g.cs");
 
         Assert.NotNull(result.GeneratedSource);
-        Assert.Contains("app.MapAgentCore<TestApp.MyRequest>(", result.GeneratedSource);
+        Assert.Equal(expected, result.GeneratedSource);
     }
 
     [Fact]
-    public void Generator_GlobalNamespace_HandlesCorrectly()
+    public async Task Generator_GlobalNamespace_HandlesCorrectly()
     {
         var source = @"
 using AWS.AgentCore;
@@ -200,9 +197,16 @@ public Task<string> Handle(PromptRequest request)
 }";
 
         var result = GeneratorTestHelper.RunGenerator(source);
+        var expected = await ReadSnapshot("GlobalNamespace.g.cs");
 
         Assert.NotNull(result.GeneratedSource);
-        Assert.Contains("builder.Services.AddTransient<MyAgent>();", result.GeneratedSource);
-        Assert.Contains("app.MapAgentCore<PromptRequest>(", result.GeneratedSource);
+        Assert.Equal(expected, result.GeneratedSource);
+    }
+
+    private static async Task<string> ReadSnapshot(string fileName)
+    {
+        var path = Path.Combine("Snapshots", fileName);
+        var content = await File.ReadAllTextAsync(path);
+        return content.ReplaceLineEndings("\n");
     }
 }

test/AWS.AgentCore.SourceGenerator.UnitTests/GeneratorTestHelper.cs

@@ -69,7 +69,8 @@ public class WebApplicationBuilder { }
         var generatedSource = runResult.GeneratedTrees
             .FirstOrDefault(t => t.FilePath.Contains("AgentCore_GeneratedProgram"))
             ?.GetText()
-            .ToString();
+            .ToString()
+            ?.ReplaceLineEndings("\n");
 
         return new GeneratorResult
         {

test/AWS.AgentCore.SourceGenerator.UnitTests/Snapshots/GlobalNamespace.g.cs

@@ -0,0 +1,22 @@
+// <auto-generated/>
+// Generated by AWS.AgentCore.SourceGenerator
+
+using AWS.AgentCore;
+using AWS.AgentCore.Extensions;
+using Microsoft.Extensions.DependencyInjection;
+
+var builder = WebApplication.CreateBuilder(args);
+
+builder.Services.AddTransient<MyAgent>();
+
+var app = builder.Build();
+
+app.MapAgentCore<PromptRequest>(
+    handler: (PromptRequest request) =>
+    {
+        var agent = app.Services.GetRequiredService<MyAgent>();
+        return agent.Handle(request);
+    }
+);
+
+app.Run();

test/AWS.AgentCore.SourceGenerator.UnitTests/Snapshots/RequestOnly.g.cs

@@ -0,0 +1,22 @@
+// <auto-generated/>
+// Generated by AWS.AgentCore.SourceGenerator
+
+using AWS.AgentCore;
+using AWS.AgentCore.Extensions;
+using Microsoft.Extensions.DependencyInjection;
+
+var builder = WebApplication.CreateBuilder(args);
+
+builder.Services.AddTransient<TestApp.MyAgent>();
+
+var app = builder.Build();
+
+app.MapAgentCore<TestApp.MyRequest>(
+    handler: (TestApp.MyRequest request) =>
+    {
+        var agent = app.Services.GetRequiredService<TestApp.MyAgent>();
+        return agent.Handle(request);
+    }
+);
+
+app.Run();

test/AWS.AgentCore.SourceGenerator.UnitTests/Snapshots/StartupAndAgent.g.cs

@@ -0,0 +1,25 @@
+// <auto-generated/>
+// Generated by AWS.AgentCore.SourceGenerator
+
+using AWS.AgentCore;
+using AWS.AgentCore.Extensions;
+using Microsoft.Extensions.DependencyInjection;
+
+var builder = WebApplication.CreateBuilder(args);
+
+var startup = new TestApp.Startup();
+startup.ConfigureServices(builder);
+
+builder.Services.AddTransient<TestApp.MyAgent>();
+
+var app = builder.Build();
+
+app.MapAgentCore<TestApp.PromptRequest>(
+    handler: (TestApp.PromptRequest request, AWS.AgentCore.AgentCoreRuntimeContext context, System.Threading.CancellationToken ct) =>
+    {
+        var agent = app.Services.GetRequiredService<TestApp.MyAgent>();
+        return agent.Handle(request, context, ct);
+    }
+);
+
+app.Run();

test/AWS.AgentCore.SourceGenerator.UnitTests/Snapshots/StreamingHandler.g.cs

@@ -0,0 +1,22 @@
+// <auto-generated/>
+// Generated by AWS.AgentCore.SourceGenerator
+
+using AWS.AgentCore;
+using AWS.AgentCore.Extensions;
+using Microsoft.Extensions.DependencyInjection;
+
+var builder = WebApplication.CreateBuilder(args);
+
+builder.Services.AddTransient<TestApp.MyAgent>();
+
+var app = builder.Build();
+
+app.MapAgentCore<TestApp.PromptRequest>(
+    handler: (TestApp.PromptRequest request, System.Threading.CancellationToken ct) =>
+    {
+        var agent = app.Services.GetRequiredService<TestApp.MyAgent>();
+        return agent.Handle(request, ct);
+    }
+);
+
+app.Run();

test/AWS.AgentCore.SourceGenerator.UnitTests/Snapshots/WithPingHandler.g.cs

@@ -0,0 +1,27 @@
+// <auto-generated/>
+// Generated by AWS.AgentCore.SourceGenerator
+
+using AWS.AgentCore;
+using AWS.AgentCore.Extensions;
+using Microsoft.Extensions.DependencyInjection;
+
+var builder = WebApplication.CreateBuilder(args);
+
+builder.Services.AddTransient<TestApp.MyAgent>();
+
+var app = builder.Build();
+
+app.MapAgentCore<TestApp.PromptRequest>(
+    handler: (TestApp.PromptRequest request, System.Threading.CancellationToken ct) =>
+    {
+        var agent = app.Services.GetRequiredService<TestApp.MyAgent>();
+        return agent.Handle(request, ct);
+    }
+    , pingHandler: () =>
+    {
+        var agent = app.Services.GetRequiredService<TestApp.MyAgent>();
+        return agent.Ping();
+    }
+);
+
+app.Run();

test/AWS.AgentCore.SourceGenerator.UnitTests/Snapshots/WithoutStartup.g.cs

@@ -0,0 +1,22 @@
+// <auto-generated/>
+// Generated by AWS.AgentCore.SourceGenerator
+
+using AWS.AgentCore;
+using AWS.AgentCore.Extensions;
+using Microsoft.Extensions.DependencyInjection;
+
+var builder = WebApplication.CreateBuilder(args);
+
+builder.Services.AddTransient<TestApp.MyAgent>();
+
+var app = builder.Build();
+
+app.MapAgentCore<TestApp.PromptRequest>(
+    handler: (TestApp.PromptRequest request, System.Threading.CancellationToken ct) =>
+    {
+        var agent = app.Services.GetRequiredService<TestApp.MyAgent>();
+        return agent.Handle(request, ct);
+    }
+);
+
+app.Run();