From 2a1631dadfc0f78eb3b7768f9bca3f9fa1ea619e Mon Sep 17 00:00:00 2001
From: Linnea Gräf <nea@nea.moe>
Date: Wed, 4 Jun 2025 01:06:51 +0200
Subject: feat: Add basic combo buttons (without editor for now)

---
 .../kotlin/features/macros/KeyComboTrieCreation.kt | 103 +++++++++++++++++++++
 1 file changed, 103 insertions(+)
 create mode 100644 src/test/kotlin/features/macros/KeyComboTrieCreation.kt

(limited to 'src/test/kotlin')

diff --git a/src/test/kotlin/features/macros/KeyComboTrieCreation.kt b/src/test/kotlin/features/macros/KeyComboTrieCreation.kt
new file mode 100644
index 0000000..f0e7a1b
--- /dev/null
+++ b/src/test/kotlin/features/macros/KeyComboTrieCreation.kt
@@ -0,0 +1,103 @@
+package moe.nea.firmament.test.features.macros
+
+import org.junit.jupiter.api.Assertions
+import org.junit.jupiter.api.Test
+import net.minecraft.client.util.InputUtil
+import moe.nea.firmament.features.macros.Branch
+import moe.nea.firmament.features.macros.ComboKeyAction
+import moe.nea.firmament.features.macros.CommandAction
+import moe.nea.firmament.features.macros.KeyComboTrie
+import moe.nea.firmament.features.macros.Leaf
+import moe.nea.firmament.keybindings.SavedKeyBinding
+
+class KeyComboTrieCreation {
+	val basicAction = CommandAction("ac Hello")
+	val aPress = SavedKeyBinding(InputUtil.GLFW_KEY_A)
+	val bPress = SavedKeyBinding(InputUtil.GLFW_KEY_B)
+	val cPress = SavedKeyBinding(InputUtil.GLFW_KEY_C)
+
+	@Test
+	fun testValidShortTrie() {
+		val actions = listOf(
+			ComboKeyAction(basicAction, listOf(aPress)),
+			ComboKeyAction(basicAction, listOf(bPress)),
+			ComboKeyAction(basicAction, listOf(cPress)),
+		)
+		Assertions.assertEquals(
+			Branch(
+				mapOf(
+					aPress to Leaf(basicAction),
+					bPress to Leaf(basicAction),
+					cPress to Leaf(basicAction),
+				),
+			), KeyComboTrie.fromComboList(actions)
+		)
+	}
+
+	@Test
+	fun testOverlappingLeafs() {
+		Assertions.assertThrows(IllegalStateException::class.java) {
+			KeyComboTrie.fromComboList(
+				listOf(
+					ComboKeyAction(basicAction, listOf(aPress, aPress)),
+					ComboKeyAction(basicAction, listOf(aPress, aPress)),
+				)
+			)
+		}
+		Assertions.assertThrows(IllegalStateException::class.java) {
+			KeyComboTrie.fromComboList(
+				listOf(
+					ComboKeyAction(basicAction, listOf(aPress)),
+					ComboKeyAction(basicAction, listOf(aPress)),
+				)
+			)
+		}
+	}
+
+	@Test
+	fun testBranchOverlappingLeaf() {
+		Assertions.assertThrows(IllegalStateException::class.java) {
+			KeyComboTrie.fromComboList(
+				listOf(
+					ComboKeyAction(basicAction, listOf(aPress)),
+					ComboKeyAction(basicAction, listOf(aPress, aPress)),
+				)
+			)
+		}
+	}
+	@Test
+	fun testLeafOverlappingBranch() {
+		Assertions.assertThrows(IllegalStateException::class.java) {
+			KeyComboTrie.fromComboList(
+				listOf(
+					ComboKeyAction(basicAction, listOf(aPress, aPress)),
+					ComboKeyAction(basicAction, listOf(aPress)),
+				)
+			)
+		}
+	}
+
+
+	@Test
+	fun testValidNestedTrie() {
+		val actions = listOf(
+			ComboKeyAction(basicAction, listOf(aPress, aPress)),
+			ComboKeyAction(basicAction, listOf(aPress, bPress)),
+			ComboKeyAction(basicAction, listOf(cPress)),
+		)
+		Assertions.assertEquals(
+			Branch(
+				mapOf(
+					aPress to Branch(
+						mapOf(
+							aPress to Leaf(basicAction),
+							bPress to Leaf(basicAction),
+						)
+					),
+					cPress to Leaf(basicAction),
+				),
+			), KeyComboTrie.fromComboList(actions)
+		)
+	}
+
+}
-- 
cgit