Refactor testhelper.PrepareTestRootDir using t.Cleanup

2 files changed, 11 insertions, 32 deletions

diff --git a/internal/gitlabnet/accessverifier/client_test.go b/internal/gitlabnet/accessverifier/client_test.go
index fdf5705..d3f34f6 100644
--- a/internal/gitlabnet/accessverifier/client_test.go
+++ b/internal/gitlabnet/accessverifier/client_test.go
@@ -163,9 +163,7 @@ func TestErrorResponses(t *testing.T) {
 }
 
 func setup(t *testing.T, allowedPayload string) *Client {
-	testDirCleanup, err := testhelper.PrepareTestRootDir()
-	require.NoError(t, err)
-	defer testDirCleanup()
+	testhelper.PrepareTestRootDir(t)
 
 	body, err := ioutil.ReadFile(path.Join(testhelper.TestRoot, "responses/allowed.json"))
 	require.NoError(t, err)
diff --git a/internal/testhelper/testhelper.go b/internal/testhelper/testhelper.go
index 7361cbb..c2c7826 100644
--- a/internal/testhelper/testhelper.go
+++ b/internal/testhelper/testhelper.go
@@ -7,10 +7,12 @@ import (
 	"path"
 	"runtime"
 	"time"
+	"testing"
 
 	"github.com/otiai10/copy"
 	"github.com/sirupsen/logrus"
 	"github.com/sirupsen/logrus/hooks/test"
+	"github.com/stretchr/testify/require"
 )
 
 var (
@@ -31,42 +33,21 @@ func TempEnv(env map[string]string) func() {
 	}
 }
 
-func PrepareTestRootDir() (func(), error) {
-	if err := os.MkdirAll(TestRoot, 0700); err != nil {
-		return nil, err
-	}
+func PrepareTestRootDir(t *testing.T) {
+	t.Helper()
 
-	var oldWd string
-	cleanup := func() {
-		if oldWd != "" {
-			err := os.Chdir(oldWd)
-			if err != nil {
-				panic(err)
-			}
-		}
+	require.NoError(t, os.MkdirAll(TestRoot, 0700))
 
-		if err := os.RemoveAll(TestRoot); err != nil {
-			panic(err)
-		}
-	}
+	t.Cleanup(func() { os.RemoveAll(TestRoot) })
 
-	if err := copyTestData(); err != nil {
-		cleanup()
-		return nil, err
-	}
+	require.NoError(t, copyTestData())
 
 	oldWd, err := os.Getwd()
-	if err != nil {
-		cleanup()
-		return nil, err
-	}
+	require.NoError(t, err)
 
-	if err := os.Chdir(TestRoot); err != nil {
-		cleanup()
-		return nil, err
-	}
+	t.Cleanup(func() { os.Chdir(oldWd) })
 
-	return cleanup, nil
+	require.NoError(t, os.Chdir(TestRoot))
 }
 
 func copyTestData() error {