From d152289331954a43ef76a10acf5ea053b082aa39 Mon Sep 17 00:00:00 2001
From: Ryland Herrick <ryalnd@gmail.com>
Date: Fri, 28 Jun 2024 17:00:30 -0500
Subject: [PATCH] Add ML rules to Basic/Essentials workflow tests

---
 .../common_flows_suppression_serverless_essentials.cy.ts   | 7 ++++++-
 .../rule_creation/common_flows_supression_ess_basic.cy.ts  | 5 +++++
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/x-pack/test/security_solution_cypress/cypress/e2e/detection_response/detection_engine/rule_creation/common_flows_suppression_serverless_essentials.cy.ts b/x-pack/test/security_solution_cypress/cypress/e2e/detection_response/detection_engine/rule_creation/common_flows_suppression_serverless_essentials.cy.ts
index d6f23687cf418..946e0190bc1f8 100644
--- a/x-pack/test/security_solution_cypress/cypress/e2e/detection_response/detection_engine/rule_creation/common_flows_suppression_serverless_essentials.cy.ts
+++ b/x-pack/test/security_solution_cypress/cypress/e2e/detection_response/detection_engine/rule_creation/common_flows_suppression_serverless_essentials.cy.ts
@@ -15,6 +15,7 @@ import { login } from '../../../../tasks/login';
 import { visit } from '../../../../tasks/navigation';
 import {
   ALERT_SUPPRESSION_FIELDS_INPUT,
+  MACHINE_LEARNING_TYPE,
   THRESHOLD_ENABLE_SUPPRESSION_CHECKBOX,
 } from '../../../../screens/create_new_rule';
 import { CREATE_RULE_URL } from '../../../../urls/navigation';
@@ -22,7 +23,7 @@ import { CREATE_RULE_URL } from '../../../../urls/navigation';
 describe(
   'Detection rules, Alert Suppression for Essentials tier',
   {
-    // skipped in MKI as it depends on feature flag alertSuppressionForEsqlRuleEnabled
+    // skipped in MKI as it depends on feature flag alertSuppressionForEsqlRuleEnabled, alertSuppressionForMachineLearningRuleEnabled
     tags: ['@serverless', '@skipInServerlessMKI'],
     env: {
       ftrConfig: {
@@ -35,6 +36,7 @@ describe(
       kbnServerArgs: [
         `--xpack.securitySolution.enableExperimental=${JSON.stringify([
           'alertSuppressionForEsqlRuleEnabled',
+          'alertSuppressionForMachineLearningRuleEnabled',
         ])}`,
       ],
     },
@@ -60,6 +62,9 @@ describe(
 
       selectEsqlRuleType();
       cy.get(ALERT_SUPPRESSION_FIELDS_INPUT).should('be.enabled');
+
+      // ML Rules require Complete tier
+      cy.get(MACHINE_LEARNING_TYPE).get('button').should('be.disabled');
     });
   }
 );
diff --git a/x-pack/test/security_solution_cypress/cypress/e2e/detection_response/detection_engine/rule_creation/common_flows_supression_ess_basic.cy.ts b/x-pack/test/security_solution_cypress/cypress/e2e/detection_response/detection_engine/rule_creation/common_flows_supression_ess_basic.cy.ts
index 1f86d6d0dd789..eb921e66d231c 100644
--- a/x-pack/test/security_solution_cypress/cypress/e2e/detection_response/detection_engine/rule_creation/common_flows_supression_ess_basic.cy.ts
+++ b/x-pack/test/security_solution_cypress/cypress/e2e/detection_response/detection_engine/rule_creation/common_flows_supression_ess_basic.cy.ts
@@ -8,6 +8,7 @@
 import {
   THRESHOLD_ENABLE_SUPPRESSION_CHECKBOX,
   ALERT_SUPPRESSION_DURATION_INPUT,
+  MACHINE_LEARNING_TYPE,
 } from '../../../../screens/create_new_rule';
 
 import {
@@ -16,6 +17,7 @@ import {
   selectThresholdRuleType,
   selectEsqlRuleType,
   openSuppressionFieldsTooltipAndCheckLicense,
+  selectMachineLearningRuleType,
 } from '../../../../tasks/create_new_rule';
 import { startBasicLicense } from '../../../../tasks/api_calls/licensing';
 import { login } from '../../../../tasks/login';
@@ -52,6 +54,9 @@ describe(
         selectEsqlRuleType();
         openSuppressionFieldsTooltipAndCheckLicense();
 
+        // ML Rules require Platinum license
+        cy.get(MACHINE_LEARNING_TYPE).get('button').should('be.disabled');
+
         selectThresholdRuleType();
         cy.get(THRESHOLD_ENABLE_SUPPRESSION_CHECKBOX).should('be.disabled');
         cy.get(THRESHOLD_ENABLE_SUPPRESSION_CHECKBOX).parent().trigger('mouseover');