1use super::{RandomizedTest, TestClient, TestError, TestServer, UserTestPlan};
2use crate::{db::UserId, tests::run_randomized_test};
3use anyhow::{anyhow, Result};
4use async_trait::async_trait;
5use call::ActiveCall;
6use collections::{BTreeMap, HashMap};
7use editor::Bias;
8use fs::{FakeFs, Fs as _};
9use git::status::{FileStatus, StatusCode, TrackedStatus, UnmergedStatus, UnmergedStatusCode};
10use gpui::{BackgroundExecutor, Entity, TestAppContext};
11use language::{
12 range_to_lsp, FakeLspAdapter, Language, LanguageConfig, LanguageMatcher, PointUtf16,
13};
14use lsp::FakeLanguageServer;
15use pretty_assertions::assert_eq;
16use project::{
17 search::SearchQuery, search::SearchResult, Project, ProjectPath, DEFAULT_COMPLETION_CONTEXT,
18};
19use rand::{
20 distributions::{Alphanumeric, DistString},
21 prelude::*,
22};
23use serde::{Deserialize, Serialize};
24use std::{
25 ops::{Deref, Range},
26 path::{Path, PathBuf},
27 rc::Rc,
28 sync::Arc,
29};
30use util::ResultExt;
31
32#[gpui::test(
33 iterations = 100,
34 on_failure = "crate::tests::save_randomized_test_plan"
35)]
36async fn test_random_project_collaboration(
37 cx: &mut TestAppContext,
38 executor: BackgroundExecutor,
39 rng: StdRng,
40) {
41 run_randomized_test::<ProjectCollaborationTest>(cx, executor, rng).await;
42}
43
44#[derive(Clone, Debug, Serialize, Deserialize)]
45enum ClientOperation {
46 AcceptIncomingCall,
47 RejectIncomingCall,
48 LeaveCall,
49 InviteContactToCall {
50 user_id: UserId,
51 },
52 OpenLocalProject {
53 first_root_name: String,
54 },
55 OpenRemoteProject {
56 host_id: UserId,
57 first_root_name: String,
58 },
59 AddWorktreeToProject {
60 project_root_name: String,
61 new_root_path: PathBuf,
62 },
63 CloseRemoteProject {
64 project_root_name: String,
65 },
66 OpenBuffer {
67 project_root_name: String,
68 is_local: bool,
69 full_path: PathBuf,
70 },
71 SearchProject {
72 project_root_name: String,
73 is_local: bool,
74 query: String,
75 detach: bool,
76 },
77 EditBuffer {
78 project_root_name: String,
79 is_local: bool,
80 full_path: PathBuf,
81 edits: Vec<(Range<usize>, Arc<str>)>,
82 },
83 CloseBuffer {
84 project_root_name: String,
85 is_local: bool,
86 full_path: PathBuf,
87 },
88 SaveBuffer {
89 project_root_name: String,
90 is_local: bool,
91 full_path: PathBuf,
92 detach: bool,
93 },
94 RequestLspDataInBuffer {
95 project_root_name: String,
96 is_local: bool,
97 full_path: PathBuf,
98 offset: usize,
99 kind: LspRequestKind,
100 detach: bool,
101 },
102 CreateWorktreeEntry {
103 project_root_name: String,
104 is_local: bool,
105 full_path: PathBuf,
106 is_dir: bool,
107 },
108 WriteFsEntry {
109 path: PathBuf,
110 is_dir: bool,
111 content: String,
112 },
113 GitOperation {
114 operation: GitOperation,
115 },
116}
117
118#[derive(Clone, Debug, Serialize, Deserialize)]
119enum GitOperation {
120 WriteGitIndex {
121 repo_path: PathBuf,
122 contents: Vec<(PathBuf, String)>,
123 },
124 WriteGitBranch {
125 repo_path: PathBuf,
126 new_branch: Option<String>,
127 },
128 WriteGitStatuses {
129 repo_path: PathBuf,
130 statuses: Vec<(PathBuf, FileStatus)>,
131 },
132}
133
134#[derive(Clone, Debug, Serialize, Deserialize)]
135enum LspRequestKind {
136 Rename,
137 Completion,
138 CodeAction,
139 Definition,
140 Highlights,
141}
142
143struct ProjectCollaborationTest;
144
145#[async_trait(?Send)]
146impl RandomizedTest for ProjectCollaborationTest {
147 type Operation = ClientOperation;
148
149 async fn initialize(server: &mut TestServer, users: &[UserTestPlan]) {
150 let db = &server.app_state.db;
151 for (ix, user_a) in users.iter().enumerate() {
152 for user_b in &users[ix + 1..] {
153 db.send_contact_request(user_a.user_id, user_b.user_id)
154 .await
155 .unwrap();
156 db.respond_to_contact_request(user_b.user_id, user_a.user_id, true)
157 .await
158 .unwrap();
159 }
160 }
161 }
162
163 fn generate_operation(
164 client: &TestClient,
165 rng: &mut StdRng,
166 plan: &mut UserTestPlan,
167 cx: &TestAppContext,
168 ) -> ClientOperation {
169 let call = cx.read(ActiveCall::global);
170 loop {
171 match rng.gen_range(0..100_u32) {
172 // Mutate the call
173 0..=29 => {
174 // Respond to an incoming call
175 if call.read_with(cx, |call, _| call.incoming().borrow().is_some()) {
176 break if rng.gen_bool(0.7) {
177 ClientOperation::AcceptIncomingCall
178 } else {
179 ClientOperation::RejectIncomingCall
180 };
181 }
182
183 match rng.gen_range(0..100_u32) {
184 // Invite a contact to the current call
185 0..=70 => {
186 let available_contacts =
187 client.user_store().read_with(cx, |user_store, _| {
188 user_store
189 .contacts()
190 .iter()
191 .filter(|contact| contact.online && !contact.busy)
192 .cloned()
193 .collect::<Vec<_>>()
194 });
195 if !available_contacts.is_empty() {
196 let contact = available_contacts.choose(rng).unwrap();
197 break ClientOperation::InviteContactToCall {
198 user_id: UserId(contact.user.id as i32),
199 };
200 }
201 }
202
203 // Leave the current call
204 71.. => {
205 if plan.allow_client_disconnection
206 && call.read_with(cx, |call, _| call.room().is_some())
207 {
208 break ClientOperation::LeaveCall;
209 }
210 }
211 }
212 }
213
214 // Mutate projects
215 30..=59 => match rng.gen_range(0..100_u32) {
216 // Open a new project
217 0..=70 => {
218 // Open a remote project
219 if let Some(room) = call.read_with(cx, |call, _| call.room().cloned()) {
220 let existing_dev_server_project_ids = cx.read(|cx| {
221 client
222 .dev_server_projects()
223 .iter()
224 .map(|p| p.read(cx).remote_id().unwrap())
225 .collect::<Vec<_>>()
226 });
227 let new_dev_server_projects = room.read_with(cx, |room, _| {
228 room.remote_participants()
229 .values()
230 .flat_map(|participant| {
231 participant.projects.iter().filter_map(|project| {
232 if existing_dev_server_project_ids.contains(&project.id)
233 {
234 None
235 } else {
236 Some((
237 UserId::from_proto(participant.user.id),
238 project.worktree_root_names[0].clone(),
239 ))
240 }
241 })
242 })
243 .collect::<Vec<_>>()
244 });
245 if !new_dev_server_projects.is_empty() {
246 let (host_id, first_root_name) =
247 new_dev_server_projects.choose(rng).unwrap().clone();
248 break ClientOperation::OpenRemoteProject {
249 host_id,
250 first_root_name,
251 };
252 }
253 }
254 // Open a local project
255 else {
256 let first_root_name = plan.next_root_dir_name();
257 break ClientOperation::OpenLocalProject { first_root_name };
258 }
259 }
260
261 // Close a remote project
262 71..=80 => {
263 if !client.dev_server_projects().is_empty() {
264 let project = client.dev_server_projects().choose(rng).unwrap().clone();
265 let first_root_name = root_name_for_project(&project, cx);
266 break ClientOperation::CloseRemoteProject {
267 project_root_name: first_root_name,
268 };
269 }
270 }
271
272 // Mutate project worktrees
273 81.. => match rng.gen_range(0..100_u32) {
274 // Add a worktree to a local project
275 0..=50 => {
276 let Some(project) = client.local_projects().choose(rng).cloned() else {
277 continue;
278 };
279 let project_root_name = root_name_for_project(&project, cx);
280 let mut paths = client.fs().paths(false);
281 paths.remove(0);
282 let new_root_path = if paths.is_empty() || rng.gen() {
283 Path::new("/").join(plan.next_root_dir_name())
284 } else {
285 paths.choose(rng).unwrap().clone()
286 };
287 break ClientOperation::AddWorktreeToProject {
288 project_root_name,
289 new_root_path,
290 };
291 }
292
293 // Add an entry to a worktree
294 _ => {
295 let Some(project) = choose_random_project(client, rng) else {
296 continue;
297 };
298 let project_root_name = root_name_for_project(&project, cx);
299 let is_local = project.read_with(cx, |project, _| project.is_local());
300 let worktree = project.read_with(cx, |project, cx| {
301 project
302 .worktrees(cx)
303 .filter(|worktree| {
304 let worktree = worktree.read(cx);
305 worktree.is_visible()
306 && worktree.entries(false, 0).any(|e| e.is_file())
307 && worktree.root_entry().map_or(false, |e| e.is_dir())
308 })
309 .choose(rng)
310 });
311 let Some(worktree) = worktree else { continue };
312 let is_dir = rng.gen::<bool>();
313 let mut full_path =
314 worktree.read_with(cx, |w, _| PathBuf::from(w.root_name()));
315 full_path.push(gen_file_name(rng));
316 if !is_dir {
317 full_path.set_extension("rs");
318 }
319 break ClientOperation::CreateWorktreeEntry {
320 project_root_name,
321 is_local,
322 full_path,
323 is_dir,
324 };
325 }
326 },
327 },
328
329 // Query and mutate buffers
330 60..=90 => {
331 let Some(project) = choose_random_project(client, rng) else {
332 continue;
333 };
334 let project_root_name = root_name_for_project(&project, cx);
335 let is_local = project.read_with(cx, |project, _| project.is_local());
336
337 match rng.gen_range(0..100_u32) {
338 // Manipulate an existing buffer
339 0..=70 => {
340 let Some(buffer) = client
341 .buffers_for_project(&project)
342 .iter()
343 .choose(rng)
344 .cloned()
345 else {
346 continue;
347 };
348
349 let full_path = buffer
350 .read_with(cx, |buffer, cx| buffer.file().unwrap().full_path(cx));
351
352 match rng.gen_range(0..100_u32) {
353 // Close the buffer
354 0..=15 => {
355 break ClientOperation::CloseBuffer {
356 project_root_name,
357 is_local,
358 full_path,
359 };
360 }
361 // Save the buffer
362 16..=29 if buffer.read_with(cx, |b, _| b.is_dirty()) => {
363 let detach = rng.gen_bool(0.3);
364 break ClientOperation::SaveBuffer {
365 project_root_name,
366 is_local,
367 full_path,
368 detach,
369 };
370 }
371 // Edit the buffer
372 30..=69 => {
373 let edits = buffer
374 .read_with(cx, |buffer, _| buffer.get_random_edits(rng, 3));
375 break ClientOperation::EditBuffer {
376 project_root_name,
377 is_local,
378 full_path,
379 edits,
380 };
381 }
382 // Make an LSP request
383 _ => {
384 let offset = buffer.read_with(cx, |buffer, _| {
385 buffer.clip_offset(
386 rng.gen_range(0..=buffer.len()),
387 language::Bias::Left,
388 )
389 });
390 let detach = rng.gen();
391 break ClientOperation::RequestLspDataInBuffer {
392 project_root_name,
393 full_path,
394 offset,
395 is_local,
396 kind: match rng.gen_range(0..5_u32) {
397 0 => LspRequestKind::Rename,
398 1 => LspRequestKind::Highlights,
399 2 => LspRequestKind::Definition,
400 3 => LspRequestKind::CodeAction,
401 4.. => LspRequestKind::Completion,
402 },
403 detach,
404 };
405 }
406 }
407 }
408
409 71..=80 => {
410 let query = rng.gen_range('a'..='z').to_string();
411 let detach = rng.gen_bool(0.3);
412 break ClientOperation::SearchProject {
413 project_root_name,
414 is_local,
415 query,
416 detach,
417 };
418 }
419
420 // Open a buffer
421 81.. => {
422 let worktree = project.read_with(cx, |project, cx| {
423 project
424 .worktrees(cx)
425 .filter(|worktree| {
426 let worktree = worktree.read(cx);
427 worktree.is_visible()
428 && worktree.entries(false, 0).any(|e| e.is_file())
429 })
430 .choose(rng)
431 });
432 let Some(worktree) = worktree else { continue };
433 let full_path = worktree.read_with(cx, |worktree, _| {
434 let entry = worktree
435 .entries(false, 0)
436 .filter(|e| e.is_file())
437 .choose(rng)
438 .unwrap();
439 if entry.path.as_ref() == Path::new("") {
440 Path::new(worktree.root_name()).into()
441 } else {
442 Path::new(worktree.root_name()).join(&entry.path)
443 }
444 });
445 break ClientOperation::OpenBuffer {
446 project_root_name,
447 is_local,
448 full_path,
449 };
450 }
451 }
452 }
453
454 // Update a git related action
455 91..=95 => {
456 break ClientOperation::GitOperation {
457 operation: generate_git_operation(rng, client),
458 };
459 }
460
461 // Create or update a file or directory
462 96.. => {
463 let is_dir = rng.gen::<bool>();
464 let content;
465 let mut path;
466 let dir_paths = client.fs().directories(false);
467
468 if is_dir {
469 content = String::new();
470 path = dir_paths.choose(rng).unwrap().clone();
471 path.push(gen_file_name(rng));
472 } else {
473 content = Alphanumeric.sample_string(rng, 16);
474
475 // Create a new file or overwrite an existing file
476 let file_paths = client.fs().files();
477 if file_paths.is_empty() || rng.gen_bool(0.5) {
478 path = dir_paths.choose(rng).unwrap().clone();
479 path.push(gen_file_name(rng));
480 path.set_extension("rs");
481 } else {
482 path = file_paths.choose(rng).unwrap().clone()
483 };
484 }
485 break ClientOperation::WriteFsEntry {
486 path,
487 is_dir,
488 content,
489 };
490 }
491 }
492 }
493 }
494
495 async fn apply_operation(
496 client: &TestClient,
497 operation: ClientOperation,
498 cx: &mut TestAppContext,
499 ) -> Result<(), TestError> {
500 match operation {
501 ClientOperation::AcceptIncomingCall => {
502 let active_call = cx.read(ActiveCall::global);
503 if active_call.read_with(cx, |call, _| call.incoming().borrow().is_none()) {
504 Err(TestError::Inapplicable)?;
505 }
506
507 log::info!("{}: accepting incoming call", client.username);
508 active_call
509 .update(cx, |call, cx| call.accept_incoming(cx))
510 .await?;
511 }
512
513 ClientOperation::RejectIncomingCall => {
514 let active_call = cx.read(ActiveCall::global);
515 if active_call.read_with(cx, |call, _| call.incoming().borrow().is_none()) {
516 Err(TestError::Inapplicable)?;
517 }
518
519 log::info!("{}: declining incoming call", client.username);
520 active_call.update(cx, |call, cx| call.decline_incoming(cx))?;
521 }
522
523 ClientOperation::LeaveCall => {
524 let active_call = cx.read(ActiveCall::global);
525 if active_call.read_with(cx, |call, _| call.room().is_none()) {
526 Err(TestError::Inapplicable)?;
527 }
528
529 log::info!("{}: hanging up", client.username);
530 active_call.update(cx, |call, cx| call.hang_up(cx)).await?;
531 }
532
533 ClientOperation::InviteContactToCall { user_id } => {
534 let active_call = cx.read(ActiveCall::global);
535
536 log::info!("{}: inviting {}", client.username, user_id,);
537 active_call
538 .update(cx, |call, cx| call.invite(user_id.to_proto(), None, cx))
539 .await
540 .log_err();
541 }
542
543 ClientOperation::OpenLocalProject { first_root_name } => {
544 log::info!(
545 "{}: opening local project at {:?}",
546 client.username,
547 first_root_name
548 );
549
550 let root_path = Path::new("/").join(&first_root_name);
551 client.fs().create_dir(&root_path).await.unwrap();
552 client
553 .fs()
554 .create_file(&root_path.join("main.rs"), Default::default())
555 .await
556 .unwrap();
557 let project = client.build_local_project(root_path, cx).await.0;
558 ensure_project_shared(&project, client, cx).await;
559 client.local_projects_mut().push(project.clone());
560 }
561
562 ClientOperation::AddWorktreeToProject {
563 project_root_name,
564 new_root_path,
565 } => {
566 let project = project_for_root_name(client, &project_root_name, cx)
567 .ok_or(TestError::Inapplicable)?;
568
569 log::info!(
570 "{}: finding/creating local worktree at {:?} to project with root path {}",
571 client.username,
572 new_root_path,
573 project_root_name
574 );
575
576 ensure_project_shared(&project, client, cx).await;
577 if !client.fs().paths(false).contains(&new_root_path) {
578 client.fs().create_dir(&new_root_path).await.unwrap();
579 }
580 project
581 .update(cx, |project, cx| {
582 project.find_or_create_worktree(&new_root_path, true, cx)
583 })
584 .await
585 .unwrap();
586 }
587
588 ClientOperation::CloseRemoteProject { project_root_name } => {
589 let project = project_for_root_name(client, &project_root_name, cx)
590 .ok_or(TestError::Inapplicable)?;
591
592 log::info!(
593 "{}: closing remote project with root path {}",
594 client.username,
595 project_root_name,
596 );
597
598 let ix = client
599 .dev_server_projects()
600 .iter()
601 .position(|p| p == &project)
602 .unwrap();
603 cx.update(|_| {
604 client.dev_server_projects_mut().remove(ix);
605 client.buffers().retain(|p, _| *p != project);
606 drop(project);
607 });
608 }
609
610 ClientOperation::OpenRemoteProject {
611 host_id,
612 first_root_name,
613 } => {
614 let active_call = cx.read(ActiveCall::global);
615 let project = active_call
616 .update(cx, |call, cx| {
617 let room = call.room().cloned()?;
618 let participant = room
619 .read(cx)
620 .remote_participants()
621 .get(&host_id.to_proto())?;
622 let project_id = participant
623 .projects
624 .iter()
625 .find(|project| project.worktree_root_names[0] == first_root_name)?
626 .id;
627 Some(room.update(cx, |room, cx| {
628 room.join_project(
629 project_id,
630 client.language_registry().clone(),
631 FakeFs::new(cx.background_executor().clone()),
632 cx,
633 )
634 }))
635 })
636 .ok_or(TestError::Inapplicable)?;
637
638 log::info!(
639 "{}: joining remote project of user {}, root name {}",
640 client.username,
641 host_id,
642 first_root_name,
643 );
644
645 let project = project.await?;
646 client.dev_server_projects_mut().push(project.clone());
647 }
648
649 ClientOperation::CreateWorktreeEntry {
650 project_root_name,
651 is_local,
652 full_path,
653 is_dir,
654 } => {
655 let project = project_for_root_name(client, &project_root_name, cx)
656 .ok_or(TestError::Inapplicable)?;
657 let project_path = project_path_for_full_path(&project, &full_path, cx)
658 .ok_or(TestError::Inapplicable)?;
659
660 log::info!(
661 "{}: creating {} at path {:?} in {} project {}",
662 client.username,
663 if is_dir { "dir" } else { "file" },
664 full_path,
665 if is_local { "local" } else { "remote" },
666 project_root_name,
667 );
668
669 ensure_project_shared(&project, client, cx).await;
670 project
671 .update(cx, |p, cx| p.create_entry(project_path, is_dir, cx))
672 .await?;
673 }
674
675 ClientOperation::OpenBuffer {
676 project_root_name,
677 is_local,
678 full_path,
679 } => {
680 let project = project_for_root_name(client, &project_root_name, cx)
681 .ok_or(TestError::Inapplicable)?;
682 let project_path = project_path_for_full_path(&project, &full_path, cx)
683 .ok_or(TestError::Inapplicable)?;
684
685 log::info!(
686 "{}: opening buffer {:?} in {} project {}",
687 client.username,
688 full_path,
689 if is_local { "local" } else { "remote" },
690 project_root_name,
691 );
692
693 ensure_project_shared(&project, client, cx).await;
694 let buffer = project
695 .update(cx, |project, cx| project.open_buffer(project_path, cx))
696 .await?;
697 client.buffers_for_project(&project).insert(buffer);
698 }
699
700 ClientOperation::EditBuffer {
701 project_root_name,
702 is_local,
703 full_path,
704 edits,
705 } => {
706 let project = project_for_root_name(client, &project_root_name, cx)
707 .ok_or(TestError::Inapplicable)?;
708 let buffer = buffer_for_full_path(client, &project, &full_path, cx)
709 .ok_or(TestError::Inapplicable)?;
710
711 log::info!(
712 "{}: editing buffer {:?} in {} project {} with {:?}",
713 client.username,
714 full_path,
715 if is_local { "local" } else { "remote" },
716 project_root_name,
717 edits
718 );
719
720 ensure_project_shared(&project, client, cx).await;
721 buffer.update(cx, |buffer, cx| {
722 let snapshot = buffer.snapshot();
723 buffer.edit(
724 edits.into_iter().map(|(range, text)| {
725 let start = snapshot.clip_offset(range.start, Bias::Left);
726 let end = snapshot.clip_offset(range.end, Bias::Right);
727 (start..end, text)
728 }),
729 None,
730 cx,
731 );
732 });
733 }
734
735 ClientOperation::CloseBuffer {
736 project_root_name,
737 is_local,
738 full_path,
739 } => {
740 let project = project_for_root_name(client, &project_root_name, cx)
741 .ok_or(TestError::Inapplicable)?;
742 let buffer = buffer_for_full_path(client, &project, &full_path, cx)
743 .ok_or(TestError::Inapplicable)?;
744
745 log::info!(
746 "{}: closing buffer {:?} in {} project {}",
747 client.username,
748 full_path,
749 if is_local { "local" } else { "remote" },
750 project_root_name
751 );
752
753 ensure_project_shared(&project, client, cx).await;
754 cx.update(|_| {
755 client.buffers_for_project(&project).remove(&buffer);
756 drop(buffer);
757 });
758 }
759
760 ClientOperation::SaveBuffer {
761 project_root_name,
762 is_local,
763 full_path,
764 detach,
765 } => {
766 let project = project_for_root_name(client, &project_root_name, cx)
767 .ok_or(TestError::Inapplicable)?;
768 let buffer = buffer_for_full_path(client, &project, &full_path, cx)
769 .ok_or(TestError::Inapplicable)?;
770
771 log::info!(
772 "{}: saving buffer {:?} in {} project {}, {}",
773 client.username,
774 full_path,
775 if is_local { "local" } else { "remote" },
776 project_root_name,
777 if detach { "detaching" } else { "awaiting" }
778 );
779
780 ensure_project_shared(&project, client, cx).await;
781 let requested_version = buffer.read_with(cx, |buffer, _| buffer.version());
782 let save =
783 project.update(cx, |project, cx| project.save_buffer(buffer.clone(), cx));
784 let save = cx.spawn(|cx| async move {
785 save.await
786 .map_err(|err| anyhow!("save request failed: {:?}", err))?;
787 assert!(buffer
788 .read_with(&cx, |buffer, _| { buffer.saved_version().to_owned() })
789 .expect("App should not be dropped")
790 .observed_all(&requested_version));
791 anyhow::Ok(())
792 });
793 if detach {
794 cx.update(|cx| save.detach_and_log_err(cx));
795 } else {
796 save.await?;
797 }
798 }
799
800 ClientOperation::RequestLspDataInBuffer {
801 project_root_name,
802 is_local,
803 full_path,
804 offset,
805 kind,
806 detach,
807 } => {
808 let project = project_for_root_name(client, &project_root_name, cx)
809 .ok_or(TestError::Inapplicable)?;
810 let buffer = buffer_for_full_path(client, &project, &full_path, cx)
811 .ok_or(TestError::Inapplicable)?;
812
813 log::info!(
814 "{}: request LSP {:?} for buffer {:?} in {} project {}, {}",
815 client.username,
816 kind,
817 full_path,
818 if is_local { "local" } else { "remote" },
819 project_root_name,
820 if detach { "detaching" } else { "awaiting" }
821 );
822
823 use futures::{FutureExt as _, TryFutureExt as _};
824 let offset = buffer.read_with(cx, |b, _| b.clip_offset(offset, Bias::Left));
825
826 let process_lsp_request = project.update(cx, |project, cx| match kind {
827 LspRequestKind::Rename => project
828 .prepare_rename(buffer, offset, cx)
829 .map_ok(|_| ())
830 .boxed(),
831 LspRequestKind::Completion => project
832 .completions(&buffer, offset, DEFAULT_COMPLETION_CONTEXT, cx)
833 .map_ok(|_| ())
834 .boxed(),
835 LspRequestKind::CodeAction => project
836 .code_actions(&buffer, offset..offset, None, cx)
837 .map(|_| Ok(()))
838 .boxed(),
839 LspRequestKind::Definition => project
840 .definition(&buffer, offset, cx)
841 .map_ok(|_| ())
842 .boxed(),
843 LspRequestKind::Highlights => project
844 .document_highlights(&buffer, offset, cx)
845 .map_ok(|_| ())
846 .boxed(),
847 });
848 let request = cx.foreground_executor().spawn(process_lsp_request);
849 if detach {
850 request.detach();
851 } else {
852 request.await?;
853 }
854 }
855
856 ClientOperation::SearchProject {
857 project_root_name,
858 is_local,
859 query,
860 detach,
861 } => {
862 let project = project_for_root_name(client, &project_root_name, cx)
863 .ok_or(TestError::Inapplicable)?;
864
865 log::info!(
866 "{}: search {} project {} for {:?}, {}",
867 client.username,
868 if is_local { "local" } else { "remote" },
869 project_root_name,
870 query,
871 if detach { "detaching" } else { "awaiting" }
872 );
873
874 let search = project.update(cx, |project, cx| {
875 project.search(
876 SearchQuery::text(
877 query,
878 false,
879 false,
880 false,
881 Default::default(),
882 Default::default(),
883 None,
884 )
885 .unwrap(),
886 cx,
887 )
888 });
889 drop(project);
890 let search = cx.executor().spawn(async move {
891 let mut results = HashMap::default();
892 while let Ok(result) = search.recv().await {
893 if let SearchResult::Buffer { buffer, ranges } = result {
894 results.entry(buffer).or_insert(ranges);
895 }
896 }
897 results
898 });
899 search.await;
900 }
901
902 ClientOperation::WriteFsEntry {
903 path,
904 is_dir,
905 content,
906 } => {
907 if !client
908 .fs()
909 .directories(false)
910 .contains(&path.parent().unwrap().to_owned())
911 {
912 return Err(TestError::Inapplicable);
913 }
914
915 if is_dir {
916 log::info!("{}: creating dir at {:?}", client.username, path);
917 client.fs().create_dir(&path).await.unwrap();
918 } else {
919 let exists = client.fs().metadata(&path).await?.is_some();
920 let verb = if exists { "updating" } else { "creating" };
921 log::info!("{}: {} file at {:?}", verb, client.username, path);
922
923 client
924 .fs()
925 .save(&path, &content.as_str().into(), text::LineEnding::Unix)
926 .await
927 .unwrap();
928 }
929 }
930
931 ClientOperation::GitOperation { operation } => match operation {
932 GitOperation::WriteGitIndex {
933 repo_path,
934 contents,
935 } => {
936 if !client.fs().directories(false).contains(&repo_path) {
937 return Err(TestError::Inapplicable);
938 }
939
940 for (path, _) in contents.iter() {
941 if !client.fs().files().contains(&repo_path.join(path)) {
942 return Err(TestError::Inapplicable);
943 }
944 }
945
946 log::info!(
947 "{}: writing git index for repo {:?}: {:?}",
948 client.username,
949 repo_path,
950 contents
951 );
952
953 let dot_git_dir = repo_path.join(".git");
954 let contents = contents
955 .into_iter()
956 .map(|(path, contents)| (path.into(), contents))
957 .collect::<Vec<_>>();
958 if client.fs().metadata(&dot_git_dir).await?.is_none() {
959 client.fs().create_dir(&dot_git_dir).await?;
960 }
961 client.fs().set_index_for_repo(&dot_git_dir, &contents);
962 }
963 GitOperation::WriteGitBranch {
964 repo_path,
965 new_branch,
966 } => {
967 if !client.fs().directories(false).contains(&repo_path) {
968 return Err(TestError::Inapplicable);
969 }
970
971 log::info!(
972 "{}: writing git branch for repo {:?}: {:?}",
973 client.username,
974 repo_path,
975 new_branch
976 );
977
978 let dot_git_dir = repo_path.join(".git");
979 if client.fs().metadata(&dot_git_dir).await?.is_none() {
980 client.fs().create_dir(&dot_git_dir).await?;
981 }
982 client
983 .fs()
984 .set_branch_name(&dot_git_dir, new_branch.clone());
985 }
986 GitOperation::WriteGitStatuses {
987 repo_path,
988 statuses,
989 } => {
990 if !client.fs().directories(false).contains(&repo_path) {
991 return Err(TestError::Inapplicable);
992 }
993 for (path, _) in statuses.iter() {
994 if !client.fs().files().contains(&repo_path.join(path)) {
995 return Err(TestError::Inapplicable);
996 }
997 }
998
999 log::info!(
1000 "{}: writing git statuses for repo {:?}: {:?}",
1001 client.username,
1002 repo_path,
1003 statuses
1004 );
1005
1006 let dot_git_dir = repo_path.join(".git");
1007
1008 let statuses = statuses
1009 .iter()
1010 .map(|(path, val)| (path.as_path(), *val))
1011 .collect::<Vec<_>>();
1012
1013 if client.fs().metadata(&dot_git_dir).await?.is_none() {
1014 client.fs().create_dir(&dot_git_dir).await?;
1015 }
1016
1017 client
1018 .fs()
1019 .set_status_for_repo(&dot_git_dir, statuses.as_slice());
1020 }
1021 },
1022 }
1023 Ok(())
1024 }
1025
1026 async fn on_client_added(client: &Rc<TestClient>, _: &mut TestAppContext) {
1027 client.language_registry().add(Arc::new(Language::new(
1028 LanguageConfig {
1029 name: "Rust".into(),
1030 matcher: LanguageMatcher {
1031 path_suffixes: vec!["rs".to_string()],
1032 ..Default::default()
1033 },
1034 ..Default::default()
1035 },
1036 None,
1037 )));
1038 client.language_registry().register_fake_lsp(
1039 "Rust",
1040 FakeLspAdapter {
1041 name: "the-fake-language-server",
1042 capabilities: lsp::LanguageServer::full_capabilities(),
1043 initializer: Some(Box::new({
1044 let fs = client.app_state.fs.clone();
1045 move |fake_server: &mut FakeLanguageServer| {
1046 fake_server.handle_request::<lsp::request::Completion, _, _>(
1047 |_, _| async move {
1048 Ok(Some(lsp::CompletionResponse::Array(vec![
1049 lsp::CompletionItem {
1050 text_edit: Some(lsp::CompletionTextEdit::Edit(
1051 lsp::TextEdit {
1052 range: lsp::Range::new(
1053 lsp::Position::new(0, 0),
1054 lsp::Position::new(0, 0),
1055 ),
1056 new_text: "the-new-text".to_string(),
1057 },
1058 )),
1059 ..Default::default()
1060 },
1061 ])))
1062 },
1063 );
1064
1065 fake_server.handle_request::<lsp::request::CodeActionRequest, _, _>(
1066 |_, _| async move {
1067 Ok(Some(vec![lsp::CodeActionOrCommand::CodeAction(
1068 lsp::CodeAction {
1069 title: "the-code-action".to_string(),
1070 ..Default::default()
1071 },
1072 )]))
1073 },
1074 );
1075
1076 fake_server.handle_request::<lsp::request::PrepareRenameRequest, _, _>(
1077 |params, _| async move {
1078 Ok(Some(lsp::PrepareRenameResponse::Range(lsp::Range::new(
1079 params.position,
1080 params.position,
1081 ))))
1082 },
1083 );
1084
1085 fake_server.handle_request::<lsp::request::GotoDefinition, _, _>({
1086 let fs = fs.clone();
1087 move |_, cx| {
1088 let background = cx.background_executor();
1089 let mut rng = background.rng();
1090 let count = rng.gen_range::<usize, _>(1..3);
1091 let files = fs.as_fake().files();
1092 let files = (0..count)
1093 .map(|_| files.choose(&mut rng).unwrap().clone())
1094 .collect::<Vec<_>>();
1095 async move {
1096 log::info!("LSP: Returning definitions in files {:?}", &files);
1097 Ok(Some(lsp::GotoDefinitionResponse::Array(
1098 files
1099 .into_iter()
1100 .map(|file| lsp::Location {
1101 uri: lsp::Url::from_file_path(file).unwrap(),
1102 range: Default::default(),
1103 })
1104 .collect(),
1105 )))
1106 }
1107 }
1108 });
1109
1110 fake_server.handle_request::<lsp::request::DocumentHighlightRequest, _, _>(
1111 move |_, cx| {
1112 let mut highlights = Vec::new();
1113 let background = cx.background_executor();
1114 let mut rng = background.rng();
1115
1116 let highlight_count = rng.gen_range(1..=5);
1117 for _ in 0..highlight_count {
1118 let start_row = rng.gen_range(0..100);
1119 let start_column = rng.gen_range(0..100);
1120 let end_row = rng.gen_range(0..100);
1121 let end_column = rng.gen_range(0..100);
1122 let start = PointUtf16::new(start_row, start_column);
1123 let end = PointUtf16::new(end_row, end_column);
1124 let range = if start > end { end..start } else { start..end };
1125 highlights.push(lsp::DocumentHighlight {
1126 range: range_to_lsp(range.clone()).unwrap(),
1127 kind: Some(lsp::DocumentHighlightKind::READ),
1128 });
1129 }
1130 highlights.sort_unstable_by_key(|highlight| {
1131 (highlight.range.start, highlight.range.end)
1132 });
1133 async move { Ok(Some(highlights)) }
1134 },
1135 );
1136 }
1137 })),
1138 ..Default::default()
1139 },
1140 );
1141 }
1142
1143 async fn on_quiesce(_: &mut TestServer, clients: &mut [(Rc<TestClient>, TestAppContext)]) {
1144 for (client, client_cx) in clients.iter() {
1145 for guest_project in client.dev_server_projects().iter() {
1146 guest_project.read_with(client_cx, |guest_project, cx| {
1147 let host_project = clients.iter().find_map(|(client, cx)| {
1148 let project = client
1149 .local_projects()
1150 .iter()
1151 .find(|host_project| {
1152 host_project.read_with(cx, |host_project, _| {
1153 host_project.remote_id() == guest_project.remote_id()
1154 })
1155 })?
1156 .clone();
1157 Some((project, cx))
1158 });
1159
1160 if !guest_project.is_disconnected(cx) {
1161 if let Some((host_project, host_cx)) = host_project {
1162 let host_worktree_snapshots =
1163 host_project.read_with(host_cx, |host_project, cx| {
1164 host_project
1165 .worktrees(cx)
1166 .map(|worktree| {
1167 let worktree = worktree.read(cx);
1168 (worktree.id(), worktree.snapshot())
1169 })
1170 .collect::<BTreeMap<_, _>>()
1171 });
1172 let guest_worktree_snapshots = guest_project
1173 .worktrees(cx)
1174 .map(|worktree| {
1175 let worktree = worktree.read(cx);
1176 (worktree.id(), worktree.snapshot())
1177 })
1178 .collect::<BTreeMap<_, _>>();
1179
1180 assert_eq!(
1181 guest_worktree_snapshots.values().map(|w| w.abs_path()).collect::<Vec<_>>(),
1182 host_worktree_snapshots.values().map(|w| w.abs_path()).collect::<Vec<_>>(),
1183 "{} has different worktrees than the host for project {:?}",
1184 client.username, guest_project.remote_id(),
1185 );
1186
1187 for (id, host_snapshot) in &host_worktree_snapshots {
1188 let guest_snapshot = &guest_worktree_snapshots[id];
1189 assert_eq!(
1190 guest_snapshot.root_name(),
1191 host_snapshot.root_name(),
1192 "{} has different root name than the host for worktree {}, project {:?}",
1193 client.username,
1194 id,
1195 guest_project.remote_id(),
1196 );
1197 assert_eq!(
1198 guest_snapshot.abs_path(),
1199 host_snapshot.abs_path(),
1200 "{} has different abs path than the host for worktree {}, project: {:?}",
1201 client.username,
1202 id,
1203 guest_project.remote_id(),
1204 );
1205 assert_eq!(
1206 guest_snapshot.entries(false, 0).collect::<Vec<_>>(),
1207 host_snapshot.entries(false, 0).collect::<Vec<_>>(),
1208 "{} has different snapshot than the host for worktree {:?} ({:?}) and project {:?}",
1209 client.username,
1210 host_snapshot.abs_path(),
1211 id,
1212 guest_project.remote_id(),
1213 );
1214 assert_eq!(guest_snapshot.repositories().iter().collect::<Vec<_>>(), host_snapshot.repositories().iter().collect::<Vec<_>>(),
1215 "{} has different repositories than the host for worktree {:?} and project {:?}",
1216 client.username,
1217 host_snapshot.abs_path(),
1218 guest_project.remote_id(),
1219 );
1220 assert_eq!(guest_snapshot.scan_id(), host_snapshot.scan_id(),
1221 "{} has different scan id than the host for worktree {:?} and project {:?}",
1222 client.username,
1223 host_snapshot.abs_path(),
1224 guest_project.remote_id(),
1225 );
1226 }
1227 }
1228 }
1229
1230 for buffer in guest_project.opened_buffers(cx) {
1231 let buffer = buffer.read(cx);
1232 assert_eq!(
1233 buffer.deferred_ops_len(),
1234 0,
1235 "{} has deferred operations for buffer {:?} in project {:?}",
1236 client.username,
1237 buffer.file().unwrap().full_path(cx),
1238 guest_project.remote_id(),
1239 );
1240 }
1241 });
1242 }
1243
1244 let buffers = client.buffers().clone();
1245 for (guest_project, guest_buffers) in &buffers {
1246 let project_id = if guest_project.read_with(client_cx, |project, cx| {
1247 project.is_local() || project.is_disconnected(cx)
1248 }) {
1249 continue;
1250 } else {
1251 guest_project
1252 .read_with(client_cx, |project, _| project.remote_id())
1253 .unwrap()
1254 };
1255 let guest_user_id = client.user_id().unwrap();
1256
1257 let host_project = clients.iter().find_map(|(client, cx)| {
1258 let project = client
1259 .local_projects()
1260 .iter()
1261 .find(|host_project| {
1262 host_project.read_with(cx, |host_project, _| {
1263 host_project.remote_id() == Some(project_id)
1264 })
1265 })?
1266 .clone();
1267 Some((client.user_id().unwrap(), project, cx))
1268 });
1269
1270 let (host_user_id, host_project, host_cx) =
1271 if let Some((host_user_id, host_project, host_cx)) = host_project {
1272 (host_user_id, host_project, host_cx)
1273 } else {
1274 continue;
1275 };
1276
1277 for guest_buffer in guest_buffers {
1278 let buffer_id =
1279 guest_buffer.read_with(client_cx, |buffer, _| buffer.remote_id());
1280 let host_buffer = host_project.read_with(host_cx, |project, cx| {
1281 project.buffer_for_id(buffer_id, cx).unwrap_or_else(|| {
1282 panic!(
1283 "host does not have buffer for guest:{}, peer:{:?}, id:{}",
1284 client.username,
1285 client.peer_id(),
1286 buffer_id
1287 )
1288 })
1289 });
1290 let path = host_buffer
1291 .read_with(host_cx, |buffer, cx| buffer.file().unwrap().full_path(cx));
1292
1293 assert_eq!(
1294 guest_buffer.read_with(client_cx, |buffer, _| buffer.deferred_ops_len()),
1295 0,
1296 "{}, buffer {}, path {:?} has deferred operations",
1297 client.username,
1298 buffer_id,
1299 path,
1300 );
1301 assert_eq!(
1302 guest_buffer.read_with(client_cx, |buffer, _| buffer.text()),
1303 host_buffer.read_with(host_cx, |buffer, _| buffer.text()),
1304 "{}, buffer {}, path {:?}, differs from the host's buffer",
1305 client.username,
1306 buffer_id,
1307 path
1308 );
1309
1310 let host_file = host_buffer.read_with(host_cx, |b, _| b.file().cloned());
1311 let guest_file = guest_buffer.read_with(client_cx, |b, _| b.file().cloned());
1312 match (host_file, guest_file) {
1313 (Some(host_file), Some(guest_file)) => {
1314 assert_eq!(guest_file.path(), host_file.path());
1315 assert_eq!(guest_file.disk_state(), host_file.disk_state(),
1316 "guest {} disk_state does not match host {} for path {:?} in project {}",
1317 guest_user_id,
1318 host_user_id,
1319 guest_file.path(),
1320 project_id,
1321 );
1322 }
1323 (None, None) => {}
1324 (None, _) => panic!("host's file is None, guest's isn't"),
1325 (_, None) => panic!("guest's file is None, hosts's isn't"),
1326 }
1327
1328 let host_diff_base = host_project.read_with(host_cx, |project, cx| {
1329 project
1330 .git_store()
1331 .read(cx)
1332 .get_unstaged_diff(host_buffer.read(cx).remote_id(), cx)
1333 .unwrap()
1334 .read(cx)
1335 .base_text_string()
1336 });
1337 let guest_diff_base = guest_project.read_with(client_cx, |project, cx| {
1338 project
1339 .git_store()
1340 .read(cx)
1341 .get_unstaged_diff(guest_buffer.read(cx).remote_id(), cx)
1342 .unwrap()
1343 .read(cx)
1344 .base_text_string()
1345 });
1346 assert_eq!(
1347 guest_diff_base, host_diff_base,
1348 "guest {} diff base does not match host's for path {path:?} in project {project_id}",
1349 client.username
1350 );
1351
1352 let host_saved_version =
1353 host_buffer.read_with(host_cx, |b, _| b.saved_version().clone());
1354 let guest_saved_version =
1355 guest_buffer.read_with(client_cx, |b, _| b.saved_version().clone());
1356 assert_eq!(
1357 guest_saved_version, host_saved_version,
1358 "guest {} saved version does not match host's for path {path:?} in project {project_id}",
1359 client.username
1360 );
1361
1362 let host_is_dirty = host_buffer.read_with(host_cx, |b, _| b.is_dirty());
1363 let guest_is_dirty = guest_buffer.read_with(client_cx, |b, _| b.is_dirty());
1364 assert_eq!(
1365 guest_is_dirty, host_is_dirty,
1366 "guest {} dirty state does not match host's for path {path:?} in project {project_id}",
1367 client.username
1368 );
1369
1370 let host_saved_mtime = host_buffer.read_with(host_cx, |b, _| b.saved_mtime());
1371 let guest_saved_mtime =
1372 guest_buffer.read_with(client_cx, |b, _| b.saved_mtime());
1373 assert_eq!(
1374 guest_saved_mtime, host_saved_mtime,
1375 "guest {} saved mtime does not match host's for path {path:?} in project {project_id}",
1376 client.username
1377 );
1378
1379 let host_is_dirty = host_buffer.read_with(host_cx, |b, _| b.is_dirty());
1380 let guest_is_dirty = guest_buffer.read_with(client_cx, |b, _| b.is_dirty());
1381 assert_eq!(guest_is_dirty, host_is_dirty,
1382 "guest {} dirty status does not match host's for path {path:?} in project {project_id}",
1383 client.username
1384 );
1385
1386 let host_has_conflict = host_buffer.read_with(host_cx, |b, _| b.has_conflict());
1387 let guest_has_conflict =
1388 guest_buffer.read_with(client_cx, |b, _| b.has_conflict());
1389 assert_eq!(guest_has_conflict, host_has_conflict,
1390 "guest {} conflict status does not match host's for path {path:?} in project {project_id}",
1391 client.username
1392 );
1393 }
1394 }
1395 }
1396 }
1397}
1398
1399fn generate_git_operation(rng: &mut StdRng, client: &TestClient) -> GitOperation {
1400 fn generate_file_paths(
1401 repo_path: &Path,
1402 rng: &mut StdRng,
1403 client: &TestClient,
1404 ) -> Vec<PathBuf> {
1405 let mut paths = client
1406 .fs()
1407 .files()
1408 .into_iter()
1409 .filter(|path| path.starts_with(repo_path))
1410 .collect::<Vec<_>>();
1411
1412 let count = rng.gen_range(0..=paths.len());
1413 paths.shuffle(rng);
1414 paths.truncate(count);
1415
1416 paths
1417 .iter()
1418 .map(|path| path.strip_prefix(repo_path).unwrap().to_path_buf())
1419 .collect::<Vec<_>>()
1420 }
1421
1422 let repo_path = client.fs().directories(false).choose(rng).unwrap().clone();
1423
1424 match rng.gen_range(0..100_u32) {
1425 0..=25 => {
1426 let file_paths = generate_file_paths(&repo_path, rng, client);
1427
1428 let contents = file_paths
1429 .into_iter()
1430 .map(|path| (path, Alphanumeric.sample_string(rng, 16)))
1431 .collect();
1432
1433 GitOperation::WriteGitIndex {
1434 repo_path,
1435 contents,
1436 }
1437 }
1438 26..=63 => {
1439 let new_branch = (rng.gen_range(0..10) > 3).then(|| Alphanumeric.sample_string(rng, 8));
1440
1441 GitOperation::WriteGitBranch {
1442 repo_path,
1443 new_branch,
1444 }
1445 }
1446 64..=100 => {
1447 let file_paths = generate_file_paths(&repo_path, rng, client);
1448 let statuses = file_paths
1449 .into_iter()
1450 .map(|path| (path, gen_status(rng)))
1451 .collect::<Vec<_>>();
1452 GitOperation::WriteGitStatuses {
1453 repo_path,
1454 statuses,
1455 }
1456 }
1457 _ => unreachable!(),
1458 }
1459}
1460
1461fn buffer_for_full_path(
1462 client: &TestClient,
1463 project: &Entity<Project>,
1464 full_path: &PathBuf,
1465 cx: &TestAppContext,
1466) -> Option<Entity<language::Buffer>> {
1467 client
1468 .buffers_for_project(project)
1469 .iter()
1470 .find(|buffer| {
1471 buffer.read_with(cx, |buffer, cx| {
1472 buffer.file().unwrap().full_path(cx) == *full_path
1473 })
1474 })
1475 .cloned()
1476}
1477
1478fn project_for_root_name(
1479 client: &TestClient,
1480 root_name: &str,
1481 cx: &TestAppContext,
1482) -> Option<Entity<Project>> {
1483 if let Some(ix) = project_ix_for_root_name(client.local_projects().deref(), root_name, cx) {
1484 return Some(client.local_projects()[ix].clone());
1485 }
1486 if let Some(ix) = project_ix_for_root_name(client.dev_server_projects().deref(), root_name, cx)
1487 {
1488 return Some(client.dev_server_projects()[ix].clone());
1489 }
1490 None
1491}
1492
1493fn project_ix_for_root_name(
1494 projects: &[Entity<Project>],
1495 root_name: &str,
1496 cx: &TestAppContext,
1497) -> Option<usize> {
1498 projects.iter().position(|project| {
1499 project.read_with(cx, |project, cx| {
1500 let worktree = project.visible_worktrees(cx).next().unwrap();
1501 worktree.read(cx).root_name() == root_name
1502 })
1503 })
1504}
1505
1506fn root_name_for_project(project: &Entity<Project>, cx: &TestAppContext) -> String {
1507 project.read_with(cx, |project, cx| {
1508 project
1509 .visible_worktrees(cx)
1510 .next()
1511 .unwrap()
1512 .read(cx)
1513 .root_name()
1514 .to_string()
1515 })
1516}
1517
1518fn project_path_for_full_path(
1519 project: &Entity<Project>,
1520 full_path: &Path,
1521 cx: &TestAppContext,
1522) -> Option<ProjectPath> {
1523 let mut components = full_path.components();
1524 let root_name = components.next().unwrap().as_os_str().to_str().unwrap();
1525 let path = components.as_path().into();
1526 let worktree_id = project.read_with(cx, |project, cx| {
1527 project.worktrees(cx).find_map(|worktree| {
1528 let worktree = worktree.read(cx);
1529 if worktree.root_name() == root_name {
1530 Some(worktree.id())
1531 } else {
1532 None
1533 }
1534 })
1535 })?;
1536 Some(ProjectPath { worktree_id, path })
1537}
1538
1539async fn ensure_project_shared(
1540 project: &Entity<Project>,
1541 client: &TestClient,
1542 cx: &mut TestAppContext,
1543) {
1544 let first_root_name = root_name_for_project(project, cx);
1545 let active_call = cx.read(ActiveCall::global);
1546 if active_call.read_with(cx, |call, _| call.room().is_some())
1547 && project.read_with(cx, |project, _| project.is_local() && !project.is_shared())
1548 {
1549 match active_call
1550 .update(cx, |call, cx| call.share_project(project.clone(), cx))
1551 .await
1552 {
1553 Ok(project_id) => {
1554 log::info!(
1555 "{}: shared project {} with id {}",
1556 client.username,
1557 first_root_name,
1558 project_id
1559 );
1560 }
1561 Err(error) => {
1562 log::error!(
1563 "{}: error sharing project {}: {:?}",
1564 client.username,
1565 first_root_name,
1566 error
1567 );
1568 }
1569 }
1570 }
1571}
1572
1573fn choose_random_project(client: &TestClient, rng: &mut StdRng) -> Option<Entity<Project>> {
1574 client
1575 .local_projects()
1576 .deref()
1577 .iter()
1578 .chain(client.dev_server_projects().iter())
1579 .choose(rng)
1580 .cloned()
1581}
1582
1583fn gen_file_name(rng: &mut StdRng) -> String {
1584 let mut name = String::new();
1585 for _ in 0..10 {
1586 let letter = rng.gen_range('a'..='z');
1587 name.push(letter);
1588 }
1589 name
1590}
1591
1592fn gen_status(rng: &mut StdRng) -> FileStatus {
1593 fn gen_tracked_status(rng: &mut StdRng) -> TrackedStatus {
1594 match rng.gen_range(0..3) {
1595 0 => TrackedStatus {
1596 index_status: StatusCode::Unmodified,
1597 worktree_status: StatusCode::Unmodified,
1598 },
1599 1 => TrackedStatus {
1600 index_status: StatusCode::Modified,
1601 worktree_status: StatusCode::Modified,
1602 },
1603 2 => TrackedStatus {
1604 index_status: StatusCode::Added,
1605 worktree_status: StatusCode::Modified,
1606 },
1607 3 => TrackedStatus {
1608 index_status: StatusCode::Added,
1609 worktree_status: StatusCode::Unmodified,
1610 },
1611 _ => unreachable!(),
1612 }
1613 }
1614
1615 fn gen_unmerged_status_code(rng: &mut StdRng) -> UnmergedStatusCode {
1616 match rng.gen_range(0..3) {
1617 0 => UnmergedStatusCode::Updated,
1618 1 => UnmergedStatusCode::Added,
1619 2 => UnmergedStatusCode::Deleted,
1620 _ => unreachable!(),
1621 }
1622 }
1623
1624 match rng.gen_range(0..2) {
1625 0 => FileStatus::Unmerged(UnmergedStatus {
1626 first_head: gen_unmerged_status_code(rng),
1627 second_head: gen_unmerged_status_code(rng),
1628 }),
1629 1 => FileStatus::Tracked(gen_tracked_status(rng)),
1630 _ => unreachable!(),
1631 }
1632}