Don't run until all the way until parked when waiting for a model's next event

crates/gpui2/src/app/test_context.rs

@@ -354,10 +354,18 @@ impl<T: Send> Model<T> {
             })
         });
 
-        cx.executor().run_until_parked();
+        loop {
-        rx.try_next()
+            match rx.try_next() {
-            .expect("no event received")
+                Ok(Some(event)) => return event,
-            .expect("model was dropped")
+                Ok(None) => panic!("model was dropped"),
+                Err(_) => {
+                    if !cx.executor().run_step() {
+                        break;
+                    }
+                }
+            }
+        }
+        panic!("no event received")
     }
 }