From 00b35461bd1a141220cc3c425ce94f10c41f5631 Mon Sep 17 00:00:00 2001
From: Alvaro Almeida Freire Stivi <astivi@google.com>
Date: Wed, 3 Jul 2024 18:10:27 +0000
Subject: [PATCH] Fixes Draining Behavior tests

---
 .../components/drain/behaviors/draining_behavior_test.dart   | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/test/game/components/drain/behaviors/draining_behavior_test.dart b/test/game/components/drain/behaviors/draining_behavior_test.dart
index 484c6ecf..c91697e5 100644
--- a/test/game/components/drain/behaviors/draining_behavior_test.dart
+++ b/test/game/components/drain/behaviors/draining_behavior_test.dart
@@ -49,10 +49,12 @@ void main() {
       flameTester.testGameWidget(
         'loads',
         setUp: (game, _) async {
+          await game.onLoad();
           final parent = Drain.test();
           final behavior = DrainingBehavior();
           await parent.add(behavior);
           await game.ensureAdd(parent);
+          await game.ready();
         },
         verify: (game, _) async {
           final parent = game.descendants().whereType<Drain>().single;
@@ -75,6 +77,7 @@ void main() {
         flameBlocTester.testGameWidget(
           'adds RoundLost when no balls left',
           setUp: (game, _) async {
+            await game.onLoad();
             final drain = Drain.test();
             final behavior = DrainingBehavior();
             final ball = Ball.test();
@@ -97,6 +100,7 @@ void main() {
         flameBlocTester.testGameWidget(
           "doesn't add RoundLost when there are balls left",
           setUp: (game, _) async {
+            await game.onLoad();
             final drain = Drain.test();
             final behavior = DrainingBehavior();
             final ball1 = Ball.test();
@@ -120,6 +124,7 @@ void main() {
         flameBlocTester.testGameWidget(
           'removes the Ball',
           setUp: (game, _) async {
+            await game.onLoad();
             final drain = Drain.test();
             final behavior = DrainingBehavior();
             final ball = Ball.test();