Add test to variable-assignment case for object spread

This gives us more assurance that this code is working as expected.

Author: lencioni

package.json

@@ -36,6 +36,7 @@
     "@babel/generator": "7.0.0-beta.42",
     "@babel/plugin-external-helpers": "7.0.0-beta.42",
     "@babel/plugin-proposal-class-properties": "7.0.0-beta.42",
+    "@babel/plugin-proposal-object-rest-spread": "^7.0.0-rc.1",
     "@babel/plugin-transform-flow-strip-types": "7.0.0-beta.42",
     "@babel/preset-env": "7.0.0-beta.42",
     "@babel/preset-flow": "7.0.0-beta.42",

test/fixtures.test.js

@@ -9,6 +9,7 @@ import { transformFileSync } from '@babel/core'
 import babelPluginSyntaxJsx from '@babel/plugin-syntax-jsx'
 import babelPluginExternalHelpers from '@babel/plugin-external-helpers'
 import babelPluginTransformClassProperties from '@babel/plugin-proposal-class-properties'
+import babelPluginProposalObjectRestSpread from '@babel/plugin-proposal-object-rest-spread'
 import babelPluginTransformReactRemovePropTypes from '../src/index'
 import { trim } from './utils'
 
@@ -55,6 +56,7 @@ describe('fixtures', () => {
               babelConfig = {
                 plugins: [
                   babelPluginExternalHelpers,
+                  babelPluginProposalObjectRestSpread,
                   [
                     babelPluginTransformReactRemovePropTypes,
                     {
@@ -70,6 +72,7 @@ describe('fixtures', () => {
               babelConfig = {
                 plugins: [
                   babelPluginExternalHelpers,
+                  babelPluginProposalObjectRestSpread,
                   [
                     babelPluginTransformReactRemovePropTypes,
                     {
@@ -88,6 +91,7 @@ describe('fixtures', () => {
                   babelPluginExternalHelpers,
                   babelPluginSyntaxJsx,
                   babelPluginTransformClassProperties,
+                  babelPluginProposalObjectRestSpread,
                   [
                     babelPluginTransformReactRemovePropTypes,
                     {
@@ -106,6 +110,7 @@ describe('fixtures', () => {
                   babelPluginExternalHelpers,
                   babelPluginSyntaxJsx,
                   babelPluginTransformClassProperties,
+                  babelPluginProposalObjectRestSpread,
                   [
                     babelPluginTransformReactRemovePropTypes,
                     {
@@ -121,6 +126,7 @@ describe('fixtures', () => {
               babelConfig = {
                 plugins: [
                   babelPluginExternalHelpers,
+                  babelPluginProposalObjectRestSpread,
                   [babelPluginTransformReactRemovePropTypes, options],
                 ],
               }

test/fixtures/variable-assignment/actual.js

@@ -22,6 +22,18 @@ const FooExtraReference = () => (
 
 FooExtraReference.propTypes = propTypesWithExtraReference;
 
+const propTypesWithExtraReferenceSpread = {
+  ...extraReference,
+  ...({ foo: PropTypes.string }),
+  bar: PropTypes.string
+};
+
+const FooExtraReferenceSpread = () => (
+  <div />
+);
+
+FooExtraReferenceSpread.propTypes = propTypesWithExtraReferenceSpread;
+
 const propTypesWrapped = {
   foo: PropTypes.string
 };

test/fixtures/variable-assignment/expected-remove-es5.js

@@ -13,6 +13,10 @@ var FooExtraReference = function FooExtraReference() {
   return React.createElement("div", null);
 };
 
+var FooExtraReferenceSpread = function FooExtraReferenceSpread() {
+  return React.createElement("div", null);
+};
+
 var FooWrapped = function FooWrapped() {
   return React.createElement("div", null);
 };

test/fixtures/variable-assignment/expected-remove-es6.js

@@ -2,6 +2,8 @@ const FooBasic = () => <div />;
 
 const FooExtraReference = () => <div />;
 
+const FooExtraReferenceSpread = () => <div />;
+
 const FooWrapped = () => <div />;
 
 const propTypesReferenced = {

test/fixtures/variable-assignment/expected-wrap-es5.js

@@ -25,6 +25,17 @@ var FooExtraReference = function FooExtraReference() {
 };
 
 FooExtraReference.propTypes = process.env.NODE_ENV !== "production" ? propTypesWithExtraReference : {};
+var propTypesWithExtraReferenceSpread = process.env.NODE_ENV !== "production" ? babelHelpers.objectSpread({}, extraReference, {
+  foo: PropTypes.string
+}, {
+  bar: PropTypes.string
+}) : {};;
+
+var FooExtraReferenceSpread = function FooExtraReferenceSpread() {
+  return React.createElement("div", null);
+};
+
+FooExtraReferenceSpread.propTypes = process.env.NODE_ENV !== "production" ? propTypesWithExtraReferenceSpread : {};
 var propTypesWrapped = process.env.NODE_ENV !== "production" ? {
   foo: PropTypes.string
 } : {};;

test/fixtures/variable-assignment/expected-wrap-es6.js

@@ -15,6 +15,15 @@ const propTypesWithExtraReference = process.env.NODE_ENV !== "production" ? Obje
 const FooExtraReference = () => <div />;
 
 FooExtraReference.propTypes = process.env.NODE_ENV !== "production" ? propTypesWithExtraReference : {};
+const propTypesWithExtraReferenceSpread = process.env.NODE_ENV !== "production" ? babelHelpers.objectSpread({}, extraReference, {
+  foo: PropTypes.string
+}, {
+  bar: PropTypes.string
+}) : {};;
+
+const FooExtraReferenceSpread = () => <div />;
+
+FooExtraReferenceSpread.propTypes = process.env.NODE_ENV !== "production" ? propTypesWithExtraReferenceSpread : {};
 const propTypesWrapped = process.env.NODE_ENV !== "production" ? {
   foo: PropTypes.string
 } : {};;