' 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.
 
Imports System.Threading.Tasks
Imports Microsoft.CodeAnalysis.CodeRefactorings
Imports Microsoft.CodeAnalysis.VisualBasic.AddFileBanner
Imports Microsoft.CodeAnalysis.Editor.VisualBasic.UnitTests.CodeRefactorings
Imports Roslyn.Test.Utilities
Imports Xunit
Imports Microsoft.CodeAnalysis.Editor.UnitTests.Workspaces
 
Namespace Microsoft.CodeAnalysis.Editor.VisualBasic.UnitTests.AddFileBanner
    <Trait(Traits.Feature, Traits.Features.CodeActionsAddFileBanner)>
    Partial Public Class AddFileBannerTests
        Inherits AbstractVisualBasicCodeActionTest_NoEditor
 
        Protected Overrides Function CreateCodeRefactoringProvider(workspace As TestWorkspace, parameters As TestParameters) As CodeRefactoringProvider
            Return New VisualBasicAddFileBannerCodeRefactoringProvider()
        End Function
 
        <Fact>
        Public Async Function TestBanner1() As Task
            Await TestInRegularAndScriptAsync(
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>[||]Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>' This is the banner
 
class Program2
end class
        </Document>
    </Project>
</Workspace>",
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>' This is the banner
 
Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>' This is the banner
 
class Program2
end class
        </Document>
    </Project>
</Workspace>")
        End Function
 
        <Fact>
        Public Async Function TestMultiLineBanner1() As Task
            Await TestInRegularAndScriptAsync(
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>[||]Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>' This is the banner
' It goes over multiple lines
 
class Program2
end class
        </Document>
    </Project>
</Workspace>",
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>' This is the banner
' It goes over multiple lines
 
Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>' This is the banner
' It goes over multiple lines
 
class Program2
end class
        </Document>
    </Project>
</Workspace>")
        End Function
 
        <Fact, WorkItem("https://github.com/dotnet/roslyn/issues/33251")>
        Public Async Function TestSingleLineDocCommentBanner() As Task
            Await TestInRegularAndScriptAsync(
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>[||]Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>''' This is the banner
''' It goes over multiple lines
 
class Program2
end class
        </Document>
    </Project>
</Workspace>",
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>''' This is the banner
''' It goes over multiple lines
 
Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>''' This is the banner
''' It goes over multiple lines
 
class Program2
end class
        </Document>
    </Project>
</Workspace>")
        End Function
 
        <Fact, WorkItem("https://github.com/dotnet/roslyn/issues/32792")>
        Public Async Function TestUpdateFileNameInComment() As Task
            Await TestInRegularAndScriptAsync(
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document FilePath=""Goo.vb"">[||]Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document FilePath=""Bar.vb"">' This is the banner in Bar.vb
' It goes over multiple lines.  This line has Baz.vb
' The last line includes Bar.vb
 
class Program2
end class
        </Document>
    </Project>
</Workspace>",
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document FilePath=""Goo.vb"">' This is the banner in Goo.vb
' It goes over multiple lines.  This line has Baz.vb
' The last line includes Goo.vb
 
Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document FilePath=""Bar.vb"">' This is the banner in Bar.vb
' It goes over multiple lines.  This line has Baz.vb
' The last line includes Bar.vb
 
class Program2
end class
        </Document>
    </Project>
</Workspace>")
        End Function
 
        <Fact, WorkItem("https://github.com/dotnet/roslyn/issues/33251")>
        Public Async Function TestUpdateFileNameInComment2() As Task
            Await TestInRegularAndScriptAsync(
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document FilePath=""Goo.vb"">[||]Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document FilePath=""Bar.vb"">''' This is the banner in Bar.vb
''' It goes over multiple lines.  This line has Baz.vb
''' The last line includes Bar.vb
 
class Program2
end class
        </Document>
    </Project>
</Workspace>",
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document FilePath=""Goo.vb"">''' This is the banner in Goo.vb
''' It goes over multiple lines.  This line has Baz.vb
''' The last line includes Goo.vb
 
Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document FilePath=""Bar.vb"">''' This is the banner in Bar.vb
''' It goes over multiple lines.  This line has Baz.vb
''' The last line includes Bar.vb
 
class Program2
end class
        </Document>
    </Project>
</Workspace>")
        End Function
 
        <Fact>
        Public Async Function TestMissingWhenAlreadyThere() As Task
            Await TestMissingAsync(
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>[||]' I already have a banner
 
Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>' This is the banner
 
class Program2
end class
        </Document>
    </Project>
</Workspace>")
        End Function
 
        <Theory>
        <InlineData("", 1)>
        <InlineData("file_header_template =", 1)>
        <InlineData("file_header_template = unset", 1)>
        <InlineData("file_header_template = defined file header", 0)>
        Public Async Function TestMissingWhenHandledByAnalyzer(fileHeaderTemplate As String, expectedActionCount As Integer) As Task
            Dim initialMarkup = $"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document FilePath=""/0/Test0.vb"">[||]Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document FilePath=""/0/Test1.vb"">' This is the banner
 
class Program2
end class
        </Document>
        <AnalyzerConfigDocument FilePath=""/.editorconfig"">
root = true
 
[*]
{fileHeaderTemplate}
        </AnalyzerConfigDocument>
    </Project>
</Workspace>"
 
            Await TestActionCountAsync(initialMarkup, expectedActionCount)
        End Function
 
        <Fact>
        Public Async Function TestMissingIfOtherFileDoesNotHaveBanner() As Task
            Await TestMissingAsync(
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>[||]
 
Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>
 
class Program2
end class
        </Document>
    </Project>
</Workspace>")
        End Function
 
        <Fact>
        Public Async Function TestMissingIfOtherFileIsAutoGenerated() As Task
            Await TestMissingAsync(
"
<Workspace>
    <Project Language=""Visual Basic"" AssemblyName=""Assembly1"" CommonReferences=""true"">
        <Document>[||]
 
Imports System
 
class Program1
    sub Main()
    end sub
end class
        </Document>
        <Document>' &lt;autogenerated /&gt;
 
class Program2
end class
        </Document>
    </Project>
</Workspace>")
        End Function
    End Class
End Namespace