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

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()).toBeTruthy();
    expect(appWithoutFlash.find("FlashMessage").html()).toBeFalsy();
  });

  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"
    );
  });
});