tests

3 files changed, 22 insertions, 10 deletions

diff --git a/frontend/components/EditUser.js b/frontend/components/EditUser.js
index e763616..5259d54 100644
--- a/frontend/components/EditUser.js
+++ b/frontend/components/EditUser.js
@@ -1,9 +1,15 @@
 import React from 'react';
+import { compose } from 'react-apollo';
+import PropTypes from 'prop-types';
 import Form from './styles/Form';
 import { userEnhancer, updateUserEnhancer } from '../enhancers/enhancers';
-import { compose } from 'react-apollo';
 
 class EditUser extends React.Component {
+  static propTypes = {
+    currentUser: PropTypes.object.isRequired,
+    updateUser: PropTypes.func.isRequired,
+  };
+
   state = {
     changes: {},
   };
@@ -22,7 +28,7 @@ class EditUser extends React.Component {
 
   handleSubmit = async e => {
     e.preventDefault();
-    const user = await this.props.updateUser({
+    await this.props.updateUser({
       variables: this.state.changes,
     });
     this.props.currentUser.refetch();
diff --git a/frontend/components/Page.js b/frontend/components/Page.js
index 51d6b09..36854ef 100644
--- a/frontend/components/Page.js
+++ b/frontend/components/Page.js
@@ -1,8 +1,8 @@
+import React from 'react';
 import styled, { ThemeProvider, injectGlobal } from 'styled-components';
+import PropTypes from 'prop-types';
 import Header from './Header';
 import Meta from './Meta';
-import Nav from './Nav';
-import { Component } from 'react';
 import { UIProvider } from './UIContext';
 // Global Style
 
@@ -54,4 +54,8 @@ const Page = ({ children }) => (
   </UIProvider>
 );
 
+Page.propTypes = {
+  children: PropTypes.object.isRequired,
+};
+
 export default Page;
diff --git a/frontend/components/RemoveFromCart.js b/frontend/components/RemoveFromCart.js
index 8d0d214..235ecce 100644
--- a/frontend/components/RemoveFromCart.js
+++ b/frontend/components/RemoveFromCart.js
@@ -1,6 +1,5 @@
 import { Component } from 'react';
-import { graphql, compose } from 'react-apollo';
-import Transition from 'react-transition-group/Transition';
+import { compose } from 'react-apollo';
 import styled from 'styled-components';
 import PropTypes from 'prop-types';
 import { removeFromCartEnhancer } from '../enhancers/enhancers';
@@ -16,15 +15,17 @@ const BigButton = styled.button`
 `;
 
 class RemoveFromCart extends Component {
+  static propTypes = {
+    removeFromCart: PropTypes.func.isRequired,
+    id: PropTypes.string.isRequired,
+  };
+
   handleRemoveFromCart = async () => {
-    console.log(`gonna remove item from cart`);
-    const res = await this.props.removeFromCart({
+    await this.props.removeFromCart({
       variables: {
         id: this.props.id,
       },
     });
-    console.log('This came back from removeFromCart:', res);
-    // this.props.currentUserQuery.refetch();
   };
 
   render() {
@@ -37,3 +38,4 @@ class RemoveFromCart extends Component {
 }
 
 export default compose(removeFromCartEnhancer)(RemoveFromCart);
+export { RemoveFromCart };