-
-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
12171d3
commit 7a4b0f3
Showing
4 changed files
with
185 additions
and
8 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
160 changes: 160 additions & 0 deletions
160
src/test/java/com/github/minecraftschurlimods/potionbundles/test/PotionBundleTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,160 @@ | ||
| package com.github.minecraftschurlimods.potionbundles.test; | ||
|
|
||
| import com.github.minecraftschurlimods.potionbundles.*; | ||
| import net.minecraft.core.BlockPos; | ||
| import net.minecraft.gametest.framework.GameTest; | ||
| import net.minecraft.gametest.framework.GameTestAssertException; | ||
| import net.minecraft.gametest.framework.GameTestHelper; | ||
| import net.minecraft.gametest.framework.GameTestSequence; | ||
| import net.minecraft.world.InteractionHand; | ||
| import net.minecraft.world.effect.MobEffectInstance; | ||
| import net.minecraft.world.effect.MobEffects; | ||
| import net.minecraft.world.entity.EntityType; | ||
| import net.minecraft.world.entity.player.Inventory; | ||
| import net.minecraft.world.entity.player.Player; | ||
| import net.minecraft.world.entity.projectile.ThrownPotion; | ||
| import net.minecraft.world.item.ItemStack; | ||
| import net.minecraft.world.item.Items; | ||
| import net.minecraft.world.item.alchemy.PotionUtils; | ||
| import net.minecraft.world.item.alchemy.Potions; | ||
| import net.minecraft.world.phys.Vec3; | ||
| import net.minecraftforge.gametest.GameTestHolder; | ||
| import net.minecraftforge.gametest.PrefixGameTestTemplate; | ||
| import org.jetbrains.annotations.Contract; | ||
| import org.jetbrains.annotations.NotNull; | ||
|
|
||
| import java.util.List; | ||
|
|
||
| @PrefixGameTestTemplate(false) | ||
| @GameTestHolder(PotionBundles.MODID) | ||
| public class PotionBundleTest { | ||
| @GameTest(template = "empty_3x3") | ||
| public static void testPotionBundle(GameTestHelper helper) { | ||
| ItemStack string = new ItemStack(Items.STRING); | ||
| PotionBundle potionBundle = PotionBundles.POTION_BUNDLE.get(); | ||
| int maxUses = potionBundle.getMaxUses(); | ||
| MobEffectInstance customEffect = new MobEffectInstance(MobEffects.FIRE_RESISTANCE, 20); | ||
| ItemStack bundle = potionBundle.createStack(string, Potions.WATER, List.of(customEffect), null); | ||
| assertUses(bundle, maxUses, "Uses don't use items max uses"); | ||
| int duration = potionBundle.getUseDuration(bundle); | ||
| Player player = setupPlayer(helper, bundle); | ||
| GameTestSequence sequence = helper.startSequence(); | ||
| for (int i = 0; i < maxUses; i++) { | ||
| int index = i + 1; | ||
| sequence.thenExecute(() -> player.setItemInHand(InteractionHand.MAIN_HAND, bundle.use(helper.getLevel(), player, InteractionHand.MAIN_HAND).getObject())) | ||
| .thenExecuteAfter(duration + 1, () -> { | ||
| player.stopUsingItem(); | ||
| assertBundleUsed(player, index, maxUses, string, bundle); | ||
| assertItemsInInventory(player.getInventory(), new ItemStack(Items.GLASS_BOTTLE), index, "Did not return glass bottle"); | ||
| MobEffectInstance effect = player.getEffect(customEffect.getEffect()); | ||
| if (effect == null || effect.getDuration() + 1 != customEffect.getDuration()) { | ||
| fail("Did not apply effect"); | ||
| } | ||
| }); | ||
| } | ||
| sequence.thenSucceed(); | ||
| } | ||
|
|
||
| @GameTest(template = "empty_3x3") | ||
| public static void testSplashPotionBundle(GameTestHelper helper) { | ||
| AbstractThrowablePotionBundle potionBundle = PotionBundles.SPLASH_POTION_BUNDLE.get(); | ||
| testThrownPotionBundle(helper, potionBundle); | ||
| } | ||
|
|
||
| @GameTest(template = "empty_3x3") | ||
| public static void testLingeringPotionBundle(GameTestHelper helper) { | ||
| AbstractThrowablePotionBundle potionBundle = PotionBundles.LINGERING_POTION_BUNDLE.get(); | ||
| testThrownPotionBundle(helper, potionBundle); | ||
| } | ||
|
|
||
| private static void testThrownPotionBundle(GameTestHelper helper, AbstractThrowablePotionBundle potionBundle) { | ||
| ItemStack string = new ItemStack(Items.STRING); | ||
| MobEffectInstance customEffect = new MobEffectInstance(MobEffects.FIRE_RESISTANCE, 20); | ||
| int maxUses = potionBundle.getMaxUses(); | ||
| ItemStack bundle = potionBundle.createStack(string, Potions.WATER, List.of(customEffect), null); | ||
| assertUses(bundle, maxUses, "Uses don't use items max uses"); | ||
| Player player = setupPlayer(helper, bundle); | ||
| GameTestSequence sequence = helper.startSequence(); | ||
| for (int i = 0; i < maxUses; i++) { | ||
| int index = i + 1; | ||
| sequence.thenExecute(() -> player.setItemInHand(InteractionHand.MAIN_HAND, bundle.use(helper.getLevel(), player, InteractionHand.MAIN_HAND).getObject())) | ||
| .thenExecute(() -> { | ||
| assertBundleUsed(player, index, maxUses, string, bundle); | ||
| List<ThrownPotion> potions = helper.getEntities(EntityType.POTION, new BlockPos(1, 1, 1), 2); | ||
| if (potions.isEmpty()) { | ||
| fail("Did not throw potion"); | ||
| } | ||
| if (potions.size() > 1) { | ||
| fail("Threw too many potions"); | ||
| } | ||
| ThrownPotion thrownPotion = potions.get(0); | ||
| ItemStack thrownPotionItem = thrownPotion.getItem(); | ||
| List<MobEffectInstance> mobEffects = PotionUtils.getMobEffects(thrownPotionItem); | ||
| if (mobEffects.size() != 1 || !mobEffects.get(0).equals(customEffect)) { | ||
| fail("Wrong potion thrown"); | ||
| } | ||
| if (potionBundle == PotionBundles.SPLASH_POTION_BUNDLE.get()) { | ||
| if (!thrownPotionItem.is(Items.SPLASH_POTION)) { | ||
| helper.fail("Wrong potion type thrown"); | ||
| } | ||
| } | ||
| if (potionBundle == PotionBundles.LINGERING_POTION_BUNDLE.get()) { | ||
| if (!thrownPotionItem.is(Items.LINGERING_POTION)) { | ||
| helper.fail("Wrong potion type thrown"); | ||
| } | ||
| } | ||
| thrownPotion.discard(); | ||
| }); | ||
| } | ||
| sequence.thenSucceed(); | ||
| } | ||
|
|
||
| private static void assertBundleUsed(Player player, int index, int maxUses, ItemStack string, ItemStack bundle) { | ||
| ItemStack inHand = player.getItemInHand(InteractionHand.MAIN_HAND); | ||
| assertNotEmpty(inHand, "Bundle should not be empty"); | ||
| if (index == maxUses) { | ||
| assertSameItem(inHand, string, "Bundle should be empty and returned the string"); | ||
| } else { | ||
| assertSameItem(inHand, bundle, "Wrong item in hand"); | ||
| assertUses(inHand, maxUses - index, "Uses not decremented"); | ||
| } | ||
| } | ||
|
|
||
| private static void assertSameItem(ItemStack stack, ItemStack expectes, String message) { | ||
| if (!ItemStack.isSameItem(stack, expectes)) { | ||
| fail(message); | ||
| } | ||
| } | ||
|
|
||
| private static void assertNotEmpty(ItemStack stack, String message) { | ||
| if (stack.isEmpty()) { | ||
| fail(message); | ||
| } | ||
| } | ||
|
|
||
| private static void assertUses(ItemStack bundle, int expectedUses, String message) { | ||
| if (PotionBundleUtils.getUses(bundle) != expectedUses) { | ||
| fail(message); | ||
| } | ||
| } | ||
|
|
||
| private static void assertItemsInInventory(Inventory inventory, ItemStack item, int count, String message) { | ||
| int slot = inventory.findSlotMatchingItem(item); | ||
| if (slot == -1 || inventory.getItem(slot).getCount() != count) { | ||
| fail(message); | ||
| } | ||
| } | ||
|
|
||
| @Contract("_ -> fail") | ||
| private static void fail(String message) { | ||
| throw new GameTestAssertException(message); | ||
| } | ||
|
|
||
| private static @NotNull Player setupPlayer(GameTestHelper helper, ItemStack bundle) { | ||
| Player player = helper.makeMockSurvivalPlayer(); | ||
| player.setPos(helper.absoluteVec(Vec3.ZERO)); | ||
| helper.getLevel().addFreshEntity(player); | ||
| player.setItemInHand(InteractionHand.MAIN_HAND, bundle); | ||
| return player; | ||
| } | ||
| } |
Binary file not shown.