Handle interactive add menu commands

app/src/test/java/solutions/tretter/githugandroid/GitSandboxEngineTest.kt

@@ -369,6 +369,73 @@ class GitSandboxEngineTest {
         assertTrue(quitOutput.any { it.contains("Bye.") })
     }
 
+    @Test
+    fun interactiveAddHandlesEveryDisplayedMenuCommand() {
+        val menuCommands = listOf(
+            "1" to "What now> 1",
+            "2" to "Update>>",
+            "3" to "Revert>>",
+            "4" to "Add untracked>>",
+            "5" to "Patch update>>",
+            "6" to "Diff>>",
+            "7" to "Bye.",
+            "8" to "What now> 8",
+        )
+
+        menuCommands.forEach { (command, expectedOutput) ->
+            val repo = RepoState(initialized = true, files = listOf(GitFile("README")))
+            val (menuRepo, _) = GitSandboxEngine.execute(repo, "git add -i")
+            val (updatedRepo, output) = GitSandboxEngine.execute(menuRepo, command)
+
+            assertFalse("$command should not be rejected", output.any { it.contains("Huh ($command)?") })
+            assertTrue("$command should produce $expectedOutput", output.any { it.contains(expectedOutput) })
+            if (command in listOf("2", "3", "4", "5", "6")) {
+                assertTrue(updatedRepo.interactiveAddSession?.awaitingUpdateSelection == true)
+            }
+        }
+    }
+
+    @Test
+    fun interactiveAddHandlesMenuCommandAliases() {
+        val aliases = listOf(
+            "status" to "What now> status",
+            "update" to "Update>>",
+            "revert" to "Revert>>",
+            "add untracked" to "Add untracked>>",
+            "patch" to "Patch update>>",
+            "diff" to "Diff>>",
+            "quit" to "Bye.",
+            "help" to "What now> help",
+        )
+
+        aliases.forEach { (command, expectedOutput) ->
+            val repo = RepoState(initialized = true, files = listOf(GitFile("README")))
+            val (menuRepo, _) = GitSandboxEngine.execute(repo, "git add -i")
+            val (updatedRepo, output) = GitSandboxEngine.execute(menuRepo, command)
+
+            assertFalse("$command should not be rejected", output.any { it.contains("Huh ($command)?") })
+            assertTrue("$command should produce $expectedOutput", output.any { it.contains(expectedOutput) })
+            if (command in listOf("update", "revert", "add untracked", "patch", "diff")) {
+                assertTrue(updatedRepo.interactiveAddSession?.awaitingUpdateSelection == true)
+            }
+        }
+    }
+
+    @Test
+    fun interactiveAddPatchSelectionStagesSelectedPath() {
+        val repo = RepoState(initialized = true, files = listOf(GitFile("README")))
+
+        val (menuRepo, _) = GitSandboxEngine.execute(repo, "git add -i")
+        val (patchRepo, patchOutput) = GitSandboxEngine.execute(menuRepo, "patch")
+        val (selectedRepo, selectionOutput) = GitSandboxEngine.execute(patchRepo, "1")
+
+        assertTrue(patchRepo.interactiveAddSession?.awaitingUpdateSelection == true)
+        assertTrue(patchOutput.any { it.contains("Patch update>>") })
+        assertTrue(selectedRepo.files.single { it.name == "README" }.staged)
+        assertTrue(selectionOutput.any { it.contains("Patch update>> 1") })
+        assertTrue(selectionOutput.any { it.contains("updated 1 path(s)") })
+    }
+
     @Test
     fun nativeInteractiveAddSelectionUpdatesGitIndex() {
         val git = testGitBinary()