mirror of
https://github.com/ipfs/kubo.git
synced 2025-06-19 09:52:03 +08:00
125 lines
2.5 KiB
Go
125 lines
2.5 KiB
Go
package dagutils
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
key "github.com/ipfs/go-ipfs/blocks/key"
|
|
dag "github.com/ipfs/go-ipfs/merkledag"
|
|
mdtest "github.com/ipfs/go-ipfs/merkledag/test"
|
|
|
|
context "github.com/ipfs/go-ipfs/Godeps/_workspace/src/golang.org/x/net/context"
|
|
)
|
|
|
|
func TestAddLink(t *testing.T) {
|
|
ds := mdtest.Mock()
|
|
fishnode := &dag.Node{
|
|
Data: []byte("fishcakes!"),
|
|
}
|
|
|
|
fk, err := ds.Add(fishnode)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
nd := new(dag.Node)
|
|
nnode, err := addLink(context.Background(), ds, nd, "fish", fishnode)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
fnprime, err := nnode.GetLinkedNode(context.Background(), ds, "fish")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
fnpkey, err := fnprime.Key()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if fnpkey != fk {
|
|
t.Fatal("wrong child node found!")
|
|
}
|
|
}
|
|
|
|
func assertNodeAtPath(t *testing.T, ds dag.DAGService, root *dag.Node, path string, exp key.Key) {
|
|
parts := strings.Split(path, "/")
|
|
cur := root
|
|
for _, e := range parts {
|
|
nxt, err := cur.GetLinkedNode(context.Background(), ds, e)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cur = nxt
|
|
}
|
|
|
|
curk, err := cur.Key()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if curk != exp {
|
|
t.Fatal("node not as expected at end of path")
|
|
}
|
|
}
|
|
|
|
func TestInsertNode(t *testing.T) {
|
|
ds := mdtest.Mock()
|
|
root := new(dag.Node)
|
|
e := NewDagEditor(ds, root)
|
|
|
|
testInsert(t, e, "a", "anodefortesting", false, "")
|
|
testInsert(t, e, "a/b", "data", false, "")
|
|
testInsert(t, e, "a/b/c/d/e", "blah", false, "merkledag: not found")
|
|
testInsert(t, e, "a/b/c/d/e", "foo", true, "")
|
|
testInsert(t, e, "a/b/c/d/f", "baz", true, "")
|
|
testInsert(t, e, "a/b/c/d/f", "bar", true, "")
|
|
|
|
testInsert(t, e, "", "bar", true, "cannot create link with no name!")
|
|
testInsert(t, e, "////", "slashes", true, "cannot create link with no name!")
|
|
|
|
k, err := e.GetNode().Key()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if k.B58String() != "QmThorWojP6YzLJwDukxiYCoKQSwyrMCvdt4WZ6rPm221t" {
|
|
t.Fatal("output was different than expected")
|
|
}
|
|
}
|
|
|
|
func testInsert(t *testing.T, e *Editor, path, data string, create bool, experr string) {
|
|
child := &dag.Node{Data: []byte(data)}
|
|
ck, err := e.ds.Add(child)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var c func() *dag.Node
|
|
if create {
|
|
c = func() *dag.Node {
|
|
return &dag.Node{}
|
|
}
|
|
}
|
|
|
|
err = e.InsertNodeAtPath(context.Background(), path, child, c)
|
|
if experr != "" {
|
|
var got string
|
|
if err != nil {
|
|
got = err.Error()
|
|
}
|
|
if got != experr {
|
|
t.Fatalf("expected '%s' but got '%s'", experr, got)
|
|
}
|
|
return
|
|
}
|
|
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
assertNodeAtPath(t, e.ds, e.root, path, ck)
|
|
}
|