Adapting tests for computedValues attribute. We are now removing all None values before comparison.

Author: apetenchea

arango/formatter.py

@@ -217,22 +217,18 @@ def format_collection(body: Json) -> Json:
         result["schema"] = body["schema"]
 
     # New in 3.10
-    if "computedValues" in body:
-        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 body.get("computedValues") is not None
-            else None
-        )
+    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/conftest.py

@@ -232,6 +232,12 @@ def mock_verify_format(body, result):
         body.pop("error", None)
         body.pop("code", None)
         result.pop("edge", None)
+
+        # Remove all None values
+        # Sometimes they are expected to be excluded from the body (see computedValues)
+        result = {k: v for k, v in result.items() if v is not None}
+        body = {k: v for k, v in body.items() if v is not None}
+
         if len(body) != len(result):
             before = sorted(body, key=lambda x: x.strip("_"))
             after = sorted(result, key=lambda x: x.strip("_"))