Regenerate client from commit ebaa65e of spec repo

Author: ci.datadog-api-spec

.generator/schemas/v2/openapi.yaml

@@ -61289,6 +61289,129 @@ components:
       - name
       - options
       type: object
+    UpdateFlakyTestStateRequest:
+      description: Request to update the state of multiple flaky tests.
+      properties:
+        data:
+          $ref: '#/components/schemas/UpdateFlakyTestStateRequestData'
+      required:
+      - data
+      type: object
+    UpdateFlakyTestStateRequestAttributes:
+      description: Attributes for updating flaky test states.
+      properties:
+        tests:
+          description: List of flaky tests to update.
+          items:
+            $ref: '#/components/schemas/UpdateFlakyTestStateRequestTest'
+          type: array
+      required:
+      - tests
+      type: object
+    UpdateFlakyTestStateRequestData:
+      description: The JSON:API data for updating flaky test states.
+      properties:
+        attributes:
+          $ref: '#/components/schemas/UpdateFlakyTestStateRequestAttributes'
+        type:
+          $ref: '#/components/schemas/UpdateFlakyTestStateRequestDataType'
+      required:
+      - type
+      - attributes
+      type: object
+    UpdateFlakyTestStateRequestDataType:
+      description: The definition of `UpdateFlakyTestStateRequestDataType` object.
+      enum:
+      - update_flaky_test_state_request
+      example: update_flaky_test_state_request
+      type: string
+      x-enum-varnames:
+      - UPDATE_FLAKY_TEST_STATE_REQUEST
+    UpdateFlakyTestStateRequestTest:
+      description: A flaky test to update.
+      properties:
+        id:
+          description: The ID of the flaky test (FQN).
+          example: 4eb1887a8adb1847
+          type: string
+        new_state:
+          $ref: '#/components/schemas/UpdateFlakyTestStateRequestTestNewState'
+      required:
+      - id
+      - new_state
+      type: object
+    UpdateFlakyTestStateRequestTestNewState:
+      description: The new state to set for the flaky test.
+      enum:
+      - active
+      - quarantined
+      - disabled
+      - fixed
+      example: active
+      type: string
+      x-enum-varnames:
+      - ACTIVE
+      - QUARANTINED
+      - DISABLED
+      - FIXED
+    UpdateFlakyTestStateResponse:
+      description: Response object for updating flaky test states.
+      properties:
+        data:
+          $ref: '#/components/schemas/UpdateFlakyTestStateResponseData'
+      type: object
+    UpdateFlakyTestStateResponseAttributes:
+      description: Attributes for the update flaky test state response.
+      properties:
+        has_errors:
+          description: Whether any errors occurred during the update operation.
+          example: false
+          type: boolean
+        results:
+          description: Results of the update operation for each test.
+          items:
+            $ref: '#/components/schemas/UpdateFlakyTestStateResponseResult'
+          type: array
+      required:
+      - has_errors
+      - results
+      type: object
+    UpdateFlakyTestStateResponseData:
+      description: The JSON:API data for the update flaky test state response.
+      properties:
+        attributes:
+          $ref: '#/components/schemas/UpdateFlakyTestStateResponseAttributes'
+        id:
+          description: The ID of the response.
+          type: string
+        type:
+          $ref: '#/components/schemas/UpdateFlakyTestStateResponseDataType'
+      type: object
+    UpdateFlakyTestStateResponseDataType:
+      description: The definition of `UpdateFlakyTestStateResponseDataType` object.
+      enum:
+      - update_flaky_test_state_response
+      type: string
+      x-enum-varnames:
+      - UPDATE_FLAKY_TEST_STATE_RESPONSE
+    UpdateFlakyTestStateResponseResult:
+      description: Result of updating a single flaky test state.
+      properties:
+        error:
+          description: Error message if the update failed.
+          type: string
+        id:
+          description: The ID of the flaky test (FQN).
+          example: 4eb1887a8adb1847
+          type: string
+        success:
+          description: Whether the update was successful.
+          example: false
+          type: boolean
+      required:
+      - id
+      - success
+      type: object
     UpdateOnCallNotificationRuleRequest:
       description: A top-level wrapper for updating a notification rule for a user
       example:
