Enable evaluateCanonicalTypesToNativeValues by default

PiperOrigin-RevId: 808650209

Author: l46kok

bundle/src/test/java/dev/cel/bundle/CelImplTest.java

@@ -72,7 +72,6 @@
 import dev.cel.common.CelVarDecl;
 import dev.cel.common.ast.CelExpr;
 import dev.cel.common.ast.CelExpr.CelList;
-import dev.cel.common.internal.ProtoTimeUtils;
 import dev.cel.common.testing.RepeatedTestProvider;
 import dev.cel.common.types.CelKind;
 import dev.cel.common.types.CelProtoMessageTypes;
@@ -108,6 +107,7 @@
 import dev.cel.runtime.CelVariableResolver;
 import dev.cel.runtime.UnknownContext;
 import dev.cel.testing.testdata.proto3.StandaloneGlobalEnum;
+import java.time.Instant;
 import java.util.ArrayList;
 import java.util.LinkedHashMap;
 import java.util.List;
@@ -820,7 +820,8 @@ public void program_duplicateTypeDescriptor() throws Exception {
             .build();
     CelRuntime.Program program =
         cel.createProgram(cel.compile("protobuf.Timestamp{seconds: 12}").getAst());
-    assertThat(program.eval()).isEqualTo(ProtoTimeUtils.fromSecondsToTimestamp(12));
+
+    assertThat(program.eval()).isEqualTo(Instant.ofEpochSecond(12));
   }
 
   @Test
@@ -833,7 +834,7 @@ public void program_hermeticDescriptors_wellKnownProtobuf() throws Exception {
             .build();
     CelRuntime.Program program =
         cel.createProgram(cel.compile("protobuf.Timestamp{seconds: 12}").getAst());
-    assertThat(program.eval()).isEqualTo(ProtoTimeUtils.fromSecondsToTimestamp(12));
+    assertThat(program.eval()).isEqualTo(Instant.ofEpochSecond(12));
   }
 
   @Test
@@ -962,7 +963,7 @@ public void program_typeProvider() throws Exception {
             .build();
     CelRuntime.Program program =
         cel.createProgram(cel.compile("protobuf.Timestamp{seconds: 12}").getAst());
-    assertThat(program.eval()).isEqualTo(ProtoTimeUtils.fromSecondsToTimestamp(12));
+    assertThat(program.eval()).isEqualTo(Instant.ofEpochSecond(12));
   }
 
   @Test

codelab/README.md

@@ -1065,8 +1065,8 @@ public void validate_invalidTimestampLiteral_returnsError() throws Exception {
   assertThat(validationResult.hasError()).isTrue();
   assertThat(validationResult.getErrorString())
       .isEqualTo(
-          "ERROR: <input>:1:11: timestamp validation failed. Reason: evaluation error: Failed to"
-              + " parse timestamp: invalid timestamp \"bad\"\n"
+          "ERROR: <input>:1:11: timestamp validation failed. Reason: evaluation error: Text 'bad'"
+              + " could not be parsed at index 0\n"
               + " | timestamp('bad')\n"
               + " | ..........^");
 }

codelab/src/test/codelab/Exercise8Test.java

@@ -43,8 +43,8 @@ public void validate_invalidTimestampLiteral_returnsError() throws Exception {
     assertThat(validationResult.hasError()).isTrue();
     assertThat(validationResult.getErrorString())
         .isEqualTo(
-            "ERROR: <input>:1:11: timestamp validation failed. Reason: evaluation error: Failed to"
-                + " parse timestamp: invalid timestamp \"bad\"\n"
+            "ERROR: <input>:1:11: timestamp validation failed. Reason: evaluation error: Text 'bad'"
+                + " could not be parsed at index 0\n"
                 + " | timestamp('bad')\n"
                 + " | ..........^");
   }

codelab/src/test/codelab/solutions/Exercise8Test.java

@@ -43,8 +43,8 @@ public void validate_invalidTimestampLiteral_returnsError() throws Exception {
     assertThat(validationResult.hasError()).isTrue();
     assertThat(validationResult.getErrorString())
         .isEqualTo(
-            "ERROR: <input>:1:11: timestamp validation failed. Reason: evaluation error: Failed to"
-                + " parse timestamp: invalid timestamp \"bad\"\n"
+            "ERROR: <input>:1:11: timestamp validation failed. Reason: evaluation error: Text 'bad'"
+                + " could not be parsed at index 0\n"
                 + " | timestamp('bad')\n"
                 + " | ..........^");
   }

common/src/main/java/dev/cel/common/CelOptions.java

