Search in sources :

Example 6 with HealEntityEvent

use of org.spongepowered.api.event.entity.HealEntityEvent in project SpongeAPI by SpongePowered.

the class SpongeAbstractHealEntityEventTest method testSetModifiers.

@Test
public void testSetModifiers() {
    Entity targetEntity = mockParam(Entity.class);
    final int originalDamage = 1;
    final int originalFinalDamage = 18;
    final int firstModifierDamage = 2;
    final int secondModifierDamage = 15;
    final int firstChangedDamage = 1;
    final int secondChangedDamage = 10;
    final int modifiedFinalDamage = 12;
    HealthModifier firstModifer = mockParam(HealthModifier.class);
    HealthModifier secondModifier = mockParam(HealthModifier.class);
    List<HealthFunction> originalFunctions = Lists.newArrayList(HealthFunction.of(firstModifer, p -> p * 2), HealthFunction.of(secondModifier, p -> p * 5));
    HealEntityEvent event = SpongeEventFactory.createHealEntityEvent(Cause.of(EventContext.empty(), "none"), originalFunctions, targetEntity, originalDamage);
    assertThat(event.getOriginalFunctions(), is(Matchers.equalTo(originalFunctions)));
    DoubleUnaryOperator newFunction = p -> p;
    event.setHealAmount(firstModifer, newFunction);
    assertThat(event.getHealAmount(firstModifer), is(closeTo(firstChangedDamage, ERROR)));
    assertThat(event.getHealAmount(secondModifier), is(closeTo(secondChangedDamage, ERROR)));
    assertThat(event.getOriginalHealingModifierAmount(firstModifer), is(closeTo(firstModifierDamage, ERROR)));
    assertThat(event.getOriginalHealingModifierAmount(secondModifier), is(closeTo(secondModifierDamage, ERROR)));
    assertThat(event.getOriginalHealAmount(), is(closeTo(originalDamage, ERROR)));
    assertThat(event.getOriginalFinalHealAmount(), is(closeTo(originalFinalDamage, ERROR)));
    assertThat(event.getFinalHealAmount(), is(closeTo(modifiedFinalDamage, ERROR)));
    assertThat(event.getOriginalFunctions(), is(Matchers.equalTo(originalFunctions)));
    assertThat(event.getModifiers(), is(Matchers.equalTo(Lists.newArrayList(HealthFunction.of(firstModifer, newFunction), originalFunctions.get(1)))));
}
Also used : CoreMatchers.is(org.hamcrest.CoreMatchers.is) Matchers(org.hamcrest.Matchers) Test(org.junit.Test) Entity(org.spongepowered.api.entity.Entity) HealEntityEvent(org.spongepowered.api.event.entity.HealEntityEvent) DoubleUnaryOperator(java.util.function.DoubleUnaryOperator) HealthFunction(org.spongepowered.api.event.cause.entity.health.HealthFunction) HealthModifier(org.spongepowered.api.event.cause.entity.health.HealthModifier) Assert.assertThat(org.junit.Assert.assertThat) Cause(org.spongepowered.api.event.cause.Cause) List(java.util.List) Lists(com.google.common.collect.Lists) Matchers.closeTo(org.hamcrest.Matchers.closeTo) Map(java.util.Map) EventContext(org.spongepowered.api.event.cause.EventContext) Entity(org.spongepowered.api.entity.Entity) HealEntityEvent(org.spongepowered.api.event.entity.HealEntityEvent) HealthModifier(org.spongepowered.api.event.cause.entity.health.HealthModifier) HealthFunction(org.spongepowered.api.event.cause.entity.health.HealthFunction) DoubleUnaryOperator(java.util.function.DoubleUnaryOperator) Test(org.junit.Test)

Example 7 with HealEntityEvent

use of org.spongepowered.api.event.entity.HealEntityEvent in project SpongeAPI by SpongePowered.

the class SpongeAbstractHealEntityEventTest method testNotApplicableModifer.

@Test(expected = IllegalArgumentException.class)
public void testNotApplicableModifer() {
    HealEntityEvent event = SpongeEventFactory.createHealEntityEvent(Cause.of(EventContext.empty(), "none"), Lists.newArrayList(), mockParam(Entity.class), 0);
    HealthModifier modifier = mockParam(HealthModifier.class);
    assertThat(event.isModifierApplicable(modifier), is(false));
    event.getOriginalHealingModifierAmount(modifier);
}
Also used : HealEntityEvent(org.spongepowered.api.event.entity.HealEntityEvent) Entity(org.spongepowered.api.entity.Entity) HealthModifier(org.spongepowered.api.event.cause.entity.health.HealthModifier) Test(org.junit.Test)

Example 8 with HealEntityEvent

use of org.spongepowered.api.event.entity.HealEntityEvent in project LanternServer by LanternPowered.

the class LanternEntity method heal.

/**
 * Heals the entity for the specified amount.
 *
 * <p>Will not heal them if they are dead and will not set
 * them above their maximum health.</p>
 *
 * @param amount The amount to heal for
 * @param source The healing source
 */
public boolean heal(double amount, HealingSource source) {
    if (isDead()) {
        return false;
    }
    final MutableBoundedValue<Double> health = getValue(Keys.HEALTH).orElse(null);
    if (health == null || health.get() >= health.getMaxValue()) {
        return false;
    }
    final CauseStack causeStack = CauseStack.current();
    try (CauseStack.Frame frame = causeStack.pushCauseFrame()) {
        frame.pushCause(source);
        frame.addContext(LanternEventContextKeys.HEALING_TYPE, source.getHealingType());
        final HealEntityEvent event = SpongeEventFactory.createHealEntityEvent(frame.getCurrentCause(), new ArrayList<>(), this, amount);
        Sponge.getEventManager().post(event);
        if (event.isCancelled()) {
            return false;
        }
        amount = event.getFinalHealAmount();
        if (amount > 0) {
            offer(Keys.HEALTH, Math.min(health.get() + amount, health.getMaxValue()));
        }
    }
    return true;
}
Also used : CauseStack(org.lanternpowered.server.event.CauseStack) HealEntityEvent(org.spongepowered.api.event.entity.HealEntityEvent)

Aggregations

HealEntityEvent (org.spongepowered.api.event.entity.HealEntityEvent)8 Test (org.junit.Test)7 Entity (org.spongepowered.api.entity.Entity)7 HealthModifier (org.spongepowered.api.event.cause.entity.health.HealthModifier)5 Lists (com.google.common.collect.Lists)4 List (java.util.List)4 Map (java.util.Map)4 DoubleUnaryOperator (java.util.function.DoubleUnaryOperator)4 CoreMatchers.is (org.hamcrest.CoreMatchers.is)4 Matchers (org.hamcrest.Matchers)4 Matchers.closeTo (org.hamcrest.Matchers.closeTo)4 Assert.assertThat (org.junit.Assert.assertThat)4 Cause (org.spongepowered.api.event.cause.Cause)4 EventContext (org.spongepowered.api.event.cause.EventContext)4 HealthFunction (org.spongepowered.api.event.cause.entity.health.HealthFunction)4 CauseStack (org.lanternpowered.server.event.CauseStack)1