pass context to callback

MZanggl · Jan 4, 2020 · 0123a4d · 0123a4d
1 parent 2d6079c
commit 0123a4d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -41,7 +41,7 @@ function promistate<T>(action: (...args: CallbackArgs) => Promise<T>, options: P
             this.isEmpty = false
             this.error = null
 
-            return action(...args)
+            return action.apply(this, args)
                 .then((result: T) => {
                     this.isEmpty = isEmpty<T>(result)
                     this.value = result

diff --git a/test/promistate.spec.js b/test/promistate.spec.js
@@ -152,4 +152,14 @@ test('can pass custom isEmpty check', async (assert) => {
     })
 
     assert.isTrue(state.isEmpty)
-})
+})
+
+test('can access state in load function', async (assert) => {
+    const state = promistate(async function() {
+        return this.value + 1
+    }, { defaultValue: 1 })
+
+    await state.load()
+
+    assert.equal(state.value, 2)
+})