Add pumpUntil

Author: Amar3tto

playground/frontend/playground_components_dev/lib/src/widget_tester.dart

@@ -31,33 +31,6 @@ import 'examples/example_descriptor.dart';
 import 'expect.dart';
 
 extension WidgetTesterExtension on WidgetTester {
-  Future<RunFinishedAnalyticsEvent> waitForRunFinishedAnalyticsEvent({
-      Duration timeout = const Duration(seconds: 30),
-      Duration step = const Duration(milliseconds: 100),
-      Object? previousEvent, // optional guard
-    }) async {
-      final deadline = DateTime.now().add(timeout);
-
-      while (DateTime.now().isBefore(deadline)) {
-        // Drive frames/timers on Web-server device.
-        await pump(step);
-
-        final e = PlaygroundComponents.analyticsService.lastEvent;
-
-        // Optional: ensure we don't accept an old event from a previous run.
-        if (previousEvent != null && identical(e, previousEvent)) {
-          continue;
-        }
-
-        if (e is RunFinishedAnalyticsEvent) return e;
-      }
-
-      throw TestFailure(
-        'Timed out waiting for RunFinishedAnalyticsEvent. '
-        'Last event: ${PlaygroundComponents.analyticsService.lastEvent}',
-      );
-    }
-
   //workaround for https://github.com/flutter/flutter/issues/120060
   Future<void> enterCodeFieldText(String text) async {
     final codeField = widget(find.snippetCodeField());
@@ -175,6 +148,19 @@ extension WidgetTesterExtension on WidgetTester {
     expectOutputIfDeployed(example, this);
   }
 
+  Future<void> pumpUntil(
+    bool Function() condition, {
+    Duration timeout = const Duration(seconds: 60),
+    Duration step = const Duration(milliseconds: 100),
+  }) async {
+    final deadline = DateTime.now().add(timeout);
+    while (DateTime.now().isBefore(deadline)) {
+      await pump(step);
+      if (condition()) return;
+    }
+    throw TestFailure('Timed out waiting for condition');
+  }
+
   Future<void> modifyRunExpectReal(ExampleDescriptor example) async {
     modifyCodeController();
 
@@ -186,16 +172,12 @@ extension WidgetTesterExtension on WidgetTester {
     await tap(find.runOrCancelButton());
     await pump(); // start the click handler
 
+    await pumpUntil(() => find.runOrCancelButton().evaluate().isNotEmpty);
+
+    // Now check output
     final actualText = findOutputText();
     expect(actualText, isNot(startsWith(kCachedResultsLog)));
     expectOutputIfDeployed(example, this);
-
-    final finishedEvent = await waitForRunFinishedAnalyticsEvent(
-      previousEvent: before,
-      timeout: const Duration(seconds: 60), // optional bump for web
-    );
-
-    expect(finishedEvent.snippetContext, eventSnippetContext);
   }
 
   /// Modifies the code controller in a unique way by inserting timestamp