branch_test.go

package git

import (
"testing"
)

funcTestBranchIterator(t*testing.T) {
t.Parallel()
repo:=createTestRepo(t)
defercleanupTestRepo(t, repo)

seedTestRepo(t, repo)

i, err:=repo.NewBranchIterator(BranchLocal)
checkFatal(t, err)

b, bt, err:=i.Next()
checkFatal(t, err)
ifname, _:=b.Name(); name!="master" {
t.Fatalf("expected master")
 } elseifbt!=BranchLocal {
t.Fatalf("expected BranchLocal, not %v", t)
 }
b, bt, err=i.Next()
if!IsErrorCode(err, ErrorCodeIterOver) {
t.Fatal("expected iterover")
 }
}

funcTestBranchIteratorEach(t*testing.T) {
t.Parallel()
repo:=createTestRepo(t)
defercleanupTestRepo(t, repo)

seedTestRepo(t, repo)

i, err:=repo.NewBranchIterator(BranchLocal)
checkFatal(t, err)

varnames []string
f:=func(b*Branch, tBranchType) error {
name, err:=b.Name()
iferr!=nil {
returnerr
 }

names=append(names, name)
returnnil
 }

err=i.ForEach(f)
iferr!=nil&&!IsErrorCode(err, ErrorCodeIterOver) {
t.Fatal(err)
 }

iflen(names) !=1 {
t.Fatalf("expect 1 branch, but it was %d\n", len(names))
 }

ifnames[0] !="master" {
t.Fatalf("expect branch master, but it was %s\n", names[0])
 }
}