Allow effects to be flushed before TestAppContext::update callback completes

This commit is contained in:
Nathan Sobo 2021-04-10 00:11:13 -06:00
parent 97a8a8ed43
commit 620eedb727
2 changed files with 6 additions and 5 deletions

View File

@ -313,8 +313,11 @@ impl TestAppContext {
pub fn update<T, F: FnOnce(&mut MutableAppContext) -> T>(&mut self, callback: F) -> T {
let mut state = self.0.borrow_mut();
state.pending_flushes += 1;
// Don't increment pending flushes in order to effects to be flushed before the callback
// completes, which is helpful in tests.
let result = callback(&mut *state);
// Flush effects after the callback just in case there are any. This can happen in edge
// cases such as the closure dropping handles.
state.flush_effects();
result
}
@ -895,7 +898,7 @@ impl MutableAppContext {
}
fn flush_effects(&mut self) {
self.pending_flushes -= 1;
self.pending_flushes = self.pending_flushes.saturating_sub(1);
if !self.flushing_effects && self.pending_flushes == 0 {
self.flushing_effects = true;

View File

@ -515,13 +515,11 @@ mod tests {
);
ctx.dispatch_action(window_id, vec![pane_2.id()], "pane:close_active_item", ());
});
app.read(|ctx| {
let w = workspace_view.read(ctx);
assert_eq!(w.panes.len(), 1);
assert_eq!(w.active_pane(), &pane_1);
})
});
});
}
}