From bd45d4c7aac6d1c6b054185058a96b2e01669828 Mon Sep 17 00:00:00 2001 From: Wes Bos Date: Wed, 14 Mar 2018 21:00:08 -0400 Subject: tests --- frontend/components/TakeMyMoney.js | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) (limited to 'frontend/components/TakeMyMoney.js') diff --git a/frontend/components/TakeMyMoney.js b/frontend/components/TakeMyMoney.js index 333f31b..011451a 100644 --- a/frontend/components/TakeMyMoney.js +++ b/frontend/components/TakeMyMoney.js @@ -1,10 +1,11 @@ import { Component } from 'react'; import StripeCheckout from 'react-stripe-checkout'; -import { graphql, compose } from 'react-apollo'; -import { CREATE_ORDER_MUTATION, CURRENT_USER_QUERY } from '../queries'; -import formatMoney from '../lib/formatMoney'; +import { compose } from 'react-apollo'; import Router from 'next/router'; import NProgress from 'nprogress'; +import PropTypes from 'prop-types'; +// import { CREATE_ORDER_MUTATION, CURRENT_USER_QUERY } from '../queries'; +import { userEnhancer, createOrderEnhancer } from '../enhancers/enhancers'; class TakeMyMoney extends Component { onToken = async res => { @@ -23,7 +24,7 @@ class TakeMyMoney extends Component { }; render() { - const { me } = this.props.currentUserQuery; + const { me } = this.props.currentUser; if (!me || !me.cart.length) return null; const total = me.cart.reduce((tally, cartItem) => tally + cartItem.item.price * cartItem.quantity, 0); const totalItems = me.cart.reduce((tally, cartItem) => tally + cartItem.quantity, 0); @@ -44,6 +45,10 @@ class TakeMyMoney extends Component { } } -const userEnhancer = graphql(CURRENT_USER_QUERY, { name: 'currentUserQuery' }); -const createOrderEnhancer = graphql(CREATE_ORDER_MUTATION, { name: 'createOrder' }); +TakeMyMoney.propTypes = { + currentUser: PropTypes.object.isRequired, + createOrder: PropTypes.func.isRequired, +}; + export default compose(userEnhancer, createOrderEnhancer)(TakeMyMoney); +export { TakeMyMoney }; -- cgit v1.3