summaryrefslogtreecommitdiff
path: root/go/internal/testhelper/testhelper.go
diff options
context:
space:
mode:
Diffstat (limited to 'go/internal/testhelper/testhelper.go')
-rw-r--r--go/internal/testhelper/testhelper.go72
1 files changed, 71 insertions, 1 deletions
diff --git a/go/internal/testhelper/testhelper.go b/go/internal/testhelper/testhelper.go
index 5cdab89..5c900aa 100644
--- a/go/internal/testhelper/testhelper.go
+++ b/go/internal/testhelper/testhelper.go
@@ -1,6 +1,18 @@
package testhelper
-import "os"
+import (
+ "fmt"
+ "io/ioutil"
+ "os"
+ "path"
+ "runtime"
+
+ "github.com/otiai10/copy"
+)
+
+var (
+ TestRoot, _ = ioutil.TempDir("", "test-gitlab-shell")
+)
func TempEnv(env map[string]string) func() {
var original = make(map[string]string)
@@ -15,3 +27,61 @@ func TempEnv(env map[string]string) func() {
}
}
}
+
+func PrepareTestRootDir() (func(), error) {
+ if err := os.MkdirAll(TestRoot, 0700); err != nil {
+ return nil, err
+ }
+
+ var oldWd string
+ cleanup := func() {
+ if oldWd != "" {
+ err := os.Chdir(oldWd)
+ if err != nil {
+ panic(err)
+ }
+ }
+
+ if err := os.RemoveAll(TestRoot); err != nil {
+ panic(err)
+ }
+ }
+
+ if err := copyTestData(); err != nil {
+ cleanup()
+ return nil, err
+ }
+
+ oldWd, err := os.Getwd()
+ if err != nil {
+ cleanup()
+ return nil, err
+ }
+
+ if err := os.Chdir(TestRoot); err != nil {
+ cleanup()
+ return nil, err
+ }
+
+ return cleanup, nil
+}
+
+func copyTestData() error {
+ testDataDir, err := getTestDataDir()
+ if err != nil {
+ return err
+ }
+
+ testdata := path.Join(testDataDir, "testroot")
+
+ return copy.Copy(testdata, TestRoot)
+}
+
+func getTestDataDir() (string, error) {
+ _, currentFile, _, ok := runtime.Caller(0)
+ if !ok {
+ return "", fmt.Errorf("Could not get caller info")
+ }
+
+ return path.Join(path.Dir(currentFile), "testdata"), nil
+}