mirror of
https://github.com/restic/restic.git
synced 2024-08-19 19:03:22 +02:00
186 lines
3.9 KiB
Go
186 lines
3.9 KiB
Go
package test
|
|
|
|
import (
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/restic/restic/internal/restic"
|
|
"github.com/restic/restic/internal/test"
|
|
)
|
|
|
|
// Suite implements a test suite for restic backends.
|
|
type Suite struct {
|
|
// Config should be used to configure the backend.
|
|
Config interface{}
|
|
|
|
// NewConfig returns a config for a new temporary backend that will be used in tests.
|
|
NewConfig func() (interface{}, error)
|
|
|
|
// CreateFn is a function that creates a temporary repository for the tests.
|
|
Create func(cfg interface{}) (restic.Backend, error)
|
|
|
|
// OpenFn is a function that opens a previously created temporary repository.
|
|
Open func(cfg interface{}) (restic.Backend, error)
|
|
|
|
// CleanupFn removes data created during the tests.
|
|
Cleanup func(cfg interface{}) error
|
|
|
|
// MinimalData instructs the tests to not use excessive data.
|
|
MinimalData bool
|
|
|
|
// WaitForDelayedRemoval is set to a non-zero value to instruct the test
|
|
// suite to wait for this amount of time until a file that was removed
|
|
// really disappeared.
|
|
WaitForDelayedRemoval time.Duration
|
|
|
|
// ErrorHandler allows ignoring certain errors.
|
|
ErrorHandler func(testing.TB, restic.Backend, error) error
|
|
}
|
|
|
|
// RunTests executes all defined tests as subtests of t.
|
|
func (s *Suite) RunTests(t *testing.T) {
|
|
var err error
|
|
s.Config, err = s.NewConfig()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// test create/open functions first
|
|
be := s.create(t)
|
|
s.close(t, be)
|
|
|
|
for _, test := range s.testFuncs(t) {
|
|
t.Run(test.Name, test.Fn)
|
|
}
|
|
|
|
if !test.TestCleanupTempDirs {
|
|
t.Logf("not cleaning up backend")
|
|
return
|
|
}
|
|
|
|
if err = s.Cleanup(s.Config); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
type testFunction struct {
|
|
Name string
|
|
Fn func(*testing.T)
|
|
}
|
|
|
|
func (s *Suite) testFuncs(t testing.TB) (funcs []testFunction) {
|
|
tpe := reflect.TypeOf(s)
|
|
v := reflect.ValueOf(s)
|
|
|
|
for i := 0; i < tpe.NumMethod(); i++ {
|
|
methodType := tpe.Method(i)
|
|
name := methodType.Name
|
|
|
|
// discard functions which do not have the right name
|
|
if !strings.HasPrefix(name, "Test") {
|
|
continue
|
|
}
|
|
|
|
iface := v.Method(i).Interface()
|
|
f, ok := iface.(func(*testing.T))
|
|
if !ok {
|
|
t.Logf("warning: function %v of *Suite has the wrong signature for a test function\nwant: func(*testing.T),\nhave: %T",
|
|
name, iface)
|
|
continue
|
|
}
|
|
|
|
funcs = append(funcs, testFunction{
|
|
Name: name,
|
|
Fn: f,
|
|
})
|
|
}
|
|
|
|
return funcs
|
|
}
|
|
|
|
type benchmarkFunction struct {
|
|
Name string
|
|
Fn func(*testing.B)
|
|
}
|
|
|
|
func (s *Suite) benchmarkFuncs(t testing.TB) (funcs []benchmarkFunction) {
|
|
tpe := reflect.TypeOf(s)
|
|
v := reflect.ValueOf(s)
|
|
|
|
for i := 0; i < tpe.NumMethod(); i++ {
|
|
methodType := tpe.Method(i)
|
|
name := methodType.Name
|
|
|
|
// discard functions which do not have the right name
|
|
if !strings.HasPrefix(name, "Benchmark") {
|
|
continue
|
|
}
|
|
|
|
iface := v.Method(i).Interface()
|
|
f, ok := iface.(func(*testing.B))
|
|
if !ok {
|
|
t.Logf("warning: function %v of *Suite has the wrong signature for a test function\nwant: func(*testing.T),\nhave: %T",
|
|
name, iface)
|
|
continue
|
|
}
|
|
|
|
funcs = append(funcs, benchmarkFunction{
|
|
Name: name,
|
|
Fn: f,
|
|
})
|
|
}
|
|
|
|
return funcs
|
|
}
|
|
|
|
// RunBenchmarks executes all defined benchmarks as subtests of b.
|
|
func (s *Suite) RunBenchmarks(b *testing.B) {
|
|
var err error
|
|
s.Config, err = s.NewConfig()
|
|
if err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
// test create/open functions first
|
|
be := s.create(b)
|
|
s.close(b, be)
|
|
|
|
for _, test := range s.benchmarkFuncs(b) {
|
|
b.Run(test.Name, test.Fn)
|
|
}
|
|
|
|
if !test.TestCleanupTempDirs {
|
|
b.Logf("not cleaning up backend")
|
|
return
|
|
}
|
|
|
|
if err = s.Cleanup(s.Config); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func (s *Suite) create(t testing.TB) restic.Backend {
|
|
be, err := s.Create(s.Config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return be
|
|
}
|
|
|
|
func (s *Suite) open(t testing.TB) restic.Backend {
|
|
be, err := s.Open(s.Config)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return be
|
|
}
|
|
|
|
func (s *Suite) close(t testing.TB, be restic.Backend) {
|
|
err := be.Close()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|