From fa8a3bb850915aa0a4beed2be402bbc91db21b94 Mon Sep 17 00:00:00 2001
From: Ethan P <eth-p+git@hidden.email>
Date: Tue, 22 Oct 2019 15:21:12 -0700
Subject: [PATCH] Improve tests

---
 test/run.sh                         | 7 +++++++
 test/tests/batgrep.pattern.sh       | 1 -
 test/tests/batgrep.pattern.snapshot | 7 -------
 test/tests/batgrep.symlink.sh       | 3 +++
 test/tests/batgrep.symlink.snapshot | 7 +++++++
 test/util/test-exec.sh              | 7 +++++++
 6 files changed, 24 insertions(+), 8 deletions(-)
 create mode 100644 test/tests/batgrep.symlink.sh
 create mode 100644 test/tests/batgrep.symlink.snapshot

diff --git a/test/run.sh b/test/run.sh
index cd72f62..02657bc 100755
--- a/test/run.sh
+++ b/test/run.sh
@@ -79,6 +79,13 @@ display_test_summary() {
 # -----------------------------------------------------------------------------
 
 if [ -n "$1" ]; then
+	test_types="$("$HERE/util/test-exec.sh" --supports)"
+	if ! grep -F "$1" <<< "$test_types" &>/dev/null; then
+		printc "%{RED}Unknown test type: %s%{CLEAR}\n" "$1"
+		printc "%{RED}Supported types:%{CLEAR}\n%s\n" "$test_types"
+		exit 2
+	fi
+
 	run_all_tests "$1"
 	exit_status=$?
 	display_test_summary
diff --git a/test/tests/batgrep.pattern.sh b/test/tests/batgrep.pattern.sh
index 5829e52..0d2105c 100644
--- a/test/tests/batgrep.pattern.sh
+++ b/test/tests/batgrep.pattern.sh
@@ -1,4 +1,3 @@
 set -e
 "$TEST_RUNNER" batgrep "ca" file.txt
-"$TEST_RUNNER" batgrep "ca" link.txt
 
diff --git a/test/tests/batgrep.pattern.snapshot b/test/tests/batgrep.pattern.snapshot
index a0e63d9..19d2bdb 100644
--- a/test/tests/batgrep.pattern.snapshot
+++ b/test/tests/batgrep.pattern.snapshot
@@ -5,10 +5,3 @@ car
 frog
 fox
 ────────────────────────────────────────────────────────────────────────────────
-────────────────────────────────────────────────────────────────────────────────
-cat
-dog
-car
-frog
-fox
-────────────────────────────────────────────────────────────────────────────────
diff --git a/test/tests/batgrep.symlink.sh b/test/tests/batgrep.symlink.sh
new file mode 100644
index 0000000..a191108
--- /dev/null
+++ b/test/tests/batgrep.symlink.sh
@@ -0,0 +1,3 @@
+set -e
+"$TEST_RUNNER" batgrep "ca" link.txt
+
diff --git a/test/tests/batgrep.symlink.snapshot b/test/tests/batgrep.symlink.snapshot
new file mode 100644
index 0000000..19d2bdb
--- /dev/null
+++ b/test/tests/batgrep.symlink.snapshot
@@ -0,0 +1,7 @@
+────────────────────────────────────────────────────────────────────────────────
+cat
+dog
+car
+frog
+fox
+────────────────────────────────────────────────────────────────────────────────
diff --git a/test/util/test-exec.sh b/test/util/test-exec.sh
index 6bb4d22..9ba278d 100755
--- a/test/util/test-exec.sh
+++ b/test/util/test-exec.sh
@@ -80,6 +80,13 @@ case "$1" in
 		fi
 	};;
 
+	--supports) {
+		echo "snapshot-generate"
+		echo "snapshot-test"
+		echo "consistency-test"
+		exit 0
+	};;
+
 	*) {
 		printf "\x1B[31mUnknown subcommand.\x1B[0m\n"
 		printf " - \x1B[33msnapshot-generate\x1B[0m  -- generate new snapshots\n"