0ca2bbaeb3
This provides a way to create a CID without having to read all data into memory before passing it into Sum. Instead, call SumStream passing in an io.Reader.
142 lines
3.1 KiB
Go
142 lines
3.1 KiB
Go
package cid
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
mh "github.com/multiformats/go-multihash"
|
|
)
|
|
|
|
func TestV0Builder(t *testing.T) {
|
|
data := []byte("this is some test content")
|
|
|
|
// Construct c1
|
|
format := V0Builder{}
|
|
c1, err := format.Sum(data)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
reader := bytes.NewReader(data)
|
|
c1a, err := format.SumStream(reader)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !c1a.Equals(c1) {
|
|
t.Fatal("Sum and SumStream create different cids")
|
|
}
|
|
|
|
// Construct c2
|
|
hash, err := mh.Sum(data, mh.SHA2_256, -1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
c2 := NewCidV0(hash)
|
|
|
|
if !c1.Equals(c2) {
|
|
t.Fatal("cids mismatch")
|
|
}
|
|
if c1.Prefix() != c2.Prefix() {
|
|
t.Fatal("prefixes mismatch")
|
|
}
|
|
|
|
reader.Seek(0, 0)
|
|
hash, err = mh.SumStream(reader, mh.SHA2_256, -1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
c2 = NewCidV0(hash)
|
|
if !c1.Equals(c2) {
|
|
t.Fatal("Sum and SumStream create different cids")
|
|
}
|
|
}
|
|
|
|
func TestV1Builder(t *testing.T) {
|
|
data := []byte("this is some test content")
|
|
|
|
// Construct c1
|
|
format := V1Builder{Codec: DagCBOR, MhType: mh.SHA2_256}
|
|
c1, err := format.Sum(data)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
reader := bytes.NewReader(data)
|
|
c1a, err := format.SumStream(reader)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !c1a.Equals(c1) {
|
|
t.Fatal("Sum and SumStream create different cids")
|
|
}
|
|
|
|
// Construct c2
|
|
hash, err := mh.Sum(data, mh.SHA2_256, -1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
c2 := NewCidV1(DagCBOR, hash)
|
|
|
|
if !c1.Equals(c2) {
|
|
t.Fatal("cids mismatch")
|
|
}
|
|
if c1.Prefix() != c2.Prefix() {
|
|
t.Fatal("prefixes mismatch")
|
|
}
|
|
}
|
|
|
|
func TestCodecChange(t *testing.T) {
|
|
t.Run("Prefix-CidV0", func(t *testing.T) {
|
|
p := Prefix{Version: 0, Codec: DagProtobuf, MhType: mh.SHA2_256, MhLength: mh.DefaultLengths[mh.SHA2_256]}
|
|
testCodecChange(t, p)
|
|
})
|
|
t.Run("Prefix-CidV1", func(t *testing.T) {
|
|
p := Prefix{Version: 1, Codec: DagProtobuf, MhType: mh.SHA2_256, MhLength: mh.DefaultLengths[mh.SHA2_256]}
|
|
testCodecChange(t, p)
|
|
})
|
|
t.Run("Prefix-NoChange", func(t *testing.T) {
|
|
p := Prefix{Version: 0, Codec: DagProtobuf, MhType: mh.SHA2_256, MhLength: mh.DefaultLengths[mh.SHA2_256]}
|
|
if p.GetCodec() != DagProtobuf {
|
|
t.Fatal("original builder not using Protobuf codec")
|
|
}
|
|
pn := p.WithCodec(DagProtobuf)
|
|
if pn != p {
|
|
t.Fatal("should have returned same builder")
|
|
}
|
|
})
|
|
t.Run("V0Builder", func(t *testing.T) {
|
|
testCodecChange(t, V0Builder{})
|
|
})
|
|
t.Run("V0Builder-NoChange", func(t *testing.T) {
|
|
b := V0Builder{}
|
|
if b.GetCodec() != DagProtobuf {
|
|
t.Fatal("original builder not using Protobuf codec")
|
|
}
|
|
bn := b.WithCodec(DagProtobuf)
|
|
if bn != b {
|
|
t.Fatal("should have returned same builder")
|
|
}
|
|
})
|
|
t.Run("V1Builder", func(t *testing.T) {
|
|
testCodecChange(t, V1Builder{Codec: DagProtobuf, MhType: mh.SHA2_256})
|
|
})
|
|
}
|
|
|
|
func testCodecChange(t *testing.T, b Builder) {
|
|
data := []byte("this is some test content")
|
|
|
|
if b.GetCodec() != DagProtobuf {
|
|
t.Fatal("original builder not using Protobuf codec")
|
|
}
|
|
|
|
b = b.WithCodec(Raw)
|
|
c, err := b.Sum(data)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if c.Type() != Raw {
|
|
t.Fatal("new cid codec did not change to Raw")
|
|
}
|
|
}
|