restic/archiver_test.go

57 lines
983 B
Go
Raw Normal View History

2014-11-16 20:41:05 +00:00
package khepri_test
import (
"bytes"
"io"
"math/rand"
"testing"
"github.com/fd0/khepri"
2014-11-16 20:41:05 +00:00
"github.com/fd0/khepri/chunker"
)
func get_random(seed, count int) []byte {
buf := make([]byte, count)
rnd := rand.New(rand.NewSource(23))
for i := 0; i < count; i += 4 {
r := rnd.Uint32()
buf[i] = byte(r)
buf[i+1] = byte(r >> 8)
buf[i+2] = byte(r >> 16)
buf[i+3] = byte(r >> 24)
}
return buf
}
func BenchmarkChunkEncrypt(b *testing.B) {
data := get_random(23, 10<<20) // 10MiB
be := setupBackend(b)
defer teardownBackend(b, be)
key := setupKey(b, be, "geheim")
chunkBuf := make([]byte, chunker.MaxSize)
2014-11-16 20:41:05 +00:00
b.ResetTimer()
b.SetBytes(int64(len(data)))
for i := 0; i < b.N; i++ {
ch := chunker.New(bytes.NewReader(data))
for {
chunk_data, err := ch.Next(chunkBuf)
2014-11-16 20:41:05 +00:00
if err == io.EOF {
break
}
ok(b, err)
buf := make([]byte, khepri.CiphertextExtension+chunker.MaxSize)
_, err = key.Encrypt(buf, chunk_data.Data)
2014-11-16 20:41:05 +00:00
ok(b, err)
}
}
}