tests(850): add tests for comment ls, add and edit

1 files changed, 160 insertions, 0 deletions

diff --git a/commands/comment_test.go b/commands/comment_test.go
new file mode 100644
index 00000000..c2e7cb36
--- /dev/null
+++ b/commands/comment_test.go
@@ -0,0 +1,160 @@
+package commands
+
+import (
+	"fmt"
+	"io/ioutil"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestComment(t *testing.T) {
+	const golden = "testdata/comment/message-only"
+
+	env, bug := newTestEnvAndBug(t)
+
+	require.NoError(t, runComment(env.env, []string{bug}))
+
+	requireCommentsEqual(t, golden, env)
+}
+
+const gitDateFormat = "Mon Jan 2 15:04:05 2006 -0700"
+
+type parsedComment struct {
+	author  string
+	id      string
+	date    time.Time
+	message string
+}
+
+type parseFunc func(*parsedComment, string)
+
+type commentParser struct {
+	t        *testing.T
+	fn       parseFunc
+	comments []parsedComment
+}
+
+func parseComments(t *testing.T, env *testEnv) []parsedComment {
+	t.Helper()
+
+	parser := &commentParser{
+		t:        t,
+		comments: []parsedComment{},
+	}
+
+	comment := &parsedComment{}
+	parser.fn = parser.parseAuthor
+
+	for _, line := range strings.Split(env.out.String(), "\n") {
+		parser.fn(comment, line)
+	}
+
+	parser.comments = append(parser.comments, *comment)
+
+	return parser.comments
+}
+
+func (p *commentParser) parseAuthor(comment *parsedComment, line string) {
+	p.t.Helper()
+
+	tkns := strings.Split(line, ": ")
+	require.Len(p.t, tkns, 2)
+	require.Equal(p.t, "Author", tkns[0])
+
+	comment.author = tkns[1]
+	p.fn = p.parseID
+}
+
+func (p *commentParser) parseID(comment *parsedComment, line string) {
+	p.t.Helper()
+
+	tkns := strings.Split(line, ": ")
+	require.Len(p.t, tkns, 2)
+	require.Equal(p.t, "Id", tkns[0])
+
+	comment.id = tkns[1]
+	p.fn = p.parseDate
+}
+
+func (p *commentParser) parseDate(comment *parsedComment, line string) {
+	p.t.Helper()
+
+	tkns := strings.Split(line, ": ")
+	require.Len(p.t, tkns, 2)
+	require.Equal(p.t, "Date", tkns[0])
+
+	date, err := time.Parse(gitDateFormat, tkns[1])
+	require.NoError(p.t, err)
+
+	comment.date = date
+	p.fn = p.parseMessage
+}
+
+func (p *commentParser) parseMessage(comment *parsedComment, line string) {
+	p.t.Helper()
+
+	if strings.HasPrefix(line, "Author: ") {
+		p.comments = append(p.comments, *comment)
+		comment = &parsedComment{}
+		p.parseAuthor(comment, line)
+
+		return
+	}
+
+	require.True(p.t, line == "" || strings.HasPrefix(line, "    "))
+
+	comment.message = strings.Join([]string{comment.message, line}, "\n")
+}
+
+func normalizeParsedComments(t *testing.T, comments []parsedComment) []parsedComment {
+	t.Helper()
+
+	prefix := 0x1234567
+	date, err := time.Parse(gitDateFormat, "Fri Aug 19 07:00:00 2022 +1900")
+	require.NoError(t, err)
+
+	out := []parsedComment{}
+
+	for i, comment := range comments {
+		comment.id = fmt.Sprintf("%7x", prefix+i)
+		comment.date = date.Add(time.Duration(i) * time.Minute)
+		out = append(out, comment)
+	}
+
+	return out
+}
+
+func requireCommentsEqual(t *testing.T, golden string, env *testEnv) {
+	t.Helper()
+
+	const goldenFilePatter = "%s-%d-golden.txt"
+
+	comments := parseComments(t, env)
+	comments = normalizeParsedComments(t, comments)
+
+	if *update {
+		t.Log("Got here")
+		for i, comment := range comments {
+			fileName := fmt.Sprintf(goldenFilePatter, golden, i)
+			require.NoError(t, ioutil.WriteFile(fileName, []byte(comment.message), 0644))
+		}
+	}
+
+	prefix := 0x1234567
+	date, err := time.Parse(gitDateFormat, "Fri Aug 19 07:00:00 2022 +1900")
+	require.NoError(t, err)
+
+	for i, comment := range comments {
+		require.Equal(t, "John Doe", comment.author)
+		require.Equal(t, fmt.Sprintf("%7x", prefix+i), comment.id)
+		require.Equal(t, date.Add(time.Duration(i)*time.Minute), comment.date)
+
+		fileName := fmt.Sprintf(goldenFilePatter, golden, i)
+		exp, err := ioutil.ReadFile(fileName)
+		require.NoError(t, err)
+		require.Equal(t, string(exp), comment.message)
+	}
+}