fleet/frontend/layouts/CoreLayout/CoreLayout.tests.jsx

import expect from 'expect';
import { mount } from 'enzyme';

import helpers from 'test/helpers';
import { userStub } from 'test/stubs';
import CoreLayout from './CoreLayout';

const {
  connectedComponent,
  reduxMockStore,
} = helpers;

describe('CoreLayout - layouts', () => {
  const store = {
    app: { config: {} },
    auth: { user: userStub },
    notifications: {},
    persistentFlash: {
      showFlash: false,
      message: '',
    },
  };
  const mockStore = reduxMockStore(store);

  it('renders the FlashMessage component when notifications are present', () => {
    const storeWithNotifications = {
      app: { config: {} },
      auth: {
        user: userStub,
      },
      notifications: {
        alertType: 'success',
        isVisible: true,
        message: 'nice jerb!',
      },
      persistentFlash: {
        showFlash: false,
        message: '',
      },
    };
    const mockStoreWithNotifications = reduxMockStore(storeWithNotifications);
    const componentWithFlash = connectedComponent(CoreLayout, {
      mockStore: mockStoreWithNotifications,
    });
    const componentWithoutFlash = connectedComponent(CoreLayout, {
      mockStore,
    });

    const appWithFlash = mount(componentWithFlash);
    const appWithoutFlash = mount(componentWithoutFlash);

    expect(appWithFlash.length).toEqual(1);
    expect(appWithoutFlash.length).toEqual(1);

    expect(appWithFlash.find('FlashMessage').html()).toExist();
    expect(appWithoutFlash.find('FlashMessage').html()).toNotExist();
  });

  it('renders the PersistentFlash component when showFlash is true', () => {
    const storeWithPersistentFlash = {
      ...store,
      persistentFlash: {
        showFlash: true,
        message: 'This is the flash message',
      },
    };

    const mockStoreWithPersistentFlash = reduxMockStore(storeWithPersistentFlash);

    const Layout = connectedComponent(CoreLayout, {
      mockStore: mockStoreWithPersistentFlash,
    });
    const MountedLayout = mount(Layout);

    expect(MountedLayout.find('PersistentFlash').length).toEqual(1, 'Expected the Persistent Flash to be on the page');
  });
});