From 0ad6bdbc0952a313ebf0a081b0f1fc303a47939a Mon Sep 17 00:00:00 2001
From: Alexander Neumann <alexander@bumpern.de>
Date: Sat, 4 Jul 2015 16:51:59 +0200
Subject: [PATCH] Add more test helpers

---
 test/helpers.go | 58 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)

diff --git a/test/helpers.go b/test/helpers.go
index b99f56dec..544195d9b 100644
--- a/test/helpers.go
+++ b/test/helpers.go
@@ -3,13 +3,18 @@ package test_helper
 import (
 	"bytes"
 	"fmt"
+	"io/ioutil"
 	"math/rand"
+	"os"
+	"os/exec"
 	"path/filepath"
 	"reflect"
 	"runtime"
 	"testing"
 
 	"github.com/restic/restic/backend"
+	"github.com/restic/restic/backend/local"
+	"github.com/restic/restic/repository"
 )
 
 // Assert fails the test if the condition is false.
@@ -65,3 +70,56 @@ func Random(seed, count int) []byte {
 func RandomReader(seed, size int) *bytes.Reader {
 	return bytes.NewReader(Random(seed, size))
 }
+
+// SetupTarTestFixture extracts the tarFile to outputDir.
+func SetupTarTestFixture(t testing.TB, outputDir, tarFile string) {
+	err := System("sh", "-c", `(cd "$1" && tar xz) < "$2"`,
+		"sh", outputDir, tarFile)
+	OK(t, err)
+}
+
+// System runs the command and returns the exit code. Output is passed on to
+// stdout/stderr.
+func System(command string, args ...string) error {
+	cmd := exec.Command(command, args...)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	return cmd.Run()
+}
+
+// WithTestEnvironment creates a test environment, extracts the repository
+// fixture and and calls f with the repository dir.
+func WithTestEnvironment(t testing.TB, repoFixture string, f func(repodir string)) {
+	tempdir, err := ioutil.TempDir(TestTempDir, "restic-test-")
+	OK(t, err)
+
+	fd, err := os.Open(repoFixture)
+	if err != nil {
+		panic(err)
+	}
+	OK(t, fd.Close())
+
+	SetupTarTestFixture(t, tempdir, repoFixture)
+
+	f(filepath.Join(tempdir, "repo"))
+
+	if !TestCleanup {
+		t.Logf("leaving temporary directory %v used for test", tempdir)
+		return
+	}
+
+	OK(t, os.RemoveAll(tempdir))
+}
+
+// OpenLocalRepo opens the local repository located at dir.
+func OpenLocalRepo(t testing.TB, dir string) *repository.Repository {
+	be, err := local.Open(dir)
+	OK(t, err)
+
+	repo := repository.New(be)
+	err = repo.SearchKey(TestPassword)
+	OK(t, err)
+
+	return repo
+}