@@ -184,6 +184,7 @@ public static Builder current() {
         .enableUnsignedLongs(true)
         .enableRegexPartialMatch(true)
         .errorOnDuplicateMapKeys(true)
+        .evaluateCanonicalTypesToNativeValues(true)
         .errorOnIntWrap(true)
         .resolveTypeDependencies(true)
         .disableCelStandardEquality(false);
@@ -462,6 +463,8 @@ public abstract static class Builder {
      *       com.google.protobuf.ByteString}.
      *   <li>CEL null: {@code dev.cel.common.values.NullValue} instead of {@code
      *       com.google.protobuf.NullValue}.
+     *   <li>Timestamp: {@code java.time.Instant} instead of {@code com.google.protobuf.Timestamp}.
+     *   <li>Duration: {@code java.time.Duration} instead of {@code com.google.protobuf.Duration}.
      * </ul>
      */
     public abstract Builder evaluateCanonicalTypesToNativeValues(boolean value);

common/src/test/java/dev/cel/common/internal/ProtoAdapterTest.java

@@ -41,6 +41,7 @@
 import com.google.type.Expr;
 import dev.cel.common.CelOptions;
 import dev.cel.common.values.CelByteString;
+import java.time.Instant;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Optional;
@@ -88,11 +89,10 @@ public static List<Object[]> data() {
             {1.5D, Any.pack(DoubleValue.of(1.5D))},
             {1.5D, Value.newBuilder().setNumberValue(1.5D).build()},
             {
-              Duration.newBuilder().setSeconds(123).build(),
-              Duration.newBuilder().setSeconds(123).build(),
+              java.time.Duration.ofSeconds(123), Duration.newBuilder().setSeconds(123).build(),
             },
             {
-              Duration.newBuilder().setSeconds(123).build(),
+              java.time.Duration.ofSeconds(123),
               Any.pack(Duration.newBuilder().setSeconds(123).build()),
             },
             {1L, Int64Value.of(1L)},
@@ -132,12 +132,10 @@ public static List<Object[]> data() {
                   .build(),
             },
             {
-              Timestamp.newBuilder().setSeconds(123).build(),
-              Timestamp.newBuilder().setSeconds(123).build(),
+              Instant.ofEpochSecond(123), Timestamp.newBuilder().setSeconds(123).build(),
             },
             {
-              Timestamp.newBuilder().setSeconds(123).build(),
-              Any.pack(Timestamp.newBuilder().setSeconds(123).build()),
+              Instant.ofEpochSecond(123), Any.pack(Timestamp.newBuilder().setSeconds(123).build()),
             },
             {UnsignedLong.valueOf(1L), UInt64Value.of(1L)},
             {UnsignedLong.valueOf(1L), Any.pack(UInt64Value.of(1L))},

common/src/test/java/dev/cel/common/values/ProtoMessageValueProviderTest.java

@@ -28,7 +28,6 @@
 import dev.cel.common.internal.DefaultMessageFactory;
 import dev.cel.common.internal.DynamicProto;
 import dev.cel.common.internal.ProtoMessageFactory;
-import dev.cel.common.internal.ProtoTimeUtils;
 import dev.cel.expr.conformance.proto2.TestAllTypes;
 import dev.cel.expr.conformance.proto2.TestAllTypesExtensions;
 import java.time.Duration;
@@ -89,9 +88,9 @@ public void newValue_createProtoMessage_fieldsPopulated() {
                         "single_string",
                         "hello",
                         "single_timestamp",
-                        ProtoTimeUtils.fromSecondsToTimestamp(50),
+                        Instant.ofEpochSecond(50),
                         "single_duration",
-                        ProtoTimeUtils.fromSecondsToDuration(100)))
+                        Duration.ofSeconds(100)))
                 .get();
 
     assertThat(protoMessageValue.isZeroValue()).isFalse();

validator/src/main/java/dev/cel/validator/validators/DurationLiteralValidator.java

@@ -14,7 +14,7 @@
 
 package dev.cel.validator.validators;
 
-import com.google.protobuf.Duration;
+import java.time.Duration;
 
 /** DurationLiteralValidator ensures that duration literal arguments are valid. */
 public final class DurationLiteralValidator extends LiteralValidator {

validator/src/main/java/dev/cel/validator/validators/TimestampLiteralValidator.java

@@ -14,12 +14,12 @@
 
 package dev.cel.validator.validators;
 
-import com.google.protobuf.Timestamp;
+import java.time.Instant;
 
 /** TimestampLiteralValidator ensures that timestamp literal arguments are valid. */
 public final class TimestampLiteralValidator extends LiteralValidator {
   public static final TimestampLiteralValidator INSTANCE =
-      new TimestampLiteralValidator("timestamp", Timestamp.class);
+      new TimestampLiteralValidator("timestamp", Instant.class);
 
   private TimestampLiteralValidator(String functionName, Class<?> expectedResultType) {
     super(functionName, expectedResultType);

validator/src/test/java/dev/cel/validator/validators/DurationLiteralValidatorTest.java

@@ -20,7 +20,6 @@
 import static org.junit.Assert.assertThrows;
 
 import com.google.common.collect.ImmutableMap;
-import com.google.protobuf.Duration;
 import com.google.testing.junit.testparameterinjector.TestParameterInjector;
 import com.google.testing.junit.testparameterinjector.TestParameters;
 import dev.cel.bundle.Cel;
@@ -35,6 +34,7 @@
 import dev.cel.validator.CelValidator;
 import dev.cel.validator.CelValidatorFactory;
 import java.text.ParseException;
+import java.time.Duration;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 
@@ -174,7 +174,7 @@ public void duration_unexpectedResultType_throws() throws Exception {
     assertThat(result.getAllIssues().get(0).toDisplayString(ast.getSource()))
         .isEqualTo(
             "ERROR: <input>:1:10: duration validation failed. Reason: Expected"
-                + " com.google.protobuf.Duration type but got java.lang.Integer instead\n"
+                + " java.time.Duration type but got java.lang.Integer instead\n"
                 + " | duration('1h')\n"
                 + " | .........^");
   }