56 lines
2.1 KiB
Java
56 lines
2.1 KiB
Java
package com.comphenix.protocol.injector;
|
|
|
|
import com.comphenix.protocol.BukkitInitialization;
|
|
import com.comphenix.protocol.reflect.accessors.Accessors;
|
|
|
|
import net.minecraft.server.v1_16_R2.ChunkProviderServer;
|
|
import net.minecraft.server.v1_16_R2.Entity;
|
|
import net.minecraft.server.v1_16_R2.PlayerChunkMap;
|
|
import net.minecraft.server.v1_16_R2.PlayerChunkMap.EntityTracker;
|
|
import net.minecraft.server.v1_16_R2.WorldServer;
|
|
|
|
import org.bukkit.craftbukkit.libs.it.unimi.dsi.fastutil.ints.Int2ObjectMap;
|
|
import org.bukkit.craftbukkit.libs.it.unimi.dsi.fastutil.ints.Int2ObjectOpenHashMap;
|
|
import org.bukkit.craftbukkit.v1_16_R2.CraftWorld;
|
|
import org.bukkit.craftbukkit.v1_16_R2.entity.CraftEntity;
|
|
import org.junit.BeforeClass;
|
|
import org.junit.Test;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.mockito.Mockito.mock;
|
|
import static org.mockito.Mockito.when;
|
|
|
|
public class EntityUtilitiesTest {
|
|
|
|
@BeforeClass
|
|
public static void beforeClass() {
|
|
BukkitInitialization.initializeItemMeta();
|
|
}
|
|
|
|
@Test
|
|
public void testReflection() {
|
|
CraftWorld bukkit = mock(CraftWorld.class);
|
|
WorldServer world = mock(WorldServer.class);
|
|
when(bukkit.getHandle()).thenReturn(world);
|
|
|
|
ChunkProviderServer provider = mock(ChunkProviderServer.class);
|
|
when(world.getChunkProvider()).thenReturn(provider);
|
|
|
|
PlayerChunkMap chunkMap = mock(PlayerChunkMap.class);
|
|
Accessors.getFieldAccessor(ChunkProviderServer.class, "playerChunkMap", true).set(provider, chunkMap);
|
|
|
|
CraftEntity bukkitEntity = mock(CraftEntity.class);
|
|
Entity fakeEntity = mock(Entity.class);
|
|
when(fakeEntity.getBukkitEntity()).thenReturn(bukkitEntity);
|
|
|
|
EntityTracker tracker = mock(EntityTracker.class);
|
|
Accessors.getFieldAccessor(EntityTracker.class, "tracker", true).set(tracker, fakeEntity);
|
|
|
|
Int2ObjectMap<EntityTracker> trackerMap = new Int2ObjectOpenHashMap<>();
|
|
trackerMap.put(1, tracker);
|
|
Accessors.getFieldAccessor(PlayerChunkMap.class, "trackedEntities", true).set(chunkMap, trackerMap);
|
|
|
|
assertEquals(bukkitEntity, EntityUtilities.getInstance().getEntityFromID(bukkit, 1));
|
|
}
|
|
}
|