package storer import ( "io" . "gopkg.in/check.v1" "srcd.works/go-git.v4/plumbing" ) type ReferenceSuite struct{} var _ = Suite(&ReferenceSuite{}) func (s *ReferenceSuite) TestReferenceSliceIterNext(c *C) { slice := []*plumbing.Reference{ plumbing.NewReferenceFromStrings("foo", "foo"), plumbing.NewReferenceFromStrings("bar", "bar"), } i := NewReferenceSliceIter(slice) foo, err := i.Next() c.Assert(err, IsNil) c.Assert(foo == slice[0], Equals, true) bar, err := i.Next() c.Assert(err, IsNil) c.Assert(bar == slice[1], Equals, true) empty, err := i.Next() c.Assert(err, Equals, io.EOF) c.Assert(empty, IsNil) } func (s *ReferenceSuite) TestReferenceSliceIterForEach(c *C) { slice := []*plumbing.Reference{ plumbing.NewReferenceFromStrings("foo", "foo"), plumbing.NewReferenceFromStrings("bar", "bar"), } i := NewReferenceSliceIter(slice) var count int i.ForEach(func(r *plumbing.Reference) error { c.Assert(r == slice[count], Equals, true) count++ return nil }) c.Assert(count, Equals, 2) } func (s *ReferenceSuite) TestReferenceSliceIterForEachStop(c *C) { slice := []*plumbing.Reference{ plumbing.NewReferenceFromStrings("foo", "foo"), plumbing.NewReferenceFromStrings("bar", "bar"), } i := NewReferenceSliceIter(slice) var count int i.ForEach(func(r *plumbing.Reference) error { c.Assert(r == slice[count], Equals, true) count++ return ErrStop }) c.Assert(count, Equals, 1) }