refactor: constructor item drop test

refinedmods · May 26, 2024 · f0ec6ca · f0ec6ca
1 parent b6162d5
commit f0ec6ca
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/...nedmods/refinedstorage2/platform/common/constructordestructor/ConstructorBlockEntity.java b/...nedmods/refinedstorage2/platform/common/constructordestructor/ConstructorBlockEntity.java
@@ -117,7 +117,7 @@ boolean isDropItems() {
         return dropItems;
     }
 
-    void setDropItems(final boolean dropItems) {
+    public void setDropItems(final boolean dropItems) {
         this.dropItems = dropItems;
         setChanged();
         if (level instanceof ServerLevel serverLevel) {

diff --git a/...m-forge/src/test/java/com/refinedmods/refinedstorage2/platform/forge/ConstructorTest.java b/...m-forge/src/test/java/com/refinedmods/refinedstorage2/platform/forge/ConstructorTest.java
@@ -12,6 +12,7 @@
 import net.minecraft.gametest.framework.GameTest;
 import net.minecraft.gametest.framework.GameTestHelper;
 import net.minecraft.gametest.framework.GameTestSequence;
+import net.minecraft.world.item.Items;
 import net.minecraft.world.level.block.Blocks;
 import net.neoforged.neoforge.gametest.GameTestHolder;
 import net.neoforged.neoforge.gametest.PrefixGameTestTemplate;
@@ -72,4 +73,31 @@ public static void shouldPlaceBlock(final GameTestHelper helper) {
                 .thenSucceed();
         });
     }
+
+    @GameTest(template = "empty_15x15")
+    public static void shouldDropItem(final GameTestHelper helper) {
+        prepareConstructorPlot(helper, Direction.EAST, (constructor, pos, sequence) -> {
+            // Arrange
+            sequence.thenWaitUntil(networkIsAvailable(helper, pos, network -> {
+                insertItem(helper, network, DIRT, 10);
+                insertItem(helper, network, STONE, 15);
+            }));
+
+            // Act
+            constructor.setDropItems(true);
+            constructor.setFilters(List.of(DIRT));
+
+            // Assert
+            sequence
+                .thenWaitUntil(() -> helper.assertBlockNotPresent(Blocks.DIRT, pos.east()))
+                .thenWaitUntil(() -> helper.assertItemEntityPresent(Items.DIRT, pos.east(), 1))
+                .thenWaitUntil(storageContainsExactly(
+                    helper,
+                    pos,
+                    new ResourceAmount(DIRT, 9),
+                    new ResourceAmount(STONE, 15)
+                ))
+                .thenSucceed();
+        });
+    }
 }