fix: restoreCtx before init

wooksjs · Jul 28, 2023 · 176969e · 176969e
1 parent d1e99bf
commit 176969e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/packages/event-wf/src/wf-adapter.ts b/packages/event-wf/src/wf-adapter.ts
@@ -73,9 +73,10 @@ export class WooksWf<T> extends WooksAdapterBase {
                     restoreCtx()
                     const { id, init } = (await handler()) as { init?: () => void | Promise<void>, id: string }
                     if (init) {
-                        await init()
                         restoreCtx()
+                        await init()
                     }
+                    restoreCtx()
                     if (resume) {
                         result = await this.wf.resume<I>(id, { context: inputContext, indexes }, input as I)
                         break

diff --git a/packages/event-wf/src/wf.spec.ts b/packages/event-wf/src/wf.spec.ts
@@ -2,7 +2,7 @@ import { useRouteParams } from '@wooksjs/event-core'
 import { createWfApp } from './wf-adapter'
 import { useWfState } from './composables'
 
-const app = createWfApp<{ result: number }>()
+const app = createWfApp<{ result?: number }>()
 
 app.step('add', {
     input: 'number',
@@ -42,6 +42,14 @@ app.flow('parametric/*', [
     },
 ])
 
+let c = -100
+
+app.flow('init', ['add/1'], () => {
+    const ctx = useWfState().ctx<{ result: number }>()
+    c = ctx.result
+    ctx.result = 0
+})
+
 describe('event-wf', () => {
     it('must run simple wf', async () => {
         const result = await app.start('adding', { result: 0 })
@@ -61,4 +69,9 @@ describe('event-wf', () => {
         const result2 = await app.start('parametric/asdf/asdf12', { result: 10 })
         expect(result2.state.context.result).toBe(17)
     })
+    it('must call init fn', async () => {
+        const result = await app.start('init', {})
+        expect(c).toBeUndefined()
+        expect(result.state.context.result).toBe(1)
+    })
 })