Upstream/bentobox
1
0
Fork 0
mirror of https://github.com/BentoBoxWorld/BentoBox.git synced 2025-01-10 10:28:21 +01:00
Code Issues Projects Releases Wiki Activity

Check passengers of jockeys when spawned. #2195

Browse Source
This commit is contained in:
tastybento 2023-10-07 11:15:23 -07:00
parent 6d2f79881b
commit 7fbd041be9
2 changed files with 136 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/main/java/world/bentobox/bentobox/listeners/flags/worldsettings/LimitMobsListener.java
View File

@ -1,8 +1,10 @@
package world.bentobox.bentobox.listeners.flags.worldsettings;
import org.bukkit.entity.EntityType;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.entity.CreatureSpawnEvent;
import org.bukkit.event.entity.CreatureSpawnEvent.SpawnReason;
import world.bentobox.bentobox.api.flags.FlagListener;
@ -19,7 +21,14 @@ public class LimitMobsListener extends FlagListener {
*/
@EventHandler(priority = EventPriority.LOW, ignoreCancelled = true)
public void onMobSpawn(CreatureSpawnEvent e) {
if (getIWM().inWorld(e.getLocation()) && getIWM().getMobLimitSettings(e.getLocation().getWorld()).contains(e.getEntityType().name())) {
check(e, e.getEntityType());
if (e.getSpawnReason().equals(SpawnReason.JOCKEY) ) {
e.getEntity().getPassengers().forEach(pass -> check(e, pass.getType()));
}
}
private void check(CreatureSpawnEvent e, EntityType type) {
if (getIWM().inWorld(e.getLocation()) && getIWM().getMobLimitSettings(e.getLocation().getWorld()).contains(type.name())) {
e.setCancelled(true);
}
}

126
src/test/java/world/bentobox/bentobox/listeners/flags/worldsettings/LimitMobsListenerTest.java Normal file
View File

@ -0,0 +1,126 @@
package world.bentobox.bentobox.listeners.flags.worldsettings;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import java.util.ArrayList;
import java.util.List;
import org.bukkit.Location;
import org.bukkit.World;
import org.bukkit.entity.EntityType;
import org.bukkit.entity.LivingEntity;
import org.bukkit.event.entity.CreatureSpawnEvent;
import org.bukkit.event.entity.CreatureSpawnEvent.SpawnReason;
import org.eclipse.jdt.annotation.NonNull;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.powermock.modules.junit4.PowerMockRunner;
import org.powermock.reflect.Whitebox;
import world.bentobox.bentobox.BentoBox;
import world.bentobox.bentobox.managers.IslandWorldManager;
/**
* @author tastybento
*
*/
@RunWith(PowerMockRunner.class)
public class LimitMobsListenerTest {
@Mock
private IslandWorldManager iwm;
@Mock
private @NonNull World world;
private List<String> list = new ArrayList<>();
private LimitMobsListener lml;
@Mock
private LivingEntity zombie;
@Mock
private LivingEntity skelly;
@Mock
private LivingEntity jockey;
@Mock
private Location location;
/**
* @throws java.lang.Exception
*/
@Before
public void setUp() throws Exception {
// Set up plugin
BentoBox plugin = mock(BentoBox.class);
Whitebox.setInternalState(BentoBox.class, "instance", plugin);
when(plugin.getIWM()).thenReturn(iwm);
list.add("SKELETON");
when(iwm.getMobLimitSettings(world)).thenReturn(list);
when(iwm.inWorld(world)).thenReturn(true);
when(iwm.inWorld(location)).thenReturn(true);
when(location.getWorld()).thenReturn(world);
when(zombie.getType()).thenReturn(EntityType.ZOMBIE);
when(zombie.getLocation()).thenReturn(location);
when(skelly.getType()).thenReturn(EntityType.SKELETON);
when(skelly.getLocation()).thenReturn(location);
when(jockey.getType()).thenReturn(EntityType.SPIDER);
when(jockey.getLocation()).thenReturn(location);
when(jockey.getPassengers()).thenReturn(List.of(skelly));
lml = new LimitMobsListener();
}
/**
* @throws java.lang.Exception
*/
@After
public void tearDown() throws Exception {
Mockito.framework().clearInlineMocks();
}
/**
* Test method for {@link world.bentobox.bentobox.listeners.flags.worldsettings.LimitMobsListener#onMobSpawn(org.bukkit.event.entity.CreatureSpawnEvent)}.
*/
@Test
public void testOnMobSpawn() {
CreatureSpawnEvent e = new CreatureSpawnEvent(skelly, SpawnReason.NATURAL);
lml.onMobSpawn(e);
assertTrue(e.isCancelled());
}
/**
* Test method for {@link world.bentobox.bentobox.listeners.flags.worldsettings.LimitMobsListener#onMobSpawn(org.bukkit.event.entity.CreatureSpawnEvent)}.
*/
@Test
public void testOnMobSpawnNotInWorld() {
when(location.getWorld()).thenReturn(mock(World.class));
CreatureSpawnEvent e = new CreatureSpawnEvent(skelly, SpawnReason.NATURAL);
lml.onMobSpawn(e);
assertFalse(e.isCancelled());
}
/**
* Test method for {@link world.bentobox.bentobox.listeners.flags.worldsettings.LimitMobsListener#onMobSpawn(org.bukkit.event.entity.CreatureSpawnEvent)}.
*/
@Test
public void testOnMobSpawnOkayToSpawn() {
CreatureSpawnEvent e = new CreatureSpawnEvent(zombie, SpawnReason.NATURAL);
lml.onMobSpawn(e);
assertFalse(e.isCancelled());
}
/**
* Test method for {@link world.bentobox.bentobox.listeners.flags.worldsettings.LimitMobsListener#onMobSpawn(org.bukkit.event.entity.CreatureSpawnEvent)}.
*/
@Test
public void testOnMobSpawnJockey() {
CreatureSpawnEvent e = new CreatureSpawnEvent(jockey, SpawnReason.JOCKEY);
lml.onMobSpawn(e);
assertTrue(e.isCancelled());
}
}