1 files changed, 79 insertions, 0 deletions
diff --git a/src/test/kotlin/util/skyblock/AbilityUtilsTest.kt b/src/test/kotlin/util/skyblock/AbilityUtilsTest.kt
new file mode 100644
index 0000000..abe739d
--- /dev/null
+++ b/src/test/kotlin/util/skyblock/AbilityUtilsTest.kt
@@ -0,0 +1,79 @@
+package moe.nea.firmament.test.util.skyblock
+
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.Test
+import kotlin.time.Duration.Companion.minutes
+import kotlin.time.Duration.Companion.seconds
+import net.minecraft.text.Text
+import moe.nea.firmament.test.testutil.ItemResources
+import moe.nea.firmament.util.skyblock.AbilityUtils
+import moe.nea.firmament.util.unformattedString
+
+class AbilityUtilsTest {
+
+	fun List<AbilityUtils.ItemAbility>.stripDescriptions() = map {
+		it.copy(descriptionLines = it.descriptionLines.map { Text.literal(it.unformattedString) })
+	}
+
+	@Test
+	fun testUnpoweredDrill() {
+		Assertions.assertEquals(
+			listOf(
+				AbilityUtils.ItemAbility(
+					"Pickobulus",
+					false,
+					AbilityUtils.AbilityActivation.RIGHT_CLICK,
+					null,
+					listOf("Throw your pickaxe to create an",
+					       "explosion mining all ores in a 3 block",
+					       "radius.").map(Text::literal),
+					48.seconds
+				)
+			),
+			AbilityUtils.getAbilities(ItemResources.loadItem("titanium-drill")).stripDescriptions()
+		)
+	}
+
+	@Test
+	fun testPoweredPickaxe() {
+		Assertions.assertEquals(
+			listOf(
+				AbilityUtils.ItemAbility(
+					"Mining Speed Boost",
+					true,
+					AbilityUtils.AbilityActivation.RIGHT_CLICK,
+					null,
+					listOf("Grants +200% ⸕ Mining Speed for",
+					       "10s.").map(Text::literal),
+					2.minutes
+				)
+			),
+			AbilityUtils.getAbilities(ItemResources.loadItem("diamond-pickaxe")).stripDescriptions()
+		)
+	}
+
+	@Test
+	fun testAOTV() {
+		Assertions.assertEquals(
+			listOf(
+				AbilityUtils.ItemAbility(
+					"Instant Transmission", true, AbilityUtils.AbilityActivation.RIGHT_CLICK, 23,
+					listOf("Teleport 12 blocks ahead of you and",
+					       "gain +50 ✦ Speed for 3 seconds.").map(Text::literal),
+					null
+				),
+				AbilityUtils.ItemAbility(
+					"Ether Transmission",
+					false,
+					AbilityUtils.AbilityActivation.SNEAK_RIGHT_CLICK,
+					90,
+					listOf("Teleport to your targeted block up",
+					       "to 61 blocks away.",
+					       "Soulflow Cost: 1").map(Text::literal),
+					null
+				)
+			),
+			AbilityUtils.getAbilities(ItemResources.loadItem("aspect-of-the-void")).stripDescriptions()
+		)
+	}
+}