agent-framework/dotnet/tests/Microsoft.Agents.AI.Foundry.Hosting.UnitTests/HostedFoundryMemoryProviderScopesTests.cs

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

using System;

namespace Microsoft.Agents.AI.Foundry.Hosting.UnitTests;

/// <summary>
/// Tests for <see cref="HostedFoundryMemoryProviderScopes"/> built-in stateInitializer factories.
/// </summary>
public class HostedFoundryMemoryProviderScopesTests
{
    private const string TestUserId = "user-isolation-key-1";
    private const string TestChatId = "chat-isolation-key-1";

    [Fact]
    public void PerUser_UsesUserIdAsScope()
    {
        // Arrange
        var session = CreateTaggedSession(TestUserId, TestChatId);
        var initializer = HostedFoundryMemoryProviderScopes.PerUser();

        // Act
        var state = initializer(session);

        // Assert
        Assert.NotNull(state);
        Assert.Equal(TestUserId, state.Scope.Scope);
    }

    [Fact]
    public void PerChat_UsesChatIdAsScope()
    {
        // Arrange
        var session = CreateTaggedSession(TestUserId, TestChatId);
        var initializer = HostedFoundryMemoryProviderScopes.PerChat();

        // Act
        var state = initializer(session);

        // Assert
        Assert.NotNull(state);
        Assert.Equal(TestChatId, state.Scope.Scope);
    }

    [Fact]
    public void PerUserAndChat_ComposesUserAndChatWithEscapedSeparator()
    {
        // Arrange
        var session = CreateTaggedSession(TestUserId, TestChatId);
        var initializer = HostedFoundryMemoryProviderScopes.PerUserAndChat();

        // Act
        var state = initializer(session);

        // Assert
        Assert.NotNull(state);
        Assert.Equal($"{TestUserId}::{TestChatId}", state.Scope.Scope);
    }

    [Fact]
    public void PerUserAndChat_EscapesColonsInUserAndChat()
    {
        // Arrange
        var session = CreateTaggedSession("alice:finance", "q2:final");
        var initializer = HostedFoundryMemoryProviderScopes.PerUserAndChat();

        // Act
        var state = initializer(session);

        // Assert - colons inside each part are escaped as \: , parts joined with ::
        Assert.Equal(@"alice\:finance::q2\:final", state.Scope.Scope);
    }

    [Fact]
    public void PerUserAndChat_EscapesBackslashesInUserAndChat()
    {
        // Arrange
        var session = CreateTaggedSession(@"alice\corp", @"chat\1");
        var initializer = HostedFoundryMemoryProviderScopes.PerUserAndChat();

        // Act
        var state = initializer(session);

        // Assert - backslashes escaped first as \\ , parts joined with ::
        Assert.Equal(@"alice\\corp::chat\\1", state.Scope.Scope);
    }

    [Fact]
    public void PerUserAndChat_DistinctContextsDoNotCollide()
    {
        // Arrange - two distinct (UserId, ChatId) pairs that collide under raw-colon composition.
        var sessionA = CreateTaggedSession("alice:finance", "q2");
        var sessionB = CreateTaggedSession("alice", "finance:q2");
        var initializer = HostedFoundryMemoryProviderScopes.PerUserAndChat();

        // Act
        var scopeA = initializer(sessionA).Scope.Scope;
        var scopeB = initializer(sessionB).Scope.Scope;

        // Assert
        Assert.NotEqual(scopeA, scopeB);
    }

    [Fact]
    public void PerUser_NullSession_Throws()
    {
        // Arrange
        var initializer = HostedFoundryMemoryProviderScopes.PerUser();

        // Act & Assert
        var ex = Assert.Throws<InvalidOperationException>(() => initializer(null));
        Assert.Contains(nameof(HostedSessionContext), ex.Message);
    }

    [Fact]
    public void PerChat_NullSession_Throws()
    {
        // Arrange
        var initializer = HostedFoundryMemoryProviderScopes.PerChat();

        // Act & Assert
        Assert.Throws<InvalidOperationException>(() => initializer(null));
    }

    [Fact]
    public void PerUserAndChat_NullSession_Throws()
    {
        // Arrange
        var initializer = HostedFoundryMemoryProviderScopes.PerUserAndChat();

        // Act & Assert
        Assert.Throws<InvalidOperationException>(() => initializer(null));
    }

    [Fact]
    public void PerUser_SessionWithoutHostedContext_Throws()
    {
        // Arrange
        var session = new BareAgentSession();
        var initializer = HostedFoundryMemoryProviderScopes.PerUser();

        // Act & Assert
        var ex = Assert.Throws<InvalidOperationException>(() => initializer(session));
        Assert.Contains(nameof(HostedFoundryMemoryProviderScopes), ex.Message);
    }

    private static BareAgentSession CreateTaggedSession(string userId, string chatId)
    {
        var session = new BareAgentSession();
        session.SetHostedContext(new HostedSessionContext(userId, chatId));
        return session;
    }

    private sealed class BareAgentSession : AgentSession
    {
        public BareAgentSession() : base(new AgentSessionStateBag()) { }
    }
}