vest 5.4.4 → 5.4.5-dev-20241111-bedd

package/dist/cjs/vest.development.js

@@ -489,6 +489,7 @@ const suiteResultCache = vestUtils.cache();
 const preAggCache = vestUtils.cache();
 function useCreateVestState({ suiteName, VestReconciler, }) {
     const stateRef = {
+        afterEachCallbacks: vestUtils.tinyState.createTinyState(() => []),
         doneCallbacks: vestUtils.tinyState.createTinyState(() => []),
         fieldCallbacks: vestUtils.tinyState.createTinyState(() => ({})),
         preAggCache,
@@ -507,6 +508,9 @@ function useDoneCallbacks() {
 function useFieldCallbacks() {
     return useX().fieldCallbacks();
 }
+function useAfterEachCallbacks() {
+    return useX().afterEachCallbacks();
+}
 function useSuiteName() {
     return useX().suiteName;
 }
@@ -532,8 +536,10 @@ function useExpireSuiteResultCache() {
 function useResetCallbacks() {
     const [, , resetDoneCallbacks] = useDoneCallbacks();
     const [, , resetFieldCallbacks] = useFieldCallbacks();
+    const [, , resetAfterEachCallbacks] = useAfterEachCallbacks();
     resetDoneCallbacks();
     resetFieldCallbacks();
+    resetAfterEachCallbacks();
 }
 function useResetSuite() {
     useResetCallbacks();
@@ -1681,6 +1687,13 @@ function useRunDoneCallbacks() {
     const [doneCallbacks] = useDoneCallbacks();
     vestUtils.callEach(doneCallbacks);
 }
+function useRunAfterEachCallbacks(fieldName) {
+    if (SuiteWalker.useHasRemainingWithTestNameMatching(fieldName)) {
+        return;
+    }
+    const [afterEachCallbacks] = useAfterEachCallbacks();
+    vestUtils.callEach(afterEachCallbacks);
+}
 
 // eslint-disable-next-line max-statements, max-lines-per-function
 function useInitVestBus() {
@@ -1704,6 +1717,7 @@ function useInitVestBus() {
             if (!VestTest.isCanceled(isolate)) {
                 const { fieldName } = VestTest.getData(isolate);
                 useRunFieldCallbacks(fieldName);
+                useRunAfterEachCallbacks(fieldName);
             }
         }
         if (!SuiteWalker.useHasPending()) {
@@ -1786,10 +1800,20 @@ function shouldSkipDoneRegistration(callback, fieldName, output) {
 }
 
 function useSuiteRunResult() {
+    const persistedDone = vestjsRuntime.VestRuntime.persist(done);
     return vestUtils.freezeAssign({
-        done: vestjsRuntime.VestRuntime.persist(done),
+        afterEach: vestjsRuntime.VestRuntime.persist(afterEach),
+        done: persistedDone,
+        after: persistedDone,
     }, useCreateSuiteResult());
 }
+// @vx-allow use-use
+function afterEach(callback) {
+    const output = useSuiteRunResult();
+    const [, setAfterEachCallbacks] = useAfterEachCallbacks();
+    setAfterEachCallbacks(prev => [...prev, callback]);
+    return output;
+}
 /**
  * Registers done callbacks.
  * @register {Object} Vest output object.