Fix state filtering in the category list

chrisronline · chrisronline · commit ae015fb8744c · 2020-12-12T08:31:42.000-05:00
diff --git a/x-pack/plugins/monitoring/public/alerts/lib/__snapshots__/get_alert_panels_by_category.test.tsx.snap b/x-pack/plugins/monitoring/public/alerts/lib/__snapshots__/get_alert_panels_by_category.test.tsx.snap
diff --git a/x-pack/plugins/monitoring/public/alerts/lib/get_alert_panels_by_category.test.tsx b/x-pack/plugins/monitoring/public/alerts/lib/get_alert_panels_by_category.test.tsx
@@ -181,6 +181,19 @@ describe('getAlertPanelsByCategory', () => {
       );
       expect(result).toMatchSnapshot();
     });
+
+    it('should allow for state filtering', () => {
+      const alerts = [getAlert(ALERT_CPU_USAGE, 2)];
+      const customStateFilter = (state: AlertState) => state.nodeName === 'es_name_0';
+      const result = getAlertPanelsByCategory(
+        panelTitle,
+        false,
+        alerts,
+        alertsContext,
+        customStateFilter
+      );
+      expect(result).toMatchSnapshot();
+    });
   });
 
   describe('setup mode', () => {
diff --git a/x-pack/plugins/monitoring/public/alerts/lib/get_alert_panels_by_category.tsx b/x-pack/plugins/monitoring/public/alerts/lib/get_alert_panels_by_category.tsx
@@ -61,7 +61,7 @@ export function getAlertPanelsByCategory(
               states: foundAlert.states,
               alertName,
             });
-            categoryFiringAlertCount += foundAlert.states.length;
+            categoryFiringAlertCount += states.length;
           }
         }
       }
@@ -144,12 +144,13 @@ export function getAlertPanelsByCategory(
         id: nodeIndex + 1,
         title: `${category.label}`,
         items: category.alerts.map(({ alertName, states }) => {
+          const filteredStates = states.filter(({ state }) => stateFilter(state));
           const alertStatus = alertsContext.allAlerts[alertName];
           const name = inSetupMode ? (
             <EuiText>{alertStatus.rawAlert.name}</EuiText>
           ) : (
             <EuiText>
-              {alertStatus.rawAlert.name} ({states.length})
+              {alertStatus.rawAlert.name} ({filteredStates.length})
             </EuiText>
           );
           return {
@@ -169,7 +170,7 @@ export function getAlertPanelsByCategory(
     for (const category of menu) {
       for (const { alert, states } of category.alerts) {
         const items = [];
-        for (const alertState of states) {
+        for (const alertState of states.filter(({ state }) => stateFilter(state))) {
           items.push({
             name: (
               <Fragment>
@@ -211,7 +212,7 @@ export function getAlertPanelsByCategory(
     }, menu.length);
     for (const category of menu) {
       for (const { alert, states } of category.alerts) {
-        for (const state of states) {
+        for (const state of states.filter(({ state: _state }) => stateFilter(_state))) {
           panels.push({
             id: ++tertiaryPanelIndex2,
             title: `${alert.name}`,