@@ -63638,6 +63761,7 @@ components:
             teams_read: Read Teams data. A User with this permission can view Team
               names, metadata, and which Users are on each Team.
             test_optimization_read: View Test Optimization.
+            test_optimization_write: Modify Test Optimization.
             timeseries_query: Query Timeseries data.
             usage_read: View your organization's usage and usage attribution.
             user_access_invite: Invite other users to your organization.
@@ -94850,6 +94974,44 @@ paths:
         - incident_settings_write
       x-unstable: '**Note**: This endpoint is deprecated. See the [Teams API endpoints](https://docs.datadoghq.com/api/latest/teams/).'
   /api/v2/test/flaky-test-management/tests:
+    patch:
+      description: Update the state of multiple flaky tests in Flaky Test Management.
+      operationId: UpdateFlakyTestState
+      requestBody:
+        content:
+          application/json:
+            schema:
+              $ref: '#/components/schemas/UpdateFlakyTestStateRequest'
+        required: true
+      responses:
+        '200':
+          content:
+            application/json:
+              schema:
+                $ref: '#/components/schemas/UpdateFlakyTestStateResponse'
+          description: OK
+        '400':
+          $ref: '#/components/responses/BadRequestResponse'
+        '403':
+          $ref: '#/components/responses/NotAuthorizedResponse'
+        '429':
+          $ref: '#/components/responses/TooManyRequestsResponse'
+      security:
+      - apiKeyAuth: []
+        appKeyAuth: []
+      - AuthZ:
+        - test_optimization_write
+      summary: Update flaky test states
+      tags:
+      - Test Optimization
+      x-codegen-request-body-name: body
+      x-permission:
+        operator: OR
+        permissions:
+        - test_optimization_write
+      x-unstable: '**Note**: This endpoint is in preview and may be subject to change.
+
+        If you have any feedback, contact [Datadog support](https://docs.datadoghq.com/help/).'
     post:
       description: List endpoint returning flaky tests from Flaky Test Management.
         Results are paginated.

docs/datadog_api_client.v2.model.rst

@@ -26975,6 +26975,83 @@ datadog\_api\_client.v2.model.update\_deployment\_rule\_params\_data\_attributes
    :members:
    :show-inheritance:
 
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_request module
+------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_request
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_request\_attributes module
+------------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_request_attributes
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_request\_data module
+------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_request_data
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_request\_data\_type module
+------------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_request_data_type
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_request\_test module
+------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_request_test
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_request\_test\_new\_state module
+------------------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_request_test_new_state
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_response module
+-------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_response
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_response\_attributes module
+-------------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_response_attributes
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_response\_data module
+-------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_response_data
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_response\_data\_type module
+-------------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_response_data_type
+   :members:
+   :show-inheritance:
+
+datadog\_api\_client.v2.model.update\_flaky\_test\_state\_response\_result module
+---------------------------------------------------------------------------------
+
+.. automodule:: datadog_api_client.v2.model.update_flaky_test_state_response_result
+   :members:
+   :show-inheritance:
+
 datadog\_api\_client.v2.model.update\_on\_call\_notification\_rule\_request module
 ----------------------------------------------------------------------------------
 

examples/v2/test-optimization/UpdateFlakyTestState.py

@@ -0,0 +1,36 @@
+"""
+Update flaky test states returns "OK" response
+"""
+
+from datadog_api_client import ApiClient, Configuration
+from datadog_api_client.v2.api.test_optimization_api import TestOptimizationApi
+from datadog_api_client.v2.model.update_flaky_test_state_request import UpdateFlakyTestStateRequest
+from datadog_api_client.v2.model.update_flaky_test_state_request_attributes import UpdateFlakyTestStateRequestAttributes
+from datadog_api_client.v2.model.update_flaky_test_state_request_data import UpdateFlakyTestStateRequestData
+from datadog_api_client.v2.model.update_flaky_test_state_request_data_type import UpdateFlakyTestStateRequestDataType
+from datadog_api_client.v2.model.update_flaky_test_state_request_test import UpdateFlakyTestStateRequestTest
+from datadog_api_client.v2.model.update_flaky_test_state_request_test_new_state import (
+    UpdateFlakyTestStateRequestTestNewState,
+)
+
+body = UpdateFlakyTestStateRequest(
+    data=UpdateFlakyTestStateRequestData(
+        attributes=UpdateFlakyTestStateRequestAttributes(
+            tests=[
+                UpdateFlakyTestStateRequestTest(
+                    id="4eb1887a8adb1847",
+                    new_state=UpdateFlakyTestStateRequestTestNewState.ACTIVE,
+                ),
+            ],
+        ),
+        type=UpdateFlakyTestStateRequestDataType.UPDATE_FLAKY_TEST_STATE_REQUEST,
+    ),
+)
+
+configuration = Configuration()
+configuration.unstable_operations["update_flaky_test_state"] = True
+with ApiClient(configuration) as api_client:
+    api_instance = TestOptimizationApi(api_client)
+    response = api_instance.update_flaky_test_state(body=body)
+
+    print(response)

src/datadog_api_client/configuration.py

@@ -415,6 +415,7 @@ def __init__(
                 "v2.list_incident_teams": False,
                 "v2.update_incident_team": False,
                 "v2.search_flaky_tests": False,
+                "v2.update_flaky_test_state": False,
             }
         )
 

