107 lines
4.1 KiB
Java
107 lines
4.1 KiB
Java
package com.comphenix.protocol.wrappers;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
import com.comphenix.protocol.BukkitInitialization;
|
|
import com.comphenix.protocol.PacketType;
|
|
import com.comphenix.protocol.events.PacketContainer;
|
|
import com.comphenix.protocol.wrappers.WrappedAttributeModifier.Operation;
|
|
import com.google.common.collect.Lists;
|
|
import net.minecraft.core.IRegistry;
|
|
import net.minecraft.core.registries.BuiltInRegistries;
|
|
import net.minecraft.network.protocol.game.PacketPlayOutUpdateAttributes.AttributeSnapshot;
|
|
import net.minecraft.resources.MinecraftKey;
|
|
import net.minecraft.world.entity.ai.attributes.AttributeBase;
|
|
import net.minecraft.world.entity.ai.attributes.AttributeModifier;
|
|
import org.junit.jupiter.api.BeforeAll;
|
|
import org.junit.jupiter.api.BeforeEach;
|
|
import org.junit.jupiter.api.Test;
|
|
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
import static org.junit.jupiter.api.Assertions.assertNotSame;
|
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
|
|
|
public class WrappedAttributeTest {
|
|
|
|
private WrappedAttributeModifier doubleModifier;
|
|
private WrappedAttributeModifier constantModifier;
|
|
private WrappedAttribute attribute;
|
|
|
|
@BeforeAll
|
|
public static void initializeBukkit() {
|
|
BukkitInitialization.initializeAll();
|
|
}
|
|
|
|
@BeforeEach
|
|
public void setUp() {
|
|
// Create a couple of modifiers
|
|
this.doubleModifier =
|
|
WrappedAttributeModifier.newBuilder().
|
|
name("Double Damage").
|
|
amount(1).
|
|
operation(Operation.ADD_PERCENTAGE).
|
|
build();
|
|
this.constantModifier =
|
|
WrappedAttributeModifier.newBuilder().
|
|
name("Damage Bonus").
|
|
amount(5).
|
|
operation(Operation.ADD_NUMBER).
|
|
build();
|
|
|
|
// Create attribute
|
|
this.attribute = WrappedAttribute.newBuilder().
|
|
attributeKey("generic.attackDamage").
|
|
baseValue(2).
|
|
packet(new PacketContainer(PacketType.Play.Server.UPDATE_ATTRIBUTES)).
|
|
modifiers(Lists.newArrayList(this.constantModifier, this.doubleModifier)).
|
|
build();
|
|
}
|
|
|
|
@Test
|
|
public void testEquality() {
|
|
// Check wrapped equality
|
|
assertEquals(this.doubleModifier, this.doubleModifier);
|
|
assertNotSame(this.constantModifier, this.doubleModifier);
|
|
|
|
assertEquals(this.doubleModifier.getHandle(), this.getModifierCopy(this.doubleModifier));
|
|
assertEquals(this.constantModifier.getHandle(), this.getModifierCopy(this.constantModifier));
|
|
}
|
|
|
|
@Test
|
|
public void testAttribute() {
|
|
assertEquals(this.attribute, WrappedAttribute.fromHandle(this.getAttributeCopy(this.attribute)));
|
|
|
|
assertTrue(this.attribute.hasModifier(this.doubleModifier.getUUID()));
|
|
assertTrue(this.attribute.hasModifier(this.constantModifier.getUUID()));
|
|
}
|
|
|
|
@Test
|
|
public void testFromTemplate() {
|
|
assertEquals(this.attribute, WrappedAttribute.newBuilder(this.attribute).build());
|
|
}
|
|
|
|
/**
|
|
* Retrieve the equivalent NMS attribute.
|
|
*
|
|
* @param attribute - the wrapped attribute.
|
|
* @return The equivalent NMS attribute.
|
|
*/
|
|
private AttributeSnapshot getAttributeCopy(WrappedAttribute attribute) {
|
|
List<AttributeModifier> modifiers = new ArrayList<>();
|
|
|
|
for (WrappedAttributeModifier wrapper : attribute.getModifiers()) {
|
|
modifiers.add((AttributeModifier) wrapper.getHandle());
|
|
}
|
|
|
|
IRegistry<AttributeBase> registry = BuiltInRegistries.u;
|
|
AttributeBase base = registry.a(MinecraftKey.a(attribute.getAttributeKey()));
|
|
return new AttributeSnapshot(base, attribute.getBaseValue(), modifiers);
|
|
}
|
|
|
|
private AttributeModifier getModifierCopy(WrappedAttributeModifier modifier) {
|
|
AttributeModifier.Operation operation = AttributeModifier.Operation.values()[modifier.getOperation().getId()];
|
|
return new AttributeModifier(modifier.getUUID(), modifier.getName(), modifier.getAmount(), operation);
|
|
}
|
|
}
|