add test case to check for data within retention period

elastic · Oct 10, 2024 · 3786500 · 3786500
1 parent 8ca483a
commit 3786500
Showing 1 changed file with 81 additions and 1 deletion.
diff --git a/x-pack/test/cloud_security_posture_functional/pages/findings_old_data.ts b/x-pack/test/cloud_security_posture_functional/pages/findings_old_data.ts
@@ -11,7 +11,8 @@ import type { FtrProviderContext } from '../ftr_provider_context';
 import { vulnerabilitiesLatestMock } from '../mocks/vulnerabilities_latest_mock';
 
 // eslint-disable-next-line import/no-default-export
-export default function ({ getPageObjects }: FtrProviderContext) {
+export default function ({ getPageObjects, getService }: FtrProviderContext) {
+  const retry = getService('retry');
   const pageObjects = getPageObjects(['common', 'findings', 'header']);
   const chance = new Chance();
   const daysToMillisecond = (days: number) => days * 24 * 60 * 60 * 1000;
@@ -36,6 +37,25 @@ export default function ({ getPageObjects }: FtrProviderContext) {
       cluster_id: 'Upper case cluster id',
     },
   ];
+  const dataWithinRetentionKspm = [
+    {
+      '@timestamp': (Date.now() - daysToMillisecond(RETENTION - 1)).toString(),
+      resource: { id: chance.guid(), name: `kubelet`, sub_type: 'lower case sub type' },
+      result: { evaluation: chance.integer() % 2 === 0 ? 'passed' : 'failed' },
+      rule: {
+        name: 'Upper case rule name',
+        section: 'Upper case section',
+        benchmark: {
+          id: 'cis_k8s',
+          posture_type: 'kspm',
+          name: 'CIS Kubernetes V1.23',
+          version: 'v1.0.0',
+        },
+        type: 'process',
+      },
+      cluster_id: 'Upper case cluster id',
+    },
+  ];
 
   const dataOldCspm = [
     {
@@ -56,20 +76,49 @@ export default function ({ getPageObjects }: FtrProviderContext) {
       cluster_id: 'Upper case cluster id',
     },
   ];
+  const dataWithinRetentionCspm = [
+    {
+      '@timestamp': (Date.now() - daysToMillisecond(RETENTION - 1)).toString(),
+      resource: { id: chance.guid(), name: `kubelet`, sub_type: 'lower case sub type' },
+      result: { evaluation: chance.integer() % 2 === 0 ? 'passed' : 'failed' },
+      rule: {
+        name: 'Upper case rule name',
+        section: 'Upper case section',
+        benchmark: {
+          id: 'cis_aws',
+          posture_type: 'cspm',
+          name: 'CIS AWS V1.23',
+          version: 'v1.0.0',
+        },
+        type: 'process',
+      },
+      cluster_id: 'Upper case cluster id',
+    },
+  ];
 
   const dataOldCnvm = [
     {
       ...vulnerabilitiesLatestMock[0],
       '@timestamp': (Date.now() - daysToMillisecond(RETENTION + 1)).toString(),
     },
   ];
+  const dataWithinRetentionCnvm = [
+    {
+      ...vulnerabilitiesLatestMock[0],
+      '@timestamp': (Date.now() - daysToMillisecond(RETENTION - 1)).toString(),
+    },
+  ];
 
   describe('Old Data', function () {
     this.tags(['cloud_security_posture_findings']);
     let findings: typeof pageObjects.findings;
+    let latestFindingsTable: typeof findings.latestFindingsTable;
+    let latestVulnerabilitiesTable: typeof findings.latestVulnerabilitiesTable;
 
     before(async () => {
       findings = pageObjects.findings;
+      latestFindingsTable = findings.latestFindingsTable;
+      latestVulnerabilitiesTable = findings.latestVulnerabilitiesTable;
 
       // Before we start any test we must wait for cloud_security_posture plugin to complete its initialization
       await findings.waitForPluginInitialized();
@@ -102,6 +151,37 @@ export default function ({ getPageObjects }: FtrProviderContext) {
         await pageObjects.header.waitUntilLoadingHasFinished();
         expect(await findings.isLatestFindingsTableThere()).to.be(false);
       });
+      it('returns data grid with only data within retention KSPM', async () => {
+        await findings.index.add([...dataOldKspm, ...dataWithinRetentionKspm]);
+
+        await findings.navigateToLatestFindingsPage();
+        await retry.waitFor(
+          'Findings table to be loaded',
+          async () => (await latestFindingsTable.getRowsCount()) === dataWithinRetentionKspm.length
+        );
+        await pageObjects.header.waitUntilLoadingHasFinished();
+      });
+      it('returns data grid with only data within retention CSPM', async () => {
+        await findings.index.add([...dataOldCspm, ...dataWithinRetentionCspm]);
+
+        await findings.navigateToLatestFindingsPage();
+        await retry.waitFor(
+          'Findings table to be loaded',
+          async () => (await latestFindingsTable.getRowsCount()) === dataWithinRetentionCspm.length
+        );
+        await pageObjects.header.waitUntilLoadingHasFinished();
+      });
+      it('returns data grid with only data within retention CSPM', async () => {
+        await findings.vulnerabilitiesIndex.add([...dataOldCnvm, ...dataWithinRetentionCnvm]);
+
+        await findings.navigateToLatestVulnerabilitiesPage();
+        await retry.waitFor(
+          'Findings table to be loaded',
+          async () =>
+            (await latestVulnerabilitiesTable.getRowsCount()) === dataWithinRetentionCnvm.length
+        );
+        await pageObjects.header.waitUntilLoadingHasFinished();
+      });
     });
   });
 }