From f9650b3111f7a1ae071fd14adc4bc9554cf0545b Mon Sep 17 00:00:00 2001
From: Max Brunsfeld <maxbrunsfeld@gmail.com>
Date: Thu, 16 Nov 2023 14:09:33 -0800
Subject: [PATCH] Don't run until all the way until parked when waiting for a
 model's next event

---
 crates/gpui2/src/app/test_context.rs | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/crates/gpui2/src/app/test_context.rs b/crates/gpui2/src/app/test_context.rs
index 5397a2214d..37f81da15b 100644
--- a/crates/gpui2/src/app/test_context.rs
+++ b/crates/gpui2/src/app/test_context.rs
@@ -354,10 +354,18 @@ impl<T: Send> Model<T> {
             })
         });
 
-        cx.executor().run_until_parked();
-        rx.try_next()
-            .expect("no event received")
-            .expect("model was dropped")
+        loop {
+            match rx.try_next() {
+                Ok(Some(event)) => return event,
+                Ok(None) => panic!("model was dropped"),
+                Err(_) => {
+                    if !cx.executor().run_step() {
+                        break;
+                    }
+                }
+            }
+        }
+        panic!("no event received")
     }
 }