mirror of
https://github.com/ipfs/kubo.git
synced 2026-03-02 23:08:07 +08:00
f4d7369c4a
This change adds the /ipfs/bitswap/1.1.0 protocol. The new protocol adds a 'payload' field to the protobuf message and deprecates the existing 'blocks' field. The 'payload' field is an array of pairs of cid prefixes and block data. The cid prefixes are used to ensure the correct codecs and hash functions are used to handle the block on the receiving end. License: MIT Signed-off-by: Jeromy <why@ipfs.io>
100 lines
1.8 KiB
Go
100 lines
1.8 KiB
Go
package blocks
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
cid "gx/ipfs/QmXUuRadqDq5BuFWzVU6VuKaSjTcNm1gNCtLvvP1TJCW4z/go-cid"
|
|
mh "gx/ipfs/QmYDds3421prZgqKbLpEK7T9Aa2eVdQ7o3YarX1LVLdP2J/go-multihash"
|
|
u "gx/ipfs/Qmb912gdngC1UWwTkhuW8knyRbcWeu5kqkxBpveLmW8bSr/go-ipfs-util"
|
|
)
|
|
|
|
func TestBlocksBasic(t *testing.T) {
|
|
|
|
// Test empty data
|
|
empty := []byte{}
|
|
NewBlock(empty)
|
|
|
|
// Test nil case
|
|
NewBlock(nil)
|
|
|
|
// Test some data
|
|
NewBlock([]byte("Hello world!"))
|
|
}
|
|
|
|
func TestData(t *testing.T) {
|
|
data := []byte("some data")
|
|
block := NewBlock(data)
|
|
|
|
if !bytes.Equal(block.RawData(), data) {
|
|
t.Error("data is wrong")
|
|
}
|
|
}
|
|
|
|
func TestHash(t *testing.T) {
|
|
data := []byte("some other data")
|
|
block := NewBlock(data)
|
|
|
|
hash, err := mh.Sum(data, mh.SHA2_256, -1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !bytes.Equal(block.Multihash(), hash) {
|
|
t.Error("wrong multihash")
|
|
}
|
|
}
|
|
|
|
func TestCid(t *testing.T) {
|
|
data := []byte("yet another data")
|
|
block := NewBlock(data)
|
|
c := block.Cid()
|
|
|
|
if !bytes.Equal(block.Multihash(), c.Hash()) {
|
|
t.Error("key contains wrong data")
|
|
}
|
|
}
|
|
|
|
func TestManualHash(t *testing.T) {
|
|
oldDebugState := u.Debug
|
|
defer (func() {
|
|
u.Debug = oldDebugState
|
|
})()
|
|
|
|
data := []byte("I can't figure out more names .. data")
|
|
hash, err := mh.Sum(data, mh.SHA2_256, -1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
c := cid.NewCidV0(hash)
|
|
|
|
u.Debug = false
|
|
block, err := NewBlockWithCid(data, c)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !bytes.Equal(block.Multihash(), hash) {
|
|
t.Error("wrong multihash")
|
|
}
|
|
|
|
data[5] = byte((uint32(data[5]) + 5) % 256) // Transfrom hash to be different
|
|
block, err = NewBlockWithCid(data, c)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !bytes.Equal(block.Multihash(), hash) {
|
|
t.Error("wrong multihash")
|
|
}
|
|
|
|
u.Debug = true
|
|
|
|
block, err = NewBlockWithCid(data, c)
|
|
if err != ErrWrongHash {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
}
|