feat: add determinism level for javascript UDFs

Author: plamut

google/cloud/bigquery/routine.py

@@ -50,6 +50,7 @@ class Routine(object):
         "return_type": "returnType",
         "type_": "routineType",
         "description": "description",
+        "determinism_level": "determinismLevel",
     }
 
     def __init__(self, routine_ref, **kwargs):
@@ -253,6 +254,17 @@ def description(self):
     def description(self, value):
         self._properties[self._PROPERTY_TO_API_FIELD["description"]] = value
 
+    @property
+    def determinism_level(self):
+        """Optional[str]: (experimental) The determinism level of the JavaScript UDF
+        if defined.
+        """
+        return self._properties.get(self._PROPERTY_TO_API_FIELD["determinism_level"])
+
+    @determinism_level.setter
+    def determinism_level(self, value):
+        self._properties[self._PROPERTY_TO_API_FIELD["determinism_level"]] = value
+
     @classmethod
     def from_api_repr(cls, resource):
         """Factory: construct a routine given its API representation.

tests/system/test_client.py

@@ -2682,6 +2682,7 @@ def test_create_routine(self):
             )
         ]
         routine.body = "return maxValue(arr)"
+        routine.determinism_level = "DETERMINISTIC"
         query_string = "SELECT `{}`([-100.0, 3.14, 100.0, 42.0]) as max_value;".format(
             str(routine.reference)
         )

tests/unit/routine/test_routine.py

@@ -73,6 +73,7 @@ def test_ctor_w_properties(target_class):
     )
     type_ = "SCALAR_FUNCTION"
     description = "A routine description."
+    determinism_level = "NOT_DETERMINISTIC"
 
     actual_routine = target_class(
         routine_id,
@@ -82,6 +83,7 @@ def test_ctor_w_properties(target_class):
         return_type=return_type,
         type_=type_,
         description=description,
+        determinism_level=determinism_level,
     )
 
     ref = RoutineReference.from_string(routine_id)
@@ -92,6 +94,7 @@ def test_ctor_w_properties(target_class):
     assert actual_routine.return_type == return_type
     assert actual_routine.type_ == type_
     assert actual_routine.description == description
+    assert actual_routine.determinism_level == "NOT_DETERMINISTIC"
 
 
 def test_from_api_repr(target_class):
@@ -120,6 +123,7 @@ def test_from_api_repr(target_class):
         "routineType": "SCALAR_FUNCTION",
         "someNewField": "someValue",
         "description": "A routine description.",
+        "determinismLevel": "DETERMINISTIC",
     }
     actual_routine = target_class.from_api_repr(resource)
 
@@ -152,6 +156,7 @@ def test_from_api_repr(target_class):
     assert actual_routine.type_ == "SCALAR_FUNCTION"
     assert actual_routine._properties["someNewField"] == "someValue"
     assert actual_routine.description == "A routine description."
+    assert actual_routine.determinism_level == "DETERMINISTIC"
 
 
 def test_from_api_repr_w_minimal_resource(target_class):
@@ -177,6 +182,7 @@ def test_from_api_repr_w_minimal_resource(target_class):
     assert actual_routine.return_type is None
     assert actual_routine.type_ is None
     assert actual_routine.description is None
+    assert actual_routine.determinism_level is None
 
 
 def test_from_api_repr_w_unknown_fields(target_class):
@@ -208,6 +214,7 @@ def test_from_api_repr_w_unknown_fields(target_class):
                 "returnType": {"typeKind": "INT64"},
                 "routineType": "SCALAR_FUNCTION",
                 "description": "A routine description.",
+                "determinismLevel": "DETERMINISM_LEVEL_UNSPECIFIED",
             },
             ["arguments"],
             {"arguments": [{"name": "x", "dataType": {"typeKind": "INT64"}}]},
@@ -220,6 +227,7 @@ def test_from_api_repr_w_unknown_fields(target_class):
                 "returnType": {"typeKind": "INT64"},
                 "routineType": "SCALAR_FUNCTION",
                 "description": "A routine description.",
+                "determinismLevel": "DETERMINISM_LEVEL_UNSPECIFIED",
             },
             ["body"],
             {"definitionBody": "x * 3"},
@@ -232,6 +240,7 @@ def test_from_api_repr_w_unknown_fields(target_class):
                 "returnType": {"typeKind": "INT64"},
                 "routineType": "SCALAR_FUNCTION",
                 "description": "A routine description.",
+                "determinismLevel": "DETERMINISM_LEVEL_UNSPECIFIED",
             },
             ["language"],
             {"language": "SQL"},
@@ -244,6 +253,7 @@ def test_from_api_repr_w_unknown_fields(target_class):
                 "returnType": {"typeKind": "INT64"},
                 "routineType": "SCALAR_FUNCTION",
                 "description": "A routine description.",
+                "determinismLevel": "DETERMINISM_LEVEL_UNSPECIFIED",
             },
             ["return_type"],
             {"returnType": {"typeKind": "INT64"}},
@@ -256,6 +266,7 @@ def test_from_api_repr_w_unknown_fields(target_class):
                 "returnType": {"typeKind": "INT64"},
                 "routineType": "SCALAR_FUNCTION",
                 "description": "A routine description.",
+                "determinismLevel": "DETERMINISM_LEVEL_UNSPECIFIED",
             },
             ["type_"],
             {"routineType": "SCALAR_FUNCTION"},
@@ -268,20 +279,43 @@ def test_from_api_repr_w_unknown_fields(target_class):
                 "returnType": {"typeKind": "INT64"},
                 "routineType": "SCALAR_FUNCTION",
                 "description": "A routine description.",
+                "determinismLevel": "DETERMINISM_LEVEL_UNSPECIFIED",
             },
             ["description"],
             {"description": "A routine description."},
         ),
+        (
+            {
+                "arguments": [{"name": "x", "dataType": {"typeKind": "INT64"}}],
+                "definitionBody": "x * 3",
+                "language": "SQL",
+                "returnType": {"typeKind": "INT64"},
+                "routineType": "SCALAR_FUNCTION",
+                "description": "A routine description.",
+                "determinismLevel": "DETERMINISM_LEVEL_UNSPECIFIED",
+            },
+            ["determinism_level"],
+            {"determinismLevel": "DETERMINISM_LEVEL_UNSPECIFIED"},
+        ),
         (
             {},
-            ["arguments", "language", "body", "type_", "return_type", "description"],
+            [
+                "arguments",
+                "language",
+                "body",
+                "type_",
+                "return_type",
+                "description",
+                "determinism_level",
+            ],
             {
                 "arguments": None,
                 "definitionBody": None,
                 "language": None,
                 "returnType": None,
                 "routineType": None,
                 "description": None,
+                "determinismLevel": None,
             },
         ),
         (