fix(testing): properly clear effect mocks when clearing all mocks

Author: MrWolfZ

packages/testing/src/effects.spec.ts

@@ -1,4 +1,5 @@
 import { createEffect } from '@simplux/core'
+import { clearAllSimpluxMocks } from './cleanup'
 import { mockEffect } from './effects'
 
 describe(mockEffect.name, () => {
@@ -35,6 +36,25 @@ describe(mockEffect.name, () => {
         expect(mockSpy).toHaveBeenCalledTimes(1)
       })
 
+      it('can be removed all at once', () => {
+        const spy = jest.fn()
+        const mockSpy1 = jest.fn()
+        const mockSpy2 = jest.fn()
+        const effect1 = createEffect(spy)
+        const effect2 = createEffect(spy)
+        mockEffect(effect1, mockSpy1)
+        mockEffect(effect2, mockSpy2)
+        effect1()
+        effect2()
+        expect(mockSpy1).toHaveBeenCalledTimes(1)
+        expect(mockSpy2).toHaveBeenCalledTimes(1)
+        clearAllSimpluxMocks()
+        effect1()
+        effect2()
+        expect(mockSpy1).toHaveBeenCalledTimes(1)
+        expect(mockSpy2).toHaveBeenCalledTimes(1)
+      })
+
       it('can safely be removed twice', () => {
         const spy = jest.fn()
         const mockSpy1 = jest.fn()

packages/testing/src/effects.ts

@@ -1,4 +1,5 @@
 import { EffectMockDefinition, getMockDefinitionsInternal } from '@simplux/core'
+import { registerMockCleanupFunction } from './cleanup'
 
 /**
  * Specify a mock function that should be called instead of the
@@ -23,7 +24,14 @@ export function mockEffect<TEffect extends Function>(
     mockFn,
   })
 
-  return () => removeMock(mockDefinitions, effectToMock)
+  const cleanup = () => {
+    removeMock(mockDefinitions, effectToMock)
+    clearCleanup()
+  }
+
+  const clearCleanup = registerMockCleanupFunction(cleanup)
+
+  return cleanup
 }
 
 function removeMock<TEffect extends Function>(