restic/test/backend.go

53 lines
1.4 KiB
Go
Raw Normal View History

2015-04-26 12:46:15 +00:00
package test_helper
import (
"flag"
"io/ioutil"
"os"
"path/filepath"
"testing"
"github.com/restic/restic"
"github.com/restic/restic/backend"
"github.com/restic/restic/backend/local"
2015-05-09 11:21:28 +00:00
"github.com/restic/restic/repo"
2015-04-26 12:46:15 +00:00
)
2015-05-03 14:36:52 +00:00
var TestPassword = flag.String("test.password", "", `use this password for repositories created during tests (default: "geheim")`)
2015-04-26 12:46:15 +00:00
var TestCleanup = flag.Bool("test.cleanup", true, "clean up after running tests (remove local backend directory with all content)")
var TestTempDir = flag.String("test.tempdir", "", "use this directory for temporary storage (default: system temp dir)")
2015-05-09 11:25:52 +00:00
func SetupBackend(t testing.TB) *repo.Repository {
2015-04-26 12:46:15 +00:00
tempdir, err := ioutil.TempDir(*TestTempDir, "restic-test-")
OK(t, err)
// create repository below temp dir
b, err := local.Create(filepath.Join(tempdir, "repo"))
OK(t, err)
2015-05-03 14:36:52 +00:00
// set cache dir below temp dir
2015-04-26 12:46:15 +00:00
err = os.Setenv("RESTIC_CACHE", filepath.Join(tempdir, "cache"))
OK(t, err)
2015-05-09 11:25:52 +00:00
s := repo.New(b)
OK(t, s.Init(*TestPassword))
2015-05-03 14:36:52 +00:00
return s
2015-04-26 12:46:15 +00:00
}
2015-05-09 11:25:52 +00:00
func TeardownBackend(t testing.TB, s *repo.Repository) {
2015-04-26 12:46:15 +00:00
if !*TestCleanup {
l := s.Backend().(*local.Local)
t.Logf("leaving local backend at %s\n", l.Location())
return
}
OK(t, s.Delete())
}
2015-05-09 11:25:52 +00:00
func SnapshotDir(t testing.TB, server *repo.Repository, path string, parent backend.ID) *restic.Snapshot {
2015-04-30 01:41:51 +00:00
arch := restic.NewArchiver(server)
2015-04-26 12:46:15 +00:00
sn, _, err := arch.Snapshot(nil, []string{path}, parent)
OK(t, err)
return sn
}