starter files

1 files changed, 66 insertions, 0 deletions

diff --git a/finished-application/frontend/__tests__/Nav.test.js b/finished-application/frontend/__tests__/Nav.test.js
new file mode 100644
index 0000000..5c61c56
--- /dev/null
+++ b/finished-application/frontend/__tests__/Nav.test.js
@@ -0,0 +1,66 @@
+import React from 'react';
+import { mount } from 'enzyme';
+import toJSON from 'enzyme-to-json';
+import wait from 'waait';
+import Router from 'next/router';
+import Nav from '../components/Nav';
+import { MockedProvider } from 'react-apollo/test-utils';
+import { fakeUser } from '../lib/testUtils';
+import { CURRENT_USER_QUERY } from '../components/User';
+
+// Mock the router
+Router.router = { push() {} };
+
+const loggedOutMocks = [
+  {
+    request: { query: CURRENT_USER_QUERY },
+    delay: 0,
+    result: {
+      data: {
+        me: null,
+      },
+    },
+  },
+];
+
+const loggedInMocks = [
+  {
+    request: { query: CURRENT_USER_QUERY },
+    result: {
+      data: {
+        what: 'yah',
+        me: {
+          ...fakeUser(),
+          orders: [],
+          cart: [],
+        },
+      },
+    },
+  },
+];
+
+describe('<Nav></Nav>', () => {
+  it('Renders minimal nav when logged out', () => {
+    const wrapper = mount(
+      <MockedProvider mocks={loggedOutMocks}>
+        <Nav />
+      </MockedProvider>
+    );
+    const nav = wrapper.find('ul[data-test="nav"]');
+    expect(toJSON(nav)).toMatchSnapshot();
+  });
+
+  it('renders full nav when logged in', async () => {
+    const wrapper = mount(
+      <MockedProvider mocks={loggedInMocks}>
+        <Nav />
+      </MockedProvider>
+    );
+    await wait();
+    wrapper.update();
+    const nav = wrapper.find('ul[data-test="nav"]');
+    // account link
+    const accountLink = wrapper.find('a[href="/me"]');
+    expect(accountLink.length).toBe(1);
+  });
+});