Moves files

internal/backend/utils_test.go

@@ -0,0 +1,91 @@
+package backend_test
+
+import (
+	"bytes"
+	"context"
+	"math/rand"
+	"restic"
+	"testing"
+
+	"restic/backend"
+	"restic/backend/mem"
+	. "restic/test"
+)
+
+const KiB = 1 << 10
+const MiB = 1 << 20
+
+func TestLoadAll(t *testing.T) {
+	b := mem.New()
+
+	for i := 0; i < 20; i++ {
+		data := Random(23+i, rand.Intn(MiB)+500*KiB)
+
+		id := restic.Hash(data)
+		err := b.Save(context.TODO(), restic.Handle{Name: id.String(), Type: restic.DataFile}, bytes.NewReader(data))
+		OK(t, err)
+
+		buf, err := backend.LoadAll(context.TODO(), b, restic.Handle{Type: restic.DataFile, Name: id.String()})
+		OK(t, err)
+
+		if len(buf) != len(data) {
+			t.Errorf("length of returned buffer does not match, want %d, got %d", len(data), len(buf))
+			continue
+		}
+
+		if !bytes.Equal(buf, data) {
+			t.Errorf("wrong data returned")
+			continue
+		}
+	}
+}
+
+func TestLoadSmallBuffer(t *testing.T) {
+	b := mem.New()
+
+	for i := 0; i < 20; i++ {
+		data := Random(23+i, rand.Intn(MiB)+500*KiB)
+
+		id := restic.Hash(data)
+		err := b.Save(context.TODO(), restic.Handle{Name: id.String(), Type: restic.DataFile}, bytes.NewReader(data))
+		OK(t, err)
+
+		buf, err := backend.LoadAll(context.TODO(), b, restic.Handle{Type: restic.DataFile, Name: id.String()})
+		OK(t, err)
+
+		if len(buf) != len(data) {
+			t.Errorf("length of returned buffer does not match, want %d, got %d", len(data), len(buf))
+			continue
+		}
+
+		if !bytes.Equal(buf, data) {
+			t.Errorf("wrong data returned")
+			continue
+		}
+	}
+}
+
+func TestLoadLargeBuffer(t *testing.T) {
+	b := mem.New()
+
+	for i := 0; i < 20; i++ {
+		data := Random(23+i, rand.Intn(MiB)+500*KiB)
+
+		id := restic.Hash(data)
+		err := b.Save(context.TODO(), restic.Handle{Name: id.String(), Type: restic.DataFile}, bytes.NewReader(data))
+		OK(t, err)
+
+		buf, err := backend.LoadAll(context.TODO(), b, restic.Handle{Type: restic.DataFile, Name: id.String()})
+		OK(t, err)
+
+		if len(buf) != len(data) {
+			t.Errorf("length of returned buffer does not match, want %d, got %d", len(data), len(buf))
+			continue
+		}
+
+		if !bytes.Equal(buf, data) {
+			t.Errorf("wrong data returned")
+			continue
+		}
+	}
+}