autogen/dotnet/Microsoft.AutoGen.Agents.Tests/AgentBaseTests.cs

// Copyright (c) Microsoft Corporation. All rights reserved.
// AgentBaseTests.cs

using FluentAssertions;
using Google.Protobuf.Reflection;
using Microsoft.AutoGen.Abstractions;
using Moq;
using Xunit;

namespace Microsoft.AutoGen.Agents.Tests;

public class AgentBaseTests
{
    [Fact]
    public async Task ItInvokeRightHandlerTestAsync()
    {
        var mockContext = new Mock<IAgentContext>();
        var agent = new TestAgent(mockContext.Object, new EventTypes(TypeRegistry.Empty, [], []));

        await agent.HandleObject("hello world");
        await agent.HandleObject(42);

        agent.ReceivedItems.Should().HaveCount(2);
        agent.ReceivedItems[0].Should().Be("hello world");
        agent.ReceivedItems[1].Should().Be(42);
    }

    /// <summary>
    /// The test agent is a simple agent that is used for testing purposes.
    /// </summary>
    public class TestAgent : AgentBase, IHandle<string>, IHandle<int>
    {
        public TestAgent(IAgentContext context, EventTypes eventTypes) : base(context, eventTypes)
        {
        }

        public Task Handle(string item)
        {
            ReceivedItems.Add(item);
            return Task.CompletedTask;
        }

        public Task Handle(int item)
        {
            ReceivedItems.Add(item);
            return Task.CompletedTask;
        }

        public List<object> ReceivedItems { get; private set; } = [];
    }
}