From e32f1f8b8034234a523ab450d0d1fd7814d834bf Mon Sep 17 00:00:00 2001 From: Antonio Scandurra Date: Thu, 23 Sep 2021 18:17:49 +0200 Subject: [PATCH] Create `test_app_state` with `FakeFs` instead of `RealFs` Co-Authored-By: Nathan Sobo --- zed/src/file_finder.rs | 59 ++++++++++++++--------- zed/src/fs.rs | 12 +++++ zed/src/test.rs | 4 +- zed/src/workspace.rs | 104 +++++++++++++++++++++++------------------ 4 files changed, 110 insertions(+), 69 deletions(-) diff --git a/zed/src/file_finder.rs b/zed/src/file_finder.rs index 8f3217b2e5..2749bbe59d 100644 --- a/zed/src/file_finder.rs +++ b/zed/src/file_finder.rs @@ -438,29 +438,37 @@ mod tests { use crate::{ editor::{self, Insert}, fs::FakeFs, - test::{temp_tree, test_app_state}, + test::test_app_state, workspace::Workspace, }; use serde_json::json; - use std::fs; - use tempdir::TempDir; + use std::path::PathBuf; #[gpui::test] async fn test_matching_paths(mut cx: gpui::TestAppContext) { - let tmp_dir = TempDir::new("example").unwrap(); - fs::create_dir(tmp_dir.path().join("a")).unwrap(); - fs::write(tmp_dir.path().join("a/banana"), "banana").unwrap(); - fs::write(tmp_dir.path().join("a/bandana"), "bandana").unwrap(); + let app_state = cx.update(test_app_state); + app_state + .fs + .as_fake() + .insert_tree( + "/root", + json!({ + "a": { + "banana": "", + "bandana": "", + } + }), + ) + .await; cx.update(|cx| { super::init(cx); editor::init(cx); }); - let app_state = cx.update(test_app_state); let (window_id, workspace) = cx.add_window(|cx| Workspace::new(&app_state, cx)); workspace .update(&mut cx, |workspace, cx| { - workspace.add_worktree(tmp_dir.path(), cx) + workspace.add_worktree(Path::new("/root"), cx) }) .await .unwrap(); @@ -572,17 +580,17 @@ mod tests { #[gpui::test] async fn test_single_file_worktrees(mut cx: gpui::TestAppContext) { - let temp_dir = TempDir::new("test-single-file-worktrees").unwrap(); - let dir_path = temp_dir.path().join("the-parent-dir"); - let file_path = dir_path.join("the-file"); - fs::create_dir(&dir_path).unwrap(); - fs::write(&file_path, "").unwrap(); - let app_state = cx.update(test_app_state); + app_state + .fs + .as_fake() + .insert_tree("/root", json!({ "the-parent-dir": { "the-file": "" } })) + .await; + let (_, workspace) = cx.add_window(|cx| Workspace::new(&app_state, cx)); workspace .update(&mut cx, |workspace, cx| { - workspace.add_worktree(&file_path, cx) + workspace.add_worktree(Path::new("/root/the-parent-dir/the-file"), cx) }) .await .unwrap(); @@ -620,18 +628,25 @@ mod tests { #[gpui::test(retries = 5)] async fn test_multiple_matches_with_same_relative_path(mut cx: gpui::TestAppContext) { - let tmp_dir = temp_tree(json!({ - "dir1": { "a.txt": "" }, - "dir2": { "a.txt": "" } - })); - let app_state = cx.update(test_app_state); + app_state + .fs + .as_fake() + .insert_tree( + "/root", + json!({ + "dir1": { "a.txt": "" }, + "dir2": { "a.txt": "" } + }), + ) + .await; + let (_, workspace) = cx.add_window(|cx| Workspace::new(&app_state, cx)); workspace .update(&mut cx, |workspace, cx| { workspace.open_paths( - &[tmp_dir.path().join("dir1"), tmp_dir.path().join("dir2")], + &[PathBuf::from("/root/dir1"), PathBuf::from("/root/dir2")], cx, ) }) diff --git a/zed/src/fs.rs b/zed/src/fs.rs index e9d6d9230b..7f67d6ef02 100644 --- a/zed/src/fs.rs +++ b/zed/src/fs.rs @@ -29,6 +29,8 @@ pub trait Fs: Send + Sync { latency: Duration, ) -> Pin>>>; fn is_fake(&self) -> bool; + #[cfg(any(test, feature = "test-support"))] + fn as_fake(&self) -> &FakeFs; } #[derive(Clone, Debug)] @@ -125,6 +127,11 @@ impl Fs for RealFs { fn is_fake(&self) -> bool { false } + + #[cfg(any(test, feature = "test-support"))] + fn as_fake(&self) -> &FakeFs { + panic!("called `RealFs::as_fake`") + } } #[derive(Clone, Debug)] @@ -413,4 +420,9 @@ impl Fs for FakeFs { fn is_fake(&self) -> bool { true } + + #[cfg(any(test, feature = "test-support"))] + fn as_fake(&self) -> &FakeFs { + self + } } diff --git a/zed/src/test.rs b/zed/src/test.rs index eb6bf20f45..59edd380d5 100644 --- a/zed/src/test.rs +++ b/zed/src/test.rs @@ -1,7 +1,7 @@ use crate::{ assets::Assets, channel::ChannelList, - fs::RealFs, + fs::FakeFs, http::{HttpClient, Request, Response, ServerResponse}, language::LanguageRegistry, rpc::{self, Client, Credentials, EstablishConnectionError}, @@ -177,7 +177,7 @@ pub fn test_app_state(cx: &mut MutableAppContext) -> Arc { channel_list: cx.add_model(|cx| ChannelList::new(user_store.clone(), rpc.clone(), cx)), rpc, user_store, - fs: Arc::new(RealFs), + fs: Arc::new(FakeFs::new()), }) } diff --git a/zed/src/workspace.rs b/zed/src/workspace.rs index c148b72916..456b9e7042 100644 --- a/zed/src/workspace.rs +++ b/zed/src/workspace.rs @@ -1205,11 +1205,9 @@ mod tests { editor::{Editor, Insert}, fs::FakeFs, test::{temp_tree, test_app_state}, - worktree::WorktreeHandle, }; use serde_json::json; - use std::{collections::HashSet, fs}; - use tempdir::TempDir; + use std::collections::HashSet; #[gpui::test] async fn test_open_paths_action(mut cx: gpui::TestAppContext) { @@ -1278,20 +1276,26 @@ mod tests { #[gpui::test] async fn test_open_entry(mut cx: gpui::TestAppContext) { - let dir = temp_tree(json!({ - "a": { - "file1": "contents 1", - "file2": "contents 2", - "file3": "contents 3", - }, - })); - let app_state = cx.update(test_app_state); + app_state + .fs + .as_fake() + .insert_tree( + "/root", + json!({ + "a": { + "file1": "contents 1", + "file2": "contents 2", + "file3": "contents 3", + }, + }), + ) + .await; let (_, workspace) = cx.add_window(|cx| Workspace::new(&app_state, cx)); workspace .update(&mut cx, |workspace, cx| { - workspace.add_worktree(dir.path(), cx) + workspace.add_worktree(Path::new("/root"), cx) }) .await .unwrap(); @@ -1455,28 +1459,30 @@ mod tests { #[gpui::test] async fn test_save_conflicting_item(mut cx: gpui::TestAppContext) { - let dir = temp_tree(json!({ - "a.txt": "", - })); - let app_state = cx.update(test_app_state); + app_state + .fs + .as_fake() + .insert_tree( + "/root", + json!({ + "a.txt": "", + }), + ) + .await; + let (window_id, workspace) = cx.add_window(|cx| Workspace::new(&app_state, cx)); workspace .update(&mut cx, |workspace, cx| { - workspace.add_worktree(dir.path(), cx) + workspace.add_worktree(Path::new("/root"), cx) }) .await .unwrap(); - let tree = cx.read(|cx| { - let mut trees = workspace.read(cx).worktrees().iter(); - trees.next().unwrap().clone() - }); - tree.flush_fs_events(&cx).await; // Open a file within an existing worktree. cx.update(|cx| { workspace.update(cx, |view, cx| { - view.open_paths(&[dir.path().join("a.txt")], cx) + view.open_paths(&[PathBuf::from("/root/a.txt")], cx) }) }) .await; @@ -1487,7 +1493,12 @@ mod tests { }); cx.update(|cx| editor.update(cx, |editor, cx| editor.insert(&Insert("x".into()), cx))); - fs::write(dir.path().join("a.txt"), "changed").unwrap(); + app_state + .fs + .as_fake() + .insert_file("/root/a.txt", "changed".to_string()) + .await + .unwrap(); editor .condition(&cx, |editor, cx| editor.has_conflict(cx)) .await; @@ -1503,12 +1514,12 @@ mod tests { #[gpui::test] async fn test_open_and_save_new_file(mut cx: gpui::TestAppContext) { - let dir = TempDir::new("test-new-file").unwrap(); let app_state = cx.update(test_app_state); + app_state.fs.as_fake().insert_dir("/root").await.unwrap(); let (_, workspace) = cx.add_window(|cx| Workspace::new(&app_state, cx)); workspace .update(&mut cx, |workspace, cx| { - workspace.add_worktree(dir.path(), cx) + workspace.add_worktree(Path::new("/root"), cx) }) .await .unwrap(); @@ -1521,7 +1532,6 @@ mod tests { .unwrap() .clone() }); - tree.flush_fs_events(&cx).await; // Create a new untitled buffer let editor = workspace.update(&mut cx, |workspace, cx| { @@ -1547,7 +1557,7 @@ mod tests { workspace.save_active_item(&Save, cx) }); cx.simulate_new_path_selection(|parent_dir| { - assert_eq!(parent_dir, dir.path()); + assert_eq!(parent_dir, Path::new("/root")); Some(parent_dir.join("the-new-name.rs")) }); cx.read(|cx| { @@ -1608,8 +1618,8 @@ mod tests { async fn test_setting_language_when_saving_as_single_file_worktree( mut cx: gpui::TestAppContext, ) { - let dir = TempDir::new("test-new-file").unwrap(); let app_state = cx.update(test_app_state); + app_state.fs.as_fake().insert_dir("/root").await.unwrap(); let (_, workspace) = cx.add_window(|cx| Workspace::new(&app_state, cx)); // Create a new untitled buffer @@ -1633,7 +1643,7 @@ mod tests { workspace.update(&mut cx, |workspace, cx| { workspace.save_active_item(&Save, cx) }); - cx.simulate_new_path_selection(|_| Some(dir.path().join("the-new-name.rs"))); + cx.simulate_new_path_selection(|_| Some(PathBuf::from("/root/the-new-name.rs"))); editor .condition(&cx, |editor, cx| !editor.is_dirty(cx)) @@ -1650,7 +1660,7 @@ mod tests { cx.update(init); let app_state = cx.update(test_app_state); - cx.dispatch_global_action(OpenNew(app_state)); + cx.dispatch_global_action(OpenNew(app_state.clone())); let window_id = *cx.window_ids().first().unwrap(); let workspace = cx.root_view::(window_id).unwrap(); let editor = workspace.update(&mut cx, |workspace, cx| { @@ -1670,10 +1680,8 @@ mod tests { workspace.save_active_item(&Save, cx) }); - let dir = TempDir::new("test-new-empty-workspace").unwrap(); - cx.simulate_new_path_selection(|_| { - Some(dir.path().canonicalize().unwrap().join("the-new-name")) - }); + app_state.fs.as_fake().insert_dir("/root").await.unwrap(); + cx.simulate_new_path_selection(|_| Some(PathBuf::from("/root/the-new-name"))); editor .condition(&cx, |editor, cx| editor.title(cx) == "the-new-name") @@ -1686,20 +1694,26 @@ mod tests { #[gpui::test] async fn test_pane_actions(mut cx: gpui::TestAppContext) { cx.update(|cx| pane::init(cx)); - - let dir = temp_tree(json!({ - "a": { - "file1": "contents 1", - "file2": "contents 2", - "file3": "contents 3", - }, - })); - let app_state = cx.update(test_app_state); + app_state + .fs + .as_fake() + .insert_tree( + "/root", + json!({ + "a": { + "file1": "contents 1", + "file2": "contents 2", + "file3": "contents 3", + }, + }), + ) + .await; + let (window_id, workspace) = cx.add_window(|cx| Workspace::new(&app_state, cx)); workspace .update(&mut cx, |workspace, cx| { - workspace.add_worktree(dir.path(), cx) + workspace.add_worktree(Path::new("/root"), cx) }) .await .unwrap();