Enforce 5 minutes default test timeout

manicminer · Jul 27, 2023 · bf10df8 · bf10df8
1 parent 01d7935
commit bf10df8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/internal/test/testing.go b/internal/test/testing.go
@@ -7,6 +7,7 @@ import (
 	"os"
 	"strconv"
 	"testing"
+	"time"
 
 	"github.com/hashicorp/go-azure-sdk/sdk/auth"
 	"github.com/hashicorp/go-azure-sdk/sdk/claims"
@@ -140,10 +141,12 @@ type Test struct {
 
 func NewTest(t *testing.T) (c *Test) {
 	ctx := context.Background()
-	var cancel context.CancelFunc = func() {}
+	var cancel context.CancelFunc
 
 	if deadline, ok := t.Deadline(); ok {
-		ctx, cancel = context.WithDeadline(context.Background(), deadline)
+		ctx, cancel = context.WithDeadline(ctx, deadline)
+	} else {
+		ctx, cancel = context.WithTimeout(ctx, 5*time.Minute)
 	}
 
 	c = &Test{