deduplicate memory quota test compile setup

Author: mgomes

vibes/memory_quota_test.go

@@ -41,16 +41,12 @@ end
 `
 
 func TestMemoryQuotaExceeded(t *testing.T) {
-	engine := MustNewEngine(Config{
+	script := compileScriptWithConfig(t, Config{
 		StepQuota:        20000,
 		MemoryQuotaBytes: 2048,
-	})
-
-	script, err := engine.Compile(quotaFixture)
-	if err != nil {
-		t.Fatalf("compile failed: %v", err)
-	}
+	}, quotaFixture)
 
+	var err error
 	_, err = script.Call(context.Background(), "run", nil, CallOptions{})
 	if err == nil {
 		t.Fatalf("expected memory quota error")
@@ -61,16 +57,12 @@ func TestMemoryQuotaExceeded(t *testing.T) {
 }
 
 func TestMemoryQuotaCountsClassVars(t *testing.T) {
-	engine := MustNewEngine(Config{
+	script := compileScriptWithConfig(t, Config{
 		StepQuota:        20000,
 		MemoryQuotaBytes: 3072,
-	})
-
-	script, err := engine.Compile(classVarFixture)
-	if err != nil {
-		t.Fatalf("compile failed: %v", err)
-	}
+	}, classVarFixture)
 
+	var err error
 	_, err = script.Call(context.Background(), "run", nil, CallOptions{})
 	if err == nil {
 		t.Fatalf("expected memory quota error")
@@ -81,16 +73,12 @@ func TestMemoryQuotaCountsClassVars(t *testing.T) {
 }
 
 func TestMemoryQuotaAllowsExecution(t *testing.T) {
-	engine := MustNewEngine(Config{
+	script := compileScriptWithConfig(t, Config{
 		StepQuota:        20000,
 		MemoryQuotaBytes: 1 << 20,
-	})
-
-	script, err := engine.Compile(quotaFixture)
-	if err != nil {
-		t.Fatalf("compile failed: %v", err)
-	}
+	}, quotaFixture)
 
+	var err error
 	result, err := script.Call(context.Background(), "run", nil, CallOptions{})
 	if err != nil {
 		t.Fatalf("unexpected error: %v", err)
@@ -101,17 +89,13 @@ func TestMemoryQuotaAllowsExecution(t *testing.T) {
 }
 
 func TestMemoryQuotaExceededOnCompletion(t *testing.T) {
-	engine := MustNewEngine(Config{
+	script := compileScriptWithConfig(t, Config{
 		StepQuota:        20000,
 		MemoryQuotaBytes: 2048,
-	})
-
-	script, err := engine.Compile(splitFixture)
-	if err != nil {
-		t.Fatalf("compile failed: %v", err)
-	}
+	}, splitFixture)
 
 	input := strings.Repeat("a,", 4000)
+	var err error
 	_, err = script.Call(context.Background(), "run", []Value{NewString(input)}, CallOptions{})
 	if err == nil {
 		t.Fatalf("expected memory quota error")
@@ -122,20 +106,18 @@ func TestMemoryQuotaExceededOnCompletion(t *testing.T) {
 }
 
 func TestMemoryQuotaExceededForEmptyBodyDefaultArg(t *testing.T) {
-	engine := MustNewEngine(Config{
+	cfg := Config{
 		StepQuota:        20000,
 		MemoryQuotaBytes: 2048,
-	})
+	}
 
 	largeCSV := strings.Repeat("abcdefghij,", 1500)
 	source := `def run(payload = "` + largeCSV + `".split(","))
 end`
 
-	script, err := engine.Compile(source)
-	if err != nil {
-		t.Fatalf("compile failed: %v", err)
-	}
+	script := compileScriptWithConfig(t, cfg, source)
 
+	var err error
 	_, err = script.Call(context.Background(), "run", nil, CallOptions{})
 	if err == nil {
 		t.Fatalf("expected memory quota error")
@@ -146,23 +128,19 @@ end`
 }
 
 func TestMemoryQuotaExceededForBoundArguments(t *testing.T) {
-	engine := MustNewEngine(Config{
+	script := compileScriptWithConfig(t, Config{
 		StepQuota:        20000,
 		MemoryQuotaBytes: 2048,
-	})
-
-	script, err := engine.Compile(`def run(payload)
+	}, `def run(payload)
 end`)
-	if err != nil {
-		t.Fatalf("compile failed: %v", err)
-	}
 
 	parts := make([]Value, 2000)
 	for i := range parts {
 		parts[i] = NewString("abcdefghij")
 	}
 	largeArg := NewArray(parts)
 
+	var err error
 	_, err = script.Call(context.Background(), "run", []Value{largeArg}, CallOptions{})
 	if err == nil {
 		t.Fatalf("expected memory quota error for positional arg")
@@ -185,22 +163,18 @@ end`)
 }
 
 func TestMemoryQuotaCountsIndependentEmptySlices(t *testing.T) {
-	engine := MustNewEngine(Config{
+	script := compileScriptWithConfig(t, Config{
 		StepQuota:        20000,
 		MemoryQuotaBytes: 4096,
-	})
-
-	script, err := engine.Compile(`def run
+	}, `def run
   items = []
   for i in 1..400
     items = items.push([])
   end
   items.size
 end`)
-	if err != nil {
-		t.Fatalf("compile failed: %v", err)
-	}
 
+	var err error
 	_, err = script.Call(context.Background(), "run", nil, CallOptions{})
 	if err == nil {
 		t.Fatalf("expected memory quota error for many independent empty slices")
@@ -211,12 +185,10 @@ end`)
 }
 
 func TestMemoryQuotaExceededWithWhileLoopAllocations(t *testing.T) {
-	engine := MustNewEngine(Config{
+	script := compileScriptWithConfig(t, Config{
 		StepQuota:        20000,
 		MemoryQuotaBytes: 2048,
-	})
-
-	script, err := engine.Compile(`def run()
+	}, `def run()
   items = []
   n = 0
   while n < 200
@@ -225,10 +197,8 @@ func TestMemoryQuotaExceededWithWhileLoopAllocations(t *testing.T) {
   end
   items.size
 end`)
-	if err != nil {
-		t.Fatalf("compile failed: %v", err)
-	}
 
+	var err error
 	_, err = script.Call(context.Background(), "run", nil, CallOptions{})
 	if err == nil {
 		t.Fatalf("expected memory quota error for while-loop allocations")