automatic start

Signed-off-by: Matteo Collina <[email protected]>

asyncforge.js

@@ -14,7 +14,11 @@ let forgeCounter = 0
 function forge (fn) {
   const sym = Symbol('forge.' + (fn.name || forgeCounter++))
   return function () {
-    const store = asyncLocalStorage.getStore()
+    let store = asyncLocalStorage.getStore()
+    if (!store) {
+      store = Object.create(null)
+      asyncLocalStorage.enterWith(store)
+    }
     if (store[sym]) {
       return store[sym]
     }
@@ -32,12 +36,12 @@ function memo (name) {
 
   function get () {
     const store = asyncLocalStorage.getStore()
-    return store[sym]
+    return store?.[sym]
   }
 
   function set (value) {
     let store = asyncLocalStorage.getStore()
-    store = Object.create(store)
+    store = Object.create(store || null)
     store[sym] = value
     asyncLocalStorage.enterWith(store)
   }

test/basic.test.js

@@ -6,6 +6,18 @@ const { start, forge } = require('../')
 const { setImmediate: immediate } = require('node:timers/promises')
 const tspl = require('@matteo.collina/tspl')
 
+test('forge', async (t) => {
+  const { a, b } = require('./fixture/basic')
+
+  assert.deepEqual(a(), { value: undefined })
+  assert.equal(a(), a())
+  assert.deepEqual(b(), {
+    fromA: {
+      value: undefined
+    }
+  })
+})
+
 test('start and forge', async (t) => {
   const { a, b } = require('./fixture/basic')
 

test/do-not-start-memo.test.js

@@ -0,0 +1,64 @@
+'use strict'
+
+const { test } = require('node:test')
+const { start, memo } = require('../')
+const tspl = require('@matteo.collina/tspl')
+
+test('memo without start', async (t) => {
+  const p = tspl(t, { plan: 6 })
+  const a = memo()
+
+  p.deepStrictEqual(a(), undefined)
+  a.set({ value: 'bar' })
+  p.deepStrictEqual(a(), { value: 'bar' })
+
+  setImmediate(() => {
+    p.deepStrictEqual(a(), { value: 'bar' })
+  })
+
+  queueMicrotask(() => {
+    p.deepStrictEqual(a(), { value: 'bar' })
+  })
+
+  start()
+
+  p.deepStrictEqual(a(), undefined)
+  a.set({ value: 'baz' })
+
+  p.deepEqual(a(), { value: 'baz' })
+
+  setImmediate(() => {
+    p.deepStrictEqual(a(), { value: 'baz' })
+  })
+
+  queueMicrotask(() => {
+    p.deepStrictEqual(a(), { value: 'baz' })
+  })
+
+  await p.completed
+})
+
+test('nested', async (t) => {
+  const p = tspl(t, { plan: 5 })
+  const a = memo()
+
+  p.deepStrictEqual(a(), undefined)
+  a.set({ value: 'bar' })
+  p.deepStrictEqual(a(), { value: 'bar' })
+
+  setImmediate(() => {
+    p.deepStrictEqual(a(), { value: 'bar' })
+
+    a.set({ value: 'baz' })
+
+    setImmediate(() => {
+      p.deepStrictEqual(a(), { value: 'baz' })
+    })
+  })
+
+  setImmediate(() => {
+    p.deepStrictEqual(a(), { value: 'bar' })
+  })
+
+  await p.completed
+})

test/fixture/basic/a.js

@@ -4,6 +4,6 @@ const { forge } = require('../../../')
 
 module.exports = forge((config) => {
   return {
-    value: config.foo
+    value: config?.foo
   }
 })