Add more tests for HealthAuthoritySystem

Terasology · May 21, 2019 · a4d033f · a4d033f
1 parent 9b6f581
commit a4d033f
Showing 1 changed file with 62 additions and 0 deletions.
diff --git a/src/test/java/org/terasology/logic/health/HealEventTest.java b/src/test/java/org/terasology/logic/health/HealEventTest.java
@@ -23,11 +23,16 @@
 import org.terasology.engine.Time;
 import org.terasology.entitySystem.entity.EntityManager;
 import org.terasology.entitySystem.entity.EntityRef;
+import org.terasology.logic.health.event.BeforeHealEvent;
 import org.terasology.logic.health.event.DoHealEvent;
 import org.terasology.logic.players.PlayerCharacterComponent;
 import org.terasology.moduletestingenvironment.ModuleTestingEnvironment;
+import org.terasology.moduletestingenvironment.TestEventReceiver;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+import java.util.List;
 import java.util.Set;
 
 public class HealEventTest extends ModuleTestingEnvironment {
@@ -67,4 +72,61 @@ public void healTest() {
         assertEquals(player.getComponent(HealthComponent.class).currentHealth, 100);
     }
 
+    @Test
+    public void healEventSentTest() {
+        TestEventReceiver<BeforeHealEvent> receiver = new TestEventReceiver<>(getHostContext(), BeforeHealEvent.class);
+        List<BeforeHealEvent> list = receiver.getEvents();
+        assertTrue(list.isEmpty());
+
+        HealthComponent healthComponent = new HealthComponent();
+        healthComponent.currentHealth = 0;
+        healthComponent.maxHealth = 100;
+
+        final EntityRef player = entityManager.create();
+        player.addComponent(new PlayerCharacterComponent());
+        player.addComponent(healthComponent);
+
+        player.send(new DoHealEvent(10));
+        assertEquals(1, list.size());
+    }
+
+    @Test
+    public void healEventCancelTest() {
+        HealthComponent healthComponent = new HealthComponent();
+        healthComponent.currentHealth = 0;
+        healthComponent.maxHealth = 100;
+
+        final EntityRef player = entityManager.create();
+        player.addComponent(new PlayerCharacterComponent());
+        player.addComponent(healthComponent);
+
+        TestEventReceiver<BeforeHealEvent> receiver = new TestEventReceiver<>(getHostContext(), BeforeHealEvent.class,
+        (event, entity) -> {
+            event.consume();
+        });
+        player.send(new DoHealEvent(10));
+        assertEquals(0, player.getComponent(HealthComponent.class).currentHealth);
+    }
+
+    @Test
+    public void healModifyEventTest() {
+        HealthComponent healthComponent = new HealthComponent();
+        healthComponent.currentHealth = 0;
+        healthComponent.maxHealth = 100;
+
+        final EntityRef player = entityManager.create();
+        player.addComponent(new PlayerCharacterComponent());
+        player.addComponent(healthComponent);
+
+        TestEventReceiver<BeforeHealEvent> receiver = new TestEventReceiver<>(getHostContext(), BeforeHealEvent.class,
+                (event, entity) -> {
+                    event.add(10);
+                    event.subtract(5);
+                    event.multiply(2);
+                });
+        // expected = ( initial:10 + 10 - 5 ) * 2 = 30
+        player.send(new DoHealEvent(10));
+        assertEquals(30, player.getComponent(HealthComponent.class).currentHealth);
+    }
+
 }