Method TestTreeIter

plumbing/object/tree_test.go:304–337

(c *C)

}

func (s *TreeSuite) TestTreeIter(c *C) {
	encIter, err := s.Storer.IterEncodedObjects(plumbing.TreeObject)
	c.Assert(err, IsNil)
	iter := NewTreeIter(s.Storer, encIter)

	trees := []*Tree{}
	iter.ForEach(func(t *Tree) error {
		t.s = nil
		trees = append(trees, t)
		return nil
	})

	c.Assert(len(trees) > 0, Equals, true)
	iter.Close()

	encIter, err = s.Storer.IterEncodedObjects(plumbing.TreeObject)
	c.Assert(err, IsNil)
	iter = NewTreeIter(s.Storer, encIter)

	i := 0
	for {
		t, err := iter.Next()
		if err == io.EOF {
			break
		}

		t.s = nil
		c.Assert(err, IsNil)
		c.Assert(t, DeepEquals, trees[i])
		i += 1
	}

	iter.Close()
}

func (s *TreeSuite) TestTreeWalkerNext(c *C) {
	commit, err := GetCommit(s.Storer, plumbing.NewHash("6ecf0ef2c2dffb796033e5a02219af86ec6584e5"))