op_edit_comment_test.go

 1package bug
 2
 3import (
 4	"encoding/json"
 5	"testing"
 6	"time"
 7
 8	"github.com/MichaelMure/git-bug/identity"
 9	"github.com/stretchr/testify/assert"
10	"github.com/stretchr/testify/require"
11)
12
13func TestEdit(t *testing.T) {
14	snapshot := Snapshot{}
15
16	rene := identity.NewBare("René Descartes", "rene@descartes.fr")
17	unix := time.Now().Unix()
18
19	create := NewCreateOp(rene, unix, "title", "create", nil)
20	create.Apply(&snapshot)
21
22	hash1, err := create.Hash()
23	require.NoError(t, err)
24
25	comment := NewAddCommentOp(rene, unix, "comment", nil)
26	comment.Apply(&snapshot)
27
28	hash2, err := comment.Hash()
29	require.NoError(t, err)
30
31	edit := NewEditCommentOp(rene, unix, hash1, "create edited", nil)
32	edit.Apply(&snapshot)
33
34	assert.Equal(t, len(snapshot.Timeline), 2)
35	assert.Equal(t, len(snapshot.Timeline[0].(*CreateTimelineItem).History), 2)
36	assert.Equal(t, len(snapshot.Timeline[1].(*AddCommentTimelineItem).History), 1)
37	assert.Equal(t, snapshot.Comments[0].Message, "create edited")
38	assert.Equal(t, snapshot.Comments[1].Message, "comment")
39
40	edit2 := NewEditCommentOp(rene, unix, hash2, "comment edited", nil)
41	edit2.Apply(&snapshot)
42
43	assert.Equal(t, len(snapshot.Timeline), 2)
44	assert.Equal(t, len(snapshot.Timeline[0].(*CreateTimelineItem).History), 2)
45	assert.Equal(t, len(snapshot.Timeline[1].(*AddCommentTimelineItem).History), 2)
46	assert.Equal(t, snapshot.Comments[0].Message, "create edited")
47	assert.Equal(t, snapshot.Comments[1].Message, "comment edited")
48}
49
50func TestEditCommentSerialize(t *testing.T) {
51	var rene = identity.NewBare("René Descartes", "rene@descartes.fr")
52	unix := time.Now().Unix()
53	before := NewEditCommentOp(rene, unix, "target", "message", nil)
54
55	data, err := json.Marshal(before)
56	assert.NoError(t, err)
57
58	var after EditCommentOperation
59	err = json.Unmarshal(data, &after)
60	assert.NoError(t, err)
61
62	assert.Equal(t, before, &after)
63}