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)
11
12func TestEdit(t *testing.T) {
13	snapshot := Snapshot{}
14
15	var rene = identity.NewIdentity("René Descartes", "rene@descartes.fr")
16
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	if err != nil {
24		t.Fatal(err)
25	}
26
27	comment := NewAddCommentOp(rene, unix, "comment", nil)
28	comment.Apply(&snapshot)
29
30	hash2, err := comment.Hash()
31	if err != nil {
32		t.Fatal(err)
33	}
34
35	edit := NewEditCommentOp(rene, unix, hash1, "create edited", nil)
36	edit.Apply(&snapshot)
37
38	assert.Equal(t, len(snapshot.Timeline), 2)
39	assert.Equal(t, len(snapshot.Timeline[0].(*CreateTimelineItem).History), 2)
40	assert.Equal(t, len(snapshot.Timeline[1].(*AddCommentTimelineItem).History), 1)
41	assert.Equal(t, snapshot.Comments[0].Message, "create edited")
42	assert.Equal(t, snapshot.Comments[1].Message, "comment")
43
44	edit2 := NewEditCommentOp(rene, unix, hash2, "comment edited", nil)
45	edit2.Apply(&snapshot)
46
47	assert.Equal(t, len(snapshot.Timeline), 2)
48	assert.Equal(t, len(snapshot.Timeline[0].(*CreateTimelineItem).History), 2)
49	assert.Equal(t, len(snapshot.Timeline[1].(*AddCommentTimelineItem).History), 2)
50	assert.Equal(t, snapshot.Comments[0].Message, "create edited")
51	assert.Equal(t, snapshot.Comments[1].Message, "comment edited")
52}
53
54func TestEditCommentSerialize(t *testing.T) {
55	var rene = identity.NewBare("René Descartes", "rene@descartes.fr")
56	unix := time.Now().Unix()
57	before := NewEditCommentOp(rene, unix, "target", "message", nil)
58
59	data, err := json.Marshal(before)
60	assert.NoError(t, err)
61
62	var after EditCommentOperation
63	err = json.Unmarshal(data, &after)
64	assert.NoError(t, err)
65
66	assert.Equal(t, before, &after)
67}