add option to use cache in test context (#19)

gov4git · Sep 18, 2023 · 0fa8cb8 · 0fa8cb8
1 parent 7205a53
commit 0fa8cb8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/id/init_test.go b/id/init_test.go
@@ -9,7 +9,7 @@ import (
 )
 
 func TestInit(t *testing.T) {
-	ctx := testutil.NewCtx()
+	ctx := testutil.NewCtx(t, false)
 	testID := NewTestID(ctx, t, git.MainBranch, true)
 	Init(ctx, testID.OwnerAddress())
 	if err := must.Try(func() { Init(ctx, testID.OwnerAddress()) }); err == nil {

diff --git a/testutil/ctx.go b/testutil/ctx.go
@@ -2,16 +2,24 @@ package testutil
 
 import (
 	"context"
+	"math/rand"
+	"path/filepath"
 	"strconv"
 	"sync"
+	"testing"
 
 	"github.com/gov4git/lib4git/git"
 )
 
-func NewCtx() context.Context {
+func NewCtx(t *testing.T, useCache bool) context.Context {
 	ctx := git.WithAuth(context.Background(), nil)
 	ctx = git.WithTTL(ctx, nil)
-	return context.WithValue(ctx, counterKey{}, &counter{})
+	ctx = context.WithValue(ctx, counterKey{}, &counter{})
+	if useCache {
+		return git.WithCache(ctx, filepath.Join(t.TempDir(), "cache-"+strconv.FormatUint(uint64(rand.Int63()), 36)))
+	} else {
+		return ctx
+	}
 }
 
 type counterKey struct{}