restic/src/restic/backend/mem/mem_backend_test.go

package mem_test

import (
	"github.com/pkg/errors"

	"restic/backend"
	"restic/backend/mem"
	"restic/backend/test"
)

var be backend.Backend

//go:generate go run ../test/generate_backend_tests.go

func init() {
	test.CreateFn = func() (backend.Backend, error) {
		if be != nil {
			return nil, errors.New("temporary memory backend dir already exists")
		}

		be = mem.New()

		return be, nil
	}

	test.OpenFn = func() (backend.Backend, error) {
		if be == nil {
			return nil, errors.New("repository not initialized")
		}

		return be, nil
	}

	test.CleanupFn = func() error {
		be = nil
		return nil
	}
}