Lazy Draft works

packages/vest/src/isolates/omitWhen.ts

@@ -6,7 +6,7 @@ import { SuiteContext, useOmitted } from 'SuiteContext';
 import { TFieldName, TGroupName } from 'SuiteResultTypes';
 import { VestIsolateType } from 'VestIsolateType';
 import { TDraftCondition } from 'getTypedMethods';
-import { useCreateSuiteResult } from 'suiteResult';
+import { LazyDraft } from 'LazyDraft';
 
 /**
  * Conditionally omits tests from the suite.
@@ -27,10 +27,7 @@ export function omitWhen<F extends TFieldName, G extends TGroupName>(
       {
         omitted:
           useWithinActiveOmitWhen() ||
-          optionalFunctionValue(
-            conditional,
-            optionalFunctionValue(useCreateSuiteResult),
-          ),
+          optionalFunctionValue(conditional, LazyDraft<F, G>()),
       },
       callback,
     );

packages/vest/src/isolates/skipWhen.ts

@@ -5,7 +5,7 @@ import { SuiteContext, useSkipped } from 'SuiteContext';
 import { TFieldName, TGroupName } from 'SuiteResultTypes';
 import { VestIsolateType } from 'VestIsolateType';
 import { TDraftCondition } from 'getTypedMethods';
-import { useCreateSuiteResult } from 'suiteResult';
+import { LazyDraft } from 'LazyDraft';
 
 /**
  * Conditionally skips running tests within the callback.
@@ -29,10 +29,7 @@ export function skipWhen<F extends TFieldName, G extends TGroupName>(
           // we should skip the test if the parent conditional is true.
           useIsExcludedIndividually() ||
           // Otherwise, we should skip the test if the conditional is true.
-          optionalFunctionValue(
-            condition,
-            optionalFunctionValue(useCreateSuiteResult),
-          ),
+          optionalFunctionValue(condition, LazyDraft<F, G>()),
       },
       callback,
     );

packages/vest/src/suiteResult/selectors/LazyDraft.ts

@@ -0,0 +1,22 @@
+import { constructSuiteResultObject, useCreateSuiteResult } from 'suiteResult';
+import {
+  SuiteResult,
+  SuiteSummary,
+  TFieldName,
+  TGroupName,
+} from 'SuiteResultTypes';
+
+export function LazyDraft<
+  F extends TFieldName,
+  G extends TGroupName,
+>(): SuiteResult<F, G> {
+  const emptySummary = constructSuiteResultObject(new SuiteSummary<F, G>());
+
+  return new Proxy(emptySummary, {
+    get: (_, prop) => {
+      const result = useCreateSuiteResult<F, G>();
+
+      return result[prop as keyof SuiteResult<F, G>];
+    },
+  }) as SuiteResult<F, G>;
+}

packages/vest/src/suiteResult/suiteResult.ts

@@ -1,7 +1,12 @@
-import { freezeAssign } from 'vest-utils';
+import { assign, Maybe } from 'vest-utils';
 
 import { useSuiteName, useSuiteResultCache } from 'Runtime';
-import { SuiteResult, TFieldName, TGroupName } from 'SuiteResultTypes';
+import {
+  SuiteResult,
+  SuiteSummary,
+  TFieldName,
+  TGroupName,
+} from 'SuiteResultTypes';
 import { suiteSelectors } from 'suiteSelectors';
 import { useProduceSuiteSummary } from 'useProduceSuiteSummary';
 
@@ -15,12 +20,16 @@ export function useCreateSuiteResult<
 
     // @vx-allow use-use
     const suiteName = useSuiteName();
-    return freezeAssign<SuiteResult<F, G>>(
-      summary,
-      suiteSelectors<F, G>(summary),
-      {
-        suiteName,
-      },
-    ) as SuiteResult<F, G>;
+
+    return Object.freeze(constructSuiteResultObject<F, G>(summary, suiteName));
   });
 }
+
+export function constructSuiteResultObject<
+  F extends TFieldName,
+  G extends TGroupName,
+>(summary: SuiteSummary<F, G>, suiteName?: Maybe<string>): SuiteResult<F, G> {
+  return assign(summary, suiteSelectors<F, G>(summary), {
+    suiteName,
+  }) as SuiteResult<F, G>;
+}