1 files changed, 41 insertions, 0 deletions
diff --git a/internal/utils/version_test.go b/internal/utils/version_test.go
new file mode 100644
index 000000000..3b8aed08a
--- /dev/null
+++ b/internal/utils/version_test.go
@@ -0,0 +1,41 @@
+package utils
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestVersionDefault(t *testing.T) {
+	v := Version()
+
+	assert.Equal(t, "untagged-unknown-dirty (master, unknown)", v)
+}
+
+func TestVersion(t *testing.T) {
+	var v string
+
+	v = version("v4.90.0", "tagged clean", "50d8b4a941c26b89482c94ab324b5a274f9ced66", "master", "")
+	assert.Equal(t, "v4.90.0", v)
+
+	v = version("v4.90.0", "tagged clean", "50d8b4a941c26b89482c94ab324b5a274f9ced66", "master", "freshports")
+	assert.Equal(t, "v4.90.0-freshports", v)
+
+	v = version("v4.90.0", "tagged dirty", "50d8b4a941c26b89482c94ab324b5a274f9ced66", "master", "")
+	assert.Equal(t, "v4.90.0-dirty", v)
+
+	v = version("v4.90.0", "untagged dirty", "50d8b4a941c26b89482c94ab324b5a274f9ced66", "master", "")
+	assert.Equal(t, "untagged-v4.90.0-dirty (master, 50d8b4a)", v)
+
+	v = version("v4.90.0", "untagged clean", "50d8b4a941c26b89482c94ab324b5a274f9ced66", "master", "")
+	assert.Equal(t, "untagged-v4.90.0 (master, 50d8b4a)", v)
+
+	v = version("v4.90.0", "untagged clean", "50d8b4a941c26b89482c94ab324b5a274f9ced66", "master", "freshports")
+	assert.Equal(t, "untagged-v4.90.0-freshports (master, 50d8b4a)", v)
+
+	v = version("v4.90.0", "untagged clean", "", "master", "")
+	assert.Equal(t, "untagged-v4.90.0 (master, unknown)", v)
+
+	v = version("v4.90.0", "", "50d8b4a941c26b89482c94ab324b5a274f9ced66", "master", "")
+	assert.Equal(t, "untagged-v4.90.0-dirty (master, 50d8b4a)", v)
+}