2019-04-30 01:24:37 +02:00
|
|
|
package utils
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2019-05-03 05:28:31 +02:00
|
|
|
"io/ioutil"
|
|
|
|
"math/rand"
|
2019-05-03 16:30:35 +02:00
|
|
|
"os"
|
2019-04-30 01:24:37 +02:00
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
)
|
|
|
|
|
2019-05-03 05:28:31 +02:00
|
|
|
func init() {
|
|
|
|
rand.Seed(0)
|
|
|
|
bigBuff := make([]byte, 75000000)
|
|
|
|
rand.Read(bigBuff)
|
|
|
|
ioutil.WriteFile("bigfile.test", bigBuff, 0666)
|
|
|
|
}
|
|
|
|
|
|
|
|
func BenchmarkMD5(b *testing.B) {
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
|
|
MD5HashFile("bigfile.test")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func BenchmarkXXHash(b *testing.B) {
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
|
|
XXHashFile("bigfile.test")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func BenchmarkImoHash(b *testing.B) {
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
|
|
IMOHashFile("bigfile.test")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-04-30 01:24:37 +02:00
|
|
|
func TestExists(t *testing.T) {
|
2019-05-03 05:28:31 +02:00
|
|
|
assert.True(t, Exists("bigfile.test"))
|
|
|
|
assert.False(t, Exists("doesnotexist"))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestMD5HashFile(t *testing.T) {
|
|
|
|
b, err := MD5HashFile("bigfile.test")
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, "9fed05acbacbc6a36555c998501c21f6", fmt.Sprintf("%x", b))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestIMOHashFile(t *testing.T) {
|
|
|
|
b, err := IMOHashFile("bigfile.test")
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, "c0d1e123a96a598ea801cc503d3db8c0", fmt.Sprintf("%x", b))
|
2019-04-30 01:24:37 +02:00
|
|
|
}
|
|
|
|
|
2019-05-03 05:28:31 +02:00
|
|
|
func TestXXHashFile(t *testing.T) {
|
|
|
|
b, err := XXHashFile("bigfile.test")
|
2019-04-30 01:24:37 +02:00
|
|
|
assert.Nil(t, err)
|
2019-05-03 05:28:31 +02:00
|
|
|
assert.Equal(t, "f2da6ee7e75e8324", fmt.Sprintf("%x", b))
|
2019-04-30 01:24:37 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestSHA256(t *testing.T) {
|
|
|
|
assert.Equal(t, "09ca7e4eaa6e8ae9c7d261167129184883644d07dfba7cbfbc4c8a2e08360d5b", SHA256("hello, world"))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestByteCountDecimal(t *testing.T) {
|
|
|
|
assert.Equal(t, "10.0 kB", ByteCountDecimal(10000))
|
|
|
|
}
|
2019-05-03 16:30:35 +02:00
|
|
|
|
|
|
|
func TestMissingChunks(t *testing.T) {
|
|
|
|
fileSize := 100
|
|
|
|
chunkSize := 10
|
|
|
|
rand.Seed(1)
|
|
|
|
bigBuff := make([]byte, fileSize)
|
|
|
|
rand.Read(bigBuff)
|
|
|
|
ioutil.WriteFile("missing.test", bigBuff, 0644)
|
|
|
|
empty := make([]byte, chunkSize)
|
|
|
|
f, err := os.OpenFile("missing.test", os.O_RDWR, 0644)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
for block := 0; block < fileSize/chunkSize; block++ {
|
|
|
|
if block == 0 || block == 4 || block == 5 || block >= 7 {
|
|
|
|
f.WriteAt(empty, int64(block*chunkSize))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
f.Close()
|
|
|
|
|
|
|
|
chunkRanges := MissingChunks("missing.test", int64(fileSize), chunkSize)
|
|
|
|
assert.Equal(t, []int64{10, 0, 1, 40, 2, 70, 3}, chunkRanges)
|
|
|
|
|
|
|
|
chunks := ChunkRangesToChunks(chunkRanges)
|
|
|
|
assert.Equal(t, []int64{0, 40, 50, 70, 80, 90}, chunks)
|
|
|
|
|
|
|
|
os.Remove("missing.test")
|
|
|
|
}
|
|
|
|
|
|
|
|
// func Test1(t *testing.T) {
|
|
|
|
// chunkRanges := MissingChunks("../../m/bigfile.test", int64(75000000), 1024*64/2)
|
|
|
|
// fmt.Println(chunkRanges)
|
|
|
|
// fmt.Println(ChunkRangesToChunks((chunkRanges)))
|
|
|
|
// assert.Nil(t, nil)
|
|
|
|
// }
|