Function TestInterpreter_RegexProgramSizeLimit

interpreter/interpreter_test.go:2116–2180

(t *testing.T)

}

func TestInterpreter_RegexProgramSizeLimit(t *testing.T) {
	tcConst := testCase{
		expr: `'hello'.matches('(a|b)*[0-9]+')`,
	}
	_, _, err := program(t, &tcConst, RegexProgramSizeLimit(5))
	if err == nil {
		t.Fatalf("expected program creation error for constant regex exceeding limit")
	}
	if !strings.Contains(err.Error(), "regex program size 8 exceeds limit of 5") {
		t.Errorf("got error %v, wanted error containing 'regex program size 8 exceeds limit of 5'", err)
	}

	tcDyn := testCase{
		expr: `'hello'.matches(pattern)`,
		vars: []*decls.VariableDecl{
			decls.NewVariable("pattern", types.StringType),
		},
		in: map[string]any{
			"pattern": "(a|b)*[0-9]+",
		},
	}
	prg, frame, err := program(t, &tcDyn, RegexProgramSizeLimit(5))
	if err != nil {
		t.Fatalf("program() failed: %v", err)
	}
	out := prg.Exec(frame)
	frame.Close()
	if !types.IsError(out) || !strings.Contains(out.(*types.Err).String(), "regex program size 8 exceeds limit of 5") {
		t.Errorf("got %v, wanted regex program size limit error", out)
	}

	tcValid := testCase{
		expr: `'hello'.matches(pattern)`,
		vars: []*decls.VariableDecl{
			decls.NewVariable("pattern", types.StringType),
		},
		in: map[string]any{
			"pattern": "el*",
		},
		out: true,
	}
	prgValid, frameValid, err := program(t, &tcValid, RegexProgramSizeLimit(5))
	if err != nil {
		t.Fatalf("program() failed: %v", err)
	}
	outValid := prgValid.Exec(frameValid)
	frameValid.Close()
	if outValid != types.True {
		t.Errorf("got %v, wanted true", outValid)
	}

	// Non-regex function should not be modified by RegexProgramSizeLimit decorator
	tcOther := testCase{
		expr: `'hello'.contains('e')`,
	}
	prgOther, frameOther, err := program(t, &tcOther, RegexProgramSizeLimit(5))
	if err != nil {
		t.Fatalf("program() failed: %v", err)