File: CodeGeneration\CodeGenerationTests.VisualBasic.cs
Web Access
Project: src\src\EditorFeatures\Test\Microsoft.CodeAnalysis.EditorFeatures.UnitTests.csproj (Microsoft.CodeAnalysis.EditorFeatures.UnitTests)
// 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;
using System.Collections.Immutable;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis;
using Microsoft.CodeAnalysis.CodeGeneration;
using Microsoft.CodeAnalysis.Editing;
using Microsoft.CodeAnalysis.Test.Utilities;
using Microsoft.CodeAnalysis.VisualBasic.Syntax;
using Roslyn.Test.Utilities;
using Roslyn.Utilities;
using Xunit;
using VB = Microsoft.CodeAnalysis.VisualBasic;
 
namespace Microsoft.CodeAnalysis.Editor.UnitTests.CodeGeneration;
 
public partial class CodeGenerationTests
{
    [UseExportProvider]
    public class VisualBasic
    {
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddNamespace()
        {
            var input = "Namespace [|N1|]\n End Namespace";
            var expected = @"Namespace N1
    Namespace N2
    End Namespace
End Namespace";
            await TestAddNamespaceAsync(input, expected,
                name: "N2");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddField()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public F As Integer
End Class";
            await TestAddFieldAsync(input, expected,
                type: GetTypeSymbol(typeof(int)));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddSharedField()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Private Shared F As String
End Class";
            await TestAddFieldAsync(input, expected,
                type: GetTypeSymbol(typeof(string)),
                accessibility: Accessibility.Private,
                modifiers: new DeclarationModifiers(isStatic: true));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddArrayField()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public F As Integer()
End Class";
            await TestAddFieldAsync(input, expected,
                type: CreateArrayType(typeof(int)));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddConstructor()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Sub New()
    End Sub
End Class";
            await TestAddConstructorAsync(input, expected);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        [WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/530785")]
        public async Task AddConstructorWithXmlComment()
        {
            var input = @"
Public Class [|C|]
''' <summary>
''' Do Nothing
''' </summary>
Public Sub GetStates()
End Sub
End Class";
            var expected = @"
Public Class C
    Public Sub New()
    End Sub
    ''' <summary>
    ''' Do Nothing
    ''' </summary>
    Public Sub GetStates()
End Sub
End Class";
            await TestAddConstructorAsync(input, expected);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddConstructorWithoutBody()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Sub New()
End Class";
            await TestAddConstructorAsync(input, expected,
                context: new CodeGenerationContext(generateMethodBodies: false));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddConstructorResolveNamespaceImport()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Imports System.Text
 
Class C
    Public Sub New(s As StringBuilder)
    End Sub
End Class";
            await TestAddConstructorAsync(input, expected,
                parameters: Parameters(Parameter(typeof(StringBuilder), "s")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddSharedConstructor()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Shared Sub New()
    End Sub
End Class";
            await TestAddConstructorAsync(input, expected,
                modifiers: new DeclarationModifiers(isStatic: true));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddChainedConstructor()
        {
            var input = "Class [|C|]\n Public Sub New(i As Integer)\n End Sub\n End Class";
            var expected = @"Class C
    Public Sub New()
        Me.New(42)
    End Sub
 
    Public Sub New(i As Integer)
 End Sub
 End Class";
            await TestAddConstructorAsync(input, expected,
                thisArguments: ImmutableArray.Create<SyntaxNode>(VB.SyntaxFactory.ParseExpression("42")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/544476")]
        public async Task AddClass()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Public Class C
    End Class
End Namespace";
            await TestAddNamedTypeAsync(input, expected);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddClassEscapeName()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Public Class [Class]
    End Class
End Namespace";
            await TestAddNamedTypeAsync(input, expected,
                name: "Class");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddClassUnicodeName()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Public Class [Class]
    End Class
End Namespace";
            await TestAddNamedTypeAsync(input, expected,
                name: "Class");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/544477")]
        public async Task AddNotInheritableClass()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Public NotInheritable Class C
    End Class
End Namespace";
            await TestAddNamedTypeAsync(input, expected,
                modifiers: new DeclarationModifiers(isSealed: true));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/544477")]
        public async Task AddMustInheritClass()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Friend MustInherit Class C
    End Class
End Namespace";
            await TestAddNamedTypeAsync(input, expected,
                accessibility: Accessibility.Internal,
                modifiers: new DeclarationModifiers(isAbstract: true));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddStructure()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Friend Structure S
    End Structure
End Namespace";
            await TestAddNamedTypeAsync(input, expected,
                name: "S",
                accessibility: Accessibility.Internal,
                typeKind: TypeKind.Struct);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/546224")]
        public async Task AddSealedStructure()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Public Structure S
    End Structure
End Namespace";
            await TestAddNamedTypeAsync(input, expected,
                name: "S",
                accessibility: Accessibility.Public,
                modifiers: new DeclarationModifiers(isSealed: true),
                typeKind: TypeKind.Struct);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddInterface()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Public Interface I
    End Interface
End Namespace";
            await TestAddNamedTypeAsync(input, expected,
                name: "I",
                typeKind: TypeKind.Interface);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/544528")]
        public async Task AddEnum()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Public Enum E
        F1
    End Enum
End Namespace";
            await TestAddNamedTypeAsync(input, expected, "E",
                typeKind: TypeKind.Enum,
                members: Members(CreateEnumField("F1", null)));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/544527")]
        public async Task AddEnumWithValues()
        {
            var input = "Namespace [|N|]\n End Namespace";
            var expected = @"Namespace N
    Public Enum E
        F1 = 1
        F2 = 2
    End Enum
End Namespace";
            await TestAddNamedTypeAsync(input, expected, "E",
                typeKind: TypeKind.Enum,
                members: Members(CreateEnumField("F1", 1), CreateEnumField("F2", 2)));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddEnumMember()
        {
            var input = "Public Enum [|E|]\n F1 = 1\n F2 = 2\n End Enum";
            var expected = @"Public Enum E
 F1 = 1
 F2 = 2
    F3
End Enum";
            await TestAddFieldAsync(input, expected,
                name: "F3");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddEnumMemberWithValue()
        {
            var input = "Public Enum [|E|]\n F1 = 1\n F2\n End Enum";
            var expected = @"Public Enum E
 F1 = 1
 F2
    F3 = 3
End Enum";
            await TestAddFieldAsync(input, expected,
                name: "F3", hasConstantValue: true, constantValue: 3);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/544529")]
        public async Task AddDelegateType()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Delegate Function D(s As String) As Integer
End Class";
            await TestAddDelegateTypeAsync(input, expected,
                returnType: typeof(int),
                parameters: Parameters(Parameter(typeof(string), "s")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/546224")]
        public async Task AddSealedDelegateType()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Delegate Function D(s As String) As Integer
End Class";
            await TestAddDelegateTypeAsync(input, expected,
                returnType: typeof(int),
                modifiers: new DeclarationModifiers(isSealed: true),
                parameters: Parameters(Parameter(typeof(string), "s")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddEvent()
        {
            var input = @"
Class [|C|]
End Class";
            var expected = @"
Class C
    Public Event E As Action
End Class";
            await TestAddEventAsync(input, expected,
                context: new CodeGenerationContext(addImports: false));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddCustomEventToClassFromSourceSymbol()
        {
            var sourceGenerated = @"Public Class [|C2|]
    Public Custom Event Click As EventHandler
        AddHandler(ByVal value As EventHandler)
            Events.AddHandler(""ClickEvent"", value)
        End AddHandler
        RemoveHandler(ByVal value As EventHandler)
            Events.RemoveHandler(""ClickEvent"", value)
        End RemoveHandler
        RaiseEvent(ByVal sender As Object, ByVal e As EventArgs)
            CType(Events(""ClickEvent""), EventHandler).Invoke(sender, e)
        End RaiseEvent
    End Event
End Class";
            var input = "Class [|C1|]\nEnd Class";
            var expected = @"Class C1
    Public Custom Event Click As EventHandler
        AddHandler(ByVal value As EventHandler)
            Events.AddHandler(""ClickEvent"", value)
        End AddHandler
        RemoveHandler(ByVal value As EventHandler)
            Events.RemoveHandler(""ClickEvent"", value)
        End RemoveHandler
        RaiseEvent(ByVal sender As Object, ByVal e As EventArgs)
            CType(Events(""ClickEvent""), EventHandler).Invoke(sender, e)
        End RaiseEvent
    End Event
End Class";
            var context = new CodeGenerationContext(reuseSyntax: true);
            await TestGenerateFromSourceSymbolAsync(sourceGenerated, input, expected, onlyGenerateMembers: true, context: context);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddEventWithAccessorAndImplementsClause()
        {
            var input = "Class [|C|] \n End Class";
            var expected = @"Class C
    Public Custom Event E As ComponentModel.PropertyChangedEventHandler Implements ComponentModel.INotifyPropertyChanged.PropertyChanged
        AddHandler(value As ComponentModel.PropertyChangedEventHandler)
        End AddHandler
        RemoveHandler(value As ComponentModel.PropertyChangedEventHandler)
        End RemoveHandler
        RaiseEvent(sender As Object, e As ComponentModel.PropertyChangedEventArgs)
        End RaiseEvent
    End Event
End Class";
            static ImmutableArray<IEventSymbol> GetExplicitInterfaceEvent(SemanticModel semanticModel)
            {
                return ImmutableArray.Create<IEventSymbol>(
                    new CodeGenerationEventSymbol(
                        GetTypeSymbol(typeof(System.ComponentModel.INotifyPropertyChanged))(semanticModel),
                        attributes: default,
                        Accessibility.Public,
                        modifiers: default,
                        GetTypeSymbol(typeof(System.ComponentModel.PropertyChangedEventHandler))(semanticModel),
                        explicitInterfaceImplementations: default,
                        nameof(System.ComponentModel.INotifyPropertyChanged.PropertyChanged), null, null, null));
            }
 
            await TestAddEventAsync(input, expected,
                addMethod: CodeGenerationSymbolFactory.CreateAccessorSymbol(
                    ImmutableArray<AttributeData>.Empty, Accessibility.NotApplicable, ImmutableArray<SyntaxNode>.Empty),
                    getExplicitInterfaceImplementations: GetExplicitInterfaceEvent,
                    type: typeof(System.ComponentModel.PropertyChangedEventHandler),
                    context: new CodeGenerationContext(addImports: false));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddEventWithAddAccessor()
        {
            var input = @"
Class [|C|]
End Class";
            var expected = @"
Class C
    Public Custom Event E As Action
        AddHandler(value As Action)
        End AddHandler
        RemoveHandler(value As Action)
        End RemoveHandler
        RaiseEvent()
        End RaiseEvent
    End Event
End Class";
            await TestAddEventAsync(input, expected,
                addMethod: CodeGenerationSymbolFactory.CreateAccessorSymbol(ImmutableArray<AttributeData>.Empty, Accessibility.NotApplicable, ImmutableArray<SyntaxNode>.Empty),
                context: new CodeGenerationContext(addImports: false));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddEventWithAccessors()
        {
            var input = @"
Class [|C|]
End Class";
            var expected = @"
Class C
    Public Custom Event E As Action
        AddHandler(value As Action)
            Console.WriteLine(0)
        End AddHandler
        RemoveHandler(value As Action)
            Console.WriteLine(1)
        End RemoveHandler
        RaiseEvent()
            Console.WriteLine(2)
        End RaiseEvent
    End Event
End Class";
            var addStatements = ImmutableArray.Create<SyntaxNode>(VB.SyntaxFactory.ParseExecutableStatement("Console.WriteLine(0)"));
            var removeStatements = ImmutableArray.Create<SyntaxNode>(VB.SyntaxFactory.ParseExecutableStatement("Console.WriteLine(1)"));
            var raiseStatements = ImmutableArray.Create<SyntaxNode>(VB.SyntaxFactory.ParseExecutableStatement("Console.WriteLine(2)"));
            await TestAddEventAsync(input, expected,
                addMethod: CodeGenerationSymbolFactory.CreateAccessorSymbol(
                    ImmutableArray<AttributeData>.Empty, Accessibility.NotApplicable, addStatements),
                removeMethod: CodeGenerationSymbolFactory.CreateAccessorSymbol(
                    ImmutableArray<AttributeData>.Empty, Accessibility.NotApplicable, removeStatements),
                raiseMethod: CodeGenerationSymbolFactory.CreateAccessorSymbol(
                    ImmutableArray<AttributeData>.Empty, Accessibility.NotApplicable, raiseStatements),
                context: new CodeGenerationContext(addImports: false));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddMethodToClass()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Sub M()
    End Sub
End Class";
            await TestAddMethodAsync(input, expected,
                returnType: typeof(void));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddMethodToClassFromSourceSymbol()
        {
            var sourceGenerated = @"Public Class [|C2|]
    Public Function FInt() As Integer
        Return 0
    End Function
End Class";
            var input = "Class [|C1|]\nEnd Class";
            var expected = @"Class C1
    Public Function FInt() As Integer
        Return 0
    End Function
End Class";
            var context = new CodeGenerationContext(reuseSyntax: true);
            await TestGenerateFromSourceSymbolAsync(sourceGenerated, input, expected, onlyGenerateMembers: true, context: context);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddMethodToClassEscapedName()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Protected Friend Sub [Sub]()
    End Sub
End Class";
            await TestAddMethodAsync(input, expected,
                accessibility: Accessibility.ProtectedOrInternal,
                name: "Sub",
                returnType: typeof(void));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration), WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/544477")]
        public async Task AddSharedMethodToStructure()
        {
            var input = "Structure [|S|]\n End Structure";
            var expected = @"Structure S
    Public Shared Function M() As Integer
        Return 0
    End Function
End Structure";
            await TestAddMethodAsync(input, expected,
                modifiers: new DeclarationModifiers(isStatic: true),
                returnType: typeof(int),
                statements: "Return 0");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddNotOverridableOverridesMethod()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public NotOverridable Overrides Function GetHashCode() As Integer
        $$
    End Function
End Class";
            await TestAddMethodAsync(input, expected,
                name: "GetHashCode",
                modifiers: new DeclarationModifiers(isOverride: true, isSealed: true),
                returnType: typeof(int),
                statements: "Return 0");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddMustOverrideMethod()
        {
            var input = "MustInherit Class [|C|]\n End Class";
            var expected = "MustInherit Class C\n    Public MustOverride Sub M()\nEnd Class";
            await TestAddMethodAsync(input, expected,
                modifiers: new DeclarationModifiers(isAbstract: true),
                returnType: typeof(void));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddMethodWithoutBody()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Sub M()
End Class";
            await TestAddMethodAsync(input, expected,
                returnType: typeof(void),
                context: new CodeGenerationContext(generateMethodBodies: false));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddGenericMethod()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Function M(Of T)() As Integer
        $$
    End Function
End Class";
            await TestAddMethodAsync(input, expected,
                returnType: typeof(int),
                typeParameters: ImmutableArray.Create(CodeGenerationSymbolFactory.CreateTypeParameterSymbol("T")),
                statements: "Return new T().GetHashCode()");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddVirtualMethod()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Protected Overridable Function M() As Integer
        $$
    End Function
End Class";
            await TestAddMethodAsync(input, expected,
                accessibility: Accessibility.Protected,
                modifiers: new DeclarationModifiers(isVirtual: true),
                returnType: typeof(int),
                statements: "Return 0");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddShadowsMethod()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Shadows Function ToString() As String
        $$
    End Function
End Class";
            await TestAddMethodAsync(input, expected,
                name: "ToString",
                accessibility: Accessibility.Public,
                modifiers: new DeclarationModifiers(isNew: true),
                returnType: typeof(string),
                statements: "Return String.Empty");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddExplicitImplementation()
        {
            var input = "Interface I\n Sub M(i As Integer)\n End Interface\n Class [|C|]\n Implements I\n End Class";
            var expected = @"Interface I
 Sub M(i As Integer)
 End Interface
 Class C
 Implements I
 
    Public Sub M(i As Integer) Implements I.M
    End Sub
End Class";
            await TestAddMethodAsync(input, expected,
                name: "M",
                returnType: typeof(void),
                parameters: Parameters(Parameter(typeof(int), "i")),
                getExplicitInterfaces: s => s.LookupSymbols(input.IndexOf('M'), null, "M").OfType<IMethodSymbol>().ToImmutableArray());
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddTrueFalseOperators()
        {
            var input = @"
Class [|C|]
End Class
";
            var expected = @"
Class C
    Public Shared Operator IsTrue(other As C) As Boolean
        $$
    End Operator
 
    Public Shared Operator IsFalse(other As C) As Boolean
        $$
    End Operator
End Class
";
            await TestAddOperatorsAsync(input, expected,
                [CodeGenerationOperatorKind.True, CodeGenerationOperatorKind.False],
                parameters: Parameters(Parameter("C", "other")),
                returnType: typeof(bool),
                statements: "Return False");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddUnaryOperators()
        {
            var input = @"
Class [|C|]
End Class
";
            var expected = @"
Class C
    Public Shared Operator +(other As C) As Object
        $$
    End Operator
 
    Public Shared Operator -(other As C) As Object
        $$
    End Operator
 
    Public Shared Operator Not(other As C) As Object
        $$
    End Operator
End Class
";
            await TestAddOperatorsAsync(input, expected,
                [
                    CodeGenerationOperatorKind.UnaryPlus,
                    CodeGenerationOperatorKind.UnaryNegation,
                    CodeGenerationOperatorKind.LogicalNot
                ],
                parameters: Parameters(Parameter("C", "other")),
                returnType: typeof(object),
                statements: "Return Nothing");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddBinaryOperators()
        {
            var input = @"
Class [|C|]
End Class
";
            var expected = @"
Class C
    Public Shared Operator +(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator -(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator *(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator /(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator \(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator ^(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator &(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator Like(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator Mod(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator And(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator Or(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator Xor(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator <<(a As C, b As C) As Object
        $$
    End Operator
 
    Public Shared Operator >>(a As C, b As C) As Object
        $$
    End Operator
End Class
";
            await TestAddOperatorsAsync(input, expected,
                [
                    CodeGenerationOperatorKind.Addition,
                    CodeGenerationOperatorKind.Subtraction,
                    CodeGenerationOperatorKind.Multiplication,
                    CodeGenerationOperatorKind.Division,
                    CodeGenerationOperatorKind.IntegerDivision,
                    CodeGenerationOperatorKind.Exponent,
                    CodeGenerationOperatorKind.Concatenate,
                    CodeGenerationOperatorKind.Like,
                    CodeGenerationOperatorKind.Modulus,
                    CodeGenerationOperatorKind.BitwiseAnd,
                    CodeGenerationOperatorKind.BitwiseOr,
                    CodeGenerationOperatorKind.ExclusiveOr,
                    CodeGenerationOperatorKind.LeftShift,
                    CodeGenerationOperatorKind.RightShift
                ],
                parameters: Parameters(Parameter("C", "a"), Parameter("C", "b")),
                returnType: typeof(object),
                statements: "Return Nothing");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddComparisonOperators()
        {
            var input = @"
Class [|C|]
End Class
";
            var expected = @"
Class C
    Public Shared Operator =(a As C, b As C) As Boolean
        $$
    End Operator
 
    Public Shared Operator <>(a As C, b As C) As Boolean
        $$
    End Operator
 
    Public Shared Operator >(a As C, b As C) As Boolean
        $$
    End Operator
 
    Public Shared Operator <(a As C, b As C) As Boolean
        $$
    End Operator
 
    Public Shared Operator >=(a As C, b As C) As Boolean
        $$
    End Operator
 
    Public Shared Operator <=(a As C, b As C) As Boolean
        $$
    End Operator
End Class
";
            await TestAddOperatorsAsync(input, expected,
                [
                    CodeGenerationOperatorKind.Equality,
                    CodeGenerationOperatorKind.Inequality,
                    CodeGenerationOperatorKind.GreaterThan,
                    CodeGenerationOperatorKind.LessThan,
                    CodeGenerationOperatorKind.GreaterThanOrEqual,
                    CodeGenerationOperatorKind.LessThanOrEqual
                ],
                parameters: Parameters(Parameter("C", "a"), Parameter("C", "b")),
                returnType: typeof(bool),
                statements: "Return True");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddUnsupportedOperator()
        {
            var input = "Class [|C|]\n End Class";
            await TestAddUnsupportedOperatorAsync(input,
                operatorKind: CodeGenerationOperatorKind.Increment,
                parameters: Parameters(Parameter("C", "other")),
                returnType: typeof(bool),
                statements: "Return True");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddExplicitConversion()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Shared Narrowing Operator CType(other As C) As Integer
        $$
    End Operator
End Class";
            await TestAddConversionAsync(input, expected,
                toType: typeof(int),
                fromType: Parameter("C", "other"),
                statements: "Return 0");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddImplicitConversion()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Shared Widening Operator CType(other As C) As Integer
        $$
    End Operator
End Class";
            await TestAddConversionAsync(input, expected,
                toType: typeof(int),
                fromType: Parameter("C", "other"),
                isImplicit: true,
                statements: "Return 0");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddStatementsToSub()
        {
            var input = "Class C\n [|Public Sub M\n Console.WriteLine(1)\n End Sub|]\n End Class";
            var expected = @"Class C
 Public Sub M
 Console.WriteLine(1)
$$ End Sub
 End Class";
            await TestAddStatementsAsync(input, expected, "Console.WriteLine(2)");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddStatementsToOperator()
        {
            var input = "Class C\n [|Shared Operator +(arg As C) As C\n Return arg\n End Operator|]\n End Class";
            var expected = @"Class C
 Shared Operator +(arg As C) As C
 Return arg
$$ End Operator
 End Class";
            await TestAddStatementsAsync(input, expected, "Return Nothing");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddStatementsToPropertySetter()
        {
            var input = "Imports System\n Class C\n WriteOnly Property P As String\n [|Set\n End Set|]\n End Property\n End Class";
            var expected = @"Imports System
 Class C
 WriteOnly Property P As String
 Set
$$ End Set
 End Property
 End Class";
            await TestAddStatementsAsync(input, expected, "Console.WriteLine(\"Setting the value\"");
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddParametersToMethod()
        {
            var input = "Class C\n Public [|Sub M()\n End Sub|]\n End Class";
            var expected = @"Class C
 Public Sub M(numAs Integer, OptionaltextAs String = ""Hello!"",OptionalfloatingAs Single = 0.5)
 End Sub
 End Class";
            await TestAddParametersAsync(input, expected,
                Parameters(Parameter(typeof(int), "num"), Parameter(typeof(string), "text", true, "Hello!"), Parameter(typeof(float), "floating", true, .5F)));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        [WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/844460")]
        public async Task AddParametersToPropertyBlock()
        {
            var input = "Class C\n [|Public Property P As String\n Get\n Return String.Empty\n End Get\n Set(value As String)\n End Set\n End Property|]\n End Class";
            var expected = @"Class C
 Public Property P (numAs Integer) As String
 Get
 Return String.Empty
 End Get
 Set(value As String)
 End Set
 End Property
 End Class";
            await TestAddParametersAsync(input, expected,
                Parameters(Parameter(typeof(int), "num")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        [WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/844460")]
        public async Task AddParametersToPropertyStatement()
        {
            var input = "Class C\n [|Public Property P As String|]\n Get\n Return String.Empty\n End Get\n Set(value As String)\n End Set\n End Property\n End Class";
            var expected = @"Class C
 Public Property P (numAs Integer) As String
 Get
 Return String.Empty
 End Get
 Set(value As String)
 End Set
 End Property
 End Class";
            await TestAddParametersAsync(input, expected,
                Parameters(Parameter(typeof(int), "num")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        [WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/844460")]
        public async Task AddParametersToPropertyGetter_ShouldNotSucceed()
        {
            var input = "Class C\n Public Property P As String\n [|Get\n Return String.Empty\n End Get|]\n Set(value As String)\n End Set\n End Property\n End Class";
            var expected = "Class C\n Public Property P As String\n Get\n Return String.Empty\n End Get\n Set(value As String)\n End Set\n End Property\n End Class";
            await TestAddParametersAsync(input, expected,
                Parameters(Parameter(typeof(int), "num")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        [WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/844460")]
        public async Task AddParametersToPropertySetter_ShouldNotSucceed()
        {
            var input = "Class C\n Public Property P As String\n Get\n Return String.Empty\n End Get\n [|Set(value As String)\n End Set|]\n End Property\n End Class";
            var expected = "Class C\n Public Property P As String\n Get\n Return String.Empty\n End Get\n Set(value As String)\n End Set\n End Property\n End Class";
            await TestAddParametersAsync(input, expected,
                Parameters(Parameter(typeof(int), "num")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddParametersToOperator()
        {
            var input = "Class C\n [|Shared Operator +(a As C) As C\n Return a\n End Operator|]\n End Class";
            var expected = @"Class C
 Shared Operator +(a As C,bAs C) As C
 Return a
 End Operator
 End Class";
            await TestAddParametersAsync(input, expected,
                Parameters(Parameter("C", "b")));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAutoProperty()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Property P As Integer
End Class";
            await TestAddPropertyAsync(input, expected,
                type: typeof(int));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddPropertyToClassFromSourceSymbol()
        {
            var sourceGenerated = @"Public Class [|C2|]
    Public Property P As Integer
        Get
            Return 0
        End Get
    End Property
End Class";
            var input = "Class [|C1|]\nEnd Class";
            var expected = @"Class C1
    Public Property P As Integer
        Get
            Return 0
        End Get
    End Property
End Class";
            var context = new CodeGenerationContext(reuseSyntax: true);
            await TestGenerateFromSourceSymbolAsync(sourceGenerated, input, expected, onlyGenerateMembers: true, context: context);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddPropertyWithoutAccessorBodies()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Public Property P As Integer
End Class";
            await TestAddPropertyAsync(input, expected,
                type: typeof(int),
                getStatements: "Return 0",
                setStatements: "Me.P = Value",
                context: new CodeGenerationContext(generateMethodBodies: false));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddIndexer()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"Class C
    Default Public ReadOnly Property Item(i As Integer) As String
        Get
            $$
        End Get
    End Property
End Class";
            await TestAddPropertyAsync(input, expected,
                name: "Item",
                type: typeof(string),
                parameters: Parameters(Parameter(typeof(int), "i")),
                getStatements: "Return String.Empty",
                isIndexer: true);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToTypes()
        {
            var input = "Class [|C|]\n End Class";
            var expected = @"<Serializable>
Class C
End Class";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromTypes()
        {
            var input = @"
<Serializable>
Class [|C|]
End Class";
            var expected = @"
Class C
End Class";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToMethods()
        {
            var input = "Class C\n Public Sub [|M()|] \n End Sub \n End Class";
            var expected = @"Class C
    <Serializable>
    Public Sub M()
    End Sub 
 End Class";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromMethods()
        {
            var input = @"
Class C
    <Serializable>
    Public Sub [|M()|]
    End Sub
End Class";
            var expected = @"
Class C
    Public Sub M()
    End Sub
End Class";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToFields()
        {
            var input = "Class C\n [|Public F As Integer|]\n End Class";
            var expected = @"Class C
    <Serializable>
    Public F As Integer
End Class";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromFields()
        {
            var input = @"
Class C
    <Serializable>
    Public [|F|] As Integer
End Class";
            var expected = @"
Class C
    Public F As Integer
End Class";
            await TestRemoveAttributeAsync<FieldDeclarationSyntax>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToProperties()
        {
            var input = "Class C \n Public Property [|P|] As Integer \n End Class";
            var expected = @"Class C
    <Serializable>
    Public Property P As Integer
End Class";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromProperties()
        {
            var input = @"
Class C
    <Serializable>
    Public Property [|P|] As Integer
End Class";
            var expected = @"
Class C
    Public Property P As Integer
End Class";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToPropertyAccessor()
        {
            var input = "Class C \n Public ReadOnly Property P As Integer \n [|Get|] \n Return 10 \n End Get \n End Property \n  End Class";
            var expected = @"Class C 
 Public ReadOnly Property P As Integer
        <Serializable>
        Get
            Return 10 
 End Get 
 End Property 
  End Class";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromPropertyAccessor()
        {
            var input = @"
Class C
    Public Property P As Integer
        <Serializable>
        [|Get|]
            Return 10
        End Get
End Class";
            var expected = @"
Class C
    Public Property P As Integer
        Get
            Return 10
        End Get
End Class";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToEnums()
        {
            var input = "Module M \n [|Enum C|] \n One \n Two \n End Enum\n End Module";
            var expected = @"Module M
    <Serializable>
    Enum C
        One 
 Two 
 End Enum
 End Module";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromEnums()
        {
            var input = @"
Module M
    <Serializable>
    Enum [|C|]
        One
        Two
    End Enum
End Module";
            var expected = @"
Module M
    Enum C
        One
        Two
    End Enum
End Module";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToEnumMembers()
        {
            var input = "Module M \n Enum C \n [|One|] \n Two \n End Enum\n End Module";
            var expected = @"Module M 
 Enum C
        <Serializable>
        One
        Two 
 End Enum
 End Module";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromEnumMembers()
        {
            var input = @"
Module M
    Enum C
        <Serializable>
        [|One|]
        Two
    End Enum
End Module";
            var expected = @"
Module M
    Enum C
        One
        Two
    End Enum
End Module";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToModule()
        {
            var input = "Module [|M|] \n End Module";
            var expected = @"<Serializable>
Module M
End Module";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromModule()
        {
            var input = @"
<Serializable>
Module [|M|]
End Module";
            var expected = @"
Module M
End Module";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToOperator()
        {
            var input = "Class C \n Public Shared Operator [|+|] (x As C, y As C) As C \n Return New C() \n End Operator \n End Class";
            var expected = @"Class C
    <Serializable>
    Public Shared Operator +(x As C, y As C) As C
        Return New C() 
 End Operator 
 End Class";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromOperator()
        {
            var input = @"
Module M
    Class C
        <Serializable>
        Public Shared Operator [|+|](x As C, y As C) As C
            Return New C()
        End Operator
    End Class
End Module";
            var expected = @"
Module M
    Class C
        Public Shared Operator +(x As C, y As C) As C
            Return New C()
        End Operator
    End Class
End Module";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToDelegate()
        {
            var input = "Module M \n Delegate Sub [|D()|]\n End Module";
            var expected = "Module M\n    <Serializable>\n    Delegate Sub D()\nEnd Module";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromDelegate()
        {
            var input = @"
Module M
    <Serializable>
    Delegate Sub [|D()|]
End Module";
            var expected = @"
Module M
    Delegate Sub D()
End Module";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToParam()
        {
            var input = "Class C \n Public Sub M([|x As Integer|]) \n End Sub \n End Class";
            var expected = "Class C \n Public Sub M(<Serializable> x As Integer) \n End Sub \n End Class";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeFromParam()
        {
            var input = @"
Class C
    Public Sub M(<Serializable> [|x As Integer|])
    End Sub
End Class";
            var expected = @"
Class C
    Public Sub M(x As Integer)
    End Sub
End Class";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeToCompilationUnit()
        {
            var input = "[|Class C \n End Class \n Class D \n End Class|]";
            var expected = "<Assembly: Serializable>\nClass C\nEnd Class\nClass D\nEnd Class";
            await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute), VB.SyntaxFactory.Token(VB.SyntaxKind.AssemblyKeyword));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task AddAttributeWithWrongTarget()
        {
            var input = "[|Class C \n End Class \n Class D \n End Class|]";
            var expected = "<Assembly: Serializable> Class C \n End Class \n Class D \n End Class";
            await Assert.ThrowsAsync<AggregateException>(async () =>
                await TestAddAttributeAsync(input, expected, typeof(SerializableAttribute), VB.SyntaxFactory.Token(VB.SyntaxKind.ReturnKeyword)));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeWithTrivia()
        {
            // With trivia.
            var input = @"' Comment 1
<System.Serializable> ' Comment 2
Class [|C|]
End Class";
            var expected = @"' Comment 1
Class C
End Class";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeWithTrivia_NewLine()
        {
            // With trivia, redundant newline at end of attribute removed.
            var input = @"' Comment 1
<System.Serializable>
Class [|C|]
End Class";
            var expected = @"' Comment 1
Class C
End Class";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeWithMultipleAttributes()
        {
            // Multiple attributes.
            var input = @"' Comment 1
<  System.Serializable   ,  System.Flags> ' Comment 2
Class [|C|]
End Class";
            var expected = @"' Comment 1
<System.Flags> ' Comment 2
Class C
End Class";
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task RemoveAttributeWithMultipleAttributeLists()
        {
            // Multiple attribute lists.
            var input = @"' Comment 1
<  System.Serializable   ,  System.Flags> ' Comment 2
<System.Obsolete> ' Comment 3
Class [|C|]
End Class";
            var expected = @"' Comment 1
<System.Flags> ' Comment 2
<System.Obsolete> ' Comment 3
Class C
End Class";
 
            await TestRemoveAttributeAsync<SyntaxNode>(input, expected, typeof(SerializableAttribute));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task TestUpdateModifiers()
        {
            var input = @"Public Shared Class [|C|] ' Comment 1
    ' Comment 2
End Class";
            var expected = @"Friend Partial NotInheritable Class C ' Comment 1
    ' Comment 2
End Class";
            var eol = VB.SyntaxFactory.EndOfLine(@"");
            var newModifiers = new[] { VB.SyntaxFactory.Token(VB.SyntaxKind.FriendKeyword).WithLeadingTrivia(eol) }.Concat(
                CreateModifierTokens(new DeclarationModifiers(isSealed: true, isPartial: true), LanguageNames.VisualBasic));
 
            await TestUpdateDeclarationAsync<ClassStatementSyntax>(input, expected, modifiers: newModifiers);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task TestUpdateAccessibility()
        {
            var input = @"' Comment 0
Public Shared Class [|C|] ' Comment 1
    ' Comment 2
End Class";
            var expected = @"' Comment 0
Protected Friend Shared Class C ' Comment 1
    ' Comment 2
End Class";
            await TestUpdateDeclarationAsync<ClassStatementSyntax>(input, expected, accessibility: Accessibility.ProtectedOrFriend);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task TestUpdateDeclarationType()
        {
            var input = @"
Public Shared Class C
    ' Comment 1
    Public Shared Function [|F|]() As Char
        Return 0
    End Function
End Class";
            var expected = @"
Public Shared Class C
    ' Comment 1
    Public Shared Function F() As Integer
        Return 0
    End Function
End Class";
            await TestUpdateDeclarationAsync<MethodStatementSyntax>(input, expected, getType: GetTypeSymbol(typeof(int)));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        public async Task TestUpdateDeclarationMembers()
        {
            var input = @"
Public Shared Class [|C|]
    ' Comment 0
    Public Shared {|RetainedMember:f|} As Integer
 
    ' Comment 1
    Public Shared Function F() As Char
        Return 0
    End Function
End Class";
            var expected = @"
Public Shared Class C
    ' Comment 0
    Public Shared f As Integer
    Public Shared f2 As Integer
End Class";
            var getField = CreateField(Accessibility.Public, new DeclarationModifiers(isStatic: true), typeof(int), "f2");
            var getMembers = ImmutableArray.Create(getField);
            await TestUpdateDeclarationAsync<ClassBlockSyntax>(input, expected, getNewMembers: getMembers);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGenerationSortDeclarations)]
        public async Task SortModules()
        {
            var generationSource = "Public Class [|C|] \n End Class";
            var initial = "Namespace [|N|] \n Module M \n End Module \n End Namespace";
            var expected = @"Namespace N
    Public Class C
    End Class
 
    Module M 
 End Module 
 End Namespace";
            await TestGenerateFromSourceSymbolAsync(generationSource, initial, expected);
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGenerationSortDeclarations)]
        public async Task SortOperators()
        {
            var generationSource = @"
Namespace N
    Public Class [|C|]
        ' Unary operators
        Public Shared Operator IsFalse(other As C) As Boolean
            Return False
        End Operator
        Public Shared Operator IsTrue(other As C) As Boolean
            Return True
        End Operator
        Public Shared Operator Not(other As C) As C
            Return Nothing
        End Operator
        Public Shared Operator -(other As C) As C
            Return Nothing
        End Operator
        Public Shared Operator + (other As C) As C
            Return Nothing
        End Operator
        Public Shared Narrowing Operator CType(c As C) As Integer
            Return 0
        End Operator
 
        ' Binary operators
        Public Shared Operator >= (a As C, b As C) As Boolean
            Return True
        End Operator
        Public Shared Operator <= (a As C, b As C) As Boolean
            Return True
        End Operator
        Public Shared Operator > (a As C, b As C) As Boolean
            Return True
        End Operator
        Public Shared Operator < (a As C, b As C) As Boolean
            Return True
        End Operator
        Public Shared Operator <> (a As C, b As C) As Boolean
            Return True
        End Operator
        Public Shared Operator = (a As C, b As C) As Boolean
            Return True
        End Operator
        Public Shared Operator >> (a As C, shift As Integer) As C
            Return Nothing
        End Operator
        Public Shared Operator << (a As C, shift As Integer) As C
            Return Nothing
        End Operator
        Public Shared Operator Xor(a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator Or(a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator And(a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator Mod(a As C, b As C) As C
            Return Nothing
        End Operator
          Public Shared Operator Like (a As C, b As C) As C
            Return Nothing
        End Operator
          Public Shared Operator & (a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator ^ (a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator \ (a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator / (a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator *(a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator -(a As C, b As C) As C
            Return Nothing
        End Operator
        Public Shared Operator + (a As C, b As C) As C
            Return Nothing
        End Operator
    End Class
End Namespace";
            var initial = "Namespace [|N|] \n End Namespace";
            var expected = @"Namespace N
    Public Class C
        Public Shared Operator +(other As C) As C
        Public Shared Operator +(a As C, b As C) As C
        Public Shared Operator -(other As C) As C
        Public Shared Operator -(a As C, b As C) As C
        Public Shared Operator *(a As C, b As C) As C
        Public Shared Operator /(a As C, b As C) As C
        Public Shared Operator \(a As C, b As C) As C
        Public Shared Operator ^(a As C, b As C) As C
        Public Shared Operator &(a As C, b As C) As C
        Public Shared Operator Not(other As C) As C
        Public Shared Operator Like(a As C, b As C) As C
        Public Shared Operator Mod(a As C, b As C) As C
        Public Shared Operator And(a As C, b As C) As C
        Public Shared Operator Or(a As C, b As C) As C
        Public Shared Operator Xor(a As C, b As C) As C
        Public Shared Operator <<(a As C, shift As Integer) As C
        Public Shared Operator >>(a As C, shift As Integer) As C
        Public Shared Operator =(a As C, b As C) As Boolean
        Public Shared Operator <>(a As C, b As C) As Boolean
        Public Shared Operator >(a As C, b As C) As Boolean
        Public Shared Operator <(a As C, b As C) As Boolean
        Public Shared Operator >=(a As C, b As C) As Boolean
        Public Shared Operator <=(a As C, b As C) As Boolean
        Public Shared Operator IsTrue(other As C) As Boolean
        Public Shared Operator IsFalse(other As C) As Boolean
        Public Shared Narrowing Operator CType(c As C) As Integer
    End Class
End Namespace";
            await TestGenerateFromSourceSymbolAsync(generationSource, initial, expected,
                forceLanguage: LanguageNames.VisualBasic,
                context: new CodeGenerationContext(generateMethodBodies: false));
        }
 
        [Fact, Trait(Traits.Feature, Traits.Features.CodeGeneration)]
        [WorkItem("http://vstfdevdiv:8080/DevDiv2/DevDiv/_workitems/edit/848357")]
        public async Task TestConstraints()
        {
            var generationSource = @"
Namespace N
    Public Class [|C|](Of T As Structure, U As Class)
        Public Sub Goo(Of Q As New, R As IComparable)()
        End Sub
        Public Delegate Sub D(Of T1 As Structure, T2 As Class)(t As T1, u As T2)
    End Class
End Namespace
";
            var initial = "Namespace [|N|] \n End Namespace";
            var expected = @"Namespace N
    Public Class C(Of T As Structure, U As Class)
        Public Sub Goo(Of Q As New, R As IComparable)()
        Public Delegate Sub D(Of T1 As Structure, T2 As Class)(t As T1, u As T2)
    End Class
End Namespace";
            await TestGenerateFromSourceSymbolAsync(generationSource, initial, expected,
                context: new CodeGenerationContext(generateMethodBodies: false),
                onlyGenerateMembers: true);
        }
    }
}