patch(vest): reduceTests

ealush · Jun 8, 2024 · 38ef2f7 · 38ef2f7
1 parent e6380eb
commit 38ef2f7
Showing 1 changed file with 25 additions and 9 deletions.
diff --git a/packages/vest/src/core/isolate/IsolateTest/TestWalker.ts b/packages/vest/src/core/isolate/IsolateTest/TestWalker.ts
@@ -18,7 +18,7 @@ export class TestWalker {
 
   static someTests(
     predicate: (test: TIsolateTest) => boolean,
-    root: MaybeRoot = TestWalker.defaultRoot()
+    root: MaybeRoot = TestWalker.defaultRoot(),
   ): boolean {
     if (!root) return false;
     return Walker.some(
@@ -28,13 +28,13 @@ export class TestWalker {
 
         return predicate(isolate);
       },
-      VestTest.is
+      VestTest.is,
     );
   }
 
   static everyTest(
     predicate: (test: TIsolateTest) => boolean,
-    root: MaybeRoot = TestWalker.defaultRoot()
+    root: MaybeRoot = TestWalker.defaultRoot(),
   ): boolean {
     if (!root) return false;
     return Walker.every(
@@ -44,27 +44,43 @@ export class TestWalker {
 
         return predicate(isolate);
       },
-      VestTest.is
+      VestTest.is,
     );
   }
 
   static walkTests<F extends TFieldName, G extends TGroupName>(
     callback: (test: TIsolateTest<F, G>, breakout: () => void) => void,
-    root: MaybeRoot = TestWalker.defaultRoot()
+    root: MaybeRoot = TestWalker.defaultRoot(),
   ): void {
     if (!root) return;
     Walker.walk(
       root,
       (isolate, breakout) => {
         callback(VestTest.cast<F, G>(isolate), breakout);
       },
-      VestTest.is
+      VestTest.is,
+    );
+  }
+
+  static reduceTests<T, I extends TIsolateTest = TIsolateTest>(
+    callback: (acc: T, test: I, breakout: () => void) => T,
+    initialValue: T,
+    root: MaybeRoot = TestWalker.defaultRoot(),
+  ): T {
+    if (!root) return initialValue;
+    return Walker.reduce(
+      root,
+      (acc, isolate, breakout) => {
+        return callback(acc, VestTest.cast(isolate) as I, breakout);
+      },
+      initialValue,
+      VestTest.is,
     );
   }
 
   static pluckTests(
     predicate: (test: TIsolateTest) => boolean,
-    root: MaybeRoot = TestWalker.defaultRoot()
+    root: MaybeRoot = TestWalker.defaultRoot(),
   ): void {
     if (!root) return;
     Walker.pluck(
@@ -74,7 +90,7 @@ export class TestWalker {
 
         return predicate(isolate);
       },
-      VestTest.is
+      VestTest.is,
     );
   }
 
@@ -88,7 +104,7 @@ export class TestWalker {
 
   static removeTestByFieldName(
     fieldName: TFieldName,
-    root: MaybeRoot = TestWalker.defaultRoot()
+    root: MaybeRoot = TestWalker.defaultRoot(),
   ): void {
     TestWalker.pluckTests(testObject => {
       return matchingFieldName(VestTest.getData(testObject), fieldName);