NuGet.Client/test/TestUtilities/VisualStudio.Test.Utility/VisualStudio/NuGetExperimentationServiceUtility.cs

// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.

using System.Collections.Generic;
using Microsoft.VisualStudio.Experimentation;
using Moq;

namespace Test.Utility.VisualStudio
{
    public static class NuGetExperimentationServiceUtility
    {
        public static IExperimentationService GetMock(Dictionary<string, bool>? flights = null)
        {
            var mock = new Mock<IExperimentationService>();
            Moq.Language.Flow.ISetup<IExperimentationService, bool> setupIsCachedFlightEnabled = mock.Setup(m => m.IsCachedFlightEnabled(It.IsAny<string>()));

            if (flights is null)
            {
                setupIsCachedFlightEnabled.Returns(false);
            }
            else
            {
                setupIsCachedFlightEnabled.Returns((string s) => flights.ContainsKey(s) && flights[s]);
            }

            return mock.Object;
        }
    }
}