1package tests
 2
 3import (
 4	"github.com/MichaelMure/git-bug/bug"
 5	"github.com/MichaelMure/git-bug/operations"
 6	"github.com/MichaelMure/git-bug/repository"
 7	"testing"
 8)
 9
10var (
11	rene = bug.Person{
12		Name:  "René Descartes",
13		Email: "rene@descartes.fr",
14	}
15
16	createOp      = operations.NewCreateOp(rene, "title", "message", nil)
17	setTitleOp    = operations.NewSetTitleOp(rene, "title2", "title1")
18	addCommentOp  = operations.NewAddCommentOp(rene, "message2", nil)
19	setStatusOp   = operations.NewSetStatusOp(rene, bug.ClosedStatus)
20	labelChangeOp = operations.NewLabelChangeOperation(rene, []bug.Label{"added"}, []bug.Label{"removed"})
21)
22
23func TestOpIterator(t *testing.T) {
24	mockRepo := repository.NewMockRepoForTest()
25
26	bug1 := bug.NewBug()
27
28	// first pack
29	bug1.Append(createOp)
30	bug1.Append(setTitleOp)
31	bug1.Append(addCommentOp)
32	bug1.Append(setStatusOp)
33	bug1.Append(labelChangeOp)
34	bug1.Commit(mockRepo)
35
36	// second pack
37	bug1.Append(setTitleOp)
38	bug1.Append(setTitleOp)
39	bug1.Append(setTitleOp)
40	bug1.Commit(mockRepo)
41
42	// staging
43	bug1.Append(setTitleOp)
44	bug1.Append(setTitleOp)
45	bug1.Append(setTitleOp)
46
47	it := bug.NewOperationIterator(bug1)
48
49	counter := 0
50	for it.Next() {
51		_ = it.Value()
52		counter++
53	}
54
55	if counter != 11 {
56		t.Fatalf("Wrong count of value iterated (%d instead of 8)", counter)
57	}
58}