From c4738d7316e9fc312ccb1cb4384705974f638284 Mon Sep 17 00:00:00 2001
From: Keith Simmons <keith@zed.dev>
Date: Tue, 10 May 2022 14:30:27 -0700
Subject: [PATCH] Add test coverage for same cursor in multiple excerpts of the
 same buffer

---
 crates/editor/src/editor.rs         | 80 ++++++++++++-----------------
 crates/util/src/test/marked_text.rs | 16 ++++--
 2 files changed, 44 insertions(+), 52 deletions(-)

diff --git a/crates/editor/src/editor.rs b/crates/editor/src/editor.rs
index 9efbd6accb..a9a95280d8 100644
--- a/crates/editor/src/editor.rs
+++ b/crates/editor/src/editor.rs
@@ -9801,65 +9801,49 @@ mod tests {
     #[gpui::test]
     fn test_editing_overlapping_excerpts(cx: &mut gpui::MutableAppContext) {
         cx.set_global(Settings::test(cx));
-        let buffer = cx.add_model(|cx| {
-            Buffer::new(
-                0,
-                indoc! {"
-                    aaaa
-                    bbbb
-                    cccc"},
-                cx,
-            )
-        });
-
+        let (initial_text, excerpt_ranges) = marked_text_ranges(indoc! {"
+                [aaaa
+                (bbbb]
+                cccc)"});
+        let buffer = cx.add_model(|cx| Buffer::new(0, initial_text, cx));
         let multibuffer = cx.add_model(|cx| {
             let mut multibuffer = MultiBuffer::new(0);
-            multibuffer.push_excerpts(
-                buffer,
-                [
-                    Point::new(0, 0)..Point::new(1, 4),
-                    Point::new(1, 0)..Point::new(2, 4),
-                ],
-                cx,
-            );
+            multibuffer.push_excerpts(buffer, excerpt_ranges, cx);
             multibuffer
         });
 
-        assert_eq!(
-            multibuffer.read(cx).read(cx).text(),
-            "aaaa\nbbbb\nbbbb\ncccc"
-        );
-
         let (_, view) = cx.add_window(Default::default(), |cx| build_editor(multibuffer, cx));
         view.update(cx, |view, cx| {
-            view.select_ranges(
-                [
-                    Point::new(1, 1)..Point::new(1, 1),
-                    Point::new(2, 3)..Point::new(2, 3),
-                ],
-                None,
-                cx,
-            );
+            let (expected_text, selection_ranges) = marked_text_ranges(indoc! {"
+                aaaa
+                b|bbb
+                b|bb|b
+                cccc"});
+            assert_eq!(view.text(cx), expected_text);
+            view.select_ranges(selection_ranges, None, cx);
 
             view.handle_input(&Input("X".to_string()), cx);
-            assert_eq!(view.text(cx), "aaaa\nbXbbXb\nbXbbXb\ncccc");
-            assert_eq!(
-                view.selected_ranges(cx),
-                [
-                    Point::new(1, 2)..Point::new(1, 2),
-                    Point::new(2, 5)..Point::new(2, 5),
-                ]
-            );
+
+            let (expected_text, expected_selections) = marked_text_ranges(indoc! {"
+                aaaa
+                bX|bbXb
+                bX|bbX|b
+                cccc"});
+            assert_eq!(view.text(cx), expected_text);
+            assert_eq!(view.selected_ranges(cx), expected_selections);
 
             view.newline(&Newline, cx);
-            assert_eq!(view.text(cx), "aaaa\nbX\nbbX\nb\nbX\nbbX\nb\ncccc");
-            assert_eq!(
-                view.selected_ranges(cx),
-                [
-                    Point::new(2, 0)..Point::new(2, 0),
-                    Point::new(6, 0)..Point::new(6, 0),
-                ]
-            );
+            let (expected_text, expected_selections) = marked_text_ranges(indoc! {"
+                aaaa
+                bX
+                |bbX
+                b
+                bX
+                |bbX
+                |b
+                cccc"});
+            assert_eq!(view.text(cx), expected_text);
+            assert_eq!(view.selected_ranges(cx), expected_selections);
         });
     }
 
diff --git a/crates/util/src/test/marked_text.rs b/crates/util/src/test/marked_text.rs
index e0f549b7df..23aa2d5806 100644
--- a/crates/util/src/test/marked_text.rs
+++ b/crates/util/src/test/marked_text.rs
@@ -55,10 +55,18 @@ pub fn marked_text_ranges_by(
     (unmarked_text, range_lookup)
 }
 
-pub fn marked_text_ranges(marked_text: &str) -> (String, Vec<Range<usize>>) {
-    let (unmarked_text, mut ranges) = marked_text_ranges_by(marked_text, vec![('[', ']')]);
+// Returns ranges delimited by (), [], and <> ranges. Ranges using the same markers
+// must not be overlapping. May also include | for empty ranges
+pub fn marked_text_ranges(full_marked_text: &str) -> (String, Vec<Range<usize>>) {
+    let (range_marked_text, empty_offsets) = marked_text(full_marked_text);
+    let (unmarked, range_lookup) =
+        marked_text_ranges_by(&range_marked_text, vec![('[', ']'), ('(', ')'), ('<', '>')]);
     (
-        unmarked_text,
-        ranges.remove(&('[', ']')).unwrap_or_else(Vec::new),
+        unmarked,
+        range_lookup
+            .into_values()
+            .flatten()
+            .chain(empty_offsets.into_iter().map(|offset| offset..offset))
+            .collect(),
     )
 }