test: fix miner to respect cooldown

eseidel · Sep 25, 2023 · 9253aaa · 9253aaa
1 parent 7b95300
commit 9253aaa
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/packages/cli/lib/behavior/behavior.dart b/packages/cli/lib/behavior/behavior.dart
@@ -160,6 +160,7 @@ class MultiJob {
         centralCommand,
         caches,
         ship,
+        getNow: getNow,
       );
       shipDetail(ship, '$name ${state.jobIndex} $result');
       if (result.isComplete) {

diff --git a/packages/cli/test/behavior/miner_test.dart b/packages/cli/test/behavior/miner_test.dart
@@ -250,6 +250,40 @@ void main() {
     when(() => db.insertExtraction(any())).thenAnswer((_) => Future.value());
 
     final logger = _MockLogger();
+
+    // With the reactor expiration, we should wait.
+    final reactorExpiration = now.add(const Duration(seconds: 10));
+    when(() => ship.cooldown).thenReturn(
+      Cooldown(
+        shipSymbol: shipSymbol.symbol,
+        totalSeconds: 0,
+        remainingSeconds: 0,
+        expiration: reactorExpiration,
+      ),
+    );
+    final reactorWait = await runWithLogger(
+      logger,
+      () => advanceMiner(
+        api,
+        db,
+        centralCommand,
+        caches,
+        state,
+        ship,
+        getNow: getNow,
+      ),
+    );
+    expect(reactorWait, reactorExpiration);
+
+    // With no wait, we should be able to complete the mining.
+    when(() => ship.cooldown).thenReturn(
+      Cooldown(
+        shipSymbol: shipSymbol.symbol,
+        totalSeconds: 0,
+        remainingSeconds: 0,
+      ),
+    );
+
     final waitUntil = await runWithLogger(
       logger,
       () => advanceMiner(