1pub mod editor_lsp_test_context;
2pub mod editor_test_context;
3
4use crate::{
5 display_map::{DisplayMap, DisplaySnapshot, ToDisplayPoint},
6 DisplayPoint, Editor, EditorMode, FoldPlaceholder, MultiBuffer,
7};
8use gpui::{Context, Font, FontFeatures, FontStyle, FontWeight, Model, Pixels, ViewContext};
9use project::Project;
10use util::test::{marked_text_offsets, marked_text_ranges};
11
12#[cfg(test)]
13#[ctor::ctor]
14fn init_logger() {
15 if std::env::var("RUST_LOG").is_ok() {
16 env_logger::init();
17 }
18}
19
20// Returns a snapshot from text containing '|' character markers with the markers removed, and DisplayPoints for each one.
21pub fn marked_display_snapshot(
22 text: &str,
23 cx: &mut gpui::AppContext,
24) -> (DisplaySnapshot, Vec<DisplayPoint>) {
25 let (unmarked_text, markers) = marked_text_offsets(text);
26
27 let font = Font {
28 family: "Courier".into(),
29 features: FontFeatures::default(),
30 weight: FontWeight::default(),
31 style: FontStyle::default(),
32 };
33 let font_size: Pixels = 14usize.into();
34
35 let buffer = MultiBuffer::build_simple(&unmarked_text, cx);
36 let display_map = cx.new_model(|cx| {
37 DisplayMap::new(
38 buffer,
39 font,
40 font_size,
41 None,
42 1,
43 1,
44 FoldPlaceholder::test(),
45 cx,
46 )
47 });
48 let snapshot = display_map.update(cx, |map, cx| map.snapshot(cx));
49 let markers = markers
50 .into_iter()
51 .map(|offset| offset.to_display_point(&snapshot))
52 .collect();
53
54 (snapshot, markers)
55}
56
57pub fn select_ranges(editor: &mut Editor, marked_text: &str, cx: &mut ViewContext<Editor>) {
58 let (unmarked_text, text_ranges) = marked_text_ranges(marked_text, true);
59 assert_eq!(editor.text(cx), unmarked_text);
60 editor.change_selections(None, cx, |s| s.select_ranges(text_ranges));
61}
62
63pub fn assert_text_with_selections(
64 editor: &mut Editor,
65 marked_text: &str,
66 cx: &mut ViewContext<Editor>,
67) {
68 let (unmarked_text, text_ranges) = marked_text_ranges(marked_text, true);
69 assert_eq!(editor.text(cx), unmarked_text);
70 assert_eq!(editor.selections.ranges(cx), text_ranges);
71}
72
73// RA thinks this is dead code even though it is used in a whole lot of tests
74#[allow(dead_code)]
75#[cfg(any(test, feature = "test-support"))]
76pub(crate) fn build_editor(buffer: Model<MultiBuffer>, cx: &mut ViewContext<Editor>) -> Editor {
77 Editor::new(EditorMode::Full, buffer, None, cx)
78}
79
80pub(crate) fn build_editor_with_project(
81 project: Model<Project>,
82 buffer: Model<MultiBuffer>,
83 cx: &mut ViewContext<Editor>,
84) -> Editor {
85 Editor::new(EditorMode::Full, buffer, Some(project), cx)
86}
87
88#[cfg(any(test, feature = "test-support"))]
89pub fn editor_hunks(
90 editor: &Editor,
91 snapshot: &DisplaySnapshot,
92 cx: &mut ViewContext<'_, Editor>,
93) -> Vec<(
94 String,
95 git::diff::DiffHunkStatus,
96 std::ops::Range<crate::DisplayRow>,
97)> {
98 use multi_buffer::MultiBufferRow;
99 use text::Point;
100
101 use crate::hunk_status;
102
103 snapshot
104 .buffer_snapshot
105 .git_diff_hunks_in_range(MultiBufferRow::MIN..MultiBufferRow::MAX)
106 .map(|hunk| {
107 let display_range = Point::new(hunk.associated_range.start.0, 0)
108 .to_display_point(snapshot)
109 .row()
110 ..Point::new(hunk.associated_range.end.0, 0)
111 .to_display_point(snapshot)
112 .row();
113 let (_, buffer, _) = editor
114 .buffer()
115 .read(cx)
116 .excerpt_containing(Point::new(hunk.associated_range.start.0, 0), cx)
117 .expect("no excerpt for expanded buffer's hunk start");
118 let diff_base = buffer
119 .read(cx)
120 .diff_base()
121 .expect("should have a diff base for expanded hunk")
122 .slice(hunk.diff_base_byte_range.clone())
123 .to_string();
124 (diff_base, hunk_status(&hunk), display_range)
125 })
126 .collect()
127}
128
129#[cfg(any(test, feature = "test-support"))]
130pub fn expanded_hunks(
131 editor: &Editor,
132 snapshot: &DisplaySnapshot,
133 cx: &mut ViewContext<'_, Editor>,
134) -> Vec<(
135 String,
136 git::diff::DiffHunkStatus,
137 std::ops::Range<crate::DisplayRow>,
138)> {
139 editor
140 .expanded_hunks
141 .hunks(false)
142 .map(|expanded_hunk| {
143 let hunk_display_range = expanded_hunk
144 .hunk_range
145 .start
146 .to_display_point(snapshot)
147 .row()
148 ..expanded_hunk
149 .hunk_range
150 .end
151 .to_display_point(snapshot)
152 .row();
153 let (_, buffer, _) = editor
154 .buffer()
155 .read(cx)
156 .excerpt_containing(expanded_hunk.hunk_range.start, cx)
157 .expect("no excerpt for expanded buffer's hunk start");
158 let diff_base = buffer
159 .read(cx)
160 .diff_base()
161 .expect("should have a diff base for expanded hunk")
162 .slice(expanded_hunk.diff_base_byte_range.clone())
163 .to_string();
164 (diff_base, expanded_hunk.status, hunk_display_range)
165 })
166 .collect()
167}
168
169#[cfg(any(test, feature = "test-support"))]
170pub fn expanded_hunks_background_highlights(
171 editor: &mut Editor,
172 cx: &mut gpui::WindowContext,
173) -> Vec<std::ops::RangeInclusive<crate::DisplayRow>> {
174 use crate::DisplayRow;
175
176 let mut highlights = Vec::new();
177
178 let mut range_start = 0;
179 let mut previous_highlighted_row = None;
180 for (highlighted_row, _) in editor.highlighted_display_rows(cx) {
181 match previous_highlighted_row {
182 Some(previous_row) => {
183 if previous_row + 1 != highlighted_row.0 {
184 highlights.push(DisplayRow(range_start)..=DisplayRow(previous_row));
185 range_start = highlighted_row.0;
186 }
187 }
188 None => {
189 range_start = highlighted_row.0;
190 }
191 }
192 previous_highlighted_row = Some(highlighted_row.0);
193 }
194 if let Some(previous_row) = previous_highlighted_row {
195 highlights.push(DisplayRow(range_start)..=DisplayRow(previous_row));
196 }
197
198 highlights
199}