added tests for raw CEL expression and CEL expression file

Author: aakash070

tools/celtest/test_runner_test.go

@@ -58,6 +58,20 @@ func setupTests() []*testCase {
 			testSuitePath: "testdata/custom_policy_tests.yaml",
 			opts:          []any{customPolicyParserOption(), compiler.PolicyMetadataEnvOption(ParsePolicyVariables)},
 		},
+		{
+			name:          "raw expression file test",
+			celExpression: "testdata/raw_expr.cel",
+			testSuitePath: "testdata/raw_expr_tests",
+			configPath:    "testdata/config.yaml",
+			opts:          []any{fnEnvOption()},
+		},
+		{
+			name:          "raw expression test",
+			celExpression: "'i + fn(j) == 42'",
+			testSuitePath: "testdata/raw_expr_tests",
+			configPath:    "testdata/config.yaml",
+			opts:          []any{fnEnvOption()},
+		},
 	}
 	return testCases
 }
@@ -161,6 +175,15 @@ func (customTagHandler) PolicyTag(ctx policy.ParserContext, id int64, tagName st
 	}
 }
 
+func fnEnvOption() cel.EnvOption {
+	return cel.Function("fn",
+		cel.Overload("fn_int", []*cel.Type{cel.IntType}, cel.IntType,
+			cel.UnaryBinding(func(in ref.Val) ref.Val {
+				i := in.(types.Int)
+				return i / types.Int(2)
+			})))
+}
+
 // TestTriggerTests tests different scenarios of the TriggerTestsFromCompiler function.
 func TestTriggerTests(t *testing.T) {
 	for _, tc := range setupTests() {

tools/celtest/testdata/config.yaml

@@ -0,0 +1,20 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+name: "simple expression config"
+variables:
+  - name: "i"
+    type_name: "int"
+  - name: "j"
+    type_name: "int"

tools/celtest/testdata/raw_expr.cel

@@ -0,0 +1 @@
+"'i + fn(j) == 42'"

tools/celtest/testdata/raw_expr_tests.yaml

@@ -0,0 +1,34 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+description: "simple expression tests"
+section:
+  - name: "valid"
+    tests:
+      - name: "true"
+        input:
+          i:
+            value: 21
+          j:
+            value: 42
+        output:
+          value: true
+      - name: "false"
+        input:
+          i:
+            value: 22
+          j:
+            value:42
+        output:
+          value: false