// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
#nullable disable
 
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.ChangeSignature;
using Microsoft.CodeAnalysis.Editor.UnitTests.ChangeSignature;
using Microsoft.CodeAnalysis.Test.Utilities;
using Microsoft.CodeAnalysis.Test.Utilities.ChangeSignature;
using Xunit;
 
namespace Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.ChangeSignature;
 
[Trait(Traits.Feature, Traits.Features.ChangeSignature)]
public partial class ChangeSignatureTests : AbstractChangeSignatureTests
{
    [Fact]
    public async Task AddParameterAddsAllImports()
    {
        var markup = """
            class C
            {
                void $$M() { }
            }
            """;
 
        var updatedSignature = new[] {
            new AddedParameterOrExistingIndex(
                new AddedParameter(
                    null,
                    "Dictionary<ConsoleColor, Task<AsyncOperation>>",
                    "test",
                    CallSiteKind.Todo),
                "System.Collections.Generic.Dictionary<System.ConsoleColor, System.Threading.Tasks.Task<System.ComponentModel.AsyncOperation>>")};
 
        var updatedCode = """
            using System;
            using System.Collections.Generic;
            using System.ComponentModel;
            using System.Threading.Tasks;
 
            class C
            {
                void M(Dictionary<ConsoleColor, Task<AsyncOperation>> test) { }
            }
            """;
 
        await TestChangeSignatureViaCommandAsync(LanguageNames.CSharp, markup, updatedSignature: updatedSignature, expectedUpdatedInvocationDocumentCode: updatedCode);
    }
 
    [Fact]
    public async Task AddParameterAddsOnlyMissingImports()
    {
        var markup = """
            using System.ComponentModel;
 
            class C
            {
                void $$M() { }
            }
            """;
 
        var updatedSignature = new[] {
            new AddedParameterOrExistingIndex(
                new AddedParameter(
                    null,
                    "Dictionary<ConsoleColor, Task<AsyncOperation>>",
                    "test",
                    CallSiteKind.Todo),
                "System.Collections.Generic.Dictionary<System.ConsoleColor, System.Threading.Tasks.Task<System.ComponentModel.AsyncOperation>>")};
 
        var updatedCode = """
            using System;
            using System.Collections.Generic;
            using System.ComponentModel;
            using System.Threading.Tasks;
 
            class C
            {
                void M(Dictionary<ConsoleColor, Task<AsyncOperation>> test) { }
            }
            """;
 
        await TestChangeSignatureViaCommandAsync(LanguageNames.CSharp, markup, updatedSignature: updatedSignature, expectedUpdatedInvocationDocumentCode: updatedCode);
    }
 
    [Fact]
    public async Task AddParameterAddsImportsOnCascading()
    {
        var markup = """
            using NS1;
 
            namespace NS1
            {
                class B
                {
                    public virtual void M() { }
                }
            }
 
            namespace NS2
            {
                using System;
                using System.Collections.Generic;
                using System.ComponentModel;
                using System.Threading.Tasks;
 
                class D : B
                {
                    public override void $$M() { }
                }
            }
            """;
 
        var updatedSignature = new[] {
            new AddedParameterOrExistingIndex(
                new AddedParameter(
                    null,
                    "Dictionary<ConsoleColor, Task<AsyncOperation>>",
                    "test",
                    CallSiteKind.Todo),
                "System.Collections.Generic.Dictionary<System.ConsoleColor, System.Threading.Tasks.Task<System.ComponentModel.AsyncOperation>>")};
 
        var updatedCode = """
            using System;
            using System.Collections.Generic;
            using System.ComponentModel;
            using System.Threading.Tasks;
            using NS1;
 
            namespace NS1
            {
                class B
                {
                    public virtual void M(Dictionary<ConsoleColor, Task<AsyncOperation>> test) { }
                }
            }
 
            namespace NS2
            {
                using System;
                using System.Collections.Generic;
                using System.ComponentModel;
                using System.Threading.Tasks;
 
                class D : B
                {
                    public override void M(Dictionary<ConsoleColor, Task<AsyncOperation>> test) { }
                }
            }
            """;
 
        await TestChangeSignatureViaCommandAsync(LanguageNames.CSharp, markup, updatedSignature: updatedSignature, expectedUpdatedInvocationDocumentCode: updatedCode);
    }
}