From 4534a5dde76d8bb19ce3e2ff09086c91e99ad13f Mon Sep 17 00:00:00 2001
From: FrankHassanabad <frank.hassanabad@elastic.co>
Date: Thu, 30 Jul 2020 19:03:49 -0600
Subject: [PATCH] Added unit tests

---
 .../alerts_utility_bar/index.test.tsx         | 188 +++++++++++++++++-
 .../alerts_table/alerts_utility_bar/index.tsx |   2 +-
 2 files changed, 186 insertions(+), 4 deletions(-)

diff --git a/x-pack/plugins/security_solution/public/detections/components/alerts_table/alerts_utility_bar/index.test.tsx b/x-pack/plugins/security_solution/public/detections/components/alerts_table/alerts_utility_bar/index.test.tsx
index cbbe43cc03568f..0ba9764cf24af4 100644
--- a/x-pack/plugins/security_solution/public/detections/components/alerts_table/alerts_utility_bar/index.test.tsx
+++ b/x-pack/plugins/security_solution/public/detections/components/alerts_table/alerts_utility_bar/index.test.tsx
@@ -5,14 +5,15 @@
  */
 
 import React from 'react';
-import { shallow } from 'enzyme';
+import { shallow, mount } from 'enzyme';
 
-import { AlertsUtilityBar } from './index';
+import { AlertsUtilityBar, AlertsUtilityBarProps } from './index';
+import { TestProviders } from '../../../../common/mock/test_providers';
 
 jest.mock('../../../../common/lib/kibana');
 
 describe('AlertsUtilityBar', () => {
-  it('renders correctly', () => {
+  test('renders correctly', () => {
     const wrapper = shallow(
       <AlertsUtilityBar
         canUserCRUD={true}
@@ -32,4 +33,185 @@ describe('AlertsUtilityBar', () => {
 
     expect(wrapper.find('[dataTestSubj="alertActionPopover"]')).toBeTruthy();
   });
+
+  describe('UtilityBarAdditionalFiltersContent', () => {
+    test('does not show the showBuildingBlockAlerts checked if the showBuildingBlockAlerts is false', () => {
+      const onShowBuildingBlockAlertsChanged = jest.fn();
+      const wrapper = mount(
+        <TestProviders>
+          <AlertsUtilityBar
+            canUserCRUD={true}
+            hasIndexWrite={true}
+            areEventsLoading={false}
+            clearSelection={jest.fn()}
+            totalCount={100}
+            selectedEventIds={{}}
+            currentFilter="closed"
+            selectAll={jest.fn()}
+            showClearSelection={true}
+            showBuildingBlockAlerts={false} // Does not show showBuildingBlockAlerts checked if this is false
+            onShowBuildingBlockAlertsChanged={onShowBuildingBlockAlertsChanged}
+            updateAlertsStatus={jest.fn()}
+          />
+        </TestProviders>
+      );
+      // click the filters button to popup the checkbox to make it visible
+      wrapper
+        .find('[data-test-subj="additionalFilters"] button')
+        .first()
+        .simulate('click')
+        .update();
+
+      // The check box should be false
+      expect(
+        wrapper
+          .find('[data-test-subj="showBuildingBlockAlertsCheckbox"] input')
+          .first()
+          .prop('checked')
+      ).toEqual(false);
+    });
+
+    test('does show the showBuildingBlockAlerts checked if the showBuildingBlockAlerts is true', () => {
+      const onShowBuildingBlockAlertsChanged = jest.fn();
+      const wrapper = mount(
+        <TestProviders>
+          <AlertsUtilityBar
+            canUserCRUD={true}
+            hasIndexWrite={true}
+            areEventsLoading={false}
+            clearSelection={jest.fn()}
+            totalCount={100}
+            selectedEventIds={{}}
+            currentFilter="closed"
+            selectAll={jest.fn()}
+            showClearSelection={true}
+            showBuildingBlockAlerts={true} // Does show showBuildingBlockAlerts checked if this is true
+            onShowBuildingBlockAlertsChanged={onShowBuildingBlockAlertsChanged}
+            updateAlertsStatus={jest.fn()}
+          />
+        </TestProviders>
+      );
+      // click the filters button to popup the checkbox to make it visible
+      wrapper
+        .find('[data-test-subj="additionalFilters"] button')
+        .first()
+        .simulate('click')
+        .update();
+
+      // The check box should be true
+      expect(
+        wrapper
+          .find('[data-test-subj="showBuildingBlockAlertsCheckbox"] input')
+          .first()
+          .prop('checked')
+      ).toEqual(true);
+    });
+
+    test('calls the onShowBuildingBlockAlertsChanged when the check box is clicked', () => {
+      const onShowBuildingBlockAlertsChanged = jest.fn();
+      const wrapper = mount(
+        <TestProviders>
+          <AlertsUtilityBar
+            canUserCRUD={true}
+            hasIndexWrite={true}
+            areEventsLoading={false}
+            clearSelection={jest.fn()}
+            totalCount={100}
+            selectedEventIds={{}}
+            currentFilter="closed"
+            selectAll={jest.fn()}
+            showClearSelection={true}
+            showBuildingBlockAlerts={false}
+            onShowBuildingBlockAlertsChanged={onShowBuildingBlockAlertsChanged}
+            updateAlertsStatus={jest.fn()}
+          />
+        </TestProviders>
+      );
+      // click the filters button to popup the checkbox to make it visible
+      wrapper
+        .find('[data-test-subj="additionalFilters"] button')
+        .first()
+        .simulate('click')
+        .update();
+
+      // check the box
+      wrapper
+        .find('[data-test-subj="showBuildingBlockAlertsCheckbox"] input')
+        .first()
+        .simulate('change', { target: { checked: true } });
+
+      // Make sure our callback is called
+      expect(onShowBuildingBlockAlertsChanged).toHaveBeenCalled();
+    });
+
+    test('can update showBuildingBlockAlerts from false to true', () => {
+      const Proxy = (props: AlertsUtilityBarProps) => (
+        <TestProviders>
+          <AlertsUtilityBar
+            canUserCRUD={true}
+            hasIndexWrite={true}
+            areEventsLoading={false}
+            clearSelection={jest.fn()}
+            totalCount={100}
+            selectedEventIds={{}}
+            currentFilter="closed"
+            selectAll={jest.fn()}
+            showClearSelection={true}
+            showBuildingBlockAlerts={props.showBuildingBlockAlerts}
+            onShowBuildingBlockAlertsChanged={jest.fn()}
+            updateAlertsStatus={jest.fn()}
+          />
+        </TestProviders>
+      );
+
+      const wrapper = mount(
+        <Proxy
+          canUserCRUD={true}
+          hasIndexWrite={true}
+          areEventsLoading={false}
+          clearSelection={jest.fn()}
+          totalCount={100}
+          selectedEventIds={{}}
+          currentFilter="closed"
+          selectAll={jest.fn()}
+          showClearSelection={true}
+          showBuildingBlockAlerts={false}
+          onShowBuildingBlockAlertsChanged={jest.fn()}
+          updateAlertsStatus={jest.fn()}
+        />
+      );
+      // click the filters button to popup the checkbox to make it visible
+      wrapper
+        .find('[data-test-subj="additionalFilters"] button')
+        .first()
+        .simulate('click')
+        .update();
+
+      // The check box should false now since we initially set the showBuildingBlockAlerts to false
+      expect(
+        wrapper
+          .find('[data-test-subj="showBuildingBlockAlertsCheckbox"] input')
+          .first()
+          .prop('checked')
+      ).toEqual(false);
+
+      wrapper.setProps({ showBuildingBlockAlerts: true });
+      wrapper.update();
+
+      // click the filters button to popup the checkbox to make it visible
+      wrapper
+        .find('[data-test-subj="additionalFilters"] button')
+        .first()
+        .simulate('click')
+        .update();
+
+      // The check box should be true now since we changed the showBuildingBlockAlerts from false to true
+      expect(
+        wrapper
+          .find('[data-test-subj="showBuildingBlockAlertsCheckbox"] input')
+          .first()
+          .prop('checked')
+      ).toEqual(true);
+    });
+  });
 });
diff --git a/x-pack/plugins/security_solution/public/detections/components/alerts_table/alerts_utility_bar/index.tsx b/x-pack/plugins/security_solution/public/detections/components/alerts_table/alerts_utility_bar/index.tsx
index 4434b65b86e97e..bdad380f59ae95 100644
--- a/x-pack/plugins/security_solution/public/detections/components/alerts_table/alerts_utility_bar/index.tsx
+++ b/x-pack/plugins/security_solution/public/detections/components/alerts_table/alerts_utility_bar/index.tsx
@@ -28,7 +28,7 @@ import { TimelineNonEcsData } from '../../../../graphql/types';
 import { UpdateAlertsStatus } from '../types';
 import { FILTER_CLOSED, FILTER_IN_PROGRESS, FILTER_OPEN } from '../alerts_filter_group';
 
-interface AlertsUtilityBarProps {
+export interface AlertsUtilityBarProps {
   canUserCRUD: boolean;
   hasIndexWrite: boolean;
   areEventsLoading: boolean;