Flatten diagnostics to count (filters) (#459)

alanzanattadev · Dec 6, 2018 · a0a788b · a0a788b
1 parent 853d9f5
commit a0a788b
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/lib/ExecutionControlEpic/DiagnosticsFeature/Presenters/DiagnosticsQuickSelector.js b/lib/ExecutionControlEpic/DiagnosticsFeature/Presenters/DiagnosticsQuickSelector.js
@@ -117,6 +117,17 @@ function DiagnosticsQuickSelector({
   severities: Set<DiagnosticSeverity>,
   onFilterClick: (severity: number) => void,
 }) {
+  const flattenDiagnostics = (
+    severity: DiagnosticSeverity,
+  ): List<MoleculeDiagnostic> =>
+    diagnostics
+      .filter(diagnosticGroup => diagnosticGroup.get(severity))
+      .reduce(
+        (acc, serverityMap) =>
+          serverityMap ? acc.concat(...serverityMap.values()) : acc,
+        List(),
+      );
+
   return (
     <DiagnosticsBox>
       {[3, 5, 2, 1].map((severity: DiagnosticSeverity) => (
@@ -125,9 +136,7 @@ function DiagnosticsQuickSelector({
           activated={severities.includes(severity)}
           key={severity}
           severity={severity}
-          count={diagnostics
-            .filter(diagnosticGroup => diagnosticGroup.get(severity))
-            .count()}
+          count={flattenDiagnostics(severity).count()}
         />
       ))}
     </DiagnosticsBox>