rename this.context to context in constructor

Author: Keyan Zhang

transforms/__testfixtures__/class-initial-state.output.js

@@ -56,7 +56,7 @@ class App2 extends React.Component {
   constructor(props, context) {
     super(props, context);
     const state = {
-      whatever: this.context.whatever, // needs context
+      whatever: context.whatever, // needs context
     };
     this.state = state;
   }

transforms/class.js

@@ -411,18 +411,30 @@ module.exports = (file, api, options) => {
       fn.value
     ), fn);
 
-  const updatePropsAccess = getInitialState =>
-    j(getInitialState)
-      .find(j.MemberExpression, {
-        object: {
-          type: 'ThisExpression',
-        },
-        property: {
-          type: 'Identifier',
-          name: 'props',
-        },
-      })
-      .forEach(path => j(path).replaceWith(j.identifier('props')));
+  const updatePropsAndContextAccess = getInitialState => {
+    const collection = j(getInitialState);
+
+    collection.find(j.MemberExpression, {
+      object: {
+        type: 'ThisExpression',
+      },
+      property: {
+        type: 'Identifier',
+        name: 'props',
+      },
+    }).forEach(path => j(path).replaceWith(j.identifier('props')));
+
+    collection.find(j.MemberExpression, {
+      object: {
+        type: 'ThisExpression',
+      },
+      property: {
+        type: 'Identifier',
+        name: 'context',
+      },
+    }).forEach(path => j(path).replaceWith(j.identifier('context')));
+  };
+
 
   const inlineGetInitialState = getInitialState => {
     const functionExpressionCollection = j(getInitialState.value);
@@ -520,7 +532,7 @@ module.exports = (file, api, options) => {
       hasContextAccess = true;
     }
 
-    updatePropsAccess(getInitialState);
+    updatePropsAndContextAccess(getInitialState);
     const constructorArgs = createConstructorArgs(hasContextAccess);
 
     return [