Fixed bug in async state handling.

IcecaveStudios · Jan 14, 2015 · 287dcba · 287dcba
1 parent 015adb4
commit 287dcba
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 10 deletions.
diff --git a/lib/AsyncBinaryState.js b/lib/AsyncBinaryState.js
diff --git a/lib/AsyncBinaryState.js.map b/lib/AsyncBinaryState.js.map
diff --git a/src/AsyncBinaryState.coffee b/src/AsyncBinaryState.coffee
@@ -17,8 +17,13 @@ module.exports = class AsyncBinaryState
 
         @isOn = isOn
 
-        promise = bluebird.resolve handler?()
+        if handler?
+            method = bluebird.method handler
+        else
+            method = -> bluebird.resolve()
 
-        promise.catch => @isOn = not @isOn
+        method()
+        .catch (error) =>
+            @isOn = not isOn
 
-        promise
+            throw error