test(types): test $let as a possible expression to $addFields

Automattic · Jul 11, 2022 · 38c53b8 · 38c53b8
1 parent db1ba6e
commit 38c53b8
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/test/types/PipelineStage.test.ts b/test/types/PipelineStage.test.ts
@@ -392,3 +392,26 @@ const stages3: PipelineStage[] = [
     }
   }
 ];
+
+const stages4: PipelineStage[] = [
+  {
+    $addFields: {
+      usersCount: {
+        $let: {
+          vars: {
+            users: { $push: '$user' }
+          },
+          in: {
+            $reduce: {
+              input: '$users',
+              initialValue: 0,
+              in: {
+                $cond: { if: { $isArray: '$$this' }, then: { $size: '$$this' }, else: '$$this' }
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+];