diff --git a/src/VisualStudio/IntegrationTest/IntegrationTests/CSharp/CSharpExtractMethod.cs b/src/VisualStudio/IntegrationTest/IntegrationTests/CSharp/CSharpExtractMethod.cs index 78068c159ce82..c53fa57cae5c6 100644 --- a/src/VisualStudio/IntegrationTest/IntegrationTests/CSharp/CSharpExtractMethod.cs +++ b/src/VisualStudio/IntegrationTest/IntegrationTests/CSharp/CSharpExtractMethod.cs @@ -72,7 +72,6 @@ public int Method() MarkupTestFile.GetSpans(expectedMarkup, out var expectedText, out ImmutableArray spans); VisualStudio.Editor.Verify.TextContains(expectedText); - VisualStudio.Workspace.WaitForAsyncOperations(Helper.HangMitigatingTimeout, FeatureAttribute.Rename); AssertEx.SetEqual(spans, VisualStudio.Editor.GetTagSpans(VisualStudio.InlineRenameDialog.ValidRenameTag)); VisualStudio.Editor.SendKeys("SayHello", VirtualKey.Enter); diff --git a/src/VisualStudio/IntegrationTest/IntegrationTests/VisualBasic/BasicExtractMethod.cs b/src/VisualStudio/IntegrationTest/IntegrationTests/VisualBasic/BasicExtractMethod.cs index b1903a1fa0765..2833d3a2ec540 100644 --- a/src/VisualStudio/IntegrationTest/IntegrationTests/VisualBasic/BasicExtractMethod.cs +++ b/src/VisualStudio/IntegrationTest/IntegrationTests/VisualBasic/BasicExtractMethod.cs @@ -80,7 +80,6 @@ End Function MarkupTestFile.GetSpans(expectedMarkup, out var expectedText, out ImmutableArray spans); VisualStudio.Editor.Verify.TextContains(expectedText); - VisualStudio.Workspace.WaitForAsyncOperations(Helper.HangMitigatingTimeout, FeatureAttribute.Rename); AssertEx.SetEqual(spans, VisualStudio.Editor.GetTagSpans(VisualStudio.InlineRenameDialog.ValidRenameTag)); VisualStudio.Editor.SendKeys("SayHello", VirtualKey.Enter); diff --git a/src/VisualStudio/IntegrationTest/TestUtilities/OutOfProcess/Editor_OutOfProc.cs b/src/VisualStudio/IntegrationTest/TestUtilities/OutOfProcess/Editor_OutOfProc.cs index ddea1064fea7d..214f565d54c5e 100644 --- a/src/VisualStudio/IntegrationTest/TestUtilities/OutOfProcess/Editor_OutOfProc.cs +++ b/src/VisualStudio/IntegrationTest/TestUtilities/OutOfProcess/Editor_OutOfProc.cs @@ -65,6 +65,11 @@ public void MoveCaret(int position) public ImmutableArray GetTagSpans(string tagId) { + if (tagId == _instance.InlineRenameDialog.ValidRenameTag) + { + _instance.Workspace.WaitForAsyncOperations(Helper.HangMitigatingTimeout, FeatureAttribute.Rename); + } + var tagInfo = _editorInProc.GetTagSpans(tagId).ToList(); // The spans are returned in an array: