Add protolite tests for Android

PiperOrigin-RevId: 749966327

Author: l46kok

.github/workflows/unwanted_deps.sh

@@ -13,16 +13,23 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Script ran as part of Github CEL-Java CI to verify that the runtime jar does not contain generated cel protos from @cel_spec.
+# Script ran as part of Github CEL-Java CI to verify that the runtime jar does not contain unwanted dependencies.
 
-runtime_deps="$(bazel query 'deps(//publish:cel_runtime)' --nohost_deps --noimplicit_deps --output graph | grep '@cel_spec')"
+UNWANTED_DEPS=(
+  '@cel_spec' # Do not include generated CEL protos in the jar
+  'protobuf_java_util' # Does not support protolite
+)
 
-if [[ ! -z $runtime_deps ]]; then
-  echo -e "Runtime contains unwanted @cel_spec dependency!\n"
-  echo "cel_spec dependency graph:"
-  echo -e "$runtime_deps"
-  exit 1
-fi
+runtime_deps="$(bazel query 'deps(//publish:cel_runtime)' --nohost_deps --noimplicit_deps --output graph)"
+
+for unwanted_dep in "${UNWANTED_DEPS[@]}"; do
+  if echo "$runtime_deps" | grep "$unwanted_dep" > /dev/null; then
+    echo -e "Runtime contains unwanted dependency: $unwanted_dep!\n"
+    echo "cel_spec dependency graph (including '$unwanted_dep'):"
+    echo "$(echo "$runtime_deps" | grep "$unwanted_dep")"
+    exit 1
+  fi
+done
 
 exit 0
 

bundle/src/test/java/dev/cel/bundle/BUILD.bazel

