From b6f5c7b7a148344163b9d3a295d6ae0411adb370 Mon Sep 17 00:00:00 2001
From: alexcrea <contact@alexcrea.xyz>
Date: Sat, 5 Jul 2025 04:03:11 +0200
Subject: [PATCH] linear xp cost pre anvil unit test

---
 .../api/CustomAnvilRecipeApiTests.java        | 86 +++++++++++++++++++
 1 file changed, 86 insertions(+)

diff --git a/src/test/java/xyz/alexcrea/cuanvil/api/CustomAnvilRecipeApiTests.java b/src/test/java/xyz/alexcrea/cuanvil/api/CustomAnvilRecipeApiTests.java
index c93b7e7..5e9e716 100644
--- a/src/test/java/xyz/alexcrea/cuanvil/api/CustomAnvilRecipeApiTests.java
+++ b/src/test/java/xyz/alexcrea/cuanvil/api/CustomAnvilRecipeApiTests.java
@@ -134,6 +134,92 @@ public class CustomAnvilRecipeApiTests extends ConfigResetCustomAnvilTest {
         AnvilFuseTestUtil.executeAnvilFuseTest(anvil, player, legalResultData2);
     }
 
+    @Test
+    public void testLinearXpCost() {
+        String recipeName = "stick_recipe";
+        ItemStack stick = new ItemStackMock(Material.STICK);
+        ItemStack stick2 = new ItemStackMock(Material.STICK, 2);
+        ItemStack stick5 = new ItemStackMock(Material.STICK, 5);
+        ItemStack stick10 = new ItemStackMock(Material.STICK, 10);
+
+        AnvilFuseTestData nullResultData = new AnvilFuseTestData(
+                stick, stick,
+                null
+        );
+
+        AnvilFuseTestData legalResultData1 = new AnvilFuseTestData(
+                stick, stick,
+                null, stick2, null,
+                1,
+                null, null
+        );
+
+        AnvilFuseTestData legalResultData2 = new AnvilFuseTestData(
+                stick5, stick,
+                null, stick10, null,
+                4,
+                null, null
+        );
+
+        AnvilFuseTestUtil.executeAnvilFuseTest(anvil, player, nullResultData);
+
+        AnvilRecipeBuilder builder = new AnvilRecipeBuilder(recipeName);
+        builder.setExactCount(false)
+                .setLeftItem(stick)
+                .setResultItem(stick2)
+                .setLevelCostPerCraft(0)
+                .setLinearXpCostPerCraft(10);
+
+        assertTrue(builder.registerIfAbsent());
+
+        // Now working test
+        AnvilFuseTestUtil.executeAnvilFuseTest(anvil, player, legalResultData1);
+        AnvilFuseTestUtil.executeAnvilFuseTest(anvil, player, legalResultData2);
+    }
+
+    @Test
+    public void testLinearXpCostRemoveExact() {
+        String recipeName = "stick_recipe";
+        ItemStack stick = new ItemStackMock(Material.STICK);
+        ItemStack stick2 = new ItemStackMock(Material.STICK, 2);
+        ItemStack stick5 = new ItemStackMock(Material.STICK, 5);
+        ItemStack stick10 = new ItemStackMock(Material.STICK, 10);
+
+        AnvilFuseTestData nullResultData = new AnvilFuseTestData(
+                stick, stick,
+                null
+        );
+
+        AnvilFuseTestData legalResultData1 = new AnvilFuseTestData(
+                stick, stick,
+                null, stick2, null,
+                2,
+                null, null
+        );
+
+        AnvilFuseTestData legalResultData2 = new AnvilFuseTestData(
+                stick5, stick,
+                null, stick10, null,
+                5,
+                null, null
+        );
+
+        AnvilFuseTestUtil.executeAnvilFuseTest(anvil, player, nullResultData);
+
+        AnvilRecipeBuilder builder = new AnvilRecipeBuilder(recipeName);
+        builder.setExactCount(false)
+                .setLeftItem(stick)
+                .setResultItem(stick2)
+                .setLinearXpCostPerCraft(10)
+                .setRemoveExactLinearXp(true);
+
+        assertTrue(builder.registerIfAbsent());
+
+        // Now working test
+        AnvilFuseTestUtil.executeAnvilFuseTest(anvil, player, legalResultData1);
+        AnvilFuseTestUtil.executeAnvilFuseTest(anvil, player, legalResultData2);
+    }
+
     @Nullable
     public static AnvilCustomRecipe getByName(String name){
         for (AnvilCustomRecipe registeredRecipe : CustomAnvilRecipeApi.getRegisteredRecipes()) {