Move memory budget test

Author: antonmedv

expr_test.go

@@ -1781,3 +1781,22 @@ func TestEval_slices_out_of_bound(t *testing.T) {
 		})
 	}
 }
+
+func TestMemoryBudget(t *testing.T) {
+	tests := []struct {
+		code string
+	}{
+		{`map(1..100, {map(1..100, {map(1..100, {0})})})`},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.code, func(t *testing.T) {
+			program, err := expr.Compile(tt.code)
+			require.NoError(t, err, "compile error")
+
+			_, err = expr.Run(program, nil)
+			assert.Error(t, err, "run error")
+			assert.Contains(t, err.Error(), "memory budget exceeded")
+		})
+	}
+}

vm/vm_test.go

@@ -133,19 +133,6 @@ func TestRun_Helpers(t *testing.T) {
 	}
 }
 
-func TestRun_MemoryBudget(t *testing.T) {
-	input := `map(1..100, {map(1..100, {map(1..100, {0})})})`
-
-	tree, err := parser.Parse(input)
-	require.NoError(t, err)
-
-	program, err := compiler.Compile(tree, nil)
-	require.NoError(t, err)
-
-	_, err = vm.Run(program, nil)
-	require.Error(t, err)
-}
-
 type ErrorEnv struct {
 	InnerEnv InnerEnv
 }