forked from TrueCloudLab/restic
103 lines
2.2 KiB
Go
103 lines
2.2 KiB
Go
package backend
|
|
|
|
import "errors"
|
|
|
|
// MockBackend implements a backend whose functions can be specified. This
|
|
// should only be used for tests.
|
|
type MockBackend struct {
|
|
CloseFn func() error
|
|
LoadFn func(h Handle, p []byte, off int64) (int, error)
|
|
SaveFn func(h Handle, p []byte) error
|
|
StatFn func(h Handle) (BlobInfo, error)
|
|
ListFn func(Type, <-chan struct{}) <-chan string
|
|
RemoveFn func(Type, string) error
|
|
TestFn func(Type, string) (bool, error)
|
|
DeleteFn func() error
|
|
LocationFn func() string
|
|
}
|
|
|
|
// Close the backend.
|
|
func (m *MockBackend) Close() error {
|
|
if m.CloseFn == nil {
|
|
return nil
|
|
}
|
|
|
|
return m.CloseFn()
|
|
}
|
|
|
|
// Location returns a location string.
|
|
func (m *MockBackend) Location() string {
|
|
if m.LocationFn == nil {
|
|
return ""
|
|
}
|
|
|
|
return m.LocationFn()
|
|
}
|
|
|
|
// Load loads data from the backend.
|
|
func (m *MockBackend) Load(h Handle, p []byte, off int64) (int, error) {
|
|
if m.LoadFn == nil {
|
|
return 0, errors.New("not implemented")
|
|
}
|
|
|
|
return m.LoadFn(h, p, off)
|
|
}
|
|
|
|
// Save data in the backend.
|
|
func (m *MockBackend) Save(h Handle, p []byte) error {
|
|
if m.SaveFn == nil {
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
return m.SaveFn(h, p)
|
|
}
|
|
|
|
// Stat an object in the backend.
|
|
func (m *MockBackend) Stat(h Handle) (BlobInfo, error) {
|
|
if m.StatFn == nil {
|
|
return BlobInfo{}, errors.New("not implemented")
|
|
}
|
|
|
|
return m.StatFn(h)
|
|
}
|
|
|
|
// List items of type t.
|
|
func (m *MockBackend) List(t Type, done <-chan struct{}) <-chan string {
|
|
if m.ListFn == nil {
|
|
ch := make(chan string)
|
|
close(ch)
|
|
return ch
|
|
}
|
|
|
|
return m.ListFn(t, done)
|
|
}
|
|
|
|
// Remove data from the backend.
|
|
func (m *MockBackend) Remove(t Type, name string) error {
|
|
if m.RemoveFn == nil {
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
return m.RemoveFn(t, name)
|
|
}
|
|
|
|
// Test for the existence of a specific item.
|
|
func (m *MockBackend) Test(t Type, name string) (bool, error) {
|
|
if m.TestFn == nil {
|
|
return false, errors.New("not implemented")
|
|
}
|
|
|
|
return m.TestFn(t, name)
|
|
}
|
|
|
|
// Delete all data.
|
|
func (m *MockBackend) Delete() error {
|
|
if m.DeleteFn == nil {
|
|
return errors.New("not implemented")
|
|
}
|
|
|
|
return m.DeleteFn()
|
|
}
|
|
|
|
// Make sure that MockBackend implements the backend interface.
|
|
var _ Backend = &MockBackend{}
|