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)
}
}