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::{path, 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.r#gen() {
283 Path::new(path!("/")).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.r#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.r#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.r#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(path!("/")).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!(
1319 guest_file.disk_state(),
1320 host_file.disk_state(),
1321 "guest {} disk_state does not match host {} for path {:?} in project {}",
1322 guest_user_id,
1323 host_user_id,
1324 guest_file.path(),
1325 project_id,
1326 );
1327 }
1328 (None, None) => {}
1329 (None, _) => panic!("host's file is None, guest's isn't"),
1330 (_, None) => panic!("guest's file is None, hosts's isn't"),
1331 }
1332
1333 let host_diff_base = host_project.read_with(host_cx, |project, cx| {
1334 project
1335 .git_store()
1336 .read(cx)
1337 .get_unstaged_diff(host_buffer.read(cx).remote_id(), cx)
1338 .unwrap()
1339 .read(cx)
1340 .base_text_string()
1341 });
1342 let guest_diff_base = guest_project.read_with(client_cx, |project, cx| {
1343 project
1344 .git_store()
1345 .read(cx)
1346 .get_unstaged_diff(guest_buffer.read(cx).remote_id(), cx)
1347 .unwrap()
1348 .read(cx)
1349 .base_text_string()
1350 });
1351 assert_eq!(
1352 guest_diff_base, host_diff_base,
1353 "guest {} diff base does not match host's for path {path:?} in project {project_id}",
1354 client.username
1355 );
1356
1357 let host_saved_version =
1358 host_buffer.read_with(host_cx, |b, _| b.saved_version().clone());
1359 let guest_saved_version =
1360 guest_buffer.read_with(client_cx, |b, _| b.saved_version().clone());
1361 assert_eq!(
1362 guest_saved_version, host_saved_version,
1363 "guest {} saved version does not match host's for path {path:?} in project {project_id}",
1364 client.username
1365 );
1366
1367 let host_is_dirty = host_buffer.read_with(host_cx, |b, _| b.is_dirty());
1368 let guest_is_dirty = guest_buffer.read_with(client_cx, |b, _| b.is_dirty());
1369 assert_eq!(
1370 guest_is_dirty, host_is_dirty,
1371 "guest {} dirty state does not match host's for path {path:?} in project {project_id}",
1372 client.username
1373 );
1374
1375 let host_saved_mtime = host_buffer.read_with(host_cx, |b, _| b.saved_mtime());
1376 let guest_saved_mtime =
1377 guest_buffer.read_with(client_cx, |b, _| b.saved_mtime());
1378 assert_eq!(
1379 guest_saved_mtime, host_saved_mtime,
1380 "guest {} saved mtime does not match host's for path {path:?} in project {project_id}",
1381 client.username
1382 );
1383
1384 let host_is_dirty = host_buffer.read_with(host_cx, |b, _| b.is_dirty());
1385 let guest_is_dirty = guest_buffer.read_with(client_cx, |b, _| b.is_dirty());
1386 assert_eq!(
1387 guest_is_dirty, host_is_dirty,
1388 "guest {} dirty status does not match host's for path {path:?} in project {project_id}",
1389 client.username
1390 );
1391
1392 let host_has_conflict = host_buffer.read_with(host_cx, |b, _| b.has_conflict());
1393 let guest_has_conflict =
1394 guest_buffer.read_with(client_cx, |b, _| b.has_conflict());
1395 assert_eq!(
1396 guest_has_conflict, host_has_conflict,
1397 "guest {} conflict status does not match host's for path {path:?} in project {project_id}",
1398 client.username
1399 );
1400 }
1401 }
1402 }
1403 }
1404}
1405
1406fn generate_git_operation(rng: &mut StdRng, client: &TestClient) -> GitOperation {
1407 fn generate_file_paths(
1408 repo_path: &Path,
1409 rng: &mut StdRng,
1410 client: &TestClient,
1411 ) -> Vec<PathBuf> {
1412 let mut paths = client
1413 .fs()
1414 .files()
1415 .into_iter()
1416 .filter(|path| path.starts_with(repo_path))
1417 .collect::<Vec<_>>();
1418
1419 let count = rng.gen_range(0..=paths.len());
1420 paths.shuffle(rng);
1421 paths.truncate(count);
1422
1423 paths
1424 .iter()
1425 .map(|path| path.strip_prefix(repo_path).unwrap().to_path_buf())
1426 .collect::<Vec<_>>()
1427 }
1428
1429 let repo_path = client.fs().directories(false).choose(rng).unwrap().clone();
1430
1431 match rng.gen_range(0..100_u32) {
1432 0..=25 => {
1433 let file_paths = generate_file_paths(&repo_path, rng, client);
1434
1435 let contents = file_paths
1436 .into_iter()
1437 .map(|path| (path, Alphanumeric.sample_string(rng, 16)))
1438 .collect();
1439
1440 GitOperation::WriteGitIndex {
1441 repo_path,
1442 contents,
1443 }
1444 }
1445 26..=63 => {
1446 let new_branch = (rng.gen_range(0..10) > 3).then(|| Alphanumeric.sample_string(rng, 8));
1447
1448 GitOperation::WriteGitBranch {
1449 repo_path,
1450 new_branch,
1451 }
1452 }
1453 64..=100 => {
1454 let file_paths = generate_file_paths(&repo_path, rng, client);
1455 let statuses = file_paths
1456 .into_iter()
1457 .map(|path| (path, gen_status(rng)))
1458 .collect::<Vec<_>>();
1459 GitOperation::WriteGitStatuses {
1460 repo_path,
1461 statuses,
1462 }
1463 }
1464 _ => unreachable!(),
1465 }
1466}
1467
1468fn buffer_for_full_path(
1469 client: &TestClient,
1470 project: &Entity<Project>,
1471 full_path: &PathBuf,
1472 cx: &TestAppContext,
1473) -> Option<Entity<language::Buffer>> {
1474 client
1475 .buffers_for_project(project)
1476 .iter()
1477 .find(|buffer| {
1478 buffer.read_with(cx, |buffer, cx| {
1479 buffer.file().unwrap().full_path(cx) == *full_path
1480 })
1481 })
1482 .cloned()
1483}
1484
1485fn project_for_root_name(
1486 client: &TestClient,
1487 root_name: &str,
1488 cx: &TestAppContext,
1489) -> Option<Entity<Project>> {
1490 if let Some(ix) = project_ix_for_root_name(client.local_projects().deref(), root_name, cx) {
1491 return Some(client.local_projects()[ix].clone());
1492 }
1493 if let Some(ix) = project_ix_for_root_name(client.dev_server_projects().deref(), root_name, cx)
1494 {
1495 return Some(client.dev_server_projects()[ix].clone());
1496 }
1497 None
1498}
1499
1500fn project_ix_for_root_name(
1501 projects: &[Entity<Project>],
1502 root_name: &str,
1503 cx: &TestAppContext,
1504) -> Option<usize> {
1505 projects.iter().position(|project| {
1506 project.read_with(cx, |project, cx| {
1507 let worktree = project.visible_worktrees(cx).next().unwrap();
1508 worktree.read(cx).root_name() == root_name
1509 })
1510 })
1511}
1512
1513fn root_name_for_project(project: &Entity<Project>, cx: &TestAppContext) -> String {
1514 project.read_with(cx, |project, cx| {
1515 project
1516 .visible_worktrees(cx)
1517 .next()
1518 .unwrap()
1519 .read(cx)
1520 .root_name()
1521 .to_string()
1522 })
1523}
1524
1525fn project_path_for_full_path(
1526 project: &Entity<Project>,
1527 full_path: &Path,
1528 cx: &TestAppContext,
1529) -> Option<ProjectPath> {
1530 let mut components = full_path.components();
1531 let root_name = components.next().unwrap().as_os_str().to_str().unwrap();
1532 let path = components.as_path().into();
1533 let worktree_id = project.read_with(cx, |project, cx| {
1534 project.worktrees(cx).find_map(|worktree| {
1535 let worktree = worktree.read(cx);
1536 if worktree.root_name() == root_name {
1537 Some(worktree.id())
1538 } else {
1539 None
1540 }
1541 })
1542 })?;
1543 Some(ProjectPath { worktree_id, path })
1544}
1545
1546async fn ensure_project_shared(
1547 project: &Entity<Project>,
1548 client: &TestClient,
1549 cx: &mut TestAppContext,
1550) {
1551 let first_root_name = root_name_for_project(project, cx);
1552 let active_call = cx.read(ActiveCall::global);
1553 if active_call.read_with(cx, |call, _| call.room().is_some())
1554 && project.read_with(cx, |project, _| project.is_local() && !project.is_shared())
1555 {
1556 match active_call
1557 .update(cx, |call, cx| call.share_project(project.clone(), cx))
1558 .await
1559 {
1560 Ok(project_id) => {
1561 log::info!(
1562 "{}: shared project {} with id {}",
1563 client.username,
1564 first_root_name,
1565 project_id
1566 );
1567 }
1568 Err(error) => {
1569 log::error!(
1570 "{}: error sharing project {}: {:?}",
1571 client.username,
1572 first_root_name,
1573 error
1574 );
1575 }
1576 }
1577 }
1578}
1579
1580fn choose_random_project(client: &TestClient, rng: &mut StdRng) -> Option<Entity<Project>> {
1581 client
1582 .local_projects()
1583 .deref()
1584 .iter()
1585 .chain(client.dev_server_projects().iter())
1586 .choose(rng)
1587 .cloned()
1588}
1589
1590fn gen_file_name(rng: &mut StdRng) -> String {
1591 let mut name = String::new();
1592 for _ in 0..10 {
1593 let letter = rng.gen_range('a'..='z');
1594 name.push(letter);
1595 }
1596 name
1597}
1598
1599fn gen_status(rng: &mut StdRng) -> FileStatus {
1600 fn gen_tracked_status(rng: &mut StdRng) -> TrackedStatus {
1601 match rng.gen_range(0..3) {
1602 0 => TrackedStatus {
1603 index_status: StatusCode::Unmodified,
1604 worktree_status: StatusCode::Unmodified,
1605 },
1606 1 => TrackedStatus {
1607 index_status: StatusCode::Modified,
1608 worktree_status: StatusCode::Modified,
1609 },
1610 2 => TrackedStatus {
1611 index_status: StatusCode::Added,
1612 worktree_status: StatusCode::Modified,
1613 },
1614 3 => TrackedStatus {
1615 index_status: StatusCode::Added,
1616 worktree_status: StatusCode::Unmodified,
1617 },
1618 _ => unreachable!(),
1619 }
1620 }
1621
1622 fn gen_unmerged_status_code(rng: &mut StdRng) -> UnmergedStatusCode {
1623 match rng.gen_range(0..3) {
1624 0 => UnmergedStatusCode::Updated,
1625 1 => UnmergedStatusCode::Added,
1626 2 => UnmergedStatusCode::Deleted,
1627 _ => unreachable!(),
1628 }
1629 }
1630
1631 match rng.gen_range(0..2) {
1632 0 => FileStatus::Unmerged(UnmergedStatus {
1633 first_head: gen_unmerged_status_code(rng),
1634 second_head: gen_unmerged_status_code(rng),
1635 }),
1636 1 => FileStatus::Tracked(gen_tracked_status(rng)),
1637 _ => unreachable!(),
1638 }
1639}