1 files changed, 21 insertions, 0 deletions
diff --git a/cmd/krino/main_test.go b/cmd/krino/main_test.go
index 1c01fa0..6e4df44 100644
--- a/cmd/krino/main_test.go
+++ b/cmd/krino/main_test.go
@@ -62,3 +62,24 @@ func TestSortNoConfig(t *testing.T) {
 		t.Fatalf("got %d %q", code, errOut)
 	}
 }
+
+func TestVersionStripsTheTagPrefix(t *testing.T) {
+	old := version
+	t.Cleanup(func() { version = old })
+	for _, tc := range []struct{ stamped, want string }{
+		{"v0.0.1", "krino 0.0.1\n"},
+		{"v0.0.1-3-gabc1234", "krino 0.0.1-3-gabc1234\n"},
+		{"v0.0.1-dirty", "krino 0.0.1-dirty\n"},
+		{"0.0.1", "krino 0.0.1\n"},
+		{"dev", "krino dev\n"},
+	} {
+		version = tc.stamped
+		var out, errOut strings.Builder
+		if code := run([]string{"--version"}, &out, &errOut); code != 0 {
+			t.Errorf("%s: exit %d", tc.stamped, code)
+		}
+		if out.String() != tc.want {
+			t.Errorf("stamped %q: got %q, want %q", tc.stamped, out.String(), tc.want)
+		}
+	}
+}