src/modules/AdvancedPaste/AdvancedPaste.UnitTests/ServicesTests/AdvancedAIProviderResolverTests.cs - main - githubmirror/PowerToys

src/modules/AdvancedPaste/AdvancedPaste.UnitTests/ServicesTests/AdvancedAIProviderResolverTests.cs

55 строк2 KB

Gleb Khmyznikov

Advanced Paste additional customizations and PhiSilica provider (#46727)

08 авг 2026, 08:43

08 авг 2026, 08:43a87f5b0

// Copyright (c) Microsoft Corporation
// The Microsoft Corporation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System.Collections.ObjectModel;
using AdvancedPaste.Services;
using Microsoft.PowerToys.Settings.UI.Library;
using Microsoft.VisualStudio.TestTools.UnitTesting;

namespace AdvancedPaste.UnitTests.ServicesTests;

[TestClass]
public sealed class AdvancedAIProviderResolverTests
{
    [TestMethod]
    public void TryResolveAdvancedProvider_WithPhiSilicaOverrideAndAdvancedActiveProvider_ReturnsFalse()
    {
        var advancedProvider = CreateAdvancedProvider("advanced", AIServiceType.OpenAI);
        var phiSilicaProvider = new PasteAIProviderDefinition { Id = "phi", ServiceTypeKind = AIServiceType.PhiSilica };
        var configuration = CreateConfiguration(advancedProvider, advancedProvider, phiSilicaProvider);

        var result = AdvancedAIProviderResolver.TryResolveAdvancedProvider(configuration, phiSilicaProvider.Id, out var provider);

        Assert.IsFalse(result);
        Assert.IsNull(provider);
    }

    [TestMethod]
    public void TryResolveAdvancedProvider_WithNonActiveAdvancedOverride_ReturnsOverride()
    {
        var activeProvider = CreateAdvancedProvider("active", AIServiceType.OpenAI);
        var overrideProvider = CreateAdvancedProvider("override", AIServiceType.AzureOpenAI);
        var configuration = CreateConfiguration(activeProvider, activeProvider, overrideProvider);

        var result = AdvancedAIProviderResolver.TryResolveAdvancedProvider(configuration, overrideProvider.Id, out var provider);

        Assert.IsTrue(result);
        Assert.AreSame(overrideProvider, provider);
    }

    private static PasteAIProviderDefinition CreateAdvancedProvider(string id, AIServiceType serviceType) =>
        new()
        {
            Id = id,
            ServiceTypeKind = serviceType,
            EnableAdvancedAI = true,
        };

    private static PasteAIConfiguration CreateConfiguration(PasteAIProviderDefinition activeProvider, params PasteAIProviderDefinition[] providers) =>
        new()
        {
            ActiveProviderId = activeProvider.Id,
            Providers = new ObservableCollection<PasteAIProviderDefinition>(providers),
        };
}