feat: controller gametest

refinedmods · Dec 13, 2024 · 9260480 · 9260480
1 parent ef25098
commit 9260480
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/...nedstorage-neoforge/src/test/java/com/refinedmods/refinedstorage/common/GameTestUtil.java b/...nedstorage-neoforge/src/test/java/com/refinedmods/refinedstorage/common/GameTestUtil.java
@@ -2,7 +2,7 @@
 
 import com.refinedmods.refinedstorage.api.core.Action;
 import com.refinedmods.refinedstorage.api.network.Network;
-import com.refinedmods.refinedstorage.api.network.energy.EnergyNetworkComponent;
+import com.refinedmods.refinedstorage.api.network.energy.EnergyStorage;
 import com.refinedmods.refinedstorage.api.network.node.NetworkNode;
 import com.refinedmods.refinedstorage.api.network.storage.StorageNetworkComponent;
 import com.refinedmods.refinedstorage.api.resource.ResourceAmount;

diff --git a/...oforge/src/test/java/com/refinedmods/refinedstorage/common/controller/ControllerTest.java b/...oforge/src/test/java/com/refinedmods/refinedstorage/common/controller/ControllerTest.java
@@ -34,16 +34,16 @@ public static void shouldConsumeEnergy(final GameTestHelper helper) {
             // Assert
             sequence
                 .thenIdle(20)
-                .thenExecute(() -> energyStoredExactly(energyStorage.getStored(), energyStorage.getCapacity()))
+                .thenExecute(energyStoredExactly(energyStorage, energyStorage.getCapacity()))
                 .thenWaitUntil(() -> helper.setBlock(pos.above(), RSBLOCKS.getGrid().getDefault()))
                 .thenIdle(1)
-                .thenExecute(() -> energyStoredExactly(
-                    energyStorage.getStored(),
+                .thenExecute(energyStoredExactly(
+                    energyStorage,
                     energyStorage.getCapacity() - Platform.INSTANCE.getConfig().getGrid().getEnergyUsage()
                 ))
                 .thenIdle(9)
-                .thenExecute(() -> energyStoredExactly(
-                    energyStorage.getStored(),
+                .thenExecute(energyStoredExactly(
+                    energyStorage,
                     energyStorage.getCapacity() - Platform.INSTANCE.getConfig().getGrid().getEnergyUsage() * 10
                 ))
                 .thenSucceed();
@@ -62,10 +62,10 @@ public static void shouldNotConsumeEnergy(final GameTestHelper helper) {
             // Assert
             sequence
                 .thenIdle(20)
-                .thenExecute(() -> energyStoredExactly(energyStorage.getStored(), energyStorage.getCapacity()))
+                .thenExecute(energyStoredExactly(energyStorage, energyStorage.getCapacity()))
                 .thenWaitUntil(() -> helper.setBlock(pos.above(), RSBLOCKS.getGrid().getDefault()))
                 .thenIdle(20)
-                .thenExecute(() -> energyStoredExactly(energyStorage.getStored(), energyStorage.getCapacity()))
+                .thenExecute(energyStoredExactly(energyStorage, energyStorage.getCapacity()))
                 .thenSucceed();
         });
     }