github-mirrors/kibana
1
0
Fork 0
mirror of https://github.com/elastic/kibana.git synced 2025-04-23 17:28:26 -04:00
Code Issues Projects Releases Packages Wiki Activity

Fix Chart Trend by Severity Bug

Browse source
This commit is contained in:
Ido Cohen 2025-02-25 13:24:50 +02:00 • committed by GitHub
parent 93931f9e3b
commit 7f2df7cd42
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 84 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
x-pack/solutions/security/plugins/cloud_security_posture/server/tasks/findings_stats_task.ts
View file

@ -339,7 +339,7 @@ const getFindingsScoresDocIndexingPromises = (
});
});
const getVulnStatsTrendDocIndexingPromises = (
export const getVulnStatsTrendDocIndexingPromises = (
esClient: ElasticsearchClient,
vulnStatsAggs?: VulnSeverityAggs
) => {
@ -348,12 +348,11 @@ const getVulnStatsTrendDocIndexingPromises = (
const scoreByCloudAccount = Object.fromEntries(
vulnStatsAggs.vulnerabilities_stats_by_cloud_account.buckets.map((accountScore) => {
const cloudAccountId = accountScore.key;
return [
cloudAccountId,
{
cloudAccountId: accountScore.key,
cloudAccountName: accountScore.cloud_account_name.buckets[0].key,
cloudAccountName: accountScore.cloud_account_name.buckets[0]?.key || '',
critical: accountScore.critical.doc_count,
high: accountScore.high.doc_count,
medium: accountScore.medium.doc_count,

82
x-pack/solutions/security/plugins/cloud_security_posture/server/tasks/task_state.test.ts
View file

@ -7,6 +7,10 @@
import { cloneDeep } from 'lodash';
import { stateSchemaByVersion } from './task_state';
import { getVulnStatsTrendDocIndexingPromises } from './findings_stats_task';
import { elasticsearchClientMock } from '@kbn/core-elasticsearch-client-server-mocks';
const esClient = elasticsearchClientMock.createClusterClient().asScoped().asInternalUser;
describe('finding stats task state', () => {
describe('v1', () => {
@ -35,5 +39,83 @@ describe('finding stats task state', () => {
const result = v1.up(state);
expect(result).not.toHaveProperty('foo');
});
it('should return undefined if vulnStatsAggs is not provided', () => {
const result = getVulnStatsTrendDocIndexingPromises(esClient);
expect(result).toBeUndefined();
expect(esClient.index).not.toHaveBeenCalled();
});
it('should correctly process vulnerability stats and call esClient.index', async () => {
const vulnStatsAggs = {
vulnerabilities_stats_by_cloud_account: {
buckets: [
{
key: 'account1',
cloud_account_name: { buckets: [{ key: 'Account One' }] },
critical: { doc_count: 5 },
high: { doc_count: 10 },
medium: { doc_count: 15 },
low: { doc_count: 20 },
},
],
},
critical: { doc_count: 50 },
high: { doc_count: 100 },
medium: { doc_count: 150 },
low: { doc_count: 200 },
};
await getVulnStatsTrendDocIndexingPromises(esClient, vulnStatsAggs);
expect(esClient.index).toHaveBeenCalledWith({
index: expect.any(String),
document: expect.objectContaining({
vulnerabilities_stats_by_cloud_account: {
account1: expect.objectContaining({
cloudAccountName: 'Account One',
critical: 5,
high: 10,
medium: 15,
low: 20,
}),
},
}),
});
});
it('should handle missing cloud account name gracefully', async () => {
const vulnStatsAggs = {
vulnerabilities_stats_by_cloud_account: {
buckets: [
{
key: 'account2',
cloud_account_name: { buckets: [] },
critical: { doc_count: 3 },
high: { doc_count: 6 },
medium: { doc_count: 9 },
low: { doc_count: 12 },
},
],
},
critical: { doc_count: 30 },
high: { doc_count: 60 },
medium: { doc_count: 90 },
low: { doc_count: 120 },
};
await getVulnStatsTrendDocIndexingPromises(esClient, vulnStatsAggs);
expect(esClient.index).toHaveBeenCalledWith({
index: expect.any(String),
document: expect.objectContaining({
vulnerabilities_stats_by_cloud_account: {
account2: expect.objectContaining({
cloudAccountName: '',
}),
},
}),
});
});
});
});