1package bug
 2
 3import (
 4	"fmt"
 5	"testing"
 6	"time"
 7
 8	"github.com/stretchr/testify/require"
 9
10	"github.com/MichaelMure/git-bug/identity"
11	"github.com/MichaelMure/git-bug/repository"
12)
13
14func ExampleOperationIterator() {
15	b := NewBug()
16
17	// add operations
18
19	it := NewOperationIterator(b)
20
21	for it.Next() {
22		// do something with each operations
23		_ = it.Value()
24	}
25}
26
27func TestOpIterator(t *testing.T) {
28	mockRepo := repository.NewMockRepoForTest()
29
30	rene := identity.NewIdentity("René Descartes", "rene@descartes.fr")
31	err := rene.Commit(mockRepo)
32	require.NoError(t, err)
33
34	unix := time.Now().Unix()
35
36	createOp := NewCreateOp(rene, unix, "title", "message", nil)
37	addCommentOp := NewAddCommentOp(rene, unix, "message2", nil)
38	setStatusOp := NewSetStatusOp(rene, unix, ClosedStatus)
39	labelChangeOp := NewLabelChangeOperation(rene, unix, []Label{"added"}, []Label{"removed"})
40
41	var i int
42	genTitleOp := func() Operation {
43		i++
44		return NewSetTitleOp(rene, unix, fmt.Sprintf("title%d", i), "")
45	}
46
47	bug1 := NewBug()
48
49	// first pack
50	bug1.Append(createOp)
51	bug1.Append(addCommentOp)
52	bug1.Append(setStatusOp)
53	bug1.Append(labelChangeOp)
54	err = bug1.Commit(mockRepo)
55	require.NoError(t, err)
56
57	// second pack
58	bug1.Append(genTitleOp())
59	bug1.Append(genTitleOp())
60	bug1.Append(genTitleOp())
61	err = bug1.Commit(mockRepo)
62	require.NoError(t, err)
63
64	// staging
65	bug1.Append(genTitleOp())
66	bug1.Append(genTitleOp())
67	bug1.Append(genTitleOp())
68
69	it := NewOperationIterator(bug1)
70
71	counter := 0
72	for it.Next() {
73		_ = it.Value()
74		counter++
75	}
76
77	require.Equal(t, 10, counter)
78}