src/datadog_api_client/v2/api/test_optimization_api.py

@@ -14,6 +14,8 @@
     UnsetType,
     unset,
 )
+from datadog_api_client.v2.model.update_flaky_test_state_response import UpdateFlakyTestStateResponse
+from datadog_api_client.v2.model.update_flaky_test_state_request import UpdateFlakyTestStateRequest
 from datadog_api_client.v2.model.flaky_tests_search_response import FlakyTestsSearchResponse
 from datadog_api_client.v2.model.flaky_tests_search_request import FlakyTestsSearchRequest
 from datadog_api_client.v2.model.flaky_test import FlakyTest
@@ -48,6 +50,26 @@ def __init__(self, api_client=None):
             api_client=api_client,
         )
 
+        self._update_flaky_test_state_endpoint = _Endpoint(
+            settings={
+                "response_type": (UpdateFlakyTestStateResponse,),
+                "auth": ["apiKeyAuth", "appKeyAuth", "AuthZ"],
+                "endpoint_path": "/api/v2/test/flaky-test-management/tests",
+                "operation_id": "update_flaky_test_state",
+                "http_method": "PATCH",
+                "version": "v2",
+            },
+            params_map={
+                "body": {
+                    "required": True,
+                    "openapi_types": (UpdateFlakyTestStateRequest,),
+                    "location": "body",
+                },
+            },
+            headers_map={"accept": ["application/json"], "content_type": ["application/json"]},
+            api_client=api_client,
+        )
+
     def search_flaky_tests(
         self,
         *,
@@ -96,3 +118,19 @@ def search_flaky_tests_with_pagination(
             "kwargs": kwargs,
         }
         return endpoint.call_with_http_info_paginated(pagination)
+
+    def update_flaky_test_state(
+        self,
+        body: UpdateFlakyTestStateRequest,
+    ) -> UpdateFlakyTestStateResponse:
+        """Update flaky test states.
+
+        Update the state of multiple flaky tests in Flaky Test Management.
+
+        :type body: UpdateFlakyTestStateRequest
+        :rtype: UpdateFlakyTestStateResponse
+        """
+        kwargs: Dict[str, Any] = {}
+        kwargs["body"] = body
+
+        return self._update_flaky_test_state_endpoint.call_with_http_info(**kwargs)