mirror of
https://github.com/ipfs/kubo.git
synced 2025-05-17 23:16:11 +08:00
98 lines
2.0 KiB
Go
98 lines
2.0 KiB
Go
package coreunix
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"testing"
|
|
|
|
bserv "github.com/ipfs/go-blockservice"
|
|
merkledag "github.com/ipfs/go-merkledag"
|
|
ft "github.com/ipfs/go-unixfs"
|
|
importer "github.com/ipfs/go-unixfs/importer"
|
|
uio "github.com/ipfs/go-unixfs/io"
|
|
core "github.com/ipfs/kubo/core"
|
|
|
|
cid "github.com/ipfs/go-cid"
|
|
ds "github.com/ipfs/go-datastore"
|
|
dssync "github.com/ipfs/go-datastore/sync"
|
|
bstore "github.com/ipfs/go-ipfs-blockstore"
|
|
chunker "github.com/ipfs/go-ipfs-chunker"
|
|
offline "github.com/ipfs/go-ipfs-exchange-offline"
|
|
u "github.com/ipfs/go-ipfs-util"
|
|
ipld "github.com/ipfs/go-ipld-format"
|
|
)
|
|
|
|
func getDagserv(t *testing.T) ipld.DAGService {
|
|
db := dssync.MutexWrap(ds.NewMapDatastore())
|
|
bs := bstore.NewBlockstore(db)
|
|
blockserv := bserv.New(bs, offline.Exchange(bs))
|
|
return merkledag.NewDAGService(blockserv)
|
|
}
|
|
|
|
func TestMetadata(t *testing.T) {
|
|
ctx := context.Background()
|
|
// Make some random node
|
|
ds := getDagserv(t)
|
|
data := make([]byte, 1000)
|
|
_, err := io.ReadFull(u.NewTimeSeededRand(), data)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r := bytes.NewReader(data)
|
|
nd, err := importer.BuildDagFromReader(ds, chunker.DefaultSplitter(r))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
c := nd.Cid()
|
|
|
|
m := new(ft.Metadata)
|
|
m.MimeType = "THIS IS A TEST"
|
|
|
|
// Such effort, many compromise
|
|
ipfsnode := &core.IpfsNode{DAG: ds}
|
|
|
|
mdk, err := AddMetadataTo(ipfsnode, c.String(), m)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
rec, err := Metadata(ipfsnode, mdk)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if rec.MimeType != m.MimeType {
|
|
t.Fatalf("something went wrong in conversion: '%s' != '%s'", rec.MimeType, m.MimeType)
|
|
}
|
|
|
|
cdk, err := cid.Decode(mdk)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
retnode, err := ds.Get(ctx, cdk)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
rtnpb, ok := retnode.(*merkledag.ProtoNode)
|
|
if !ok {
|
|
t.Fatal("expected protobuf node")
|
|
}
|
|
|
|
ndr, err := uio.NewDagReader(ctx, rtnpb, ds)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
out, err := io.ReadAll(ndr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !bytes.Equal(out, data) {
|
|
t.Fatal("read incorrect data")
|
|
}
|
|
}
|