diff --git a/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/RouteBuilderTests.cs b/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/RouteBuilderTests.cs index 1e404c8a76..824efb2adb 100644 --- a/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/RouteBuilderTests.cs +++ b/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/RouteBuilderTests.cs @@ -423,21 +423,21 @@ public sealed class RouteBuilderTests routeBuilder.AddHandler((message, context, cancellationToken) => { invocation.Capture(message, context, cancellationToken); - return message.ToUpperInvariant(); + return NormalizeHandlerResult(message); }); break; case 1: routeBuilder.AddHandler((message, context) => { invocation.Capture(message, context); - return message.ToUpperInvariant(); + return NormalizeHandlerResult(message); }); break; case 2: Func> asyncHandlerWithCancellation = (message, context, cancellationToken) => { invocation.Capture(message, context, cancellationToken); - return ValueTask.FromResult(message.ToUpperInvariant()); + return ValueTask.FromResult(NormalizeHandlerResult(message)); }; routeBuilder.AddHandler(asyncHandlerWithCancellation); break; @@ -445,7 +445,7 @@ public sealed class RouteBuilderTests Func> asyncHandler = (message, context) => { invocation.Capture(message, context); - return ValueTask.FromResult(message.ToUpperInvariant()); + return ValueTask.FromResult(NormalizeHandlerResult(message)); }; routeBuilder.AddHandler(asyncHandler); break; @@ -491,21 +491,21 @@ public sealed class RouteBuilderTests routeBuilder.AddCatchAll((message, context, cancellationToken) => { invocation.Capture(message, context, cancellationToken); - return GetPayloadValue(message).ToUpperInvariant(); + return NormalizeCatchAllResult(message); }); break; case 1: routeBuilder.AddCatchAll((message, context) => { invocation.Capture(message, context); - return GetPayloadValue(message).ToUpperInvariant(); + return NormalizeCatchAllResult(message); }); break; case 2: Func> asyncCatchAllWithCancellation = (message, context, cancellationToken) => { invocation.Capture(message, context, cancellationToken); - return ValueTask.FromResult(GetPayloadValue(message).ToUpperInvariant()); + return ValueTask.FromResult(NormalizeCatchAllResult(message)); }; routeBuilder.AddCatchAll(asyncCatchAllWithCancellation); break; @@ -513,7 +513,7 @@ public sealed class RouteBuilderTests Func> asyncCatchAll = (message, context) => { invocation.Capture(message, context); - return ValueTask.FromResult(GetPayloadValue(message).ToUpperInvariant()); + return ValueTask.FromResult(NormalizeCatchAllResult(message)); }; routeBuilder.AddCatchAll(asyncCatchAll); break; @@ -524,6 +524,10 @@ public sealed class RouteBuilderTests private static bool UsesCancellationToken(int overload) => overload is 0 or 2; + private static string NormalizeHandlerResult(string message) => message.ToUpperInvariant(); + + private static string NormalizeCatchAllResult(PortableValue message) => GetPayloadValue(message).ToUpperInvariant(); + private static string GetPayloadValue(PortableValue message) { return message.As() is TestPayload payload