1 files changed, 34 insertions, 0 deletions
diff --git a/src/test/java/de/hysky/skyblocker/skyblock/dwarven/MetalDetectorTest.java b/src/test/java/de/hysky/skyblocker/skyblock/dwarven/MetalDetectorTest.java
new file mode 100644
index 00000000..22532f96
--- /dev/null
+++ b/src/test/java/de/hysky/skyblocker/skyblock/dwarven/MetalDetectorTest.java
@@ -0,0 +1,34 @@
+package de.hysky.skyblocker.skyblock.dwarven;
+
+import net.minecraft.util.math.Vec3d;
+import net.minecraft.util.math.Vec3i;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+
+import java.util.ArrayList;
+
+public class MetalDetectorTest {
+
+    @Test
+    void testFindPossibleBlocks() {
+        //test starting without knowing middle
+        MetalDetector.updatePossibleBlocks(10.0, new Vec3d(0, 0, 0));
+        Assertions.assertEquals(MetalDetector.possibleBlocks.size(), 40);
+
+        MetalDetector.updatePossibleBlocks(11.2, new Vec3d(5, 0, 0));
+        Assertions.assertEquals(MetalDetector.possibleBlocks.size(), 2);
+
+        MetalDetector.updatePossibleBlocks(10.0, new Vec3d(10, 0, 10));
+        Assertions.assertEquals(MetalDetector.possibleBlocks.get(0), new Vec3i(0, 0, 10));
+
+        //test while knowing the middle location
+        MetalDetector.possibleBlocks = new ArrayList<>();
+        MetalDetector.newTreasure = true;
+        MetalDetector.minesCenter = new Vec3i(0, 0, 0);
+
+        MetalDetector.updatePossibleBlocks(24.9, new Vec3d(10, 1, 10));
+        Assertions.assertEquals(MetalDetector.possibleBlocks.size(), 1);
+        Assertions.assertEquals(MetalDetector.possibleBlocks.get(0), new Vec3i(1, -20, 20));
+
+    }
+}