feat: Add item type function

1 files changed, 53 insertions, 0 deletions

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"))
+		)
+	}
+}