package frame
import (
"fmt"
"testing"
"gopkg.in/src-d/go-git.v4/utils/merkletrie/internal/fsnoder"
"gopkg.in/src-d/go-git.v4/utils/merkletrie/noder"
. "gopkg.in/check.v1"
)
func Test(t *testing.T) { TestingT(t) }
type FrameSuite struct{}
var _ = Suite(&FrameSuite{})
func (s *FrameSuite) TestNewFrameFromEmptyDir(c *C) {
A, err := fsnoder.New("A()")
c.Assert(err, IsNil)
frame, err := New(A)
c.Assert(err, IsNil)
expectedString := `[]`
c.Assert(frame.String(), Equals, expectedString)
first, ok := frame.First()
c.Assert(first, IsNil)
c.Assert(ok, Equals, false)
first, ok = frame.First()
c.Assert(first, IsNil)
c.Assert(ok, Equals, false)
l := frame.Len()
c.Assert(l, Equals, 0)
}
func (s *FrameSuite) TestNewFrameFromNonEmpty(c *C) {
// _______A/________
// | / \ |
// x y B/ C/
// |
// z
root, err := fsnoder.New("A(x<> y<> B() C(z<>))")
c.Assert(err, IsNil)
frame, err := New(root)
c.Assert(err, IsNil)
expectedString := `["B", "C", "x", "y"]`
c.Assert(frame.String(), Equals, expectedString)
l := frame.Len()
c.Assert(l, Equals, 4)
checkFirstAndDrop(c, frame, "B", true)
l = frame.Len()
c.Assert(l, Equals, 3)
checkFirstAndDrop(c, frame, "C", true)
l = frame.Len()
c.Assert(l, Equals, 2)
checkFirstAndDrop(c, frame, "x", true)
l = frame.Len()
c.Assert(l, Equals, 1)
checkFirstAndDrop(c, frame, "y", true)
l = frame.Len()
c.Assert(l, Equals, 0)
checkFirstAndDrop(c, frame, "", false)
l = frame.Len()
c.Assert(l, Equals, 0)
checkFirstAndDrop(c, frame, "", false)
}
func checkFirstAndDrop(c *C, f *Frame, expectedNodeName string, expectedOK bool) {
first, ok := f.First()
c.Assert(ok, Equals, expectedOK)
if expectedOK {
c.Assert(first.Name(), Equals, expectedNodeName)
}
f.Drop()
}
// a mock noder that returns error when Children() is called
type errorNoder struct{ noder.Noder }
func (e *errorNoder) Children() ([]noder.Noder, error) {
return nil, fmt.Errorf("mock error")
}
func (s *FrameSuite) TestNewFrameErrors(c *C) {
_, err := New(&errorNoder{})
c.Assert(err, ErrorMatches, "mock error")
}
