mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-22 02:47:48 +08:00
115 lines
2.7 KiB
Go
115 lines
2.7 KiB
Go
package dagutils
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
path "gx/ipfs/QmTG5WFmAM4uAnqGskeAPijdpTmmNDLJNCQ71NqfdvC6hV/go-path"
|
|
dag "gx/ipfs/QmXhrNaxjxNLwAHnWQScc6GxvpJMyn8wfdRmGDbUQwpfth/go-merkledag"
|
|
mdtest "gx/ipfs/QmXhrNaxjxNLwAHnWQScc6GxvpJMyn8wfdRmGDbUQwpfth/go-merkledag/test"
|
|
|
|
ipld "gx/ipfs/QmUSyMZ8Vt4vTZr5HdDEgEfpwAXfQRuDdfCFTt7XBzhxpQ/go-ipld-format"
|
|
cid "gx/ipfs/Qmdu2AYUV7yMoVBQPxXNfe7FJcdx16kYtsx6jAPKWQYF1y/go-cid"
|
|
)
|
|
|
|
func TestAddLink(t *testing.T) {
|
|
ctx, context := context.WithCancel(context.Background())
|
|
defer context()
|
|
|
|
ds := mdtest.Mock()
|
|
fishnode := dag.NodeWithData([]byte("fishcakes!"))
|
|
|
|
err := ds.Add(ctx, fishnode)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
nd := new(dag.ProtoNode)
|
|
nnode, err := addLink(ctx, ds, nd, "fish", fishnode)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
fnprime, err := nnode.GetLinkedNode(ctx, ds, "fish")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
fnpkey := fnprime.Cid()
|
|
if !fnpkey.Equals(fishnode.Cid()) {
|
|
t.Fatal("wrong child node found!")
|
|
}
|
|
}
|
|
|
|
func assertNodeAtPath(t *testing.T, ds ipld.DAGService, root *dag.ProtoNode, pth string, exp *cid.Cid) {
|
|
parts := path.SplitList(pth)
|
|
cur := root
|
|
for _, e := range parts {
|
|
nxt, err := cur.GetLinkedProtoNode(context.Background(), ds, e)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cur = nxt
|
|
}
|
|
|
|
curc := cur.Cid()
|
|
if !curc.Equals(exp) {
|
|
t.Fatal("node not as expected at end of path")
|
|
}
|
|
}
|
|
|
|
func TestInsertNode(t *testing.T) {
|
|
root := new(dag.ProtoNode)
|
|
e := NewDagEditor(root, nil)
|
|
|
|
testInsert(t, e, "a", "anodefortesting", false, "")
|
|
testInsert(t, e, "a/b", "data", false, "")
|
|
testInsert(t, e, "a/b/c/d/e", "blah", false, "no link by that name")
|
|
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")
|
|
|
|
c := e.GetNode().Cid()
|
|
|
|
if c.String() != "QmZ8yeT9uD6ouJPNAYt62XffYuXBT6b4mP4obRSE9cJrSt" {
|
|
t.Fatal("output was different than expected: ", c)
|
|
}
|
|
}
|
|
|
|
func testInsert(t *testing.T, e *Editor, path, data string, create bool, experr string) {
|
|
child := dag.NodeWithData([]byte(data))
|
|
err := e.tmp.Add(context.Background(), child)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var c func() *dag.ProtoNode
|
|
if create {
|
|
c = func() *dag.ProtoNode {
|
|
return &dag.ProtoNode{}
|
|
}
|
|
}
|
|
|
|
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, path, data, create, experr)
|
|
}
|
|
|
|
assertNodeAtPath(t, e.tmp, e.root, path, child.Cid())
|
|
}
|