diff --git a/src/EditorFeatures/Test2/KeywordHighlighting/AbstractKeywordHighlightingTests.vb b/src/EditorFeatures/Test2/KeywordHighlighting/AbstractKeywordHighlightingTests.vb index 46c4bc309849b19b45908e0af6fb0dfdc34e0e1e..ac908894a9528a272fef8bf25a81d84d061035f4 100644 --- a/src/EditorFeatures/Test2/KeywordHighlighting/AbstractKeywordHighlightingTests.vb +++ b/src/EditorFeatures/Test2/KeywordHighlighting/AbstractKeywordHighlightingTests.vb @@ -18,8 +18,8 @@ Imports Roslyn.Utilities Namespace Microsoft.CodeAnalysis.Editor.UnitTests.KeywordHighlighting Public MustInherit Class AbstractKeywordHighlightingTests - Protected Sub VerifyHighlights(test As XElement, Optional optionIsEnabled As Boolean = True) - Using workspace = TestWorkspaceFactory.CreateWorkspace(test) + Protected Async Function VerifyHighlightsAsync(test As XElement, Optional optionIsEnabled As Boolean = True) As Tasks.Task + Using workspace = Await TestWorkspaceFactory.CreateWorkspaceAsync(test) Dim testDocument = workspace.Documents.Single(Function(d) d.CursorPosition.HasValue) Dim buffer = testDocument.TextBuffer Dim snapshot = testDocument.InitialTextSnapshot @@ -51,7 +51,7 @@ Namespace Microsoft.CodeAnalysis.Editor.UnitTests.KeywordHighlighting AssertEx.Equal(expectedTags, producedTags) End Using - End Sub + End Function End Class diff --git a/src/EditorFeatures/Test2/KeywordHighlighting/CSharpKeywordHighlightingTests.vb b/src/EditorFeatures/Test2/KeywordHighlighting/CSharpKeywordHighlightingTests.vb index d741812af6a6e3fc326fd66fc1faa447230bf811..c262cd651c4c2f80cad8b0d2e9e875eaf654b5d9 100644 --- a/src/EditorFeatures/Test2/KeywordHighlighting/CSharpKeywordHighlightingTests.vb +++ b/src/EditorFeatures/Test2/KeywordHighlighting/CSharpKeywordHighlightingTests.vb @@ -1,12 +1,14 @@ ' Copyright (c) Microsoft. All Rights Reserved. Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information. +Imports System.Threading.Tasks + Namespace Microsoft.CodeAnalysis.Editor.UnitTests.KeywordHighlighting Public Class CSharpKeywordHighlightingTests Inherits AbstractKeywordHighlightingTests - Public Sub VerifyNoHighlightsWhenOptionDisabled() - VerifyHighlights( + Public Async Function TestVerifyNoHighlightsWhenOptionDisabled() As Task + Await VerifyHighlightsAsync( @@ -22,11 +24,11 @@ Namespace Microsoft.CodeAnalysis.Editor.UnitTests.KeywordHighlighting , optionIsEnabled:=False) - End Sub + End Function - Public Sub VerifyHighlightsWhenOptionEnabled() - VerifyHighlights( + Public Async Function TestVerifyHighlightsWhenOptionEnabled() As Task + Await VerifyHighlightsAsync( @@ -41,6 +43,6 @@ Namespace Microsoft.CodeAnalysis.Editor.UnitTests.KeywordHighlighting ) - End Sub + End Function End Class End Namespace diff --git a/src/EditorFeatures/Test2/KeywordHighlighting/VisualBasicKeywordHighlightingTests.vb b/src/EditorFeatures/Test2/KeywordHighlighting/VisualBasicKeywordHighlightingTests.vb index ebfcf2fe831388983c007a0f4c45fe4057bd6338..f7188507f36f73d234a57fe19d9ecc36b5744a7c 100644 --- a/src/EditorFeatures/Test2/KeywordHighlighting/VisualBasicKeywordHighlightingTests.vb +++ b/src/EditorFeatures/Test2/KeywordHighlighting/VisualBasicKeywordHighlightingTests.vb @@ -5,8 +5,8 @@ Namespace Microsoft.CodeAnalysis.Editor.UnitTests.KeywordHighlighting Inherits AbstractKeywordHighlightingTests - Public Sub VerifyNoHighlightsWhenOptionDisabled() - VerifyHighlights( + Public Async Function VerifyNoHighlightsWhenOptionDisabled() As System.Threading.Tasks.Task + Await VerifyHighlightsAsync( @@ -21,11 +21,11 @@ Namespace Microsoft.CodeAnalysis.Editor.UnitTests.KeywordHighlighting , optionIsEnabled:=False) - End Sub + End Function - Public Sub VerifyHighlightsWhenOptionEnabled() - VerifyHighlights( + Public Async Function VerifyHighlightsWhenOptionEnabled() As System.Threading.Tasks.Task + Await VerifyHighlightsAsync( @@ -39,6 +39,6 @@ Namespace Microsoft.CodeAnalysis.Editor.UnitTests.KeywordHighlighting ) - End Sub + End Function End Class End Namespace