fix: better edge case handling

Automattic · Jul 21, 2024 · 08e65f1 · 08e65f1
1 parent bb2853c
commit 08e65f1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/helpers/query/cast$expr.js b/lib/helpers/query/cast$expr.js
@@ -93,8 +93,12 @@ function _castExpression(val, schema, strictQuery) {
   } else if (val.$ifNull != null) {
     val.$ifNull.map(v => _castExpression(v, schema, strictQuery));
   } else if (val.$switch != null) {
-    val.$switch.branches = val.$switch.branches.map(v => _castExpression(v, schema, strictQuery));
-    val.$switch.default = _castExpression(val.$switch.default, schema, strictQuery);
+    if (Array.isArray(val.$switch.branches)) {
+      val.$switch.branches = val.$switch.branches.map(v => _castExpression(v, schema, strictQuery));
+    }
+    if ('default' in val.$switch) {
+      val.$switch.default = _castExpression(val.$switch.default, schema, strictQuery);
+    }
   }
 
   const keys = Object.keys(val);