2
2
mirror of https://github.com/octoleo/restic.git synced 2024-12-04 19:03:46 +00:00
restic/internal/backend/test/tests.go

917 lines
24 KiB
Go
Raw Normal View History

2016-01-23 16:08:03 +00:00
package test
import (
"bytes"
2017-06-03 15:39:57 +00:00
"context"
2016-01-23 16:08:03 +00:00
"fmt"
"io"
"math/rand"
"os"
2016-01-23 16:08:03 +00:00
"reflect"
"sort"
"testing"
2017-05-28 08:16:29 +00:00
"time"
2016-01-23 16:08:03 +00:00
"github.com/minio/sha256-simd"
2017-07-23 12:21:03 +00:00
"github.com/restic/restic/internal/errors"
2017-07-24 15:42:25 +00:00
"github.com/restic/restic/internal/restic"
2017-07-23 12:21:03 +00:00
"github.com/restic/restic/internal/test"
"github.com/restic/restic/internal/backend"
2016-01-23 16:08:03 +00:00
)
2017-05-28 08:16:29 +00:00
func seedRand(t testing.TB) {
seed := time.Now().UnixNano()
rand.Seed(seed)
t.Logf("rand initialized with seed %d", seed)
}
func beTest(ctx context.Context, be backend.Backend, h backend.Handle) (bool, error) {
_, err := be.Stat(ctx, h)
if err != nil && be.IsNotExist(err) {
return false, nil
}
return err == nil, err
}
func LoadAll(ctx context.Context, be backend.Backend, h backend.Handle) ([]byte, error) {
var buf []byte
err := be.Load(ctx, h, 0, 0, func(rd io.Reader) error {
var err error
buf, err = io.ReadAll(rd)
return err
})
if err != nil {
return nil, err
}
return buf, nil
}
// TestStripPasswordCall tests that the StripPassword method of a factory can be called without crashing.
// It does not verify whether passwords are removed correctly
2023-06-08 17:35:20 +00:00
func (s *Suite[C]) TestStripPasswordCall(_ *testing.T) {
s.Factory.StripPassword("some random string")
}
// TestCreateWithConfig tests that creating a backend in a location which already
2016-01-23 17:07:15 +00:00
// has a config file fails.
func (s *Suite[C]) TestCreateWithConfig(t *testing.T) {
2017-05-01 20:23:46 +00:00
b := s.open(t)
defer s.close(t, b)
2016-01-23 17:07:15 +00:00
// remove a config if present
cfgHandle := backend.Handle{Type: backend.ConfigFile}
cfgPresent, err := beTest(context.TODO(), b, cfgHandle)
if err != nil {
t.Fatalf("unable to test for config: %+v", err)
}
if cfgPresent {
remove(t, b, cfgHandle)
}
2016-01-23 17:07:15 +00:00
// save a config
store(t, b, backend.ConfigFile, []byte("test config"))
2016-01-23 17:07:15 +00:00
// now create the backend again, this must fail
_, err = s.createOrError()
2016-01-23 17:07:15 +00:00
if err == nil {
t.Fatalf("expected error not found for creating a backend with an existing config file")
}
// remove config
err = b.Remove(context.TODO(), backend.Handle{Type: backend.ConfigFile, Name: ""})
2016-01-23 17:07:15 +00:00
if err != nil {
t.Fatalf("unexpected error removing config: %+v", err)
2016-01-23 17:07:15 +00:00
}
}
// TestConfig saves and loads a config from the backend.
func (s *Suite[C]) TestConfig(t *testing.T) {
2017-05-01 20:23:46 +00:00
b := s.open(t)
defer s.close(t, b)
2016-01-23 16:08:03 +00:00
var testString = "Config"
// create config and read it back
_, err := LoadAll(context.TODO(), b, backend.Handle{Type: backend.ConfigFile})
2016-01-23 16:08:03 +00:00
if err == nil {
t.Fatalf("did not get expected error for non-existing config")
}
test.Assert(t, b.IsNotExist(err), "IsNotExist() did not recognize error from LoadAll(): %v", err)
test.Assert(t, b.IsPermanentError(err), "IsPermanentError() did not recognize error from LoadAll(): %v", err)
2016-01-23 16:08:03 +00:00
err = b.Save(context.TODO(), backend.Handle{Type: backend.ConfigFile}, backend.NewByteReader([]byte(testString), b.Hasher()))
2016-01-23 16:08:03 +00:00
if err != nil {
t.Fatalf("Save() error: %+v", err)
2016-01-23 16:08:03 +00:00
}
// try accessing the config with different names, should all return the
// same config
for _, name := range []string{"", "foo", "bar", "0000000000000000000000000000000000000000000000000000000000000000"} {
h := backend.Handle{Type: backend.ConfigFile, Name: name}
buf, err := LoadAll(context.TODO(), b, h)
2016-01-23 16:08:03 +00:00
if err != nil {
t.Fatalf("unable to read config with name %q: %+v", name, err)
2016-01-23 16:08:03 +00:00
}
if string(buf) != testString {
t.Fatalf("wrong data returned, want %q, got %q", testString, string(buf))
}
}
// remove the config
remove(t, b, backend.Handle{Type: backend.ConfigFile})
2016-01-23 16:08:03 +00:00
}
// TestLoad tests the backend's Load function.
func (s *Suite[C]) TestLoad(t *testing.T) {
2017-05-28 08:16:29 +00:00
seedRand(t)
2017-05-01 20:23:46 +00:00
b := s.open(t)
defer s.close(t, b)
2017-01-22 21:01:12 +00:00
err := testLoad(b, backend.Handle{Type: backend.PackFile, Name: "foobar"})
2017-01-22 21:01:12 +00:00
if err == nil {
2017-01-23 17:11:10 +00:00
t.Fatalf("Load() did not return an error for non-existing blob")
2017-01-22 21:01:12 +00:00
}
test.Assert(t, b.IsNotExist(err), "IsNotExist() did not recognize non-existing blob: %v", err)
test.Assert(t, b.IsPermanentError(err), "IsPermanentError() did not recognize non-existing blob: %v", err)
2017-01-22 21:01:12 +00:00
length := rand.Intn(1<<24) + 2000
data := test.Random(23, length)
id := restic.Hash(data)
handle := backend.Handle{Type: backend.PackFile, Name: id.String()}
err = b.Save(context.TODO(), handle, backend.NewByteReader(data, b.Hasher()))
2017-01-22 21:01:12 +00:00
if err != nil {
t.Fatalf("Save() error: %+v", err)
2017-01-22 21:01:12 +00:00
}
2017-05-28 10:32:53 +00:00
t.Logf("saved %d bytes as %v", length, handle)
err = b.Load(context.TODO(), handle, 0, 0, func(rd io.Reader) error {
_, err := io.Copy(io.Discard, rd)
if err != nil {
t.Fatal(err)
}
return errors.Errorf("deliberate error")
})
if err == nil {
t.Fatalf("Load() did not propagate consumer error!")
}
if err.Error() != "deliberate error" {
t.Fatalf("Load() did not correctly propagate consumer error!")
2017-01-22 21:01:12 +00:00
}
loadTests := 50
2017-05-01 20:23:46 +00:00
if s.MinimalData {
loadTests = 10
}
for i := 0; i < loadTests; i++ {
2017-01-22 21:01:12 +00:00
l := rand.Intn(length + 2000)
o := rand.Intn(length + 2000)
d := data
if o < len(d) {
d = d[o:]
} else {
t.Logf("offset == length, skipping test")
continue
2017-01-22 21:01:12 +00:00
}
getlen := l
if l >= len(d) {
if rand.Float32() >= 0.5 {
getlen = 0
} else {
getlen = len(d)
}
2017-01-22 21:01:12 +00:00
}
if l > 0 && l < len(d) {
d = d[:l]
}
var buf []byte
err := b.Load(context.TODO(), handle, getlen, int64(o), func(rd io.Reader) (ierr error) {
buf, ierr = io.ReadAll(rd)
return ierr
})
2017-01-22 21:01:12 +00:00
if err != nil {
2017-05-28 10:32:53 +00:00
t.Logf("Load, l %v, o %v, len(d) %v, getlen %v", l, o, len(d), getlen)
t.Errorf("Load(%d, %d) returned unexpected error: %+v", l, o, err)
2017-01-22 21:01:12 +00:00
continue
}
2017-04-26 18:47:15 +00:00
if l == 0 && len(buf) != len(d) {
2017-05-28 10:32:53 +00:00
t.Logf("Load, l %v, o %v, len(d) %v, getlen %v", l, o, len(d), getlen)
2017-04-26 18:47:15 +00:00
t.Errorf("Load(%d, %d) wrong number of bytes read: want %d, got %d", l, o, len(d), len(buf))
continue
}
if l > 0 && l <= len(d) && len(buf) != l {
2017-05-28 10:32:53 +00:00
t.Logf("Load, l %v, o %v, len(d) %v, getlen %v", l, o, len(d), getlen)
2017-01-23 17:11:10 +00:00
t.Errorf("Load(%d, %d) wrong number of bytes read: want %d, got %d", l, o, l, len(buf))
2017-01-22 21:01:12 +00:00
continue
}
if l > len(d) && len(buf) != len(d) {
2017-05-28 10:32:53 +00:00
t.Logf("Load, l %v, o %v, len(d) %v, getlen %v", l, o, len(d), getlen)
2017-01-23 17:11:10 +00:00
t.Errorf("Load(%d, %d) wrong number of bytes read for overlong read: want %d, got %d", l, o, l, len(buf))
2017-01-22 21:01:12 +00:00
continue
}
if !bytes.Equal(buf, d) {
2017-05-28 10:32:53 +00:00
t.Logf("Load, l %v, o %v, len(d) %v, getlen %v", l, o, len(d), getlen)
2017-01-23 17:11:10 +00:00
t.Errorf("Load(%d, %d) returned wrong bytes", l, o)
2017-01-22 21:01:12 +00:00
continue
}
}
// test error checking for partial and fully out of bounds read
// only test for length > 0 as we currently do not need strict out of bounds handling for length==0
for _, offset := range []int{length - 99, length - 50, length, length + 100} {
err = b.Load(context.TODO(), handle, 100, int64(offset), func(rd io.Reader) (ierr error) {
_, ierr = io.ReadAll(rd)
return ierr
})
test.Assert(t, err != nil, "Load() did not return error on out of bounds read! o %v, l %v, filelength %v", offset, 100, length)
test.Assert(t, b.IsPermanentError(err), "IsPermanentError() did not recognize out of range read: %v", err)
test.Assert(t, !b.IsNotExist(err), "IsNotExist() must not recognize out of range read: %v", err)
}
2017-06-03 15:39:57 +00:00
test.OK(t, b.Remove(context.TODO(), handle))
2017-01-22 21:01:12 +00:00
}
type setter interface {
SetListMaxItems(int)
}
// TestList makes sure that the backend implements List() pagination correctly.
func (s *Suite[C]) TestList(t *testing.T) {
2017-09-17 09:09:09 +00:00
seedRand(t)
numTestFiles := rand.Intn(20) + 20
2017-09-17 09:09:09 +00:00
b := s.open(t)
defer s.close(t, b)
// Check that the backend is empty to start with
var found []string
err := b.List(context.TODO(), backend.PackFile, func(fi backend.FileInfo) error {
found = append(found, fi.Name)
return nil
})
if err != nil {
t.Fatalf("List returned error %v", err)
}
if found != nil {
t.Fatalf("backend not empty at start of test - contains: %v", found)
}
list1 := make(map[restic.ID]int64)
2017-09-17 09:09:09 +00:00
for i := 0; i < numTestFiles; i++ {
data := test.Random(rand.Int(), rand.Intn(100)+55)
id := restic.Hash(data)
h := backend.Handle{Type: backend.PackFile, Name: id.String()}
err := b.Save(context.TODO(), h, backend.NewByteReader(data, b.Hasher()))
if err != nil {
t.Fatal(err)
}
list1[id] = int64(len(data))
}
t.Logf("wrote %v files", len(list1))
var tests = []struct {
maxItems int
}{
2017-09-18 11:18:42 +00:00
{11}, {23}, {numTestFiles}, {numTestFiles + 10}, {numTestFiles + 1123},
}
for _, test := range tests {
t.Run(fmt.Sprintf("max-%v", test.maxItems), func(t *testing.T) {
list2 := make(map[restic.ID]int64)
if s, ok := b.(setter); ok {
t.Logf("setting max list items to %d", test.maxItems)
s.SetListMaxItems(test.maxItems)
}
2017-09-17 09:09:09 +00:00
err := b.List(context.TODO(), backend.PackFile, func(fi backend.FileInfo) error {
id, err := restic.ParseID(fi.Name)
if err != nil {
t.Fatal(err)
}
list2[id] = fi.Size
return nil
})
if err != nil {
t.Fatalf("List returned error %v", err)
}
2017-09-17 09:09:09 +00:00
t.Logf("loaded %v IDs from backend", len(list2))
2017-09-17 09:09:09 +00:00
for id, size := range list1 {
size2, ok := list2[id]
if !ok {
t.Errorf("id %v not returned by List()", id.Str())
}
if size != size2 {
t.Errorf("wrong size for id %v returned: want %v, got %v", id.Str(), size, size2)
}
}
for id := range list2 {
_, ok := list1[id]
if !ok {
t.Errorf("extra id %v returned by List()", id.Str())
}
}
})
2017-09-17 09:09:09 +00:00
}
t.Logf("remove %d files", numTestFiles)
handles := make([]backend.Handle, 0, len(list1))
2017-09-17 09:09:09 +00:00
for id := range list1 {
handles = append(handles, backend.Handle{Type: backend.PackFile, Name: id.String()})
2017-09-18 11:18:42 +00:00
}
err = s.delayedRemove(t, b, handles...)
2017-09-18 11:18:42 +00:00
if err != nil {
t.Fatal(err)
2017-09-17 09:09:09 +00:00
}
}
// TestListCancel tests that the context is respected and the error is returned by List.
func (s *Suite[C]) TestListCancel(t *testing.T) {
seedRand(t)
numTestFiles := 5
b := s.open(t)
defer s.close(t, b)
testFiles := make([]backend.Handle, 0, numTestFiles)
for i := 0; i < numTestFiles; i++ {
data := []byte(fmt.Sprintf("random test blob %v", i))
id := restic.Hash(data)
h := backend.Handle{Type: backend.PackFile, Name: id.String()}
err := b.Save(context.TODO(), h, backend.NewByteReader(data, b.Hasher()))
if err != nil {
t.Fatal(err)
}
testFiles = append(testFiles, h)
}
t.Run("Cancelled", func(t *testing.T) {
ctx, cancel := context.WithCancel(context.TODO())
cancel()
// pass in a cancelled context
err := b.List(ctx, backend.PackFile, func(fi backend.FileInfo) error {
t.Errorf("got FileInfo %v for cancelled context", fi)
return nil
})
if !errors.Is(err, context.Canceled) {
t.Fatalf("expected error not found, want %v, got %v", context.Canceled, err)
}
})
t.Run("First", func(t *testing.T) {
ctx, cancel := context.WithCancel(context.TODO())
defer cancel()
i := 0
err := b.List(ctx, backend.PackFile, func(fi backend.FileInfo) error {
i++
// cancel the context on the first file
if i == 1 {
cancel()
}
return nil
})
if !errors.Is(err, context.Canceled) {
t.Fatalf("expected error not found, want %v, got %v", context.Canceled, err)
}
if i != 1 {
t.Fatalf("wrong number of files returned by List, want %v, got %v", 1, i)
}
})
t.Run("Last", func(t *testing.T) {
ctx, cancel := context.WithCancel(context.TODO())
defer cancel()
i := 0
err := b.List(ctx, backend.PackFile, func(fi backend.FileInfo) error {
// cancel the context at the last file
i++
if i == numTestFiles {
cancel()
}
return nil
})
if !errors.Is(err, context.Canceled) {
t.Fatalf("expected error not found, want %v, got %v", context.Canceled, err)
}
if i != numTestFiles {
t.Fatalf("wrong number of files returned by List, want %v, got %v", numTestFiles, i)
}
})
t.Run("Timeout", func(t *testing.T) {
// rather large timeout, let's try to get at least one item
timeout := time.Second
ctxTimeout, cancel := context.WithTimeout(context.TODO(), timeout)
defer cancel()
i := 0
// pass in a context with a timeout
err := b.List(ctxTimeout, backend.PackFile, func(fi backend.FileInfo) error {
i++
// wait until the context is cancelled
<-ctxTimeout.Done()
// The cancellation of a context first closes the done channel of the context and
// _afterwards_ propagates the cancellation to child contexts. If the List
// implementation uses a child context, then it may take a moment until that context
// is also cancelled. Thus give the context cancellation a moment to propagate.
time.Sleep(time.Millisecond)
return nil
})
if !errors.Is(err, context.DeadlineExceeded) {
t.Fatalf("expected error not found, want %#v, got %#v", context.DeadlineExceeded, err)
}
2018-02-10 11:53:38 +00:00
if i > 2 {
t.Fatalf("wrong number of files returned by List, want <= 2, got %v", i)
}
})
err := s.delayedRemove(t, b, testFiles...)
if err != nil {
t.Fatal(err)
}
}
type errorCloser struct {
io.ReadSeeker
l int64
t testing.TB
h []byte
}
func (ec errorCloser) Close() error {
ec.t.Error("forbidden method close was called")
return errors.New("forbidden method close was called")
}
func (ec errorCloser) Length() int64 {
return ec.l
}
func (ec errorCloser) Hash() []byte {
return ec.h
}
func (ec errorCloser) Rewind() error {
_, err := ec.ReadSeeker.Seek(0, io.SeekStart)
return err
}
// TestSave tests saving data in the backend.
func (s *Suite[C]) TestSave(t *testing.T) {
2017-05-28 08:16:29 +00:00
seedRand(t)
2017-05-01 20:23:46 +00:00
b := s.open(t)
defer s.close(t, b)
2016-08-31 20:51:35 +00:00
var id restic.ID
2016-01-24 15:59:38 +00:00
saveTests := 10
2017-05-01 20:23:46 +00:00
if s.MinimalData {
saveTests = 2
}
for i := 0; i < saveTests; i++ {
2016-01-24 16:46:18 +00:00
length := rand.Intn(1<<23) + 200000
2016-09-04 12:38:18 +00:00
data := test.Random(23, length)
id = sha256.Sum256(data)
2016-01-24 15:59:38 +00:00
h := backend.Handle{
Type: backend.PackFile,
Name: id.String(),
2016-01-24 15:59:38 +00:00
}
err := b.Save(context.TODO(), h, backend.NewByteReader(data, b.Hasher()))
2016-09-04 12:38:18 +00:00
test.OK(t, err)
2016-01-24 15:59:38 +00:00
buf, err := LoadAll(context.TODO(), b, h)
2016-09-04 12:38:18 +00:00
test.OK(t, err)
2016-01-24 15:59:38 +00:00
if len(buf) != len(data) {
t.Fatalf("number of bytes does not match, want %v, got %v", len(data), len(buf))
}
if !bytes.Equal(buf, data) {
t.Fatalf("data not equal")
}
2017-06-03 15:39:57 +00:00
fi, err := b.Stat(context.TODO(), h)
2016-09-04 12:38:18 +00:00
test.OK(t, err)
2016-01-24 15:59:38 +00:00
if fi.Name != h.Name {
t.Errorf("Stat() returned wrong name, want %q, got %q", h.Name, fi.Name)
}
2016-01-24 15:59:38 +00:00
if fi.Size != int64(len(data)) {
t.Errorf("Stat() returned different size, want %q, got %d", len(data), fi.Size)
2016-01-24 15:59:38 +00:00
}
2017-06-03 15:39:57 +00:00
err = b.Remove(context.TODO(), h)
2016-01-24 15:59:38 +00:00
if err != nil {
t.Fatalf("error removing item: %+v", err)
2016-01-24 15:59:38 +00:00
}
}
// test saving from a tempfile
tmpfile, err := os.CreateTemp("", "restic-backend-save-test-")
if err != nil {
t.Fatal(err)
}
length := rand.Intn(1<<23) + 200000
data := test.Random(23, length)
id = sha256.Sum256(data)
if _, err = tmpfile.Write(data); err != nil {
t.Fatal(err)
}
if _, err = tmpfile.Seek(0, io.SeekStart); err != nil {
t.Fatal(err)
}
h := backend.Handle{Type: backend.PackFile, Name: id.String()}
// wrap the tempfile in an errorCloser, so we can detect if the backend
// closes the reader
var beHash []byte
if b.Hasher() != nil {
beHasher := b.Hasher()
// must never fail according to interface
_, err := beHasher.Write(data)
if err != nil {
panic(err)
}
beHash = beHasher.Sum(nil)
}
err = b.Save(context.TODO(), h, errorCloser{
t: t,
l: int64(length),
ReadSeeker: tmpfile,
h: beHash,
})
if err != nil {
t.Fatal(err)
}
err = s.delayedRemove(t, b, h)
if err != nil {
t.Fatalf("error removing item: %+v", err)
}
if err = tmpfile.Close(); err != nil {
t.Fatal(err)
}
if err = os.Remove(tmpfile.Name()); err != nil {
t.Fatal(err)
}
2016-01-24 15:59:38 +00:00
}
type incompleteByteReader struct {
backend.ByteReader
}
func (r *incompleteByteReader) Length() int64 {
return r.ByteReader.Length() + 42
}
// TestSaveError tests saving data in the backend.
func (s *Suite[C]) TestSaveError(t *testing.T) {
seedRand(t)
b := s.open(t)
defer func() {
// rclone will report an error when closing the backend. We have to ignore it
// otherwise this test will always fail
_ = b.Close()
}()
length := rand.Intn(1<<23) + 200000
2021-01-29 21:52:26 +00:00
data := test.Random(24, length)
var id restic.ID
copy(id[:], data)
// test that incomplete uploads fail
h := backend.Handle{Type: backend.PackFile, Name: id.String()}
err := b.Save(context.TODO(), h, &incompleteByteReader{ByteReader: *backend.NewByteReader(data, b.Hasher())})
2021-01-29 21:52:26 +00:00
// try to delete possible leftovers
_ = s.delayedRemove(t, b, h)
if err == nil {
t.Fatal("incomplete upload did not fail")
}
}
type wrongByteReader struct {
backend.ByteReader
}
func (b *wrongByteReader) Hash() []byte {
h := b.ByteReader.Hash()
modHash := make([]byte, len(h))
copy(modHash, h)
// flip a bit in the hash
modHash[0] ^= 0x01
return modHash
}
// TestSaveWrongHash tests that uploads with a wrong hash fail
func (s *Suite[C]) TestSaveWrongHash(t *testing.T) {
seedRand(t)
b := s.open(t)
defer s.close(t, b)
// nothing to do if the backend doesn't support external hashes
if b.Hasher() == nil {
return
}
length := rand.Intn(1<<23) + 200000
data := test.Random(25, length)
var id restic.ID
copy(id[:], data)
// test that upload with hash mismatch fails
h := backend.Handle{Type: backend.PackFile, Name: id.String()}
err := b.Save(context.TODO(), h, &wrongByteReader{ByteReader: *backend.NewByteReader(data, b.Hasher())})
exists, err2 := beTest(context.TODO(), b, h)
if err2 != nil {
t.Fatal(err2)
}
_ = s.delayedRemove(t, b, h)
if err == nil {
t.Fatal("upload with wrong hash did not fail")
}
t.Logf("%v", err)
if exists {
t.Fatal("Backend returned an error but stored the file anyways")
}
}
2016-01-23 16:08:03 +00:00
var testStrings = []struct {
id string
data string
}{
{"c3ab8ff13720e8ad9047dd39466b3c8974e592c2fa383d4a3960714caef0c4f2", "foobar"},
{"248d6a61d20638b8e5c026930c3e6039a33ce45964ff2167f6ecedd419db06c1", "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq"},
{"cc5d46bdb4991c6eae3eb739c9c8a7a46fe9654fab79c47b4fe48383b5b25e1c", "foo/bar"},
{"4e54d2c721cbdb730f01b10b62dec622962b36966ec685880effa63d71c808f2", "foo/../../baz"},
}
func store(t testing.TB, b backend.Backend, tpe backend.FileType, data []byte) backend.Handle {
2016-08-31 20:51:35 +00:00
id := restic.Hash(data)
h := backend.Handle{Name: id.String(), Type: tpe}
err := b.Save(context.TODO(), h, backend.NewByteReader([]byte(data), b.Hasher()))
2016-09-04 12:38:18 +00:00
test.OK(t, err)
return h
2016-01-23 16:08:03 +00:00
}
// testLoad loads a blob (but discards its contents).
func testLoad(b backend.Backend, h backend.Handle) error {
return b.Load(context.TODO(), h, 0, 0, func(rd io.Reader) (ierr error) {
_, ierr = io.Copy(io.Discard, rd)
return ierr
})
}
func (s *Suite[C]) delayedRemove(t testing.TB, be backend.Backend, handles ...backend.Handle) error {
// Some backend (swift, I'm looking at you) may implement delayed
// removal of data. Let's wait a bit if this happens.
2017-05-01 08:13:03 +00:00
for _, h := range handles {
err := be.Remove(context.TODO(), h)
if s.ErrorHandler != nil {
err = s.ErrorHandler(t, be, err)
}
if err != nil {
return err
}
}
for _, h := range handles {
start := time.Now()
attempt := 0
var found bool
var err error
for time.Since(start) <= s.WaitForDelayedRemoval {
found, err = beTest(context.TODO(), be, h)
if s.ErrorHandler != nil {
err = s.ErrorHandler(t, be, err)
}
if err != nil {
return err
}
if !found {
break
}
time.Sleep(2 * time.Second)
attempt++
}
if found {
t.Fatalf("removed blob %v still present after %v (%d attempts)", h, time.Since(start), attempt)
}
}
return nil
}
func delayedList(t testing.TB, b backend.Backend, tpe backend.FileType, max int, maxwait time.Duration) restic.IDs {
list := restic.NewIDSet()
start := time.Now()
for i := 0; i < max; i++ {
err := b.List(context.TODO(), tpe, func(fi backend.FileInfo) error {
id := restic.TestParseID(fi.Name)
list.Insert(id)
return nil
})
if err != nil {
t.Fatal(err)
}
if len(list) < max && time.Since(start) < maxwait {
time.Sleep(500 * time.Millisecond)
}
}
return list.List()
}
// TestBackend tests all functions of the backend.
func (s *Suite[C]) TestBackend(t *testing.T) {
2017-05-01 20:23:46 +00:00
b := s.open(t)
defer s.close(t, b)
2016-01-23 16:08:03 +00:00
test.Assert(t, !b.IsNotExist(nil), "IsNotExist() recognized nil error")
test.Assert(t, !b.IsPermanentError(nil), "IsPermanentError() recognized nil error")
for _, tpe := range []backend.FileType{
backend.PackFile, backend.KeyFile, backend.LockFile,
backend.SnapshotFile, backend.IndexFile,
2016-01-23 16:08:03 +00:00
} {
// detect non-existing files
2016-09-04 12:38:18 +00:00
for _, ts := range testStrings {
id, err := restic.ParseID(ts.id)
test.OK(t, err)
2016-01-23 16:08:03 +00:00
// test if blob is already in repository
h := backend.Handle{Type: tpe, Name: id.String()}
ret, err := beTest(context.TODO(), b, h)
2016-09-04 12:38:18 +00:00
test.OK(t, err)
test.Assert(t, !ret, "blob was found to exist before creating")
2016-01-23 16:08:03 +00:00
2016-01-24 00:00:27 +00:00
// try to stat a not existing blob
2017-06-03 15:39:57 +00:00
_, err = b.Stat(context.TODO(), h)
2016-09-04 12:38:18 +00:00
test.Assert(t, err != nil, "blob data could be extracted before creation")
test.Assert(t, b.IsNotExist(err), "IsNotExist() did not recognize Stat() error: %v", err)
test.Assert(t, b.IsPermanentError(err), "IsPermanentError() did not recognize Stat() error: %v", err)
2016-01-23 16:08:03 +00:00
// try to read not existing blob
err = testLoad(b, h)
test.Assert(t, err != nil, "blob could be read before creation")
test.Assert(t, b.IsNotExist(err), "IsNotExist() did not recognize Load() error: %v", err)
test.Assert(t, b.IsPermanentError(err), "IsPermanentError() did not recognize Load() error: %v", err)
2016-01-23 16:08:03 +00:00
// try to get string out, should fail
ret, err = beTest(context.TODO(), b, h)
2016-09-04 12:38:18 +00:00
test.OK(t, err)
test.Assert(t, !ret, "id %q was found (but should not have)", ts.id)
2016-01-23 16:08:03 +00:00
}
// add files
2016-09-04 12:38:18 +00:00
for _, ts := range testStrings {
store(t, b, tpe, []byte(ts.data))
2016-01-23 16:08:03 +00:00
2016-01-24 00:00:27 +00:00
// test Load()
h := backend.Handle{Type: tpe, Name: ts.id}
buf, err := LoadAll(context.TODO(), b, h)
2016-09-04 12:38:18 +00:00
test.OK(t, err)
test.Equals(t, ts.data, string(buf))
2016-01-23 16:08:03 +00:00
// try to read it out with an offset and a length
start := 1
2016-09-04 12:38:18 +00:00
end := len(ts.data) - 2
2016-01-23 16:08:03 +00:00
length := end - start
2016-01-24 00:00:27 +00:00
buf2 := make([]byte, length)
var n int
err = b.Load(context.TODO(), h, len(buf2), int64(start), func(rd io.Reader) (ierr error) {
n, ierr = io.ReadFull(rd, buf2)
return ierr
})
2016-09-04 12:38:18 +00:00
test.OK(t, err)
2017-01-23 16:20:08 +00:00
test.OK(t, err)
test.Equals(t, len(buf2), n)
2016-09-04 12:38:18 +00:00
test.Equals(t, ts.data[start:end], string(buf2))
2016-01-23 16:08:03 +00:00
}
// test adding the first file again
2016-09-04 12:38:18 +00:00
ts := testStrings[0]
h := backend.Handle{Type: tpe, Name: ts.id}
2016-01-23 16:08:03 +00:00
// remove and recreate
err := s.delayedRemove(t, b, h)
2016-09-04 12:38:18 +00:00
test.OK(t, err)
2016-01-23 16:08:03 +00:00
// test that the blob is gone
ok, err := beTest(context.TODO(), b, h)
2016-09-04 12:38:18 +00:00
test.OK(t, err)
2017-04-11 18:08:25 +00:00
test.Assert(t, !ok, "removed blob still present")
2016-01-23 16:08:03 +00:00
// create blob
err = b.Save(context.TODO(), h, backend.NewByteReader([]byte(ts.data), b.Hasher()))
2016-09-04 12:38:18 +00:00
test.OK(t, err)
2016-01-23 16:08:03 +00:00
// list items
2016-08-31 20:51:35 +00:00
IDs := restic.IDs{}
2016-01-23 16:08:03 +00:00
2016-09-04 12:38:18 +00:00
for _, ts := range testStrings {
id, err := restic.ParseID(ts.id)
test.OK(t, err)
2016-01-23 16:08:03 +00:00
IDs = append(IDs, id)
}
list := delayedList(t, b, tpe, len(IDs), s.WaitForDelayedRemoval)
2016-01-23 16:08:03 +00:00
if len(IDs) != len(list) {
t.Fatalf("wrong number of IDs returned: want %d, got %d", len(IDs), len(list))
}
sort.Sort(IDs)
sort.Sort(list)
if !reflect.DeepEqual(IDs, list) {
t.Fatalf("lists aren't equal, want:\n %v\n got:\n%v\n", IDs, list)
}
var handles []backend.Handle
for _, ts := range testStrings {
id, err := restic.ParseID(ts.id)
test.OK(t, err)
h := backend.Handle{Type: tpe, Name: id.String()}
2016-01-23 16:08:03 +00:00
found, err := beTest(context.TODO(), b, h)
test.OK(t, err)
test.Assert(t, found, fmt.Sprintf("id %v/%q not found", tpe, id))
handles = append(handles, h)
2016-01-23 16:08:03 +00:00
}
test.OK(t, s.delayedRemove(t, b, handles...))
2016-01-23 16:08:03 +00:00
}
}
2017-10-14 13:56:46 +00:00
// TestZZZDelete tests the Delete function. The name ensures that this test is executed last.
func (s *Suite[C]) TestZZZDelete(t *testing.T) {
2017-05-01 20:23:46 +00:00
if !test.TestCleanupTempDirs {
t.Skipf("not removing backend, TestCleanupTempDirs is false")
}
b := s.open(t)
defer s.close(t, b)
2016-01-23 16:08:03 +00:00
2017-10-14 13:56:46 +00:00
err := b.Delete(context.TODO())
2016-01-23 16:08:03 +00:00
if err != nil {
t.Fatalf("error deleting backend: %+v", err)
2016-01-23 16:08:03 +00:00
}
}