Files
microsoft--agent-framework/dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/MessageDeliveryValidation.cs
T
wehub-resource-sync db620d33df
CodeQL / Analyze (csharp) (push) Waiting to run
CodeQL / Analyze (python) (push) Waiting to run
dotnet-build-and-test / dotnet-test-functions (push) Has been cancelled
dotnet-build-and-test / paths-filter (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Debug, windows-latest, net9.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, ubuntu-latest, net10.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, ubuntu-latest, net8.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-build (Release, windows-latest, net472) (push) Has been cancelled
dotnet-build-and-test / dotnet-test (Release, integration, true, ubuntu-latest, net10.0) (push) Has been cancelled
dotnet-build-and-test / dotnet-test (Release, integration, true, windows-latest, net472) (push) Has been cancelled
dotnet-build-and-test / dotnet-foundry-hosted-it (push) Has been cancelled
dotnet-build-and-test / dotnet-build-and-test-check (push) Has been cancelled
dotnet-build-and-test / Integration Test Report (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 13:39:25 +08:00

74 lines
2.7 KiB
C#

// Copyright (c) Microsoft. All rights reserved.
using System;
using System.Collections.Generic;
using System.Linq;
using FluentAssertions;
using Microsoft.Agents.AI.Workflows.Execution;
namespace Microsoft.Agents.AI.Workflows.UnitTests;
internal static class MessageDeliveryValidation
{
public static void CheckDeliveries(this DeliveryMapping mapping, HashSet<string> receiverIds, HashSet<object> messages)
{
HashSet<string> unseenReceivers = [.. receiverIds];
HashSet<object> unseenMessages = [.. messages];
foreach (IGrouping<string, MessageDelivery> grouping in mapping.Deliveries.GroupBy(delivery => delivery.TargetId))
{
string receiverId = grouping.Key;
receiverIds.Should().Contain(receiverId);
unseenReceivers.Remove(grouping.Key);
foreach (MessageDelivery delivery in grouping)
{
object messageValue;
if (delivery.Envelope.Message is PortableValue portableValue)
{
portableValue.IsDelayedDeserialization.Should().BeFalse();
messageValue = portableValue.Value;
}
else
{
messageValue = delivery.Envelope.Message;
}
messages.Should().Contain(messageValue);
unseenMessages.Remove(messageValue);
}
}
unseenReceivers.Should().BeEmpty();
unseenMessages.Should().BeEmpty();
}
public static void CheckForwarded(Dictionary<string, List<MessageEnvelope>> queuedMessages, params (string expectedSender, List<string> expectedMessages)[] expectedForwards)
{
queuedMessages.Should().HaveCount(expectedForwards.Length);
IEnumerable<Action<string>> perSenderValidations = expectedForwards.Select(
(forward) =>
{
(string expectedSender, List<string> expectedMessages) = forward;
return (Action<string>)(
senderId =>
{
senderId.Should().Be(expectedSender);
queuedMessages[senderId].Should().HaveCount(expectedMessages.Count);
Action<MessageEnvelope>[] validations
= expectedMessages.Select(message => (Action<MessageEnvelope>)(envelope => envelope!.Message.Should().Be(message)))
.ToArray();
Assert.Collection(queuedMessages[senderId], validations);
});
}
);
Assert.Collection(queuedMessages.Keys, perSenderValidations.ToArray());
}
}