Skip to content

G
go-ipfs
Unverified 提交 452a1d28 作者: Jakub Sztandera
操作

test: refactor some utities out of mod package 

License: MIT
Signed-off-by: 's avatarJakub Sztandera <kubuxu@protonmail.ch>
上级 4f1a861b
正在显示 包含 154 行增加 和 184 行删除
unixfs/io/dagserv_test.go
package io package io
import ( import (
"bytes"
"fmt"
"io"
"io/ioutil" "io/ioutil"
"testing" "testing"
imp "github.com/ipfs/go-ipfs/importer"
"github.com/ipfs/go-ipfs/importer/chunk"
mdag "github.com/ipfs/go-ipfs/merkledag"
mdagmock "github.com/ipfs/go-ipfs/merkledag/test"
u "gx/ipfs/QmZNVWh8LLjAavuQ2JXuFmuYH3C11xo988vSgp7UQrTRj1/go-ipfs-util"
context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context" context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context"
)
func sizeSplitterGen(size int64) chunk.SplitterGen {
return func(r io.Reader) chunk.Splitter {
return chunk.NewSizeSplitter(r, size)
}
}
func getNode(t testing.TB, dserv mdag.DAGService, data []byte) *mdag.Node {
in := bytes.NewReader(data)
node, err := imp.BuildTrickleDagFromReader(dserv, sizeSplitterGen(500)(in))
if err != nil {
t.Fatal(err)
}
return node
}
func getRandomNode(t testing.TB, dserv mdag.DAGService, size int64) ([]byte, *mdag.Node) { testu "github.com/ipfs/go-ipfs/unixfs/test"
in := io.LimitReader(u.NewTimeSeededRand(), size) )
buf, err := ioutil.ReadAll(in)
if err != nil {
t.Fatal(err)
}
node := getNode(t, dserv, buf)
return buf, node
}
func arrComp(a, b []byte) error {
if len(a) != len(b) {
return fmt.Errorf("Arrays differ in length. %d != %d", len(a), len(b))
}
for i, v := range a {
if v != b[i] {
return fmt.Errorf("Arrays differ at index: %d", i)
}
}
return nil
}
func TestBasicRead(t *testing.T) { func TestBasicRead(t *testing.T) {
dserv := mdagmock.Mock() dserv := testu.GetDAGServ()
inbuf, node := getRandomNode(t, dserv, 1024) inbuf, node := testu.GetRandomNode(t, dserv, 1024)
ctx, closer := context.WithCancel(context.Background()) ctx, closer := context.WithCancel(context.Background())
defer closer() defer closer()
...@@ -71,7 +25,7 @@ func TestBasicRead(t *testing.T) { ...@@ -71,7 +25,7 @@ func TestBasicRead(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
err = arrComp(inbuf, outbuf) err = testu.ArrComp(inbuf, outbuf)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
......
unixfs/mod/dagmodifier_test.go
...@@ -2,7 +2,6 @@ package mod ...@@ -2,7 +2,6 @@ package mod
import ( import (
"fmt" "fmt"
"io"
"io/ioutil" "io/ioutil"
"os" "os"
"testing" "testing"
...@@ -10,13 +9,12 @@ import ( ...@@ -10,13 +9,12 @@ import (
"github.com/ipfs/go-ipfs/blocks/blockstore" "github.com/ipfs/go-ipfs/blocks/blockstore"
bs "github.com/ipfs/go-ipfs/blockservice" bs "github.com/ipfs/go-ipfs/blockservice"
"github.com/ipfs/go-ipfs/exchange/offline" "github.com/ipfs/go-ipfs/exchange/offline"
imp "github.com/ipfs/go-ipfs/importer"
"github.com/ipfs/go-ipfs/importer/chunk"
h "github.com/ipfs/go-ipfs/importer/helpers" h "github.com/ipfs/go-ipfs/importer/helpers"
trickle "github.com/ipfs/go-ipfs/importer/trickle" trickle "github.com/ipfs/go-ipfs/importer/trickle"
mdag "github.com/ipfs/go-ipfs/merkledag" mdag "github.com/ipfs/go-ipfs/merkledag"
ft "github.com/ipfs/go-ipfs/unixfs" ft "github.com/ipfs/go-ipfs/unixfs"
uio "github.com/ipfs/go-ipfs/unixfs/io" uio "github.com/ipfs/go-ipfs/unixfs/io"
testu "github.com/ipfs/go-ipfs/unixfs/test"
u "gx/ipfs/QmZNVWh8LLjAavuQ2JXuFmuYH3C11xo988vSgp7UQrTRj1/go-ipfs-util" u "gx/ipfs/QmZNVWh8LLjAavuQ2JXuFmuYH3C11xo988vSgp7UQrTRj1/go-ipfs-util"
context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context" context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context"
...@@ -24,14 +22,6 @@ import ( ...@@ -24,14 +22,6 @@ import (
"gx/ipfs/QmbzuUusHqaLLoNTDEVLcSF6vZDHZDLPC7p4bztRvvkXxU/go-datastore/sync" "gx/ipfs/QmbzuUusHqaLLoNTDEVLcSF6vZDHZDLPC7p4bztRvvkXxU/go-datastore/sync"
) )
func getMockDagServ(t testing.TB) mdag.DAGService {
dstore := ds.NewMapDatastore()
tsds := sync.MutexWrap(dstore)
bstore := blockstore.NewBlockstore(tsds)
bserv := bs.New(bstore, offline.Exchange(bstore))
return mdag.NewDAGService(bserv)
}
func getMockDagServAndBstore(t testing.TB) (mdag.DAGService, blockstore.GCBlockstore) { func getMockDagServAndBstore(t testing.TB) (mdag.DAGService, blockstore.GCBlockstore) {
dstore := ds.NewMapDatastore() dstore := ds.NewMapDatastore()
tsds := sync.MutexWrap(dstore) tsds := sync.MutexWrap(dstore)
...@@ -41,26 +31,6 @@ func getMockDagServAndBstore(t testing.TB) (mdag.DAGService, blockstore.GCBlocks ...@@ -41,26 +31,6 @@ func getMockDagServAndBstore(t testing.TB) (mdag.DAGService, blockstore.GCBlocks
return dserv, bstore return dserv, bstore
} }
func getNode(t testing.TB, dserv mdag.DAGService, size int64) ([]byte, *mdag.Node) {
in := io.LimitReader(u.NewTimeSeededRand(), size)
node, err := imp.BuildTrickleDagFromReader(dserv, sizeSplitterGen(500)(in))
if err != nil {
t.Fatal(err)
}
dr, err := uio.NewDagReader(context.Background(), node, dserv)
if err != nil {
t.Fatal(err)
}
b, err := ioutil.ReadAll(dr)
if err != nil {
t.Fatal(err)
}
return b, node
}
func testModWrite(t *testing.T, beg, size uint64, orig []byte, dm *DagModifier) []byte { func testModWrite(t *testing.T, beg, size uint64, orig []byte, dm *DagModifier) []byte {
newdata := make([]byte, size) newdata := make([]byte, size)
r := u.NewTimeSeededRand() r := u.NewTimeSeededRand()
...@@ -100,26 +70,20 @@ func testModWrite(t *testing.T, beg, size uint64, orig []byte, dm *DagModifier) ...@@ -100,26 +70,20 @@ func testModWrite(t *testing.T, beg, size uint64, orig []byte, dm *DagModifier)
t.Fatal(err) t.Fatal(err)
} }
err = arrComp(after, orig) err = testu.ArrComp(after, orig)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
return orig return orig
} }
func sizeSplitterGen(size int64) chunk.SplitterGen {
return func(r io.Reader) chunk.Splitter {
return chunk.NewSizeSplitter(r, size)
}
}
func TestDagModifierBasic(t *testing.T) { func TestDagModifierBasic(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
b, n := getNode(t, dserv, 50000) b, n := testu.GetRandomNode(t, dserv, 50000)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -168,13 +132,13 @@ func TestDagModifierBasic(t *testing.T) { ...@@ -168,13 +132,13 @@ func TestDagModifierBasic(t *testing.T) {
} }
func TestMultiWrite(t *testing.T) { func TestMultiWrite(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -214,20 +178,20 @@ func TestMultiWrite(t *testing.T) { ...@@ -214,20 +178,20 @@ func TestMultiWrite(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
err = arrComp(rbuf, data) err = testu.ArrComp(rbuf, data)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
} }
func TestMultiWriteAndFlush(t *testing.T) { func TestMultiWriteAndFlush(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -262,20 +226,20 @@ func TestMultiWriteAndFlush(t *testing.T) { ...@@ -262,20 +226,20 @@ func TestMultiWriteAndFlush(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
err = arrComp(rbuf, data) err = testu.ArrComp(rbuf, data)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
} }
func TestWriteNewFile(t *testing.T) { func TestWriteNewFile(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -306,19 +270,19 @@ func TestWriteNewFile(t *testing.T) { ...@@ -306,19 +270,19 @@ func TestWriteNewFile(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
if err := arrComp(data, towrite); err != nil { if err := testu.ArrComp(data, towrite); err != nil {
t.Fatal(err) t.Fatal(err)
} }
} }
func TestMultiWriteCoal(t *testing.T) { func TestMultiWriteCoal(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -351,20 +315,20 @@ func TestMultiWriteCoal(t *testing.T) { ...@@ -351,20 +315,20 @@ func TestMultiWriteCoal(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
err = arrComp(rbuf, data) err = testu.ArrComp(rbuf, data)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
} }
func TestLargeWriteChunks(t *testing.T) { func TestLargeWriteChunks(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -390,19 +354,19 @@ func TestLargeWriteChunks(t *testing.T) { ...@@ -390,19 +354,19 @@ func TestLargeWriteChunks(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
if err = arrComp(out, data); err != nil { if err = testu.ArrComp(out, data); err != nil {
t.Fatal(err) t.Fatal(err)
} }
} }
func TestDagTruncate(t *testing.T) { func TestDagTruncate(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
b, n := getNode(t, dserv, 50000) b, n := testu.GetRandomNode(t, dserv, 50000)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -430,7 +394,7 @@ func TestDagTruncate(t *testing.T) { ...@@ -430,7 +394,7 @@ func TestDagTruncate(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
if err = arrComp(out, b[:12345]); err != nil { if err = testu.ArrComp(out, b[:12345]); err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -464,12 +428,12 @@ func TestDagTruncate(t *testing.T) { ...@@ -464,12 +428,12 @@ func TestDagTruncate(t *testing.T) {
} }
func TestSparseWrite(t *testing.T) { func TestSparseWrite(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -496,18 +460,18 @@ func TestSparseWrite(t *testing.T) { ...@@ -496,18 +460,18 @@ func TestSparseWrite(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
if err = arrComp(out, buf); err != nil { if err = testu.ArrComp(out, buf); err != nil {
t.Fatal(err) t.Fatal(err)
} }
} }
func TestSeekPastEndWrite(t *testing.T) { func TestSeekPastEndWrite(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -543,18 +507,18 @@ func TestSeekPastEndWrite(t *testing.T) { ...@@ -543,18 +507,18 @@ func TestSeekPastEndWrite(t *testing.T) {
t.Fatal(err) t.Fatal(err)
} }
if err = arrComp(out, buf); err != nil { if err = testu.ArrComp(out, buf); err != nil {
t.Fatal(err) t.Fatal(err)
} }
} }
func TestRelativeSeek(t *testing.T) { func TestRelativeSeek(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -579,13 +543,12 @@ func TestRelativeSeek(t *testing.T) { ...@@ -579,13 +543,12 @@ func TestRelativeSeek(t *testing.T) {
} }
func TestInvalidSeek(t *testing.T) { func TestInvalidSeek(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
n := testu.GetEmptyNode(t, dserv)
_, n := getNode(t, dserv, 0)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -597,13 +560,13 @@ func TestInvalidSeek(t *testing.T) { ...@@ -597,13 +560,13 @@ func TestInvalidSeek(t *testing.T) {
} }
func TestEndSeek(t *testing.T) { func TestEndSeek(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -630,13 +593,13 @@ func TestEndSeek(t *testing.T) { ...@@ -630,13 +593,13 @@ func TestEndSeek(t *testing.T) {
} }
func TestReadAndSeek(t *testing.T) { func TestReadAndSeek(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -698,13 +661,13 @@ func TestReadAndSeek(t *testing.T) { ...@@ -698,13 +661,13 @@ func TestReadAndSeek(t *testing.T) {
} }
func TestCtxRead(t *testing.T) { func TestCtxRead(t *testing.T) {
dserv := getMockDagServ(t) dserv := testu.GetDAGServ()
_, n := getNode(t, dserv, 0) n := testu.GetEmptyNode(t, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -720,7 +683,7 @@ func TestCtxRead(t *testing.T) { ...@@ -720,7 +683,7 @@ func TestCtxRead(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
err = arrComp(readBuf, []byte{0, 1, 2, 3}) err = testu.ArrComp(readBuf, []byte{0, 1, 2, 3})
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
...@@ -730,14 +693,14 @@ func TestCtxRead(t *testing.T) { ...@@ -730,14 +693,14 @@ func TestCtxRead(t *testing.T) {
func BenchmarkDagmodWrite(b *testing.B) { func BenchmarkDagmodWrite(b *testing.B) {
b.StopTimer() b.StopTimer()
dserv := getMockDagServ(b) dserv := testu.GetDAGServ()
_, n := getNode(b, dserv, 0) n := testu.GetEmptyNode(b, dserv)
ctx, cancel := context.WithCancel(context.Background()) ctx, cancel := context.WithCancel(context.Background())
defer cancel() defer cancel()
wrsize := 4096 wrsize := 4096
dagmod, err := NewDagModifier(ctx, n, dserv, sizeSplitterGen(512)) dagmod, err := NewDagModifier(ctx, n, dserv, testu.SizeSplitterGen(512))
if err != nil { if err != nil {
b.Fatal(err) b.Fatal(err)
} }
...@@ -756,43 +719,3 @@ func BenchmarkDagmodWrite(b *testing.B) { ...@@ -756,43 +719,3 @@ func BenchmarkDagmodWrite(b *testing.B) {
} }
} }
} }
func arrComp(a, b []byte) error {
if len(a) != len(b) {
return fmt.Errorf("Arrays differ in length. %d != %d", len(a), len(b))
}
for i, v := range a {
if v != b[i] {
return fmt.Errorf("Arrays differ at index: %d", i)
}
}
return nil
}
func printDag(nd *mdag.Node, ds mdag.DAGService, indent int) {
pbd, err := ft.FromBytes(nd.Data())
if err != nil {
panic(err)
}
for i := 0; i < indent; i++ {
fmt.Print(" ")
}
fmt.Printf("{size = %d, type = %s, children = %d", pbd.GetFilesize(), pbd.GetType().String(), len(pbd.GetBlocksizes()))
if len(nd.Links) > 0 {
fmt.Println()
}
for _, lnk := range nd.Links {
child, err := lnk.GetNode(context.Background(), ds)
if err != nil {
panic(err)
}
printDag(child, ds, indent+1)
}
if len(nd.Links) > 0 {
for i := 0; i < indent; i++ {
fmt.Print(" ")
}
}
fmt.Println("}")
}
unixfs/test/utils.go 0 → 100644
package testu
import (
"bytes"
"fmt"
"io"
"io/ioutil"
"testing"
imp "github.com/ipfs/go-ipfs/importer"
"github.com/ipfs/go-ipfs/importer/chunk"
mdag "github.com/ipfs/go-ipfs/merkledag"
mdagmock "github.com/ipfs/go-ipfs/merkledag/test"
ft "github.com/ipfs/go-ipfs/unixfs"
u "gx/ipfs/QmZNVWh8LLjAavuQ2JXuFmuYH3C11xo988vSgp7UQrTRj1/go-ipfs-util"
context "gx/ipfs/QmZy2y8t9zQH2a1b8q2ZSLKp17ATuJoCNxxyMFG5qFExpt/go-net/context"
)
func SizeSplitterGen(size int64) chunk.SplitterGen {
return func(r io.Reader) chunk.Splitter {
return chunk.NewSizeSplitter(r, size)
}
}
func GetDAGServ() mdag.DAGService {
return mdagmock.Mock()
}
func GetNode(t testing.TB, dserv mdag.DAGService, data []byte) *mdag.Node {
in := bytes.NewReader(data)
node, err := imp.BuildTrickleDagFromReader(dserv, SizeSplitterGen(500)(in))
if err != nil {
t.Fatal(err)
}
return node
}
func GetEmptyNode(t testing.TB, dserv mdag.DAGService) *mdag.Node {
return GetNode(t, dserv, []byte{})
}
func GetRandomNode(t testing.TB, dserv mdag.DAGService, size int64) ([]byte, *mdag.Node) {
in := io.LimitReader(u.NewTimeSeededRand(), size)
buf, err := ioutil.ReadAll(in)
if err != nil {
t.Fatal(err)
}
node := GetNode(t, dserv, buf)
return buf, node
}
func ArrComp(a, b []byte) error {
if len(a) != len(b) {
return fmt.Errorf("Arrays differ in length. %d != %d", len(a), len(b))
}
for i, v := range a {
if v != b[i] {
return fmt.Errorf("Arrays differ at index: %d", i)
}
}
return nil
}
func PrintDag(nd *mdag.Node, ds mdag.DAGService, indent int) {
pbd, err := ft.FromBytes(nd.Data())
if err != nil {
panic(err)
}
for i := 0; i < indent; i++ {
fmt.Print(" ")
}
fmt.Printf("{size = %d, type = %s, children = %d", pbd.GetFilesize(), pbd.GetType().String(), len(pbd.GetBlocksizes()))
if len(nd.Links) > 0 {
fmt.Println()
}
for _, lnk := range nd.Links {
child, err := lnk.GetNode(context.Background(), ds)
if err != nil {
panic(err)
}
PrintDag(child, ds, indent+1)
}
if len(nd.Links) > 0 {
for i := 0; i < indent; i++ {
fmt.Print(" ")
}
}
fmt.Println("}")
}
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论