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.set_request_handler::<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.set_request_handler::<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
1077 .set_request_handler::<lsp::request::PrepareRenameRequest, _, _>(
1078 |params, _| async move {
1079 Ok(Some(lsp::PrepareRenameResponse::Range(lsp::Range::new(
1080 params.position,
1081 params.position,
1082 ))))
1083 },
1084 );
1085
1086 fake_server.set_request_handler::<lsp::request::GotoDefinition, _, _>({
1087 let fs = fs.clone();
1088 move |_, cx| {
1089 let background = cx.background_executor();
1090 let mut rng = background.rng();
1091 let count = rng.gen_range::<usize, _>(1..3);
1092 let files = fs.as_fake().files();
1093 let files = (0..count)
1094 .map(|_| files.choose(&mut rng).unwrap().clone())
1095 .collect::<Vec<_>>();
1096 async move {
1097 log::info!("LSP: Returning definitions in files {:?}", &files);
1098 Ok(Some(lsp::GotoDefinitionResponse::Array(
1099 files
1100 .into_iter()
1101 .map(|file| lsp::Location {
1102 uri: lsp::Url::from_file_path(file).unwrap(),
1103 range: Default::default(),
1104 })
1105 .collect(),
1106 )))
1107 }
1108 }
1109 });
1110
1111 fake_server
1112 .set_request_handler::<lsp::request::DocumentHighlightRequest, _, _>(
1113 move |_, cx| {
1114 let mut highlights = Vec::new();
1115 let background = cx.background_executor();
1116 let mut rng = background.rng();
1117
1118 let highlight_count = rng.gen_range(1..=5);
1119 for _ in 0..highlight_count {
1120 let start_row = rng.gen_range(0..100);
1121 let start_column = rng.gen_range(0..100);
1122 let end_row = rng.gen_range(0..100);
1123 let end_column = rng.gen_range(0..100);
1124 let start = PointUtf16::new(start_row, start_column);
1125 let end = PointUtf16::new(end_row, end_column);
1126 let range =
1127 if start > end { end..start } else { start..end };
1128 highlights.push(lsp::DocumentHighlight {
1129 range: range_to_lsp(range.clone()).unwrap(),
1130 kind: Some(lsp::DocumentHighlightKind::READ),
1131 });
1132 }
1133 highlights.sort_unstable_by_key(|highlight| {
1134 (highlight.range.start, highlight.range.end)
1135 });
1136 async move { Ok(Some(highlights)) }
1137 },
1138 );
1139 }
1140 })),
1141 ..Default::default()
1142 },
1143 );
1144 }
1145
1146 async fn on_quiesce(_: &mut TestServer, clients: &mut [(Rc<TestClient>, TestAppContext)]) {
1147 for (client, client_cx) in clients.iter() {
1148 for guest_project in client.dev_server_projects().iter() {
1149 guest_project.read_with(client_cx, |guest_project, cx| {
1150 let host_project = clients.iter().find_map(|(client, cx)| {
1151 let project = client
1152 .local_projects()
1153 .iter()
1154 .find(|host_project| {
1155 host_project.read_with(cx, |host_project, _| {
1156 host_project.remote_id() == guest_project.remote_id()
1157 })
1158 })?
1159 .clone();
1160 Some((project, cx))
1161 });
1162
1163 if !guest_project.is_disconnected(cx) {
1164 if let Some((host_project, host_cx)) = host_project {
1165 let host_worktree_snapshots =
1166 host_project.read_with(host_cx, |host_project, cx| {
1167 host_project
1168 .worktrees(cx)
1169 .map(|worktree| {
1170 let worktree = worktree.read(cx);
1171 (worktree.id(), worktree.snapshot())
1172 })
1173 .collect::<BTreeMap<_, _>>()
1174 });
1175 let guest_worktree_snapshots = guest_project
1176 .worktrees(cx)
1177 .map(|worktree| {
1178 let worktree = worktree.read(cx);
1179 (worktree.id(), worktree.snapshot())
1180 })
1181 .collect::<BTreeMap<_, _>>();
1182
1183 assert_eq!(
1184 guest_worktree_snapshots.values().map(|w| w.abs_path()).collect::<Vec<_>>(),
1185 host_worktree_snapshots.values().map(|w| w.abs_path()).collect::<Vec<_>>(),
1186 "{} has different worktrees than the host for project {:?}",
1187 client.username, guest_project.remote_id(),
1188 );
1189
1190 for (id, host_snapshot) in &host_worktree_snapshots {
1191 let guest_snapshot = &guest_worktree_snapshots[id];
1192 assert_eq!(
1193 guest_snapshot.root_name(),
1194 host_snapshot.root_name(),
1195 "{} has different root name than the host for worktree {}, project {:?}",
1196 client.username,
1197 id,
1198 guest_project.remote_id(),
1199 );
1200 assert_eq!(
1201 guest_snapshot.abs_path(),
1202 host_snapshot.abs_path(),
1203 "{} has different abs path than the host for worktree {}, project: {:?}",
1204 client.username,
1205 id,
1206 guest_project.remote_id(),
1207 );
1208 assert_eq!(
1209 guest_snapshot.entries(false, 0).collect::<Vec<_>>(),
1210 host_snapshot.entries(false, 0).collect::<Vec<_>>(),
1211 "{} has different snapshot than the host for worktree {:?} ({:?}) and project {:?}",
1212 client.username,
1213 host_snapshot.abs_path(),
1214 id,
1215 guest_project.remote_id(),
1216 );
1217 assert_eq!(guest_snapshot.repositories().iter().collect::<Vec<_>>(), host_snapshot.repositories().iter().collect::<Vec<_>>(),
1218 "{} has different repositories than the host for worktree {:?} and project {:?}",
1219 client.username,
1220 host_snapshot.abs_path(),
1221 guest_project.remote_id(),
1222 );
1223 assert_eq!(guest_snapshot.scan_id(), host_snapshot.scan_id(),
1224 "{} has different scan id than the host for worktree {:?} and project {:?}",
1225 client.username,
1226 host_snapshot.abs_path(),
1227 guest_project.remote_id(),
1228 );
1229 }
1230 }
1231 }
1232
1233 for buffer in guest_project.opened_buffers(cx) {
1234 let buffer = buffer.read(cx);
1235 assert_eq!(
1236 buffer.deferred_ops_len(),
1237 0,
1238 "{} has deferred operations for buffer {:?} in project {:?}",
1239 client.username,
1240 buffer.file().unwrap().full_path(cx),
1241 guest_project.remote_id(),
1242 );
1243 }
1244 });
1245 }
1246
1247 let buffers = client.buffers().clone();
1248 for (guest_project, guest_buffers) in &buffers {
1249 let project_id = if guest_project.read_with(client_cx, |project, cx| {
1250 project.is_local() || project.is_disconnected(cx)
1251 }) {
1252 continue;
1253 } else {
1254 guest_project
1255 .read_with(client_cx, |project, _| project.remote_id())
1256 .unwrap()
1257 };
1258 let guest_user_id = client.user_id().unwrap();
1259
1260 let host_project = clients.iter().find_map(|(client, cx)| {
1261 let project = client
1262 .local_projects()
1263 .iter()
1264 .find(|host_project| {
1265 host_project.read_with(cx, |host_project, _| {
1266 host_project.remote_id() == Some(project_id)
1267 })
1268 })?
1269 .clone();
1270 Some((client.user_id().unwrap(), project, cx))
1271 });
1272
1273 let (host_user_id, host_project, host_cx) =
1274 if let Some((host_user_id, host_project, host_cx)) = host_project {
1275 (host_user_id, host_project, host_cx)
1276 } else {
1277 continue;
1278 };
1279
1280 for guest_buffer in guest_buffers {
1281 let buffer_id =
1282 guest_buffer.read_with(client_cx, |buffer, _| buffer.remote_id());
1283 let host_buffer = host_project.read_with(host_cx, |project, cx| {
1284 project.buffer_for_id(buffer_id, cx).unwrap_or_else(|| {
1285 panic!(
1286 "host does not have buffer for guest:{}, peer:{:?}, id:{}",
1287 client.username,
1288 client.peer_id(),
1289 buffer_id
1290 )
1291 })
1292 });
1293 let path = host_buffer
1294 .read_with(host_cx, |buffer, cx| buffer.file().unwrap().full_path(cx));
1295
1296 assert_eq!(
1297 guest_buffer.read_with(client_cx, |buffer, _| buffer.deferred_ops_len()),
1298 0,
1299 "{}, buffer {}, path {:?} has deferred operations",
1300 client.username,
1301 buffer_id,
1302 path,
1303 );
1304 assert_eq!(
1305 guest_buffer.read_with(client_cx, |buffer, _| buffer.text()),
1306 host_buffer.read_with(host_cx, |buffer, _| buffer.text()),
1307 "{}, buffer {}, path {:?}, differs from the host's buffer",
1308 client.username,
1309 buffer_id,
1310 path
1311 );
1312
1313 let host_file = host_buffer.read_with(host_cx, |b, _| b.file().cloned());
1314 let guest_file = guest_buffer.read_with(client_cx, |b, _| b.file().cloned());
1315 match (host_file, guest_file) {
1316 (Some(host_file), Some(guest_file)) => {
1317 assert_eq!(guest_file.path(), host_file.path());
1318 assert_eq!(guest_file.disk_state(), host_file.disk_state(),
1319 "guest {} disk_state does not match host {} for path {:?} in project {}",
1320 guest_user_id,
1321 host_user_id,
1322 guest_file.path(),
1323 project_id,
1324 );
1325 }
1326 (None, None) => {}
1327 (None, _) => panic!("host's file is None, guest's isn't"),
1328 (_, None) => panic!("guest's file is None, hosts's isn't"),
1329 }
1330
1331 let host_diff_base = host_project.read_with(host_cx, |project, cx| {
1332 project
1333 .git_store()
1334 .read(cx)
1335 .get_unstaged_diff(host_buffer.read(cx).remote_id(), cx)
1336 .unwrap()
1337 .read(cx)
1338 .base_text_string()
1339 });
1340 let guest_diff_base = guest_project.read_with(client_cx, |project, cx| {
1341 project
1342 .git_store()
1343 .read(cx)
1344 .get_unstaged_diff(guest_buffer.read(cx).remote_id(), cx)
1345 .unwrap()
1346 .read(cx)
1347 .base_text_string()
1348 });
1349 assert_eq!(
1350 guest_diff_base, host_diff_base,
1351 "guest {} diff base does not match host's for path {path:?} in project {project_id}",
1352 client.username
1353 );
1354
1355 let host_saved_version =
1356 host_buffer.read_with(host_cx, |b, _| b.saved_version().clone());
1357 let guest_saved_version =
1358 guest_buffer.read_with(client_cx, |b, _| b.saved_version().clone());
1359 assert_eq!(
1360 guest_saved_version, host_saved_version,
1361 "guest {} saved version does not match host's for path {path:?} in project {project_id}",
1362 client.username
1363 );
1364
1365 let host_is_dirty = host_buffer.read_with(host_cx, |b, _| b.is_dirty());
1366 let guest_is_dirty = guest_buffer.read_with(client_cx, |b, _| b.is_dirty());
1367 assert_eq!(
1368 guest_is_dirty, host_is_dirty,
1369 "guest {} dirty state does not match host's for path {path:?} in project {project_id}",
1370 client.username
1371 );
1372
1373 let host_saved_mtime = host_buffer.read_with(host_cx, |b, _| b.saved_mtime());
1374 let guest_saved_mtime =
1375 guest_buffer.read_with(client_cx, |b, _| b.saved_mtime());
1376 assert_eq!(
1377 guest_saved_mtime, host_saved_mtime,
1378 "guest {} saved mtime does not match host's for path {path:?} in project {project_id}",
1379 client.username
1380 );
1381
1382 let host_is_dirty = host_buffer.read_with(host_cx, |b, _| b.is_dirty());
1383 let guest_is_dirty = guest_buffer.read_with(client_cx, |b, _| b.is_dirty());
1384 assert_eq!(guest_is_dirty, host_is_dirty,
1385 "guest {} dirty status does not match host's for path {path:?} in project {project_id}",
1386 client.username
1387 );
1388
1389 let host_has_conflict = host_buffer.read_with(host_cx, |b, _| b.has_conflict());
1390 let guest_has_conflict =
1391 guest_buffer.read_with(client_cx, |b, _| b.has_conflict());
1392 assert_eq!(guest_has_conflict, host_has_conflict,
1393 "guest {} conflict status does not match host's for path {path:?} in project {project_id}",
1394 client.username
1395 );
1396 }
1397 }
1398 }
1399 }
1400}
1401
1402fn generate_git_operation(rng: &mut StdRng, client: &TestClient) -> GitOperation {
1403 fn generate_file_paths(
1404 repo_path: &Path,
1405 rng: &mut StdRng,
1406 client: &TestClient,
1407 ) -> Vec<PathBuf> {
1408 let mut paths = client
1409 .fs()
1410 .files()
1411 .into_iter()
1412 .filter(|path| path.starts_with(repo_path))
1413 .collect::<Vec<_>>();
1414
1415 let count = rng.gen_range(0..=paths.len());
1416 paths.shuffle(rng);
1417 paths.truncate(count);
1418
1419 paths
1420 .iter()
1421 .map(|path| path.strip_prefix(repo_path).unwrap().to_path_buf())
1422 .collect::<Vec<_>>()
1423 }
1424
1425 let repo_path = client.fs().directories(false).choose(rng).unwrap().clone();
1426
1427 match rng.gen_range(0..100_u32) {
1428 0..=25 => {
1429 let file_paths = generate_file_paths(&repo_path, rng, client);
1430
1431 let contents = file_paths
1432 .into_iter()
1433 .map(|path| (path, Alphanumeric.sample_string(rng, 16)))
1434 .collect();
1435
1436 GitOperation::WriteGitIndex {
1437 repo_path,
1438 contents,
1439 }
1440 }
1441 26..=63 => {
1442 let new_branch = (rng.gen_range(0..10) > 3).then(|| Alphanumeric.sample_string(rng, 8));
1443
1444 GitOperation::WriteGitBranch {
1445 repo_path,
1446 new_branch,
1447 }
1448 }
1449 64..=100 => {
1450 let file_paths = generate_file_paths(&repo_path, rng, client);
1451 let statuses = file_paths
1452 .into_iter()
1453 .map(|path| (path, gen_status(rng)))
1454 .collect::<Vec<_>>();
1455 GitOperation::WriteGitStatuses {
1456 repo_path,
1457 statuses,
1458 }
1459 }
1460 _ => unreachable!(),
1461 }
1462}
1463
1464fn buffer_for_full_path(
1465 client: &TestClient,
1466 project: &Entity<Project>,
1467 full_path: &PathBuf,
1468 cx: &TestAppContext,
1469) -> Option<Entity<language::Buffer>> {
1470 client
1471 .buffers_for_project(project)
1472 .iter()
1473 .find(|buffer| {
1474 buffer.read_with(cx, |buffer, cx| {
1475 buffer.file().unwrap().full_path(cx) == *full_path
1476 })
1477 })
1478 .cloned()
1479}
1480
1481fn project_for_root_name(
1482 client: &TestClient,
1483 root_name: &str,
1484 cx: &TestAppContext,
1485) -> Option<Entity<Project>> {
1486 if let Some(ix) = project_ix_for_root_name(client.local_projects().deref(), root_name, cx) {
1487 return Some(client.local_projects()[ix].clone());
1488 }
1489 if let Some(ix) = project_ix_for_root_name(client.dev_server_projects().deref(), root_name, cx)
1490 {
1491 return Some(client.dev_server_projects()[ix].clone());
1492 }
1493 None
1494}
1495
1496fn project_ix_for_root_name(
1497 projects: &[Entity<Project>],
1498 root_name: &str,
1499 cx: &TestAppContext,
1500) -> Option<usize> {
1501 projects.iter().position(|project| {
1502 project.read_with(cx, |project, cx| {
1503 let worktree = project.visible_worktrees(cx).next().unwrap();
1504 worktree.read(cx).root_name() == root_name
1505 })
1506 })
1507}
1508
1509fn root_name_for_project(project: &Entity<Project>, cx: &TestAppContext) -> String {
1510 project.read_with(cx, |project, cx| {
1511 project
1512 .visible_worktrees(cx)
1513 .next()
1514 .unwrap()
1515 .read(cx)
1516 .root_name()
1517 .to_string()
1518 })
1519}
1520
1521fn project_path_for_full_path(
1522 project: &Entity<Project>,
1523 full_path: &Path,
1524 cx: &TestAppContext,
1525) -> Option<ProjectPath> {
1526 let mut components = full_path.components();
1527 let root_name = components.next().unwrap().as_os_str().to_str().unwrap();
1528 let path = components.as_path().into();
1529 let worktree_id = project.read_with(cx, |project, cx| {
1530 project.worktrees(cx).find_map(|worktree| {
1531 let worktree = worktree.read(cx);
1532 if worktree.root_name() == root_name {
1533 Some(worktree.id())
1534 } else {
1535 None
1536 }
1537 })
1538 })?;
1539 Some(ProjectPath { worktree_id, path })
1540}
1541
1542async fn ensure_project_shared(
1543 project: &Entity<Project>,
1544 client: &TestClient,
1545 cx: &mut TestAppContext,
1546) {
1547 let first_root_name = root_name_for_project(project, cx);
1548 let active_call = cx.read(ActiveCall::global);
1549 if active_call.read_with(cx, |call, _| call.room().is_some())
1550 && project.read_with(cx, |project, _| project.is_local() && !project.is_shared())
1551 {
1552 match active_call
1553 .update(cx, |call, cx| call.share_project(project.clone(), cx))
1554 .await
1555 {
1556 Ok(project_id) => {
1557 log::info!(
1558 "{}: shared project {} with id {}",
1559 client.username,
1560 first_root_name,
1561 project_id
1562 );
1563 }
1564 Err(error) => {
1565 log::error!(
1566 "{}: error sharing project {}: {:?}",
1567 client.username,
1568 first_root_name,
1569 error
1570 );
1571 }
1572 }
1573 }
1574}
1575
1576fn choose_random_project(client: &TestClient, rng: &mut StdRng) -> Option<Entity<Project>> {
1577 client
1578 .local_projects()
1579 .deref()
1580 .iter()
1581 .chain(client.dev_server_projects().iter())
1582 .choose(rng)
1583 .cloned()
1584}
1585
1586fn gen_file_name(rng: &mut StdRng) -> String {
1587 let mut name = String::new();
1588 for _ in 0..10 {
1589 let letter = rng.gen_range('a'..='z');
1590 name.push(letter);
1591 }
1592 name
1593}
1594
1595fn gen_status(rng: &mut StdRng) -> FileStatus {
1596 fn gen_tracked_status(rng: &mut StdRng) -> TrackedStatus {
1597 match rng.gen_range(0..3) {
1598 0 => TrackedStatus {
1599 index_status: StatusCode::Unmodified,
1600 worktree_status: StatusCode::Unmodified,
1601 },
1602 1 => TrackedStatus {
1603 index_status: StatusCode::Modified,
1604 worktree_status: StatusCode::Modified,
1605 },
1606 2 => TrackedStatus {
1607 index_status: StatusCode::Added,
1608 worktree_status: StatusCode::Modified,
1609 },
1610 3 => TrackedStatus {
1611 index_status: StatusCode::Added,
1612 worktree_status: StatusCode::Unmodified,
1613 },
1614 _ => unreachable!(),
1615 }
1616 }
1617
1618 fn gen_unmerged_status_code(rng: &mut StdRng) -> UnmergedStatusCode {
1619 match rng.gen_range(0..3) {
1620 0 => UnmergedStatusCode::Updated,
1621 1 => UnmergedStatusCode::Added,
1622 2 => UnmergedStatusCode::Deleted,
1623 _ => unreachable!(),
1624 }
1625 }
1626
1627 match rng.gen_range(0..2) {
1628 0 => FileStatus::Unmerged(UnmergedStatus {
1629 first_head: gen_unmerged_status_code(rng),
1630 second_head: gen_unmerged_status_code(rng),
1631 }),
1632 1 => FileStatus::Tracked(gen_tracked_status(rng)),
1633 _ => unreachable!(),
1634 }
1635}