diff options
Diffstat (limited to 'repository_test.go')
| -rw-r--r-- | repository_test.go | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/repository_test.go b/repository_test.go index 07fa763..b67e31d 100644 --- a/repository_test.go +++ b/repository_test.go @@ -314,24 +314,27 @@ func (s *RepositorySuite) TestTag(c *C) { func (s *RepositorySuite) TestTags(c *C) { r := NewMemoryRepository() - err := r.Clone(&CloneOptions{URL: "https://github.com/spinnaker/spinnaker.git"}) + err := r.Clone(&CloneOptions{URL: "https://github.com/git-fixtures/tags.git"}) c.Assert(err, IsNil) count := 0 tags, err := r.Tags() c.Assert(err, IsNil) - for { - tag, err := tags.Next() - if err != nil { - break - } + tags.ForEach(func(tag *Tag) error { count++ c.Assert(tag.Hash.IsZero(), Equals, false) c.Assert(tag.Type(), Equals, core.TagObject) - } - c.Assert(count, Equals, 11) + return nil + }) + + refs, _ := r.Refs() + refs.ForEach(func(ref *core.Reference) error { + return nil + }) + + c.Assert(count, Equals, 4) } func (s *RepositorySuite) TestCommitIterClosePanic(c *C) { |