Function TestVM_MemoryBudget

vm/vm_test.go:1302–1347

(t *testing.T)

}

func TestVM_MemoryBudget(t *testing.T) {
	tests := []struct {
		name        string
		expr        string
		memBudget   uint
		expectError string
	}{
		{
			name:      "under budget",
			expr:      "map(1..10, #)",
			memBudget: 100,
		},
		{
			name:        "exceeds budget",
			expr:        "map(1..1000, #)",
			memBudget:   10,
			expectError: "memory budget exceeded",
		},
		{
			name:      "zero budget uses default",
			expr:      "map(1..10, #)",
			memBudget: 0,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			node, err := parser.Parse(tt.expr)
			require.NoError(t, err)

			program, err := compiler.Compile(node, nil)
			require.NoError(t, err)

			vm := vm.VM{MemoryBudget: tt.memBudget}
			out, err := vm.Run(program, nil)

			if tt.expectError != "" {
				require.Error(t, err)
				require.Contains(t, err.Error(), tt.expectError)
			} else {
				require.NoError(t, err)
				require.NotNil(t, out)
			}
		})
	}
}

// Helper functions for creating deeply nested expressions
func createNestedArithmeticExpr(t *testing.T, depth int) string {