/*
* Copyright 2013 MovingBlocks
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.terasology.entitySystem;
import org.junit.Before;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.terasology.assets.AssetFactory;
import org.terasology.assets.management.AssetManager;
import org.terasology.assets.module.ModuleAwareAssetTypeManager;
import org.terasology.context.internal.ContextImpl;
import org.terasology.engine.bootstrap.EntitySystemSetupUtil;
import org.terasology.engine.module.ModuleManager;
import org.terasology.entitySystem.metadata.ComponentLibrary;
import org.terasology.entitySystem.prefab.Prefab;
import org.terasology.entitySystem.prefab.PrefabData;
import org.terasology.entitySystem.prefab.PrefabManager;
import org.terasology.entitySystem.prefab.internal.PojoPrefab;
import org.terasology.entitySystem.prefab.internal.PojoPrefabManager;
import org.terasology.entitySystem.prefab.internal.PrefabFormat;
import org.terasology.entitySystem.stubs.ListOfObjectComponent;
import org.terasology.entitySystem.stubs.MappedContainerComponent;
import org.terasology.entitySystem.stubs.OrderedMapTestComponent;
import org.terasology.entitySystem.stubs.StringComponent;
import org.terasology.network.NetworkMode;
import org.terasology.network.NetworkSystem;
import org.terasology.persistence.typeHandling.TypeSerializationLibrary;
import org.terasology.registry.CoreRegistry;
import org.terasology.testUtil.ModuleManagerFactory;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
/**
*/
public class PrefabTest {
private static final Logger logger = LoggerFactory.getLogger(PrefabTest.class);
private PrefabManager prefabManager;
@Before
public void setup() throws Exception {
ContextImpl context = new ContextImpl();
CoreRegistry.setContext(context);
ModuleManager moduleManager = ModuleManagerFactory.create();
context.put(ModuleManager.class, moduleManager);
EntitySystemSetupUtil.addReflectionBasedLibraries(context);
ModuleAwareAssetTypeManager assetTypeManager = new ModuleAwareAssetTypeManager();
assetTypeManager.registerCoreAssetType(Prefab.class,
(AssetFactory<Prefab, PrefabData>) PojoPrefab::new, "prefabs");
ComponentLibrary componentLibrary = context.get(ComponentLibrary.class);
TypeSerializationLibrary typeSerializationLibrary = context.get(TypeSerializationLibrary.class);
PrefabFormat prefabFormat = new PrefabFormat(componentLibrary, typeSerializationLibrary);
assetTypeManager.registerCoreFormat(Prefab.class, prefabFormat);
assetTypeManager.switchEnvironment(moduleManager.getEnvironment());
context.put(AssetManager.class, assetTypeManager.getAssetManager());
NetworkSystem networkSystem = mock(NetworkSystem.class);
when(networkSystem.getMode()).thenReturn(NetworkMode.NONE);
context.put(NetworkSystem.class, networkSystem);
EntitySystemSetupUtil.addEntityManagementRelatedClasses(context);
prefabManager = new PojoPrefabManager(context);
}
@Test
public void testGetSimplePrefab() {
Prefab prefab = prefabManager.getPrefab("unittest:simple");
assertNotNull(prefab);
assertEquals("unittest:simple", prefab.getName());
}
@Test
public void testPrefabHasDefinedComponents() {
Prefab prefab = prefabManager.getPrefab("unittest:withComponent");
assertTrue(prefab.hasComponent(StringComponent.class));
}
@Test
public void testPrefabHasDefinedComponentsWithOrderedMap() {
Prefab prefab = prefabManager.getPrefab("unittest:withComponentContainingOrderedMap");
assertTrue(prefab.hasComponent(OrderedMapTestComponent.class));
OrderedMapTestComponent component = prefab.getComponent(OrderedMapTestComponent.class);
assertNotNull(component);
Map<String, Long> orderedMap = component.orderedMap;
Set<String> keySet = orderedMap.keySet();
List<String> keyList = new ArrayList<>(keySet);
assertEquals(4, keyList.size());
assertEquals("one", keyList.get(0));
assertEquals("two", keyList.get(1));
assertEquals("three", keyList.get(2));
assertEquals("four", keyList.get(3));
assertEquals(Long.valueOf(1), orderedMap.get("one"));
assertEquals(Long.valueOf(2), orderedMap.get("two"));
assertEquals(Long.valueOf(3), orderedMap.get("three"));
assertEquals(Long.valueOf(4), orderedMap.get("four"));
}
@Test
public void testPrefabInheritsFromParent() {
Prefab prefab = prefabManager.getPrefab("unittest:inheritsComponent");
assertTrue(prefab.hasComponent(StringComponent.class));
}
@Test
public void testPrefabTransitiveInheritance() {
Prefab prefab = prefabManager.getPrefab("unittest:multilevelInheritance");
assertTrue(prefab.hasComponent(StringComponent.class));
}
@Test
public void testPrefabWithCollectionOfMappedContainers() {
Prefab prefab = prefabManager.getPrefab("unittest:withCollectionOfMappedContainers");
MappedContainerComponent mappedContainer = prefab.getComponent(MappedContainerComponent.class);
assertNotNull(mappedContainer);
assertNotNull(mappedContainer.containers);
assertEquals(1, mappedContainer.containers.size());
MappedContainerComponent.Cont cont = mappedContainer.containers.iterator().next();
assertNotNull(cont);
assertEquals("a", cont.value);
}
@Test
public void testPrefabWithListOfMappedContainers() {
Prefab prefab = prefabManager.getPrefab("unittest:withListContainer");
ListOfObjectComponent mappedContainer = prefab.getComponent(ListOfObjectComponent.class);
assertEquals(2, mappedContainer.elements.size());
assertEquals("returnHome", mappedContainer.elements.get(1).id);
}
}