From e018f5482e550a49535f4df990d3bfc09128f5d8 Mon Sep 17 00:00:00 2001
From: Wes Bos <wesbos@gmail.com>
Date: Thu, 19 Apr 2018 13:29:25 -0400
Subject: Add to cart tests

---
 frontend/components/AddToCart.js | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

(limited to 'frontend/components/AddToCart.js')

diff --git a/frontend/components/AddToCart.js b/frontend/components/AddToCart.js
index eead8f0..e2770cb 100644
--- a/frontend/components/AddToCart.js
+++ b/frontend/components/AddToCart.js
@@ -1,7 +1,7 @@
 import { Component } from 'react';
-import { Mutation } from 'react-apollo';
+import { Mutation, Query } from 'react-apollo';
 import PropTypes from 'prop-types';
-import { ADD_TO_CART_MUTATION, CURRENT_USER_QUERY } from 'queries';
+import { ADD_TO_CART_MUTATION, CURRENT_USER_QUERY } from '../queries/queries';
 
 class AddToCart extends Component {
   static propTypes = {
@@ -29,13 +29,17 @@ class AddToCart extends Component {
   render() {
     const { id } = this.props;
     return (
-      <Mutation mutation={ADD_TO_CART_MUTATION} variables={{ id }} update={this.update}>
-        {(addToCart, { loading }) => (
-          <button disabled={loading} onClick={addToCart}>
-            🛒 Add{loading && 'ing'} To Cart
-          </button>
+      <Query query={CURRENT_USER_QUERY}>
+        {() => (
+          <Mutation mutation={ADD_TO_CART_MUTATION} variables={{ id }} update={this.update}>
+            {(addToCart, { loading }) => (
+              <button disabled={loading} onClick={addToCart}>
+                🛒 Add{loading && 'ing'} To Cart
+              </button>
+            )}
+          </Mutation>
         )}
-      </Mutation>
+      </Query>
     );
   }
 }
-- 
cgit v1.3