// 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.
 
using System.Threading.Tasks;
using Xunit;
using CSharpLanguageVersion = Microsoft.CodeAnalysis.CSharp.LanguageVersion;
using VerifyCS = Test.Utilities.CSharpSecurityCodeFixVerifier<
    Roslyn.Diagnostics.Analyzers.TemporaryArrayAsRefAnalyzer,
    Microsoft.CodeAnalysis.Testing.EmptyCodeFixProvider>;
using VerifyVB = Test.Utilities.VisualBasicSecurityCodeFixVerifier<
    Roslyn.Diagnostics.Analyzers.TemporaryArrayAsRefAnalyzer,
    Microsoft.CodeAnalysis.Testing.EmptyCodeFixProvider>;
using VisualBasicLanguageVersion = Microsoft.CodeAnalysis.VisualBasic.LanguageVersion;
 
namespace Roslyn.Diagnostics.Analyzers.UnitTests
{
    public class TemporaryArrayAsRefTests
    {
        public const string TemporaryArraySource_CSharp = """
 
            namespace Microsoft.CodeAnalysis.Shared.Collections
            {
                internal struct TemporaryArray<T> : System.IDisposable
                {
                    public void Dispose() { }
                }
 
                internal static class TemporaryArrayExtensions
                {
                    public static ref TemporaryArray<T> AsRef<T>(this in TemporaryArray<T> array) => throw null;
                }
            }
 
            """;
        public const string TemporaryArraySource_VisualBasic = """
 
            Namespace Microsoft.CodeAnalysis.Shared.Collections
                Friend Structure TemporaryArray(Of T)
                    Implements System.IDisposable
 
                    Public Sub Dispose() Implements System.IDisposable.Dispose
                    End Sub
                End Structure
 
                Friend Module TemporaryArrayExtensions
                    <System.Runtime.CompilerServices.Extension>
                    Public Function AsRef(Of T)(array As TemporaryArray(Of T)) As TemporaryArray(Of T)
                        Return Nothing
                    End Function
                End Module
            End Namespace
 
            """;
 
        [Fact]
        public async Task TestUsingVariable_CSharpAsync()
        {
            var code = """
 
                using Microsoft.CodeAnalysis.Shared.Collections;
 
                class C
                {
                    void Method()
                    {
                        using (var array = new TemporaryArray<int>())
                        {
                            ref var arrayRef1 = ref array.AsRef();
                            ref var arrayRef2 = ref TemporaryArrayExtensions.AsRef(in array);
                        }
                    }
                }
 
 
                """ + TemporaryArraySource_CSharp;
 
            await new VerifyCS.Test
            {
                LanguageVersion = CSharpLanguageVersion.CSharp9,
                TestCode = code,
            }.RunAsync();
        }
 
        [Fact]
        public async Task TestUsingVariable_VisualBasicAsync()
        {
            var code = """
 
                Imports Microsoft.CodeAnalysis.Shared.Collections
 
                Class C
                    Sub Method()
                        Using array = New TemporaryArray(Of Integer)()
                            Dim arrayRef1 = array.AsRef()
                            Dim arrayRef2 = TemporaryArrayExtensions.AsRef(array)
                        End Using
                    End Sub
                End Class
 
 
                """ + TemporaryArraySource_VisualBasic;
 
            await new VerifyVB.Test
            {
                LanguageVersion = VisualBasicLanguageVersion.VisualBasic16,
                TestCode = code,
            }.RunAsync();
        }
 
        [Fact]
        public async Task TestUsingDeclarationVariable_CSharpAsync()
        {
            var code = """
 
                using Microsoft.CodeAnalysis.Shared.Collections;
 
                class C
                {
                    void Method()
                    {
                        using var array = new TemporaryArray<int>();
                        ref var arrayRef1 = ref array.AsRef();
                        ref var arrayRef2 = ref TemporaryArrayExtensions.AsRef(in array);
                    }
                }
 
 
                """ + TemporaryArraySource_CSharp;
 
            await new VerifyCS.Test
            {
                LanguageVersion = CSharpLanguageVersion.CSharp9,
                TestCode = code,
            }.RunAsync();
        }
 
        [Fact]
        public async Task TestNonUsingVariable_CSharpAsync()
        {
            var code = """
 
                using Microsoft.CodeAnalysis.Shared.Collections;
 
                class C
                {
                    void Method()
                    {
                        var array = new TemporaryArray<int>();
                        ref var arrayRef1 = ref [|array.AsRef()|];
                        ref var arrayRef2 = ref [|TemporaryArrayExtensions.AsRef(in array)|];
                    }
                }
 
 
                """ + TemporaryArraySource_CSharp;
 
            await new VerifyCS.Test
            {
                LanguageVersion = CSharpLanguageVersion.CSharp9,
                TestCode = code,
            }.RunAsync();
        }
 
        [Fact]
        public async Task TestNonUsingVariable_VisualBasicAsync()
        {
            var code = """
 
                Imports Microsoft.CodeAnalysis.Shared.Collections
 
                Class C
                    Sub Method()
                        Dim array = New TemporaryArray(Of Integer)()
                        Dim arrayRef1 = [|array.AsRef()|]
                        Dim arrayRef2 = [|TemporaryArrayExtensions.AsRef(array)|]
                    End Sub
                End Class
 
 
                """ + TemporaryArraySource_VisualBasic;
 
            await new VerifyVB.Test
            {
                LanguageVersion = VisualBasicLanguageVersion.VisualBasic16,
                TestCode = code,
            }.RunAsync();
        }
    }
}