test: fix tests

commissure · May 31, 2017 · dfddbb6 · dfddbb6
1 parent d882f2d
commit dfddbb6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/index.js b/tests/index.js
@@ -11,8 +11,12 @@ import './lib';
 
 // There’s no DOM available during the tests and the error stack is just a stub
 // that don’t need to be mapped.
+RedBoxError.prototype.mapOnConstruction = function(error) {
+  this.state = { error, mapped: true }
+}
+
 RedBoxError.prototype.mapError = function(error) {
-  this.state = { error }
+  this.setState({ error, mapped: true })
 }
 
 const beforeEach = (framesStub) => {