101 lines
3.4 KiB
Java
101 lines
3.4 KiB
Java
package com.comphenix.protocol.wrappers;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertNotSame;
|
|
import static org.junit.Assert.assertTrue;
|
|
|
|
import java.util.List;
|
|
|
|
import net.minecraft.server.v1_16_R2.AttributeBase;
|
|
import net.minecraft.server.v1_16_R2.AttributeModifier;
|
|
import net.minecraft.server.v1_16_R2.IRegistry;
|
|
import net.minecraft.server.v1_16_R2.MinecraftKey;
|
|
import net.minecraft.server.v1_16_R2.PacketPlayOutUpdateAttributes;
|
|
import net.minecraft.server.v1_16_R2.PacketPlayOutUpdateAttributes.AttributeSnapshot;
|
|
|
|
import org.junit.Before;
|
|
import org.junit.BeforeClass;
|
|
import org.junit.Test;
|
|
|
|
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;
|
|
|
|
public class WrappedAttributeTest {
|
|
private WrappedAttributeModifier doubleModifier;
|
|
private WrappedAttributeModifier constantModifier;
|
|
private WrappedAttribute attribute;
|
|
|
|
@BeforeClass
|
|
public static void initializeBukkit() {
|
|
BukkitInitialization.initializeItemMeta();
|
|
}
|
|
|
|
@Before
|
|
public void setUp() {
|
|
// Create a couple of modifiers
|
|
doubleModifier =
|
|
WrappedAttributeModifier.newBuilder().
|
|
name("Double Damage").
|
|
amount(1).
|
|
operation(Operation.ADD_PERCENTAGE).
|
|
build();
|
|
constantModifier =
|
|
WrappedAttributeModifier.newBuilder().
|
|
name("Damage Bonus").
|
|
amount(5).
|
|
operation(Operation.ADD_NUMBER).
|
|
build();
|
|
|
|
// Create attribute
|
|
attribute = WrappedAttribute.newBuilder().
|
|
attributeKey("generic.attackDamage").
|
|
baseValue(2).
|
|
packet(new PacketContainer(PacketType.Play.Server.UPDATE_ATTRIBUTES)).
|
|
modifiers(Lists.newArrayList(constantModifier, doubleModifier)).
|
|
build();
|
|
}
|
|
|
|
@Test
|
|
public void testEquality() {
|
|
// Check wrapped equality
|
|
assertEquals(doubleModifier, doubleModifier);
|
|
assertNotSame(constantModifier, doubleModifier);
|
|
|
|
assertEquals(doubleModifier.getHandle(), getModifierCopy(doubleModifier));
|
|
assertEquals(constantModifier.getHandle(), getModifierCopy(constantModifier));
|
|
}
|
|
|
|
@Test
|
|
public void testAttribute() {
|
|
assertEquals(attribute, WrappedAttribute.fromHandle(getAttributeCopy(attribute)));
|
|
|
|
assertTrue(attribute.hasModifier(doubleModifier.getUUID()));
|
|
assertTrue(attribute.hasModifier(constantModifier.getUUID()));
|
|
}
|
|
|
|
/**
|
|
* Retrieve the equivalent NMS attribute.
|
|
* @param attribute - the wrapped attribute.
|
|
* @return The equivalent NMS attribute.
|
|
*/
|
|
private AttributeSnapshot getAttributeCopy(WrappedAttribute attribute) {
|
|
List<AttributeModifier> modifiers = Lists.newArrayList();
|
|
|
|
for (WrappedAttributeModifier wrapper : attribute.getModifiers()) {
|
|
modifiers.add((AttributeModifier) wrapper.getHandle());
|
|
}
|
|
|
|
PacketPlayOutUpdateAttributes accessor = new PacketPlayOutUpdateAttributes();
|
|
AttributeBase base = IRegistry.ATTRIBUTE.get(MinecraftKey.a(attribute.getAttributeKey()));
|
|
return accessor.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);
|
|
}
|
|
}
|