restic/src/restic/mock/backend.go

package mock

import (
	"restic"

	"github.com/pkg/errors"
)

// Backend implements a mock backend.
type Backend struct {
	CloseFn    func() error
	LoadFn     func(h restic.Handle, p []byte, off int64) (int, error)
	SaveFn     func(h restic.Handle, p []byte) error
	StatFn     func(h restic.Handle) (restic.BlobInfo, error)
	ListFn     func(restic.FileType, <-chan struct{}) <-chan string
	RemoveFn   func(restic.FileType, string) error
	TestFn     func(restic.FileType, string) (bool, error)
	DeleteFn   func() error
	LocationFn func() string
}

// Close the backend.
func (m *Backend) Close() error {
	if m.CloseFn == nil {
		return nil
	}

	return m.CloseFn()
}

// Location returns a location string.
func (m *Backend) Location() string {
	if m.LocationFn == nil {
		return ""
	}

	return m.LocationFn()
}

// Load loads data from the backend.
func (m *Backend) Load(h restic.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 *Backend) Save(h restic.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 *Backend) Stat(h restic.Handle) (restic.BlobInfo, error) {
	if m.StatFn == nil {
		return restic.BlobInfo{}, errors.New("not implemented")
	}

	return m.StatFn(h)
}

// List items of type t.
func (m *Backend) List(t restic.FileType, 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 *Backend) Remove(t restic.FileType, 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 *Backend) Test(t restic.FileType, name string) (bool, error) {
	if m.TestFn == nil {
		return false, errors.New("not implemented")
	}

	return m.TestFn(t, name)
}

// Delete all data.
func (m *Backend) Delete() error {
	if m.DeleteFn == nil {
		return errors.New("not implemented")
	}

	return m.DeleteFn()
}

// Make sure that Backend implements the backend interface.
var _ restic.Backend = &Backend{}