fixup

Signed-off-by: Matteo Collina <hello@matteocollina.com>
mcollina · Apr 19, 2024 · 0add09c · 0add09c
1 parent 1ae4d82
commit 0add09c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/asyncforge.js b/asyncforge.js
@@ -54,7 +54,7 @@ function memo (name) {
 
 function setAll (memos) {
   let store = asyncLocalStorage.getStore()
-  store = Object.create(store)
+  store = Object.create(store || null)
   const keys = Object.getOwnPropertySymbols(memos)
   for (const key of keys) {
     store[key] = memos[key]

diff --git a/test/setall.test.js b/test/setall.test.js
@@ -4,6 +4,22 @@ const { test } = require('node:test')
 const { start, memo, setAll } = require('../')
 const assert = require('node:assert/strict')
 
+test('setAll no start', async (t) => {
+  const a = memo()
+  const b = memo()
+  const c = memo()
+
+  setAll({
+    [a.key]: 1,
+    [b.key]: 2,
+    [c.key]: 3
+  })
+
+  assert.equal(a(), 1)
+  assert.equal(b(), 2)
+  assert.equal(c(), 3)
+})
+
 test('setAll memos', async (t) => {
   const a = memo()
   const b = memo()