Merge pull request #78 from oslabs-beta/rydang/setStateChange

changed the way changeSetState operates

Author: mylee1995

package/__tests__/linkFiber.test.js

@@ -47,7 +47,7 @@ describe('unit test for linkFiber', () => {
   });
 
   test('linkFiber should modify the setState of the stateful component', () => {
-    expect(snapShot.tree.children[0].component.setState.name).toBe('newSetState');
+    expect(snapShot.tree.children[0].component.setState.linkFiberChanged).toBe(true);
   });
 
   // test('newSetState should still setState correctly', () => {

package/linkFiber.js

@@ -17,12 +17,14 @@ module.exports = (snap, mode) => {
 
   function changeSetState(component) {
     // check that setState hasn't been changed yet
-    if (component.setState.name === 'newSetState') return;
+    if (component.setState.linkFiberChanged) return;
 
     // make a copy of setState
     const oldSetState = component.setState.bind(component);
 
-    function newSetState(state, callback = () => {}) {
+    // replace component's setState so developer doesn't change syntax
+    // component.setState = newSetState.bind(component);
+    component.setState = (state, callback = () => {}) => {
       // dont do anything if state is locked
       // UNLESS we are currently jumping through time
       if (mode.locked && !mode.jumping) return;
@@ -32,10 +34,8 @@ module.exports = (snap, mode) => {
         sendSnapshot();
         callback();
       });
-    }
-
-    // replace component's setState so developer doesn't change syntax
-    component.setState = newSetState;
+    };
+    component.setState.linkFiberChanged = true;
   }
 
   function createTree(currentFiber, tree = new Tree('root')) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "react-time-travel",
-  "version": "1.0.4",
+  "version": "1.1.0",
   "description": "A library that helps debug React by memorizing the state of components with every render.",
   "main": "index.js",
   "repository": {