dotnet-installer/test/Microsoft.DotNet.Tools.Tests.Utilities/Commands/AddReferenceCommand.cs

// Copyright (c) .NET Foundation and contributors. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.

using Microsoft.DotNet.Cli.Utils;

namespace Microsoft.DotNet.Tools.Test.Utilities
{
    public sealed class AddReferenceCommand : TestCommand
    {
        private string _projectName = null;

        public AddReferenceCommand()
            : base("dotnet")
        {
        }

        public override CommandResult Execute(string args = "")
        {
            args = $"add {_projectName} reference {args}";
            return base.ExecuteWithCapturedOutput(args);
        }

        public AddReferenceCommand WithProject(string projectName)
        {
            _projectName = projectName;
            return this;
        }
    }
}