NuGet.Client/test/TestUtilities/Test.Utility/ProjectManagement/TestNuGetProjectContext.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.

#nullable disable

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using System.Xml.Linq;
using NuGet.Common;
using NuGet.Packaging;
using NuGet.Packaging.Core;
using NuGet.Packaging.PackageExtraction;
using NuGet.ProjectManagement;

namespace Test.Utility
{
    public class TestNuGetProjectContext : IMSBuildNuGetProjectContext
    {
        private Guid _operationId;
        public TestExecutionContext TestExecutionContext { get; set; }
        public Lazy<List<string>> Logs { get; } = new Lazy<List<string>>();
        public bool EnableLogging { get; set; }

        private object _lock = new object();

        public void Log(MessageLevel level, string message, params object[] args)
        {
            // Uncomment when you want to debug tests.
            // Console.WriteLine(message, args);

            if (EnableLogging)
            {
                lock (_lock)
                {
                    Logs.Value.Add(args != null ? message + " " + string.Join(",", args) : message);
                }
            }
        }

        public void Log(ILogMessage message)
        {
            // Uncomment when you want to debug tests.
            // Console.WriteLine(message.FormatWithCode());

            if (EnableLogging)
            {
                lock (_lock)
                {
                    Logs.Value.Add(message.Message);
                }
            }
        }

        public FileConflictAction ResolveFileConflict(string message)
        {
            return FileConflictAction.IgnoreAll;
        }

        public PackageExtractionContext PackageExtractionContext { get; set; } = new PackageExtractionContext(
                        PackageSaveMode.Defaultv2,
                        PackageExtractionBehavior.XmlDocFileSaveMode,
                        clientPolicyContext: null,
                        logger: NullLogger.Instance);

        public ISourceControlManagerProvider SourceControlManagerProvider { get; set; }

        public NuGet.ProjectManagement.ExecutionContext ExecutionContext
        {
            get { return TestExecutionContext; }
        }

        public bool SkipAssemblyReferences { get; set; }

        public bool BindingRedirectsDisabled { get; set; }

        public XDocument OriginalPackagesConfig { get; set; }

        public void ReportError(string message)
        {
        }

        public void ReportError(ILogMessage message)
        {
        }

        public NuGetActionType ActionType { get; set; }

        public Guid OperationId
        {
            get
            {
                if (_operationId == Guid.Empty)
                {
                    _operationId = Guid.NewGuid();
                }
                return _operationId;
            }
            set
            {
                _operationId = value;
            }
        }
    }

    public class TestExecutionContext : NuGet.ProjectManagement.ExecutionContext
    {
        public TestExecutionContext(PackageIdentity directInstall)
        {
            FilesOpened = new HashSet<string>();
            DirectInstall = directInstall;
        }

        public HashSet<string> FilesOpened { get; }

        public override Task OpenFile(string fullPath)
        {
            FilesOpened.Add(fullPath);
            return Task.CompletedTask;
        }
    }
}