Add smoke test for collaboration

This commit is contained in:
Mikayla Maki 2023-05-04 06:41:11 -07:00
parent ca4da52e39
commit c6d7ed33c2
No known key found for this signature in database
3 changed files with 104 additions and 10 deletions

View file

@ -619,7 +619,10 @@ impl FakeFs {
.boxed()
}
pub async fn set_index_for_repo(&self, dot_git: &Path, head_state: &[(&Path, String)]) {
pub fn with_git_state<F>(&self, dot_git: &Path, f: F)
where
F: FnOnce(&mut FakeGitRepositoryState),
{
let mut state = self.state.lock();
let entry = state.read_path(dot_git).unwrap();
let mut entry = entry.lock();
@ -628,12 +631,7 @@ impl FakeFs {
let repo_state = git_repo_state.get_or_insert_with(Default::default);
let mut repo_state = repo_state.lock();
repo_state.index_contents.clear();
repo_state.index_contents.extend(
head_state
.iter()
.map(|(path, content)| (path.to_path_buf(), content.clone())),
);
f(&mut repo_state);
state.emit_event([dot_git]);
} else {
@ -641,6 +639,21 @@ impl FakeFs {
}
}
pub async fn set_branch_name(&self, dot_git: &Path, branch: Option<impl Into<String>>) {
self.with_git_state(dot_git, |state| state.branch_name = branch.map(Into::into))
}
pub async fn set_index_for_repo(&self, dot_git: &Path, head_state: &[(&Path, String)]) {
self.with_git_state(dot_git, |state| {
state.index_contents.clear();
state.index_contents.extend(
head_state
.iter()
.map(|(path, content)| (path.to_path_buf(), content.clone())),
);
});
}
pub fn paths(&self) -> Vec<PathBuf> {
let mut result = Vec::new();
let mut queue = collections::VecDeque::new();

View file

@ -71,6 +71,7 @@ pub struct FakeGitRepository {
#[derive(Debug, Clone, Default)]
pub struct FakeGitRepositoryState {
pub index_contents: HashMap<PathBuf, String>,
pub branch_name: Option<String>,
}
impl FakeGitRepository {
@ -89,7 +90,8 @@ impl GitRepository for FakeGitRepository {
}
fn branch_name(&self) -> Option<String> {
None
let state = self.state.lock();
state.branch_name.clone()
}
}