From 5f8bddce977e33c0754ddca905600fdb23526bc9 Mon Sep 17 00:00:00 2001
From: Erick Zanardo <erickzanardoo@gmail.com>
Date: Fri, 1 Apr 2022 18:19:25 -0300
Subject: [PATCH] fix: ball spawn logic

---
 lib/game/components/controlled_ball.dart       | 2 +-
 test/game/components/controlled_ball_test.dart | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/game/components/controlled_ball.dart b/lib/game/components/controlled_ball.dart
index 257d4f1d..0e37ed7f 100644
--- a/lib/game/components/controlled_ball.dart
+++ b/lib/game/components/controlled_ball.dart
@@ -88,7 +88,7 @@ class LaunchedBallController extends BallController
   void onNewState(GameState state) {
     super.onNewState(state);
     component.shouldRemove = true;
-    if (state.balls > 1) gameRef.spawnBall();
+    if (state.balls >= 1) gameRef.spawnBall();
   }
 
   /// Removes the [Ball] from a [PinballGame]; spawning a new [Ball] if
diff --git a/test/game/components/controlled_ball_test.dart b/test/game/components/controlled_ball_test.dart
index 8417aa25..f9494543 100644
--- a/test/game/components/controlled_ball_test.dart
+++ b/test/game/components/controlled_ball_test.dart
@@ -176,7 +176,7 @@ void main() {
             await game.ensureAdd(ball);
 
             final state = MockGameState();
-            when(() => state.balls).thenReturn(1);
+            when(() => state.balls).thenReturn(0);
 
             final previousBalls = game.descendants().whereType<Ball>().toList();
             controller.onNewState(state);