mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-23 11:27:42 +08:00
92d08db7a5
- updated go-ctxgroup and goprocess ctxgroup: AddChildGroup was changed to AddChild. Used in two files: - p2p/net/mock/mock_net.go - routing/dht/dht.go - updated context from hg repo to git prev. commit in hg was ad01a6fcc8a19d3a4478c836895ffe883bd2ceab. (context: make parentCancelCtx iterative) represents commit 84f8955a887232b6308d79c68b8db44f64df455c in git repo - updated context to master (b6fdb7d8a4ccefede406f8fe0f017fb58265054c) Aaron Jacobs (2): net/context: Don't accept a context in the DoSomethingSlow example. context: Be clear that users must cancel the result of WithCancel. Andrew Gerrand (1): go.net: use golang.org/x/... import paths Bryan C. Mills (1): net/context: Don't leak goroutines in Done example. Damien Neil (1): context: fix removal of cancelled timer contexts from parent David Symonds (2): context: Fix WithValue example code. net: add import comments. Sameer Ajmani (1): context: fix TestAllocs to account for ints in interfaces
94 lines
2.3 KiB
Go
94 lines
2.3 KiB
Go
package blockservice
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
"time"
|
|
|
|
ds "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore"
|
|
dssync "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore/sync"
|
|
"github.com/jbenet/go-ipfs/Godeps/_workspace/src/golang.org/x/net/context"
|
|
blocks "github.com/jbenet/go-ipfs/blocks"
|
|
blockstore "github.com/jbenet/go-ipfs/blocks/blockstore"
|
|
blocksutil "github.com/jbenet/go-ipfs/blocks/blocksutil"
|
|
offline "github.com/jbenet/go-ipfs/exchange/offline"
|
|
u "github.com/jbenet/go-ipfs/util"
|
|
)
|
|
|
|
func TestBlocks(t *testing.T) {
|
|
bstore := blockstore.NewBlockstore(dssync.MutexWrap(ds.NewMapDatastore()))
|
|
bs, err := New(bstore, offline.Exchange(bstore))
|
|
if err != nil {
|
|
t.Error("failed to construct block service", err)
|
|
return
|
|
}
|
|
defer bs.Close()
|
|
|
|
b := blocks.NewBlock([]byte("beep boop"))
|
|
h := u.Hash([]byte("beep boop"))
|
|
if !bytes.Equal(b.Multihash, h) {
|
|
t.Error("Block Multihash and data multihash not equal")
|
|
}
|
|
|
|
if b.Key() != u.Key(h) {
|
|
t.Error("Block key and data multihash key not equal")
|
|
}
|
|
|
|
k, err := bs.AddBlock(b)
|
|
if err != nil {
|
|
t.Error("failed to add block to BlockService", err)
|
|
return
|
|
}
|
|
|
|
if k != b.Key() {
|
|
t.Error("returned key is not equal to block key", err)
|
|
}
|
|
|
|
ctx, _ := context.WithTimeout(context.TODO(), time.Second*5)
|
|
b2, err := bs.GetBlock(ctx, b.Key())
|
|
if err != nil {
|
|
t.Error("failed to retrieve block from BlockService", err)
|
|
return
|
|
}
|
|
|
|
if b.Key() != b2.Key() {
|
|
t.Error("Block keys not equal.")
|
|
}
|
|
|
|
if !bytes.Equal(b.Data, b2.Data) {
|
|
t.Error("Block data is not equal.")
|
|
}
|
|
}
|
|
|
|
func TestGetBlocksSequential(t *testing.T) {
|
|
var servs = Mocks(t, 4)
|
|
for _, s := range servs {
|
|
defer s.Close()
|
|
}
|
|
bg := blocksutil.NewBlockGenerator()
|
|
blks := bg.Blocks(50)
|
|
|
|
var keys []u.Key
|
|
for _, blk := range blks {
|
|
keys = append(keys, blk.Key())
|
|
servs[0].AddBlock(blk)
|
|
}
|
|
|
|
t.Log("one instance at a time, get blocks concurrently")
|
|
|
|
for i := 1; i < len(servs); i++ {
|
|
ctx, _ := context.WithTimeout(context.TODO(), time.Second*50)
|
|
out := servs[i].GetBlocks(ctx, keys)
|
|
gotten := make(map[u.Key]*blocks.Block)
|
|
for blk := range out {
|
|
if _, ok := gotten[blk.Key()]; ok {
|
|
t.Fatal("Got duplicate block!")
|
|
}
|
|
gotten[blk.Key()] = blk
|
|
}
|
|
if len(gotten) != len(blks) {
|
|
t.Fatalf("Didnt get enough blocks back: %d/%d", len(gotten), len(blks))
|
|
}
|
|
}
|
|
}
|