Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make stricter assertions about change events in random worktree test

Browse source
This commit is contained in:
Max Brunsfeld 2023-05-23 14:07:10 -07:00
parent f890eefdef
commit 59bfd40679
1 changed files with 17 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

35
crates/project/src/worktree.rs
View file

@ -4331,48 +4331,47 @@ mod tests {
// all changes to the worktree's snapshot.
worktree.update(cx, |tree, cx| {
let mut paths = tree
.as_local()
.unwrap()
.paths()
.cloned()
.entries(true)
.map(|e| (e.path.clone(), e.mtime))
.collect::<Vec<_>>();
cx.subscribe(&worktree, move |tree, _, event, _| {
if let Event::UpdatedEntries(changes) = event {
for ((path, _), change_type) in changes.iter() {
let mtime = tree.entry_for_path(&path).map(|e| e.mtime);
let path = path.clone();
let ix = match paths.binary_search(&path) {
let ix = match paths.binary_search_by_key(&&path, |e| &e.0) {
Ok(ix) | Err(ix) => ix,
};
match change_type {
PathChange::Loaded => {
assert_ne!(paths.get(ix), Some(&path));
paths.insert(ix, path);
paths.insert(ix, (path, mtime.unwrap()));
}
PathChange::Added => {
assert_ne!(paths.get(ix), Some(&path));
paths.insert(ix, path);
paths.insert(ix, (path, mtime.unwrap()));
}
PathChange::Removed => {
assert_eq!(paths.get(ix), Some(&path));
paths.remove(ix);
}
PathChange::Updated => {
assert_eq!(paths.get(ix), Some(&path));
let entry = paths.get_mut(ix).unwrap();
assert_eq!(entry.0, path);
entry.1 = mtime.unwrap();
}
PathChange::AddedOrUpdated => {
if paths[ix] != path {
paths.insert(ix, path);
if paths.get(ix).map(|e| &e.0) == Some(&path) {
paths.get_mut(ix).unwrap().1 = mtime.unwrap();
} else {
paths.insert(ix, (path, mtime.unwrap()));
}
}
}
}
let new_paths = tree.paths().cloned().collect::<Vec<_>>();
let new_paths = tree
.entries(true)
.map(|e| (e.path.clone(), e.mtime))
.collect::<Vec<_>>();
assert_eq!(paths, new_paths, "incorrect changes: {:?}", changes);
}
})