1package tests
 2
 3import (
 4	"github.com/MichaelMure/git-bug/bug"
 5	"github.com/MichaelMure/git-bug/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")
17	setTitleOp    = operations.NewSetTitleOp(rene, "title2")
18	addCommentOp  = operations.NewAddCommentOp(rene, "message2")
19	setStatusOp   = operations.NewSetStatusOp(rene, bug.ClosedStatus)
20	labelChangeOp = operations.NewLabelChangeOperation(rene, []bug.Label{"added"}, []bug.Label{"removed"})
21	mockRepo      = repository.NewMockRepoForTest()
22)
23
24func TestOpIterator(t *testing.T) {
25
26	bug1, err := bug.NewBug()
27
28	if err != nil {
29		t.Fatal(err)
30	}
31
32	bug1.Append(createOp)
33	bug1.Append(setTitleOp)
34	bug1.Append(addCommentOp)
35	bug1.Append(setStatusOp)
36	bug1.Append(labelChangeOp)
37	bug1.Commit(mockRepo)
38
39	bug1.Append(setTitleOp)
40	bug1.Append(setTitleOp)
41	bug1.Append(setTitleOp)
42	bug1.Commit(mockRepo)
43
44	bug1.Append(setTitleOp)
45	bug1.Append(setTitleOp)
46	bug1.Append(setTitleOp)
47
48	it := bug.NewOperationIterator(bug1)
49
50	counter := 0
51	for it.Next() {
52		_ = it.Value()
53		counter++
54	}
55
56	if counter != 11 {
57		t.Fatalf("Wrong count of value iterated (%d instead of 8)", counter)
58	}
59}