add difftree for noders (#262)

difftree for noders

1 files changed, 65 insertions, 0 deletions

diff --git a/utils/merkletrie/change_test.go b/utils/merkletrie/change_test.go
new file mode 100644
index 0000000..c899ff9
--- /dev/null
+++ b/utils/merkletrie/change_test.go
@@ -0,0 +1,65 @@
+package merkletrie_test
+
+import (
+	"srcd.works/go-git.v4/utils/merkletrie"
+	"srcd.works/go-git.v4/utils/merkletrie/internal/fsnoder"
+	"srcd.works/go-git.v4/utils/merkletrie/noder"
+
+	. "gopkg.in/check.v1"
+)
+
+type ChangeSuite struct{}
+
+var _ = Suite(&ChangeSuite{})
+
+func (s *ChangeSuite) TestUnsupportedAction(c *C) {
+	a := merkletrie.Action(42)
+	c.Assert(a.String, PanicMatches, "unsupported action.*")
+}
+
+func (s ChangeSuite) TestNewInsert(c *C) {
+	tree, err := fsnoder.New("(a(b(z<>)))")
+	c.Assert(err, IsNil)
+	path := find(c, tree, "z")
+	change := merkletrie.NewInsert(path)
+	c.Assert(change.String(), Equals, "<Insert a/b/z>")
+
+	shortPath := noder.Path([]noder.Noder{path.Last()})
+	change = merkletrie.NewInsert(shortPath)
+	c.Assert(change.String(), Equals, "<Insert z>")
+}
+
+func (s ChangeSuite) TestNewDelete(c *C) {
+	tree, err := fsnoder.New("(a(b(z<>)))")
+	c.Assert(err, IsNil)
+	path := find(c, tree, "z")
+	change := merkletrie.NewDelete(path)
+	c.Assert(change.String(), Equals, "<Delete a/b/z>")
+
+	shortPath := noder.Path([]noder.Noder{path.Last()})
+	change = merkletrie.NewDelete(shortPath)
+	c.Assert(change.String(), Equals, "<Delete z>")
+}
+
+func (s ChangeSuite) TestNewModify(c *C) {
+	tree1, err := fsnoder.New("(a(b(z<>)))")
+	c.Assert(err, IsNil)
+	path1 := find(c, tree1, "z")
+
+	tree2, err := fsnoder.New("(a(b(z<1>)))")
+	c.Assert(err, IsNil)
+	path2 := find(c, tree2, "z")
+
+	change := merkletrie.NewModify(path1, path2)
+	c.Assert(change.String(), Equals, "<Modify a/b/z>")
+
+	shortPath1 := noder.Path([]noder.Noder{path1.Last()})
+	shortPath2 := noder.Path([]noder.Noder{path2.Last()})
+	change = merkletrie.NewModify(shortPath1, shortPath2)
+	c.Assert(change.String(), Equals, "<Modify z>")
+}
+
+func (s ChangeSuite) TestMalformedChange(c *C) {
+	change := merkletrie.Change{}
+	c.Assert(change.String, PanicMatches, "malformed change.*")
+}