Add to cart tests

1 files changed, 19 insertions, 16 deletions

diff --git a/frontend/__tests__/AddToCart.test.js b/frontend/__tests__/AddToCart.test.js
index f049152..92ad0ad 100644
--- a/frontend/__tests__/AddToCart.test.js
+++ b/frontend/__tests__/AddToCart.test.js
@@ -1,28 +1,31 @@
 import React from 'react';
 import { shallow, mount } from 'enzyme';
 import toJSON from 'enzyme-to-json';
-import { AddToCart } from '../components/AddToCart';
-
-const currentUser = {
-  me: {},
-  refetch() {},
-};
+import AddToCart from '../components/AddToCart';
+import mountOptions from './mockMang';
 
 describe('<AddtoCart/>', () => {
   it('renders and matches snapshot', () => {
-    const wrapper = shallow(
-      <AddToCart id="abc123" addToCart={() => {}} currentUser={currentUser} />
-    );
-    expect(toJSON(wrapper)).toMatchSnapshot();
+    const wrapper = mount(<AddToCart id="abc123" />, mountOptions);
+    expect(toJSON(wrapper.find('button'))).toMatchSnapshot();
   });
 
   it('adds an item to the cart', () => {
-    const addToCart = jest.fn(() => Promise.resolve());
-    const wrapper = shallow(
-      <AddToCart id="abc123" addToCart={addToCart} currentUser={currentUser} />
-    );
+    const wrapper = mount(<AddToCart id="abc123" />, mountOptions);
+    const mutation = wrapper.find('Mutation').instance();
+    mutation.client.mutate = jest.fn().mockResolvedValue(null);
     wrapper.simulate('click');
-    expect(addToCart).toHaveBeenCalled();
-    expect(addToCart).toHaveBeenCalledWith({ variables: { id: 'abc123' } });
+    // check that it was called
+    expect(mutation.client.mutate).toHaveBeenCalled();
+    // check that it was called with the correct ID
+    expect(mutation.client.mutate).toHaveBeenCalledWithVariables({ id: 'abc123' });
+  });
+
+  it('changes to adding when clicked', () => {
+    const wrapper = mount(<AddToCart id="abc123" />, mountOptions);
+    const button = wrapper.find('button');
+    expect(button.text()).toContain('Add To Cart');
+    button.simulate('click');
+    expect(button.text()).toContain('Adding To Cart');
   });
 });