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

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

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

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

/// <summary>
/// Tests for <see cref="ServiceIdAgentThread"/>.
/// </summary>
public class ServiceIdAgentThreadTests
{
    #region Constructor and Property Tests  

    [Fact]
    public void Constructor_SetsDefaults()
    {
        // Arrange & Act
        var thread = new TestServiceIdAgentThread();

        // Assert
        Assert.Null(thread.GetServiceThreadId());
    }

    [Fact]
    public void Constructor_WithServiceThreadId_SetsProperty()
    {
        // Arrange & Act
        var thread = new TestServiceIdAgentThread("service-id-123");

        // Assert
        Assert.Equal("service-id-123", thread.GetServiceThreadId());
    }

    [Fact]
    public void Constructor_WithSerializedId_SetsProperty()
    {
        // Arrange
        var json = JsonSerializer.SerializeToElement(new { ServiceThreadId = "service-id-456" });

        // Act
        var thread = new TestServiceIdAgentThread(json);

        // Assert
        Assert.Equal("service-id-456", thread.GetServiceThreadId());
    }

    [Fact]
    public void Constructor_WithSerializedUndefinedId_SetsProperty()
    {
        // Arrange
        var json = JsonSerializer.SerializeToElement(new { });

        // Act
        var thread = new TestServiceIdAgentThread(json);

        // Assert
        Assert.Null(thread.GetServiceThreadId());
    }

    [Fact]
    public void Constructor_WithInvalidJson_ThrowsArgumentException()
    {
        // Arrange
        var invalidJson = JsonSerializer.SerializeToElement(42);

        // Act & Assert
        Assert.Throws<ArgumentException>(() => new TestServiceIdAgentThread(invalidJson));
    }

    #endregion

    #region SerializeAsync Tests

    [Fact]
    public async Task SerializeAsync_ReturnsCorrectJson_WhenServiceThreadIdIsSetAsync()
    {
        // Arrange
        var thread = new TestServiceIdAgentThread("service-id-789");

        // Act
        var json = await thread.SerializeAsync();

        // Assert
        Assert.Equal(JsonValueKind.Object, json.ValueKind);
        Assert.True(json.TryGetProperty("serviceThreadId", out var idProperty));
        Assert.Equal("service-id-789", idProperty.GetString());
    }

    [Fact]
    public async Task SerializeAsync_ReturnsUndefinedServiceThreadId_WhenNotSetAsync()
    {
        // Arrange
        var thread = new TestServiceIdAgentThread();

        // Act
        var json = await thread.SerializeAsync();

        // Assert
        Assert.Equal(JsonValueKind.Object, json.ValueKind);
        Assert.False(json.TryGetProperty("serviceThreadId", out _));
    }

    #endregion

    // Sealed test subclass to expose protected members for testing
    private sealed class TestServiceIdAgentThread : ServiceIdAgentThread
    {
        public TestServiceIdAgentThread() : base() { }
        public TestServiceIdAgentThread(string serviceThreadId) : base(serviceThreadId) { }
        public TestServiceIdAgentThread(JsonElement serializedThreadState) : base(serializedThreadState) { }
        public string? GetServiceThreadId() => this.ServiceThreadId;
        public override Task<JsonElement> SerializeAsync(JsonSerializerOptions? jsonSerializerOptions = null, CancellationToken cancellationToken = default) => base.SerializeAsync(jsonSerializerOptions, cancellationToken);
    }
}