From c41488139136c213fd5d799523da5f318a0014a2 Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Wed, 27 Nov 2024 16:06:52 +0100
Subject: feat: Add item type function

---
 src/test/kotlin/util/skyblock/ItemTypeTest.kt | 53 +++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100644 src/test/kotlin/util/skyblock/ItemTypeTest.kt

(limited to 'src/test/kotlin/util/skyblock')

diff --git a/src/test/kotlin/util/skyblock/ItemTypeTest.kt b/src/test/kotlin/util/skyblock/ItemTypeTest.kt
new file mode 100644
index 0000000..c89e57d
--- /dev/null
+++ b/src/test/kotlin/util/skyblock/ItemTypeTest.kt
@@ -0,0 +1,53 @@
+package moe.nea.firmament.test.util.skyblock
+
+import io.kotest.core.spec.style.AnnotationSpec
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.Test
+import moe.nea.firmament.test.testutil.ItemResources
+import moe.nea.firmament.util.skyblock.ItemType
+
+class ItemTypeTest : AnnotationSpec() {
+	@Test
+	fun testPetItem() {
+		Assertions.assertEquals(
+			ItemType.PET,
+			ItemType.fromItemStack(ItemResources.loadItem("pets/lion-item"))
+		)
+	}
+
+	@Test
+	fun testPetInUI() {
+		Assertions.assertEquals(
+			ItemType.PET,
+			ItemType.fromItemStack(ItemResources.loadItem("pets/rabbit-selected"))
+		)
+		Assertions.assertEquals(
+			ItemType.PET,
+			ItemType.fromItemStack(ItemResources.loadItem("pets/mithril-golem-not-selected"))
+		)
+	}
+
+	@Test
+	fun testAOTV() {
+		Assertions.assertEquals(
+			ItemType.SWORD,
+			ItemType.fromItemStack(ItemResources.loadItem("aspect-of-the-void"))
+		)
+	}
+
+	@Test
+	fun testDrill() {
+		Assertions.assertEquals(
+			ItemType.DRILL,
+			ItemType.fromItemStack(ItemResources.loadItem("titanium-drill"))
+		)
+	}
+
+	@Test
+	fun testPickaxe() {
+		Assertions.assertEquals(
+			ItemType.PICKAXE,
+			ItemType.fromItemStack(ItemResources.loadItem("diamond-pickaxe"))
+		)
+	}
+}
-- 
cgit