mirror of
https://github.com/ipfs/kubo.git
synced 2025-09-14 01:35:23 +08:00
88 lines
1.4 KiB
Go
88 lines
1.4 KiB
Go
package util
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
mh "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-multihash"
|
|
)
|
|
|
|
func TestKey(t *testing.T) {
|
|
|
|
h1, err := mh.Sum([]byte("beep boop"), mh.SHA2_256, -1)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
k1 := Key(h1)
|
|
h2 := mh.Multihash(k1)
|
|
k2 := Key(h2)
|
|
|
|
if !bytes.Equal(h1, h2) {
|
|
t.Error("Multihashes not equal.")
|
|
}
|
|
|
|
if k1 != k2 {
|
|
t.Error("Keys not equal.")
|
|
}
|
|
}
|
|
|
|
func TestByteChanReader(t *testing.T) {
|
|
data := make([]byte, 1024*1024)
|
|
r := NewTimeSeededRand()
|
|
r.Read(data)
|
|
dch := make(chan []byte, 8)
|
|
|
|
go func() {
|
|
beg := 0
|
|
for i := 0; i < len(data); {
|
|
i += rand.Intn(100) + 1
|
|
if i > len(data) {
|
|
i = len(data)
|
|
}
|
|
dch <- data[beg:i]
|
|
beg = i
|
|
}
|
|
close(dch)
|
|
}()
|
|
|
|
read := NewByteChanReader(dch)
|
|
out, err := ioutil.ReadAll(read)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !bytes.Equal(out, data) {
|
|
t.Fatal("Reader failed to stream correct bytes")
|
|
}
|
|
}
|
|
|
|
func TestXOR(t *testing.T) {
|
|
cases := [][3][]byte{
|
|
[3][]byte{
|
|
[]byte{0xFF, 0xFF, 0xFF},
|
|
[]byte{0xFF, 0xFF, 0xFF},
|
|
[]byte{0x00, 0x00, 0x00},
|
|
},
|
|
[3][]byte{
|
|
[]byte{0x00, 0xFF, 0x00},
|
|
[]byte{0xFF, 0xFF, 0xFF},
|
|
[]byte{0xFF, 0x00, 0xFF},
|
|
},
|
|
[3][]byte{
|
|
[]byte{0x55, 0x55, 0x55},
|
|
[]byte{0x55, 0xFF, 0xAA},
|
|
[]byte{0x00, 0xAA, 0xFF},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
r := XOR(c[0], c[1])
|
|
if !bytes.Equal(r, c[2]) {
|
|
t.Error("XOR failed")
|
|
}
|
|
}
|
|
}
|