Computed Values support for PUT /_api/collection/collection/properties

arango/collection.py

@@ -306,6 +306,7 @@ def configure(
  schema: Optional[Json] = None,
  replication_factor: Optional[int] = None,
  write_concern: Optional[int] = None,
+ computed_values: Optional[List[Json]] = None
  ) -> Result[Json]:
  """Configure collection properties.
 
@@ -341,6 +342,8 @@ def configure(
  data["replicationFactor"] = replication_factor
  if write_concern is not None:
  data["writeConcern"] = write_concern
+ if computed_values is not None:
+ data["computedValues"] = computed_values
 
  request = Request(
  method="put",

arango/formatter.py

@@ -249,20 +249,9 @@ def format_collection(body: Json) -> Json:
  result["min_revision"] = body["minRevision"]
  if "schema" in body:
  result["schema"] = body["schema"]
+ if "computedValues" in body:
+ result["computedValues"] = body["computedValues"]
 
- # New in 3.10
- if body.get("computedValues") is not None:
- result["computedValues"] = [
- {
- "name": cv["name"],
- "expression": cv["expression"],
- "overwrite": cv["overwrite"],
- "computedOn": cv["computedOn"],
- "keepNull": cv["keepNull"],
- "failOnWarning": cv["failOnWarning"],
- }
- for cv in body["computedValues"]
- ]
  if "internalValidatorType" in body:
  result["internal_validator_type"] = body["internalValidatorType"]
 

tests/test_collection.py

@@ -49,10 +49,26 @@ def test_collection_misc_methods(col, bad_col, cluster):
 
  # Test configure properties
  prev_sync = properties["sync"]
- properties = col.configure(sync=not prev_sync, schema={})
+
+ computed_values = [
+ {
+ "name": "foo",
+ "expression": "RETURN 1",
+ "computeOn": ["insert", "update", "replace"],
+ "overwrite": True,
+ "failOnWarning": False,
+ "keepNull": True,
+ }
+ ]
+
+ properties = col.configure(
+ sync=not prev_sync, schema={}, computed_values=computed_values
+ )
+
  assert properties["name"] == col.name
  assert properties["system"] is False
  assert properties["sync"] is not prev_sync
+ assert properties["computedValues"] == computed_values
 
  # Test configure properties with bad collection
  with assert_raises(CollectionConfigureError) as err:
@@ -153,6 +169,17 @@ def test_collection_management(db, bad_db, cluster):
  "type": "json",
  }
 
+ computed_values = [
+ {
+ "name": "foo",
+ "expression": "RETURN 1",
+ "computeOn": ["insert", "update", "replace"],
+ "overwrite": True,
+ "failOnWarning": False,
+ "keepNull": True,
+ }
+ ]
+
  col = db.create_collection(
  name=col_name,
  sync=True,
@@ -172,6 +199,7 @@ def test_collection_management(db, bad_db, cluster):
  smart_join_attribute="test_attr",
  write_concern=1,
  schema=schema,
+ computedValues=computed_values,
  )
  assert db.has_collection(col_name) is True
 
@@ -181,6 +209,7 @@ def test_collection_management(db, bad_db, cluster):
  assert properties["name"] == col_name
  assert properties["sync"] is True
  assert properties["system"] is False
+ assert properties["computedValues"] == computed_values
 
  # Test create duplicate collection
  with assert_raises(CollectionCreateError) as err: