3 files changed, 7 insertions, 14 deletions
diff --git a/frontend/components/Signin.js b/frontend/components/Signin.js
index f719c60..7a4db8c 100644
--- a/frontend/components/Signin.js
+++ b/frontend/components/Signin.js
@@ -13,7 +13,7 @@ class Signin extends Component {
   loginUser = async (e, signin, client) => {
     e.preventDefault();
     const res = await signin();
-    localStorage.setItem('token', res.data.signin.token);
+    // TODO instead of refetching, can we just use the data returned?
     client.query({ query: CURRENT_USER_QUERY, fetchPolicy: 'network-only' });
   };
 
diff --git a/frontend/components/Signout.js b/frontend/components/Signout.js
index 4ef8e6c..fbd7fd9 100644
--- a/frontend/components/Signout.js
+++ b/frontend/components/Signout.js
@@ -1,20 +1,13 @@
 import React, { Component } from 'react';
-import { Query } from 'react-apollo';
-import { CURRENT_USER_QUERY } from '../queries/queries.graphql';
+import { Query, Mutation } from 'react-apollo';
+import { CURRENT_USER_QUERY, SIGN_OUT_MUTATION } from '../queries/queries.graphql';
 
 class Signout extends Component {
-  signout = refetch => {
-    console.log(refetch);
-    console.log('Signing Out');
-    localStorage.removeItem('token');
-    refetch();
-  };
-
   render() {
     return (
-      <Query query={CURRENT_USER_QUERY}>
-        {({ refetch }) => <button onClick={() => this.signout(refetch)}>Sign Out</button>}
-      </Query>
+      <Mutation mutation={SIGN_OUT_MUTATION} refetchQueries={[{ query: CURRENT_USER_QUERY }]}>
+        {signout => <button onClick={signout}>Sign Out</button>}
+      </Mutation>
     );
   }
 }
diff --git a/frontend/components/Signup.js b/frontend/components/Signup.js
index e8c639b..22a729f 100644
--- a/frontend/components/Signup.js
+++ b/frontend/components/Signup.js
@@ -26,7 +26,7 @@ class Signup extends Component {
                 onSubmit={async e => {
                   e.preventDefault();
                   const res = await signup();
-                  localStorage.setItem('token', res.data.signup.token);
+                  // TODO can we return theuser from signup?
                   client.query({ query: CURRENT_USER_QUERY, fetchPolicy: 'network-only' });
                 }}
               >