Set and verify the caret position

Author: sharwell

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/Editor_InProc.cs

@@ -63,5 +63,22 @@ public void SetText(string text)
                     view.TextBuffer.Replace(replacementSpan, text);
                 });
         }
+
+        public void MoveCaret(int position)
+        {
+            ExecuteOnActiveView(
+                view =>
+                {
+                    var subjectBuffer = view.GetBufferContainingCaret();
+                    var point = new SnapshotPoint(subjectBuffer.CurrentSnapshot, position);
+
+                    view.Caret.MoveTo(point);
+                });
+        }
+
+        protected override ITextBuffer GetBufferContainingCaret(IWpfTextView view)
+        {
+            return view.GetBufferContainingCaret();
+        }
     }
 }

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/Editor_OutOfProc.cs

@@ -28,6 +28,9 @@ internal string GetText()
         internal void SetText(string value)
             => EditorInProc.SetText(value);
 
+        public void MoveCaret(int position)
+            => EditorInProc.MoveCaret(position);
+
         /// <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/IBufferGraphExtensions.cs

@@ -0,0 +1,24 @@
+// Copyright (c) Tunnel Vision Laboratories, LLC. All Rights Reserved.
+// Licensed under the Apache License, Version 2.0. See LICENSE.txt in the project root for license information.
+
+namespace Tvl.VisualStudio.MouseFastScroll.IntegrationTests
+{
+    using System;
+    using System.Linq;
+    using Microsoft.VisualStudio.Text;
+    using Microsoft.VisualStudio.Text.Projection;
+
+    internal static class IBufferGraphExtensions
+    {
+        public static SnapshotSpan? MapUpOrDownToFirstMatch(this IBufferGraph bufferGraph, SnapshotSpan span, Predicate<ITextSnapshot> match)
+        {
+            var spans = bufferGraph.MapDownToFirstMatch(span, SpanTrackingMode.EdgeExclusive, match);
+            if (!spans.Any())
+            {
+                spans = bufferGraph.MapUpToFirstMatch(span, SpanTrackingMode.EdgeExclusive, match);
+            }
+
+            return spans.Select(s => (SnapshotSpan?)s).FirstOrDefault();
+        }
+    }
+}

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/ITextViewExtensions.cs

@@ -0,0 +1,32 @@
+// Copyright (c) Tunnel Vision Laboratories, LLC. All Rights Reserved.
+// Licensed under the Apache License, Version 2.0. See LICENSE.txt in the project root for license information.
+
+namespace Tvl.VisualStudio.MouseFastScroll.IntegrationTests
+{
+    using System;
+    using Microsoft.VisualStudio.Text;
+    using Microsoft.VisualStudio.Text.Editor;
+
+    internal static class ITextViewExtensions
+    {
+        public static SnapshotPoint? GetCaretPoint(this ITextView textView, Predicate<ITextSnapshot> match)
+        {
+            var caret = textView.Caret.Position;
+            var span = textView.BufferGraph.MapUpOrDownToFirstMatch(new SnapshotSpan(caret.BufferPosition, 0), match);
+            if (span.HasValue)
+            {
+                return span.Value.Start;
+            }
+            else
+            {
+                return null;
+            }
+        }
+
+        public static ITextBuffer GetBufferContainingCaret(this ITextView textView, string contentType = "text")
+        {
+            var point = GetCaretPoint(textView, s => s.ContentType.IsOfType(contentType));
+            return point.HasValue ? point.Value.Snapshot.TextBuffer : null;
+        }
+    }
+}

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/TextViewWindow_InProc.cs

@@ -4,12 +4,26 @@
 namespace Tvl.VisualStudio.MouseFastScroll.IntegrationTests
 {
     using System;
+    using Microsoft.VisualStudio.Text;
     using Microsoft.VisualStudio.Text.Editor;
 
     internal abstract class TextViewWindow_InProc : InProcComponent
     {
         protected abstract IWpfTextView GetActiveTextView();
 
+        protected abstract ITextBuffer GetBufferContainingCaret(IWpfTextView view);
+
+        public int GetCaretPosition()
+        {
+            return ExecuteOnActiveView(
+                view =>
+                {
+                    var subjectBuffer = GetBufferContainingCaret(view);
+                    var bufferPosition = view.Caret.Position.BufferPosition;
+                    return bufferPosition.Position;
+                });
+        }
+
         protected T ExecuteOnActiveView<T>(Func<IWpfTextView, T> action)
         {
             return InvokeOnUIThread(

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/TextViewWindow_OutOfProc.cs

@@ -17,5 +17,8 @@ internal TextViewWindow_InProc TextViewWindowInProc
         }
 
         internal abstract TextViewWindow_InProc CreateInProcComponent(VisualStudioInstance visualStudioInstance);
+
+        public int GetCaretPosition()
+            => TextViewWindowInProc.GetCaretPosition();
     }
 }

Tvl.VisualStudio.MouseFastScroll.IntegrationTests/TrivialIntegrationTest.cs

@@ -43,6 +43,10 @@ public void OpenDocumentAndType()
             string expected = initialText + additionalTypedText.Replace("\n", Environment.NewLine);
             Assert.Equal(expected, VisualStudio.Editor.GetText());
 
+            Assert.Equal(expected.Length, VisualStudio.Editor.GetCaretPosition());
+            VisualStudio.Editor.MoveCaret(0);
+            Assert.Equal(0, VisualStudio.Editor.GetCaretPosition());
+
             VisualStudioInstance.RetryRpcCall(() => window.Close(vsSaveChanges.vsSaveChangesNo));
         }