diff --git a/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/RouteBuilderTests.cs b/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/RouteBuilderTests.cs index 66235bb95d..a88fd4426b 100644 --- a/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/RouteBuilderTests.cs +++ b/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/RouteBuilderTests.cs @@ -491,21 +491,21 @@ public sealed class RouteBuilderTests routeBuilder.AddCatchAll((message, context, cancellationToken) => { invocation.Capture(message, context, cancellationToken); - return message.As()!.Value.ToUpperInvariant(); + return GetPayloadValue(message).ToUpperInvariant(); }); break; case 1: routeBuilder.AddCatchAll((message, context) => { invocation.Capture(message, context); - return message.As()!.Value.ToUpperInvariant(); + return GetPayloadValue(message).ToUpperInvariant(); }); break; case 2: Func> asyncCatchAllWithCancellation = (message, context, cancellationToken) => { invocation.Capture(message, context, cancellationToken); - return ValueTask.FromResult(message.As()!.Value.ToUpperInvariant()); + return ValueTask.FromResult(GetPayloadValue(message).ToUpperInvariant()); }; routeBuilder.AddCatchAll(asyncCatchAllWithCancellation); break; @@ -513,7 +513,7 @@ public sealed class RouteBuilderTests Func> asyncCatchAll = (message, context) => { invocation.Capture(message, context); - return ValueTask.FromResult(message.As()!.Value.ToUpperInvariant()); + return ValueTask.FromResult(GetPayloadValue(message).ToUpperInvariant()); }; routeBuilder.AddCatchAll(asyncCatchAll); break; @@ -523,4 +523,11 @@ public sealed class RouteBuilderTests } private static bool UsesCancellationToken(int overload) => overload is 0 or 2; + + private static string GetPayloadValue(PortableValue message) + { + TestPayload? payload = message.As(); + payload.Should().NotBeNull(); + return payload.Value; + } }