Set and verify the text in the editor

Author: sharwell

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/Editor_InProc.cs

@@ -3,16 +3,65 @@
 
 namespace Tvl.VisualStudio.MouseFastScroll.IntegrationTests
 {
+    using System;
+    using System.Runtime.InteropServices;
+    using Microsoft.VisualStudio.Text;
+    using Microsoft.VisualStudio.Text.Editor;
+    using Microsoft.VisualStudio.TextManager.Interop;
+
     internal class Editor_InProc : TextViewWindow_InProc
     {
+        private static readonly Guid IWpfTextViewId = new Guid("8C40265E-9FDB-4F54-A0FD-EBB72B7D0476");
+
         private Editor_InProc()
         {
         }
 
         public static Editor_InProc Create()
             => new Editor_InProc();
 
+        protected override IWpfTextView GetActiveTextView()
+            => GetActiveTextViewHost().TextView;
+
+        private static IVsTextView GetActiveVsTextView()
+        {
+            var vsTextManager = GetGlobalService<SVsTextManager, IVsTextManager>();
+
+            var hresult = vsTextManager.GetActiveView(fMustHaveFocus: 1, pBuffer: null, ppView: out var vsTextView);
+            Marshal.ThrowExceptionForHR(hresult);
+
+            return vsTextView;
+        }
+
+        private static IWpfTextViewHost GetActiveTextViewHost()
+        {
+            // The active text view might not have finished composing yet, waiting for the application to 'idle'
+            // means that it is done pumping messages (including WM_PAINT) and the window should return the correct text view
+            WaitForApplicationIdle();
+
+            var activeVsTextView = (IVsUserData)GetActiveVsTextView();
+
+            var hresult = activeVsTextView.GetData(IWpfTextViewId, out var wpfTextViewHost);
+            Marshal.ThrowExceptionForHR(hresult);
+
+            return (IWpfTextViewHost)wpfTextViewHost;
+        }
+
         public void Activate()
             => GetDTE().ActiveDocument.Activate();
+
+        public string GetText()
+            => ExecuteOnActiveView(view => view.TextSnapshot.GetText());
+
+        public void SetText(string text)
+        {
+            ExecuteOnActiveView(
+                view =>
+                {
+                    var textSnapshot = view.TextSnapshot;
+                    var replacementSpan = new SnapshotSpan(textSnapshot, 0, textSnapshot.Length);
+                    view.TextBuffer.Replace(replacementSpan, text);
+                });
+        }
     }
 }

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/Editor_OutOfProc.cs

@@ -22,6 +22,12 @@ internal override TextViewWindow_InProc CreateInProcComponent(VisualStudioInstan
         public void Activate()
             => EditorInProc.Activate();
 
+        internal string GetText()
+            => EditorInProc.GetText();
+
+        internal void SetText(string value)
+            => EditorInProc.SetText(value);
+
         /// <summary>
         /// Sends key strokes to the active editor in Visual Studio. Various types are supported by this method:
         /// <see cref="string"/> (each character will be sent separately, <see cref="char"/>, and

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/TextViewWindow_InProc.cs

@@ -3,7 +3,33 @@
 
 namespace Tvl.VisualStudio.MouseFastScroll.IntegrationTests
 {
+    using System;
+    using Microsoft.VisualStudio.Text.Editor;
+
     internal abstract class TextViewWindow_InProc : InProcComponent
     {
+        protected abstract IWpfTextView GetActiveTextView();
+
+        protected T ExecuteOnActiveView<T>(Func<IWpfTextView, T> action)
+        {
+            return InvokeOnUIThread(
+                () =>
+                {
+                    var view = GetActiveTextView();
+                    return action(view);
+                });
+        }
+
+        protected void ExecuteOnActiveView(Action<IWpfTextView> action)
+            => InvokeOnUIThread(GetExecuteOnActionViewCallback(action));
+
+        protected Action GetExecuteOnActionViewCallback(Action<IWpfTextView> action)
+        {
+            return () =>
+            {
+                var view = GetActiveTextView();
+                action(view);
+            };
+        }
     }
 }

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/TrivialIntegrationTest.cs

@@ -4,7 +4,7 @@
 namespace Tvl.VisualStudio.MouseFastScroll.IntegrationTests
 {
     using System;
-    using System.Threading;
+    using System.Linq;
     using Xunit;
     using vsSaveChanges = EnvDTE.vsSaveChanges;
 
@@ -33,7 +33,16 @@ public void TestOpenAndCloseIDE()
         public void OpenDocumentAndType()
         {
             var window = VisualStudioInstance.RetryRpcCall(() => VisualStudio.Dte.ItemOperations.NewFile(Name: Guid.NewGuid() + ".txt"));
-            VisualStudio.Editor.SendKeys(Guid.NewGuid().ToString() + "\n" + Guid.NewGuid().ToString());
+
+            string initialText = string.Join(string.Empty, Enumerable.Range(0, 400).Select(i => Guid.NewGuid() + Environment.NewLine));
+            VisualStudio.Editor.SetText(initialText);
+
+            string additionalTypedText = Guid.NewGuid().ToString() + "\n" + Guid.NewGuid().ToString();
+            VisualStudio.Editor.SendKeys(additionalTypedText);
+
+            string expected = initialText + additionalTypedText.Replace("\n", Environment.NewLine);
+            Assert.Equal(expected, VisualStudio.Editor.GetText());
+
             VisualStudioInstance.RetryRpcCall(() => window.Close(vsSaveChanges.vsSaveChangesNo));
         }