org.powernukkit.tests.junit.jupiter.PowerNukkitExtension Maven / Gradle / Ivy
/*
* PowerNukkit JUnit 5 Testing Framework
* Copyright (C) 2020 José Roberto de Araújo Júnior
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see > levels = new LinkedHashMap<>();
Map> players = new LinkedHashMap<>();
List entities = new ArrayList<>();
void releaseResources() {
serverMocker.releaseResources();
}
}
@MockPlayer
@MockLevel
@MockEntity
private static final Void defaults = null;
@Override
public void beforeAll(ExtensionContext context) throws Exception {
MockServer config = context.getRequiredTestClass().getAnnotation(MockServer.class);
if (config == null) {
config = PowerNukkitExtension.class.getAnnotation(MockServer.class);
}
Session session = new Session();
ServerMocker serverMocker = new ServerMocker(config);
Server server = serverMocker.create();
session.serverMocker = serverMocker;
context.getStore(POWERNUKKIT).put(SESSION, session);
serverMocker.setActive();
initStatics(server);
}
@Override
public void postProcessTestInstance(Object testInstance, ExtensionContext context) throws Exception {
Session session = context.getStore(POWERNUKKIT).get(SESSION, Session.class);
Map> levels = session.levels;
for (Field field : AnnotationSupport.findAnnotatedFields(context.getRequiredTestClass(), MockLevel.class)) {
if (field.isAnnotationPresent(MockLevel.class) && field.getType().isAssignableFrom(Level.class)) {
LevelMocker levelMocker = new LevelMocker(field.getAnnotation(MockLevel.class));
Level level = levelMocker.create();
levels.computeIfAbsent(levelMocker.getLevelName(), it-> new ArrayList<>()).add(levelMocker);
setField(testInstance, field, level);
}
}
Function levelProvider = name-> {
if (name.isEmpty()) {
if (!levels.isEmpty()) {
return levels.values().iterator().next().get(0).getLevel();
}
LevelMocker levelMocker = new LevelMocker();
Level level = levelMocker.create();
List list = new ArrayList<>();
list.add(levelMocker);
levels.put(levelMocker.getLevelName(), list);
return level;
}
List list = levels.computeIfAbsent(name, it-> new ArrayList<>());
if (list.size() > 1) {
throw new IllegalStateException("Multiple level options: "+list);
} else if (list.size() == 1) {
return list.get(0).getLevel();
}
LevelMocker levelMocker = new LevelMocker(name);
Level level = levelMocker.create();
list.add(levelMocker);
return level;
};
for (Field field: AnnotationSupport.findAnnotatedFields(context.getRequiredTestClass(), MockEntity.class)) {
Class type = field.getType();
Class entityType;
if (!Entity.class.isAssignableFrom(type)) {
throw new AssertionError("@MockEntity must be used on fields of type Entity or higher.");
}
MockEntity annotation = field.getAnnotation(MockEntity.class);
if (!annotation.type().equals(Entity.class)) {
entityType = annotation.type();
} else {
//noinspection unchecked
entityType = (Class) field.getType();
}
EntityMocker entityMocker = new EntityMocker(levelProvider, entityType, annotation);
entityMocker.prepare();
entityMocker.createAndSet(testInstance, field);
session.entities.add(entityMocker);
}
for (Field field: AnnotationSupport.findAnnotatedFields(context.getRequiredTestClass(), MockPlayer.class)) {
PlayerMocker playerMocker = new PlayerMocker(levelProvider, field.getAnnotation(MockPlayer.class));
playerMocker.prepare();
playerMocker.setInstance(testInstance);
playerMocker.setInstanceField(field);
session.players.computeIfAbsent(playerMocker.getPlayerName(), it-> new ArrayList<>()).add(playerMocker);
}
Map loadedLevels;
if (session.serverMocker.getConfig().callsRealMethods()) {
loadedLevels = session.serverMocker.getServer().getLevels();
} else {
loadedLevels = new LinkedHashMap<>();
}
levels.values().stream().flatMap(Collection::stream)
.map(LevelMocker::getLevel)
.forEachOrdered(mock-> loadedLevels.put(mock.getId(), mock));
Level defaultLevel = levels.values().stream().flatMap(Collection::stream)
.filter(mocker-> mocker.getConfig().isDefault())
.map(LevelMocker::getLevel)
.findFirst()
.orElseGet(()-> levels.values().stream().flatMap(Collection::stream).findFirst().map(LevelMocker::getLevel).orElse(null));
if (!session.serverMocker.getConfig().callsRealMethods()) {
Server server = session.serverMocker.getServer();
lenient().when(server.getLevel(anyInt())).thenAnswer(call-> loadedLevels.get(call.getArgument(0, Integer.class)));
lenient().when(server.getLevelByName(anyString())).thenAnswer(call-> loadedLevels.values().stream()
.filter(level-> level.getName().equals(call.getArgument(0)))
.findFirst().orElse(null));
lenient().when(server.isLevelLoaded(anyString())).thenAnswer(call-> loadedLevels.values().stream()
.anyMatch(level-> level.getName().equals(call.getArgument(0))));
lenient().when(server.getLevels()).thenReturn(loadedLevels);
}
session.serverMocker.getServer().setDefaultLevel(defaultLevel);
session.players.values().stream().flatMap(Collection::stream)
.forEachOrdered(playerMocker -> execute(playerMocker::recreate));
}
@Override
public void afterEach(ExtensionContext context) {
try {
context.getStore(POWERNUKKIT).get(SESSION, Session.class).releaseResources();
} finally {
super.afterEach(context);
}
}
@Override
public void afterAll(ExtensionContext context) throws Exception {
context.getStore(POWERNUKKIT).remove(SESSION, Session.class).releaseResources();
ServerMocker.setServerInstance(null);
}
private void initStatics(Server server) {
execute(()-> {
Method method = Server.class.getDeclaredMethod("registerEntities");
method.setAccessible(true);
method.invoke(server);
method = Server.class.getDeclaredMethod("registerBlockEntities");
method.setAccessible(true);
method.invoke(server);
});
Block.init();
Enchantment.init();
Potion.init();
Effect.init();
Attribute.init();
Item.init();
EnumBiome.values(); //load class, this also registers biomes
DispenseBehaviorRegister.init();
GlobalBlockPalette.getOrCreateRuntimeId(0, 0); //Force it to load
LevelProviderManager.addProvider(null, Anvil.class);
Generator.addGenerator(Flat.class, "flat", Generator.TYPE_FLAT);
Generator.addGenerator(Normal.class, "normal", Generator.TYPE_INFINITE);
Generator.addGenerator(Normal.class, "default", Generator.TYPE_INFINITE);
Generator.addGenerator(Nether.class, "nether", Generator.TYPE_NETHER);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy