Add public POST/PATCH endpoints for Flaky Tests Management policies (#3900)

Co-authored-by: ci.datadog-api-spec <packages@datadoghq.com>

Author: api-clients-generation-pipeline[bot]

.generator/schemas/v2/openapi.yaml

@@ -33,6 +33,38 @@ Feature: Test Optimization
     When the request is sent
     Then the response status is 404 Not Found
 
+  @generated @skip @team:DataDog/ci-app-backend
+  Scenario: Get Flaky Tests Management policies returns "Bad Request" response
+    Given operation "GetFlakyTestsManagementPolicies" enabled
+    And new "GetFlakyTestsManagementPolicies" request
+    And body with value {"data": {"attributes": {"repository_id": "github.com/datadog/shopist"}, "type": "test_optimization_get_flaky_tests_management_policies_request"}}
+    When the request is sent
+    Then the response status is 400 Bad Request
+
+  @generated @skip @team:DataDog/ci-app-backend
+  Scenario: Get Flaky Tests Management policies returns "Not Found" response
+    Given operation "GetFlakyTestsManagementPolicies" enabled
+    And new "GetFlakyTestsManagementPolicies" request
+    And body with value {"data": {"attributes": {"repository_id": "github.com/datadog/shopist"}, "type": "test_optimization_get_flaky_tests_management_policies_request"}}
+    When the request is sent
+    Then the response status is 404 Not Found
+
+  @generated @skip @team:DataDog/ci-app-backend
+  Scenario: Get Flaky Tests Management policies returns "OK" response
+    Given operation "GetFlakyTestsManagementPolicies" enabled
+    And new "GetFlakyTestsManagementPolicies" request
+    And body with value {"data": {"attributes": {"repository_id": "github.com/datadog/shopist"}, "type": "test_optimization_get_flaky_tests_management_policies_request"}}
+    When the request is sent
+    Then the response status is 200 OK
+
+  @skip @team:DataDog/ci-app-backend
+  Scenario: Get Flaky Tests Management policies with empty repository_id returns bad request
+    Given operation "GetFlakyTestsManagementPolicies" enabled
+    And new "GetFlakyTestsManagementPolicies" request
+    And body with value {"data": {"attributes": {"repository_id": ""}, "type": "test_optimization_get_flaky_tests_management_policies_request"}}
+    When the request is sent
+    Then the response status is 400 Bad Request
+
   @generated @skip @team:DataDog/ci-app-backend
   Scenario: Get Test Optimization service settings returns "Bad Request" response
     Given operation "GetTestOptimizationServiceSettings" enabled
@@ -133,6 +165,38 @@ Feature: Test Optimization
     When the request with pagination is sent
     Then the response status is 200 OK
 
+  @generated @skip @team:DataDog/ci-app-backend
+  Scenario: Update Flaky Tests Management policies returns "Bad Request" response
+    Given operation "UpdateFlakyTestsManagementPolicies" enabled
+    And new "UpdateFlakyTestsManagementPolicies" request
+    And body with value {"data": {"attributes": {"attempt_to_fix": {"retries": 3}, "disabled": {"auto_disable_rule": {"enabled": false, "status": "active", "window_seconds": 3600}, "branch_rule": {"branches": ["main"], "enabled": true, "excluded_branches": [], "excluded_test_services": []}, "enabled": false, "failure_rate_rule": {"branches": [], "enabled": false, "min_runs": 10, "status": "active", "threshold": 0.5}}, "quarantined": {"auto_quarantine_rule": {"enabled": true, "window_seconds": 3600}, "branch_rule": {"branches": ["main"], "enabled": true, "excluded_branches": [], "excluded_test_services": []}, "enabled": true, "failure_rate_rule": {"branches": ["main"], "enabled": true, "min_runs": 10, "threshold": 0.5}}, "repository_id": "github.com/datadog/shopist"}, "type": "test_optimization_update_flaky_tests_management_policies_request"}}
+    When the request is sent
+    Then the response status is 400 Bad Request
+
+  @generated @skip @team:DataDog/ci-app-backend
+  Scenario: Update Flaky Tests Management policies returns "Not Found" response
+    Given operation "UpdateFlakyTestsManagementPolicies" enabled
+    And new "UpdateFlakyTestsManagementPolicies" request
+    And body with value {"data": {"attributes": {"attempt_to_fix": {"retries": 3}, "disabled": {"auto_disable_rule": {"enabled": false, "status": "active", "window_seconds": 3600}, "branch_rule": {"branches": ["main"], "enabled": true, "excluded_branches": [], "excluded_test_services": []}, "enabled": false, "failure_rate_rule": {"branches": [], "enabled": false, "min_runs": 10, "status": "active", "threshold": 0.5}}, "quarantined": {"auto_quarantine_rule": {"enabled": true, "window_seconds": 3600}, "branch_rule": {"branches": ["main"], "enabled": true, "excluded_branches": [], "excluded_test_services": []}, "enabled": true, "failure_rate_rule": {"branches": ["main"], "enabled": true, "min_runs": 10, "threshold": 0.5}}, "repository_id": "github.com/datadog/shopist"}, "type": "test_optimization_update_flaky_tests_management_policies_request"}}
+    When the request is sent
+    Then the response status is 404 Not Found
+
+  @generated @skip @team:DataDog/ci-app-backend
+  Scenario: Update Flaky Tests Management policies returns "OK" response
+    Given operation "UpdateFlakyTestsManagementPolicies" enabled
+    And new "UpdateFlakyTestsManagementPolicies" request
+    And body with value {"data": {"attributes": {"attempt_to_fix": {"retries": 3}, "disabled": {"auto_disable_rule": {"enabled": false, "status": "active", "window_seconds": 3600}, "branch_rule": {"branches": ["main"], "enabled": true, "excluded_branches": [], "excluded_test_services": []}, "enabled": false, "failure_rate_rule": {"branches": [], "enabled": false, "min_runs": 10, "status": "active", "threshold": 0.5}}, "quarantined": {"auto_quarantine_rule": {"enabled": true, "window_seconds": 3600}, "branch_rule": {"branches": ["main"], "enabled": true, "excluded_branches": [], "excluded_test_services": []}, "enabled": true, "failure_rate_rule": {"branches": ["main"], "enabled": true, "min_runs": 10, "threshold": 0.5}}, "repository_id": "github.com/datadog/shopist"}, "type": "test_optimization_update_flaky_tests_management_policies_request"}}
+    When the request is sent
+    Then the response status is 200 OK
+
+  @skip @team:DataDog/ci-app-backend
+  Scenario: Update Flaky Tests Management policies with empty repository_id returns bad request
+    Given operation "UpdateFlakyTestsManagementPolicies" enabled
+    And new "UpdateFlakyTestsManagementPolicies" request
+    And body with value {"data": {"attributes": {"repository_id": ""}, "type": "test_optimization_update_flaky_tests_management_policies_request"}}
+    When the request is sent
+    Then the response status is 400 Bad Request
+
   @generated @skip @team:DataDog/ci-app-backend
   Scenario: Update Test Optimization service settings returns "Bad Request" response
     Given operation "UpdateTestOptimizationServiceSettings" enabled

features/v2/test_optimization.feature

@@ -1036,6 +1036,18 @@
       "type": "safe"
     }
   },
+  "UpdateFlakyTestsManagementPolicies": {
+    "tag": "Test Optimization",
+    "undo": {
+      "type": "unsafe"
+    }
+  },
+  "GetFlakyTestsManagementPolicies": {
+    "tag": "Test Optimization",
+    "undo": {
+      "type": "safe"
+    }
+  },
   "DeleteTestOptimizationServiceSettings": {
     "tag": "Test Optimization",
     "undo": {

features/v2/undo.json

@@ -3925,6 +3925,22 @@ export const ScenariosModelMappings: { [key: string]: OperationMapping } = {
     },
     operationResponseType: "CIAppPipelineEventsResponse",
   },
+  "TestOptimizationApi.V2.GetFlakyTestsManagementPolicies": {
+    body: {
+      type: "TestOptimizationFlakyTestsManagementPoliciesGetRequest",
+      format: "",
+    },
+    operationResponseType:
+      "TestOptimizationFlakyTestsManagementPoliciesResponse",
+  },
+  "TestOptimizationApi.V2.UpdateFlakyTestsManagementPolicies": {
+    body: {
+      type: "TestOptimizationFlakyTestsManagementPoliciesUpdateRequest",
+      format: "",
+    },
+    operationResponseType:
+      "TestOptimizationFlakyTestsManagementPoliciesResponse",
+  },
   "TestOptimizationApi.V2.GetTestOptimizationServiceSettings": {
     body: {
       type: "TestOptimizationGetServiceSettingsRequest",

private/bdd_runner/src/support/scenarios_model_mapping.ts

@@ -28,15 +28,15 @@ const configuration = createConfiguration();
 // Enable unstable operations
 const configurationOpts = {
     unstableOperations: {
-        "TestOptimizationApi.v2.getTestOptimizationServiceSettings": true
+        "TestOptimizationApi.v2.getFlakyTestsManagementPolicies": true
     }
 }
 
 const configuration = createConfiguration(configurationOpts);
 const apiInstance = new TestOptimizationApiV2(configuration);
 const params = {/* parameters */};
 
-apiInstance.getTestOptimizationServiceSettings(params).then((data) => {
+apiInstance.getFlakyTestsManagementPolicies(params).then((data) => {
     console.log("API called successfully. Returned data: " + JSON.stringify(data));
 }).catch((error) => {
     console.error("Error calling API: " + error);