mirror of https://github.com/vbatts/tar-split.git
85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package storage
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestGetter(t *testing.T) {
|
|
fgp := NewBufferFileGetPutter()
|
|
files := map[string]map[string][]byte{
|
|
"file1.txt": {"foo": []byte{60, 60, 48, 48, 0, 0, 0, 0}},
|
|
"file2.txt": {"bar": []byte{45, 196, 22, 240, 0, 0, 0, 0}},
|
|
}
|
|
for n, b := range files {
|
|
for body, sum := range b {
|
|
_, csum, err := fgp.Put(n, bytes.NewBufferString(body))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if !bytes.Equal(csum, sum) {
|
|
t.Errorf("checksum: expected 0x%x; got 0x%x", sum, csum)
|
|
}
|
|
}
|
|
}
|
|
for n, b := range files {
|
|
for body := range b {
|
|
r, err := fgp.Get(n)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
buf, err := io.ReadAll(r)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if body != string(buf) {
|
|
t.Errorf("expected %q, got %q", body, string(buf))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPutter(t *testing.T) {
|
|
fp := NewDiscardFilePutter()
|
|
// map[filename]map[body]crc64sum
|
|
files := map[string]map[string][]byte{
|
|
"file1.txt": {"foo": []byte{60, 60, 48, 48, 0, 0, 0, 0}},
|
|
"file2.txt": {"bar": []byte{45, 196, 22, 240, 0, 0, 0, 0}},
|
|
"file3.txt": {"baz": []byte{32, 68, 22, 240, 0, 0, 0, 0}},
|
|
"file4.txt": {"bif": []byte{48, 9, 150, 240, 0, 0, 0, 0}},
|
|
}
|
|
for n, b := range files {
|
|
for body, sum := range b {
|
|
_, csum, err := fp.Put(n, bytes.NewBufferString(body))
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if !bytes.Equal(csum, sum) {
|
|
t.Errorf("checksum on %q: expected %v; got %v", n, sum, csum)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkPutter(b *testing.B) {
|
|
files := []string{
|
|
strings.Repeat("foo", 1000),
|
|
strings.Repeat("bar", 1000),
|
|
strings.Repeat("baz", 1000),
|
|
strings.Repeat("fooz", 1000),
|
|
strings.Repeat("vbatts", 1000),
|
|
strings.Repeat("systemd", 1000),
|
|
}
|
|
for i := 0; i < b.N; i++ {
|
|
fgp := NewBufferFileGetPutter()
|
|
for n, body := range files {
|
|
if _, _, err := fgp.Put(fmt.Sprintf("%d", n), bytes.NewBufferString(body)); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
}
|