custom assertion

2 files changed, 5 insertions, 4 deletions

diff --git a/frontend/__tests__/Order.test.js b/frontend/__tests__/Order.test.js
index 6bcf0b9..e86af78 100644
--- a/frontend/__tests__/Order.test.js
+++ b/frontend/__tests__/Order.test.js
@@ -1,5 +1,5 @@
 import React from 'react';
-import { mount } from 'enzyme';
+import { shallow, mount } from 'enzyme';
 import wait from 'waait';
 import toJSON from 'enzyme-to-json';
 import Order from '../components/Order';
diff --git a/frontend/__tests__/ResetRequest.test.js b/frontend/__tests__/ResetRequest.test.js
index 6d8d99f..638a2aa 100644
--- a/frontend/__tests__/ResetRequest.test.js
+++ b/frontend/__tests__/ResetRequest.test.js
@@ -4,6 +4,7 @@ import toJSON from 'enzyme-to-json';
 import { ApolloProvider } from 'react-apollo';
 import ResetRequest from '../components/ResetRequest';
 import mountOptions from './mockMang';
+import wait from 'waait';
 
 describe('<ResetRequest/>', () => {
   it('renders and matches snapshot', async () => {
@@ -21,9 +22,9 @@ describe('<ResetRequest/>', () => {
     mutation.client.mutate = jest.fn();
     mutation.forceUpdate();
     wrapper.find('Form').simulate('submit');
-    expect(mutation.client.mutate.mock.calls[0][0].variables).toEqual({
-      email: 'wesbos@gmail.com',
-    });
+
+    expect(mutation.client.mutate).toBeCalledWithVariables({ email: 'wesbos@gmail.com' });
+
     expect(toJSON(wrapper)).toMatchSnapshot();
   });