mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-22 19:07:48 +08:00
758e00bc4c
This extracts the blocks/blockstore package and renames the blocks/blockstore/util package to /blocks/blockstoreutil (because util depends on Pin and I don't plan to extract Pin and its depedencies). The history of blocks/blockstore has been preserved. It has been gx'ed and imported. Imports have been rewritten accordingly and re-ordered. License: MIT Signed-off-by: Hector Sanjuan <hector@protocol.ai>
98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package bstest
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
. "github.com/ipfs/go-ipfs/blockservice"
|
|
offline "github.com/ipfs/go-ipfs/exchange/offline"
|
|
|
|
u "gx/ipfs/QmNiJuT8Ja3hMVpBHXv3Q6dwmperaQ6JjLtpMQgMCD7xvx/go-ipfs-util"
|
|
ds "gx/ipfs/QmPpegoMqhAEqjncrzArm7KVWAkCm78rqL2DPuNjhPrshg/go-datastore"
|
|
dssync "gx/ipfs/QmPpegoMqhAEqjncrzArm7KVWAkCm78rqL2DPuNjhPrshg/go-datastore/sync"
|
|
blockstore "gx/ipfs/QmTVDM4LCSUMFNQzbDLL9zQwp8usE6QHymFdh3h8vL9v6b/go-ipfs-blockstore"
|
|
cid "gx/ipfs/QmcZfnkapfECQGcLZaf9B79NRg7cRa9EnZh4LSbkCzwNvY/go-cid"
|
|
blocks "gx/ipfs/Qmej7nf81hi2x2tvjRBF3mcp74sQyuDH4VMYDGd1YtXjb2/go-block-format"
|
|
)
|
|
|
|
func newObject(data []byte) blocks.Block {
|
|
return blocks.NewBlock(data)
|
|
}
|
|
|
|
func TestBlocks(t *testing.T) {
|
|
bstore := blockstore.NewBlockstore(dssync.MutexWrap(ds.NewMapDatastore()))
|
|
bs := New(bstore, offline.Exchange(bstore))
|
|
defer bs.Close()
|
|
|
|
o := newObject([]byte("beep boop"))
|
|
h := cid.NewCidV0(u.Hash([]byte("beep boop")))
|
|
if !o.Cid().Equals(h) {
|
|
t.Error("Block key and data multihash key not equal")
|
|
}
|
|
|
|
err := bs.AddBlock(o)
|
|
if err != nil {
|
|
t.Error("failed to add block to BlockService", err)
|
|
return
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
|
|
defer cancel()
|
|
b2, err := bs.GetBlock(ctx, o.Cid())
|
|
if err != nil {
|
|
t.Error("failed to retrieve block from BlockService", err)
|
|
return
|
|
}
|
|
|
|
if !o.Cid().Equals(b2.Cid()) {
|
|
t.Error("Block keys not equal.")
|
|
}
|
|
|
|
if !bytes.Equal(o.RawData(), b2.RawData()) {
|
|
t.Error("Block data is not equal.")
|
|
}
|
|
}
|
|
|
|
func makeObjects(n int) []blocks.Block {
|
|
var out []blocks.Block
|
|
for i := 0; i < n; i++ {
|
|
out = append(out, newObject([]byte(fmt.Sprintf("object %d", i))))
|
|
}
|
|
return out
|
|
}
|
|
|
|
func TestGetBlocksSequential(t *testing.T) {
|
|
var servs = Mocks(4)
|
|
for _, s := range servs {
|
|
defer s.Close()
|
|
}
|
|
objs := makeObjects(50)
|
|
|
|
var cids []*cid.Cid
|
|
for _, o := range objs {
|
|
cids = append(cids, o.Cid())
|
|
servs[0].AddBlock(o)
|
|
}
|
|
|
|
t.Log("one instance at a time, get blocks concurrently")
|
|
|
|
for i := 1; i < len(servs); i++ {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*50)
|
|
defer cancel()
|
|
out := servs[i].GetBlocks(ctx, cids)
|
|
gotten := make(map[string]blocks.Block)
|
|
for blk := range out {
|
|
if _, ok := gotten[blk.Cid().KeyString()]; ok {
|
|
t.Fatal("Got duplicate block!")
|
|
}
|
|
gotten[blk.Cid().KeyString()] = blk
|
|
}
|
|
if len(gotten) != len(objs) {
|
|
t.Fatalf("Didnt get enough blocks back: %d/%d", len(gotten), len(objs))
|
|
}
|
|
}
|
|
}
|