diff --git a/src/test/java/org/bitbiome/commands/LookaroundCommandTest.java b/src/test/java/org/bitbiome/commands/LookaroundCommandTest.java index 24e7ba4..b48690f 100644 --- a/src/test/java/org/bitbiome/commands/LookaroundCommandTest.java +++ b/src/test/java/org/bitbiome/commands/LookaroundCommandTest.java @@ -1,11 +1,16 @@ package org.bitbiome.commands; +import org.bitbiome.entities.Item; +import org.bitbiome.entities.Mob; import org.json.JSONArray; import org.json.JSONObject; import org.junit.jupiter.api.Test; -import static org.junit.jupiter.api.Assertions.assertEquals; +import java.util.ArrayList; +import java.util.Random; +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertTrue; public class LookaroundCommandTest { @@ -37,4 +42,31 @@ public class LookaroundCommandTest { assertEquals("Es gibt Bäume und Sträucher", result.getString("description")); } + @Test + public void testGetRandomMob() { + LookaroundCommand command = new LookaroundCommand(); + int randomNumberMobs = 2; + Random random = new Random(); + JSONArray mobs = new JSONArray(); + JSONObject mob1 = new JSONObject(); + mob1.put("name", "BigFoot"); + mob1.put("isFriendly", true); + mob1.put("hp", 10); + mob1.put("damage", 5); + JSONObject mob2 = new JSONObject(); + mob2.put("name", "Yeti"); + mob2.put("isFriendly", false); + mob2.put("hp", 20); + mob2.put("damage", 10); + mobs.put(mob1); + mobs.put(mob2); + ArrayList foundMobs = new ArrayList<>(); + ArrayList result = command.getRandomMob(randomNumberMobs, random, mobs, foundMobs); + + assertEquals(randomNumberMobs, result.size()); + for (Mob mob : result) { + assertTrue(mob.getName().equals("BigFoot") || mob.getName().equals("Yeti")); + } + } + }