@@ -30,6 +30,7 @@ java_library(
         "//common:proto_ast",
         "//common:source_location",
         "//common/ast",
+        "//common/internal:proto_time_utils",
         "//common/resources/testdata/proto3:standalone_global_enum_java_proto",
         "//common/testing",
         "//common/types",
@@ -55,7 +56,6 @@ java_library(
         "@com_google_googleapis//google/type:type_java_proto",
         "@maven//:com_google_guava_guava",
         "@maven//:com_google_protobuf_protobuf_java",
-        "@maven//:com_google_protobuf_protobuf_java_util",
         "@maven//:com_google_testparameterinjector_test_parameter_injector",
         "@maven//:com_google_truth_extensions_truth_proto_extension",
         "@maven//:junit_junit",

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

@@ -51,7 +51,6 @@
 import com.google.protobuf.Struct;
 import com.google.protobuf.TextFormat;
 import com.google.protobuf.Timestamp;
-import com.google.protobuf.util.Timestamps;
 import com.google.rpc.context.AttributeContext;
 import com.google.testing.junit.testparameterinjector.TestParameter;
 import com.google.testing.junit.testparameterinjector.TestParameterInjector;
@@ -71,6 +70,7 @@
 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;
@@ -409,6 +409,7 @@ public void program_setTypeFactoryOnAnyPackedMessage_messageConstructionSucceeds
   }
 
   @Test
+  @SuppressWarnings("unused") // testRunIndex name retained for test result readability
   public void program_concurrentMessageConstruction_succeeds(
       @TestParameter(valuesProvider = RepeatedTestProvider.class) int testRunIndex)
       throws Exception {
@@ -814,7 +815,7 @@ public void program_duplicateTypeDescriptor() throws Exception {
             .build();
     CelRuntime.Program program =
         cel.createProgram(cel.compile("protobuf.Timestamp{seconds: 12}").getAst());
-    assertThat(program.eval()).isEqualTo(Timestamps.fromSeconds(12));
+    assertThat(program.eval()).isEqualTo(ProtoTimeUtils.fromSecondsToTimestamp(12));
   }
 
   @Test
@@ -827,7 +828,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(Timestamps.fromSeconds(12));
+    assertThat(program.eval()).isEqualTo(ProtoTimeUtils.fromSecondsToTimestamp(12));
   }
 
   @Test
@@ -956,7 +957,7 @@ public void program_typeProvider() throws Exception {
             .build();
     CelRuntime.Program program =
         cel.createProgram(cel.compile("protobuf.Timestamp{seconds: 12}").getAst());
-    assertThat(program.eval()).isEqualTo(Timestamps.fromSeconds(12));
+    assertThat(program.eval()).isEqualTo(ProtoTimeUtils.fromSecondsToTimestamp(12));
   }
 
   @Test

codelab/README.md

@@ -667,7 +667,7 @@ public void evaluate_jwtWithTimeVariable_producesJsonString() throws Exception {
   @SuppressWarnings("unchecked")
   Map<String, Object> evaluatedResult =
       (Map<String, Object>)
-          exercise5.eval(ast, ImmutableMap.of("time", Timestamps.fromSeconds(1698361778)));
+          exercise5.eval(ast, ImmutableMap.of("time", ProtoTimeUtils.fromSecondsToTimestamp(1698361778)));
   String jsonOutput = exercise5.toJson(evaluatedResult);
 
   assertThat(jsonOutput)
@@ -776,7 +776,7 @@ public void evaluate_constructAttributeContext() {
           + "time: now"
           + "}";
   // Values for `now` and `jwt` variables to be passed into the runtime
-  Timestamp now = Timestamps.now();
+  Timestamp now = ProtoTimeUtils.now();
   ImmutableMap<String, Object> jwt =
       ImmutableMap.of(
           "sub", "serviceAccount:delegate@acme.co",

codelab/src/test/codelab/BUILD.bazel

@@ -78,7 +78,7 @@ java_test(
         "//codelab",
         "//common:cel_ast",
         "@maven//:com_google_guava_guava",
-        "@maven//:com_google_protobuf_protobuf_java_util",
+        "@maven//:com_google_protobuf_protobuf_java",
         "@maven//:com_google_testparameterinjector_test_parameter_injector",
         "@maven//:junit_junit",
     ],
@@ -93,10 +93,10 @@ java_test(
         "//:java_truth",
         "//codelab",
         "//common:cel_ast",
+        "//common/internal:proto_time_utils",
         "@com_google_googleapis//google/rpc/context:attribute_context_java_proto",
         "@maven//:com_google_guava_guava",
         "@maven//:com_google_protobuf_protobuf_java",
-        "@maven//:com_google_protobuf_protobuf_java_util",
         "@maven//:com_google_testparameterinjector_test_parameter_injector",
         "@maven//:junit_junit",
     ],

codelab/src/test/codelab/Exercise5Test.java

@@ -17,7 +17,7 @@
 import static com.google.common.truth.Truth.assertThat;
 
 import com.google.common.collect.ImmutableMap;
-import com.google.protobuf.util.Timestamps;
+import com.google.protobuf.Timestamp;
 import com.google.testing.junit.testparameterinjector.TestParameterInjector;
 import dev.cel.common.CelAbstractSyntaxTree;
 import java.util.Map;
@@ -48,7 +48,9 @@ public void evaluate_jwtWithTimeVariable_producesJsonString() throws Exception {
     @SuppressWarnings("unchecked")
     Map<String, Object> evaluatedResult =
         (Map<String, Object>)
-            exercise5.eval(ast, ImmutableMap.of("time", Timestamps.fromSeconds(1698361778)));
+            exercise5.eval(
+                ast,
+                ImmutableMap.of("time", Timestamp.newBuilder().setSeconds(1698361778).build()));
     String jsonOutput = exercise5.toJson(evaluatedResult);
 
     assertThat(jsonOutput)

codelab/src/test/codelab/Exercise6Test.java

@@ -20,10 +20,10 @@
 import com.google.protobuf.Struct;
 import com.google.protobuf.Timestamp;
 import com.google.protobuf.Value;
-import com.google.protobuf.util.Timestamps;
 import com.google.rpc.context.AttributeContext;
 import com.google.testing.junit.testparameterinjector.TestParameterInjector;
 import dev.cel.common.CelAbstractSyntaxTree;
+import dev.cel.common.internal.ProtoTimeUtils;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 
@@ -51,7 +51,7 @@ public void evaluate_constructAttributeContext() {
             + "time: now"
             + "}";
     // Values for `now` and `jwt` variables to be passed into the runtime
-    Timestamp now = Timestamps.now();
+    Timestamp now = ProtoTimeUtils.now();
     ImmutableMap<String, Object> jwt =
         ImmutableMap.of(
             "sub", "serviceAccount:delegate@acme.co",

codelab/src/test/codelab/solutions/BUILD.bazel

@@ -73,7 +73,7 @@ java_test(
         "//codelab:solutions",
         "//common:cel_ast",
         "@maven//:com_google_guava_guava",
-        "@maven//:com_google_protobuf_protobuf_java_util",
+        "@maven//:com_google_protobuf_protobuf_java",
         "@maven//:com_google_testparameterinjector_test_parameter_injector",
         "@maven//:junit_junit",
     ],
@@ -87,10 +87,10 @@ java_test(
         "//:java_truth",
         "//codelab:solutions",
         "//common:cel_ast",
+        "//common/internal:proto_time_utils",
         "@com_google_googleapis//google/rpc/context:attribute_context_java_proto",
         "@maven//:com_google_guava_guava",
         "@maven//:com_google_protobuf_protobuf_java",
-        "@maven//:com_google_protobuf_protobuf_java_util",
         "@maven//:com_google_testparameterinjector_test_parameter_injector",
         "@maven//:junit_junit",
     ],

codelab/src/test/codelab/solutions/Exercise5Test.java

@@ -17,7 +17,7 @@
 import static com.google.common.truth.Truth.assertThat;
 
 import com.google.common.collect.ImmutableMap;
-import com.google.protobuf.util.Timestamps;
+import com.google.protobuf.Timestamp;
 import com.google.testing.junit.testparameterinjector.TestParameterInjector;
 import dev.cel.common.CelAbstractSyntaxTree;
 import java.util.Map;
@@ -48,7 +48,9 @@ public void evaluate_jwtWithTimeVariable_producesJsonString() throws Exception {
     @SuppressWarnings("unchecked")
     Map<String, Object> evaluatedResult =
         (Map<String, Object>)
-            exercise5.eval(ast, ImmutableMap.of("time", Timestamps.fromSeconds(1698361778)));
+            exercise5.eval(
+                ast,
+                ImmutableMap.of("time", Timestamp.newBuilder().setSeconds(1698361778).build()));
     String jsonOutput = exercise5.toJson(evaluatedResult);
 
     assertThat(jsonOutput)

codelab/src/test/codelab/solutions/Exercise6Test.java

@@ -20,10 +20,10 @@
 import com.google.protobuf.Struct;
 import com.google.protobuf.Timestamp;
 import com.google.protobuf.Value;
-import com.google.protobuf.util.Timestamps;
 import com.google.rpc.context.AttributeContext;
 import com.google.testing.junit.testparameterinjector.TestParameterInjector;
 import dev.cel.common.CelAbstractSyntaxTree;
+import dev.cel.common.internal.ProtoTimeUtils;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 
@@ -51,7 +51,7 @@ public void evaluate_constructAttributeContext() {
             + "time: now"
             + "}";
     // Values for `now` and `jwt` variables to be passed into the runtime
-    Timestamp now = Timestamps.now();
+    Timestamp now = ProtoTimeUtils.now();
     ImmutableMap<String, Object> jwt =
         ImmutableMap.of(
             "sub", "serviceAccount:delegate@acme.co",