Update tests

albertoblaz · albertoblaz · commit 6bb5f17c2280 · 2025-06-04T15:52:06.000+02:00
diff --git a/x-pack/solutions/security/plugins/security_solution/public/entity_analytics/components/risk_score_donut_chart/use_risk_score_fill_color.test.tsx b/x-pack/solutions/security/plugins/security_solution/public/entity_analytics/components/risk_score_donut_chart/use_risk_score_fill_color.test.tsx
@@ -8,7 +8,7 @@
 import { renderHook } from '@testing-library/react';
 import { useRiskScoreFillColor } from './use_risk_score_fill_color';
 import { emptyDonutColor } from '../../../common/components/charts/donutchart_empty';
-import { SEVERITY_UI_SORT_ORDER } from '../../common';
+import { SEVERITY_UI_SORT_ORDER, RISK_SEVERITY_COLOUR } from '../../common';
 import { TestProviders } from '../../../common/mock';
 
 describe('useRiskScoreFillColor', () => {
@@ -17,15 +17,7 @@ describe('useRiskScoreFillColor', () => {
 
     const colors = SEVERITY_UI_SORT_ORDER.map((severity) => result.current(severity));
 
-    expect(colors).toMatchInlineSnapshot(`
-      Array [
-        "#aaa",
-        "#54B399",
-        "#D6BF57",
-        "#DA8B45",
-        "#E7664C",
-      ]
-    `);
+    expect(colors).toStrictEqual(Object.values(RISK_SEVERITY_COLOUR));
   });
 
   it('returns the empty donut color for an invalid risk severity', () => {
diff --git a/x-pack/solutions/security/plugins/security_solution/public/entity_analytics/components/severity/common/index.test.tsx b/x-pack/solutions/security/plugins/security_solution/public/entity_analytics/components/severity/common/index.test.tsx
@@ -16,7 +16,6 @@ import { EuiHealth, useEuiTheme } from '@elastic/eui';
 
 import { RiskSeverity } from '../../../../../common/search_strategy';
 import { RiskScoreLevel } from '.';
-import { SEVERITY_COLOR } from '../../../../overview/components/detection_response/utils';
 
 jest.mock('@elastic/eui', () => {
   const original = jest.requireActual('@elastic/eui');
@@ -37,7 +36,7 @@ describe('RiskScore', () => {
 
     expect(container).toHaveTextContent(RiskSeverity.Critical);
     expect(EuiHealth as jest.Mock).toHaveBeenLastCalledWith(
-      expect.objectContaining({ color: SEVERITY_COLOR.critical }),
+      expect.objectContaining({ color: euiThemeVars.euiColorSeverityDanger }),
       {}
     );
   });
@@ -52,7 +51,7 @@ describe('RiskScore', () => {
     expect(container).toHaveTextContent(RiskSeverity.High);
 
     expect(EuiHealth as jest.Mock).toHaveBeenLastCalledWith(
-      expect.objectContaining({ color: SEVERITY_COLOR.high }),
+      expect.objectContaining({ color: euiThemeVars.euiColorSeverityRisk }),
       context
     );
   });
@@ -67,7 +66,7 @@ describe('RiskScore', () => {
     expect(container).toHaveTextContent(RiskSeverity.Moderate);
 
     expect(EuiHealth as jest.Mock).toHaveBeenLastCalledWith(
-      expect.objectContaining({ color: SEVERITY_COLOR.medium }),
+      expect.objectContaining({ color: euiThemeVars.euiColorSeverityWarning }),
       context
     );
   });
@@ -82,7 +81,7 @@ describe('RiskScore', () => {
     expect(container).toHaveTextContent(RiskSeverity.Low);
 
     expect(EuiHealth as jest.Mock).toHaveBeenLastCalledWith(
-      expect.objectContaining({ color: SEVERITY_COLOR.low }),
+      expect.objectContaining({ color: euiThemeVars.euiColorSeverityNeutral }),
       context
     );
   });
diff --git a/x-pack/solutions/security/plugins/security_solution/public/entity_analytics/lens_attributes/__snapshots__/risk_score_summary.test.ts.snap b/x-pack/solutions/security/plugins/security_solution/public/entity_analytics/lens_attributes/__snapshots__/risk_score_summary.test.ts.snap
