agent-framework/dotnet/tests/Microsoft.Extensions.AI.Agents.Abstractions.UnitTests/ChatMessageStoreTests.cs

// Copyright (c) Microsoft. All rights reserved.

using System;
using System.Collections.Generic;
using System.Text.Json;
using System.Threading;
using System.Threading.Tasks;

namespace Microsoft.Extensions.AI.Agents.Abstractions.UnitTests;

/// <summary>
/// Contains tests for the <see cref="ChatMessageStore"/> class.
/// </summary>
public class ChatMessageStoreTests
{
    #region GetService Method Tests

    [Fact]
    public void GetService_RequestingExactStoreType_ReturnsStore()
    {
        var store = new TestChatMessageStore();
        var result = store.GetService(typeof(TestChatMessageStore));
        Assert.NotNull(result);
        Assert.Same(store, result);
    }

    [Fact]
    public void GetService_RequestingBaseStoreType_ReturnsStore()
    {
        var store = new TestChatMessageStore();
        var result = store.GetService(typeof(ChatMessageStore));
        Assert.NotNull(result);
        Assert.Same(store, result);
    }

    [Fact]
    public void GetService_RequestingUnrelatedType_ReturnsNull()
    {
        var store = new TestChatMessageStore();
        var result = store.GetService(typeof(string));
        Assert.Null(result);
    }

    [Fact]
    public void GetService_WithServiceKey_ReturnsNull()
    {
        var store = new TestChatMessageStore();
        var result = store.GetService(typeof(TestChatMessageStore), "some-key");
        Assert.Null(result);
    }

    [Fact]
    public void GetService_WithNullServiceType_ThrowsArgumentNullException()
    {
        var store = new TestChatMessageStore();
        Assert.Throws<ArgumentNullException>(() => store.GetService(null!));
    }

    [Fact]
    public void GetService_Generic_ReturnsCorrectType()
    {
        var store = new TestChatMessageStore();
        var result = store.GetService<TestChatMessageStore>();
        Assert.NotNull(result);
        Assert.Same(store, result);
    }

    [Fact]
    public void GetService_Generic_ReturnsNullForUnrelatedType()
    {
        var store = new TestChatMessageStore();
        var result = store.GetService<string>();
        Assert.Null(result);
    }

    #endregion

    private sealed class TestChatMessageStore : ChatMessageStore
    {
        public override Task<IEnumerable<ChatMessage>> GetMessagesAsync(CancellationToken cancellationToken = default)
            => Task.FromResult<IEnumerable<ChatMessage>>(Array.Empty<ChatMessage>());

        public override Task AddMessagesAsync(IEnumerable<ChatMessage> messages, CancellationToken cancellationToken = default)
            => Task.CompletedTask;

        public override ValueTask<JsonElement?> SerializeStateAsync(JsonSerializerOptions? jsonSerializerOptions = null, CancellationToken cancellationToken = default)
            => default;
    }
}