package server_test import ( "bytes" "crypto/rand" "crypto/sha256" "encoding/json" "flag" "io" "testing" "github.com/restic/restic" "github.com/restic/restic/backend" . "github.com/restic/restic/test" ) var benchTestDir = flag.String("test.dir", ".", "dir used in benchmarks (default: .)") type testJSONStruct struct { Foo uint32 Bar string Baz []byte } var serverTests = []testJSONStruct{ testJSONStruct{Foo: 23, Bar: "Teststring", Baz: []byte("xx")}, } func TestSaveJSON(t *testing.T) { server := SetupBackend(t) defer TeardownBackend(t, server) key := SetupKey(t, server, "geheim") server.SetKey(key) for _, obj := range serverTests { data, err := json.Marshal(obj) OK(t, err) data = append(data, '\n') h := sha256.Sum256(data) blob, err := server.SaveJSON(backend.Tree, obj) OK(t, err) Assert(t, bytes.Equal(h[:], blob.ID), "TestSaveJSON: wrong plaintext ID: expected %02x, got %02x", h, blob.ID) } } func BenchmarkSaveJSON(t *testing.B) { server := SetupBackend(t) defer TeardownBackend(t, server) key := SetupKey(t, server, "geheim") server.SetKey(key) obj := serverTests[0] data, err := json.Marshal(obj) OK(t, err) data = append(data, '\n') h := sha256.Sum256(data) t.ResetTimer() for i := 0; i < t.N; i++ { blob, err := server.SaveJSON(backend.Tree, obj) OK(t, err) Assert(t, bytes.Equal(h[:], blob.ID), "TestSaveJSON: wrong plaintext ID: expected %02x, got %02x", h, blob.ID) } } var testSizes = []int{5, 23, 2<<18 + 23, 1 << 20} func TestSaveFrom(t *testing.T) { server := SetupBackend(t) defer TeardownBackend(t, server) key := SetupKey(t, server, "geheim") server.SetKey(key) for _, size := range testSizes { data := make([]byte, size) _, err := io.ReadFull(rand.Reader, data) OK(t, err) id := sha256.Sum256(data) // save blob, err := server.SaveFrom(backend.Data, id[:], uint(size), bytes.NewReader(data)) OK(t, err) // read back buf, err := server.Load(backend.Data, blob) Assert(t, len(buf) == len(data), "number of bytes read back does not match: expected %d, got %d", len(data), len(buf)) Assert(t, bytes.Equal(buf, data), "data does not match: expected %02x, got %02x", data, buf) } } func BenchmarkSaveFrom(t *testing.B) { server := SetupBackend(t) defer TeardownBackend(t, server) key := SetupKey(t, server, "geheim") server.SetKey(key) size := 4 << 20 // 4MiB data := make([]byte, size) _, err := io.ReadFull(rand.Reader, data) OK(t, err) id := sha256.Sum256(data) t.ResetTimer() t.SetBytes(int64(size)) for i := 0; i < t.N; i++ { // save _, err := server.SaveFrom(backend.Data, id[:], uint(size), bytes.NewReader(data)) OK(t, err) } } func TestLoadJSONID(t *testing.T) { if *benchTestDir == "" { t.Skip("benchdir not set, skipping TestServerStats") } server := SetupBackend(t) defer TeardownBackend(t, server) key := SetupKey(t, server, "geheim") server.SetKey(key) // archive a few files sn := SnapshotDir(t, server, *benchTestDir, nil) t.Logf("archived snapshot %v", sn.ID()) // benchmark loading first tree done := make(chan struct{}) first, found := <-server.List(backend.Tree, done) Assert(t, found, "no Trees in repository found") close(done) id, err := backend.ParseID(first) OK(t, err) tree := restic.NewTree() err = server.LoadJSONID(backend.Tree, id, &tree) OK(t, err) } func BenchmarkLoadJSONID(t *testing.B) { if *benchTestDir == "" { t.Skip("benchdir not set, skipping TestServerStats") } server := SetupBackend(t) defer TeardownBackend(t, server) key := SetupKey(t, server, "geheim") server.SetKey(key) // archive a few files sn := SnapshotDir(t, server, *benchTestDir, nil) t.Logf("archived snapshot %v", sn.ID()) t.ResetTimer() tree := restic.NewTree() for i := 0; i < t.N; i++ { for name := range server.List(backend.Tree, nil) { id, err := backend.ParseID(name) OK(t, err) OK(t, server.LoadJSONID(backend.Tree, id, &tree)) } } }