1 files changed, 35 insertions, 0 deletions
diff --git a/cmd/generate/src_contribs_test.go b/cmd/generate/src_contribs_test.go
new file mode 100644
index 0000000..f7566c0
--- /dev/null
+++ b/cmd/generate/src_contribs_test.go
@@ -0,0 +1,35 @@
+package main
+
+import (
+	"testing"
+
+	"github.com/alecthomas/assert/v2"
+)
+
+func TestClassifyStatus(t *testing.T) {
+	testcases := map[string]struct {
+		Str string
+		Err string
+	}{
+		"merged+deployed":  {"released", ""},
+		"released in v1.2": {"released", ""},
+
+		"merged": {"merged", ""},
+
+		"open": {"open", ""},
+
+		"closed": {"closed", ""},
+		"locked": {"closed", ""},
+	}
+	for in, exp := range testcases {
+		t.Run(in, func(t *testing.T) {
+			actStr, actErr := classifyStatus(in)
+			assert.Equal(t, exp.Str, actStr)
+			if exp.Err == "" {
+				assert.NoError(t, actErr)
+			} else {
+				assert.EqualError(t, actErr, exp.Err)
+			}
+		})
+	}
+}