Adapting tests for computedValues attribute. We are now removing all …

…None values before comparison.
arangodb · May 30, 2023 · 8f7d4bd · 8f7d4bd
1 parent 47a0047
commit 8f7d4bd
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 16 deletions.
diff --git a/arango/formatter.py b/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"]
 

diff --git a/tests/conftest.py b/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("_"))