From d2405b8b3bc3c7a812e6b9390811869ade2c4ce4 Mon Sep 17 00:00:00 2001
From: The Magician <themagician@engineer.com>
Date: Fri, 8 Dec 2023 17:57:00 +0000
Subject: [PATCH] Add "edit" command

---
 tests/test_get_command.sh | 9 +++++++++
 tests/test_main.sh        | 9 +++++++++
 zk                        | 2 +-
 3 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/tests/test_get_command.sh b/tests/test_get_command.sh
index 9701c50..378da92 100644
--- a/tests/test_get_command.sh
+++ b/tests/test_get_command.sh
@@ -31,6 +31,15 @@ test_get_command_returns_new_for_substring() {
 	done
 }
 
+test_get_command_returns_edit_for_substring() {
+	expected="edit"
+
+	for arg in e ed edi edit; do
+		result="$(get_command "$ZKCOMMANDS" "$arg")"
+		assert_equals "$expected" "$result"
+	done
+}
+
 test_get_command_does_not_return_match_unless_at_start_of_command_string() {
 	commands="delete"
 	expected=""
diff --git a/tests/test_main.sh b/tests/test_main.sh
index 93db28b..7fdf590 100644
--- a/tests/test_main.sh
+++ b/tests/test_main.sh
@@ -34,3 +34,12 @@ test_main_calls_new_when_new_command_passed() {
 	assert_equals "$expected" "$result"
 }
 
+test_main_calls_edit_when_edit_command_passed() {
+	expected="_zk_edit"
+	fake get_command 'echo edit'
+	fake _zk_edit "echo $expected"
+
+	result="$(main)"
+
+	assert_equals "$expected" "$result"
+}
diff --git a/zk b/zk
index e9b8a64..db23b77 100755
--- a/zk
+++ b/zk
@@ -1,7 +1,7 @@
 #!/bin/sh
 
 ZKDIR="$HOME/.local/share/zk/"
-ZKCOMMANDS='list search new'
+ZKCOMMANDS='list search new edit'
 
 get_command() {
 	commands="$1"