@@ -96,6 +96,95 @@ func (a *TestOptimizationApi) DeleteTestOptimizationServiceSettings(ctx _context
9696 return localVarHTTPResponse , nil
9797}
9898
99+ // GetFlakyTestsManagementPolicies Get Flaky Tests Management policies.
100+ // Retrieve Flaky Tests Management repository-level policies for the given repository.
101+ func (a * TestOptimizationApi ) GetFlakyTestsManagementPolicies (ctx _context.Context , body TestOptimizationFlakyTestsManagementPoliciesGetRequest ) (TestOptimizationFlakyTestsManagementPoliciesResponse , * _nethttp.Response , error ) {
102+ var (
103+ localVarHTTPMethod = _nethttp .MethodPost
104+ localVarPostBody interface {}
105+ localVarReturnValue TestOptimizationFlakyTestsManagementPoliciesResponse
106+ )
107+
108+ operationId := "v2.GetFlakyTestsManagementPolicies"
109+ isOperationEnabled := a .Client .Cfg .IsUnstableOperationEnabled (operationId )
110+ if ! isOperationEnabled {
111+ return localVarReturnValue , nil , datadog.GenericOpenAPIError {ErrorMessage : _fmt .Sprintf ("Unstable operation '%s' is disabled" , operationId )}
112+ }
113+ if isOperationEnabled && a .Client .Cfg .Debug {
114+ _log .Printf ("WARNING: Using unstable operation '%s'" , operationId )
115+ }
116+
117+ localBasePath , err := a .Client .Cfg .ServerURLWithContext (ctx , "v2.TestOptimizationApi.GetFlakyTestsManagementPolicies" )
118+ if err != nil {
119+ return localVarReturnValue , nil , datadog.GenericOpenAPIError {ErrorMessage : err .Error ()}
120+ }
121+
122+ localVarPath := localBasePath + "/api/v2/ci/test-optimization/settings/policies"
123+
124+ localVarHeaderParams := make (map [string ]string )
125+ localVarQueryParams := _neturl.Values {}
126+ localVarFormParams := _neturl.Values {}
127+ localVarHeaderParams ["Content-Type" ] = "application/json"
128+ localVarHeaderParams ["Accept" ] = "application/json"
129+
130+ // body params
131+ localVarPostBody = & body
132+ if a .Client .Cfg .DelegatedTokenConfig != nil {
133+ err = datadog .UseDelegatedTokenAuth (ctx , & localVarHeaderParams , a .Client .Cfg .DelegatedTokenConfig )
134+ if err != nil {
135+ return localVarReturnValue , nil , err
136+ }
137+ } else {
138+ datadog .SetAuthKeys (
139+ ctx ,
140+ & localVarHeaderParams ,
141+ [2 ]string {"apiKeyAuth" , "DD-API-KEY" },
142+ [2 ]string {"appKeyAuth" , "DD-APPLICATION-KEY" },
143+ )
144+ }
145+ req , err := a .Client .PrepareRequest (ctx , localVarPath , localVarHTTPMethod , localVarPostBody , localVarHeaderParams , localVarQueryParams , localVarFormParams , nil )
146+ if err != nil {
147+ return localVarReturnValue , nil , err
148+ }
149+
150+ localVarHTTPResponse , err := a .Client .CallAPI (req )
151+ if err != nil || localVarHTTPResponse == nil {
152+ return localVarReturnValue , localVarHTTPResponse , err
153+ }
154+
155+ localVarBody , err := datadog .ReadBody (localVarHTTPResponse )
156+ if err != nil {
157+ return localVarReturnValue , localVarHTTPResponse , err
158+ }
159+
160+ if localVarHTTPResponse .StatusCode >= 300 {
161+ newErr := datadog.GenericOpenAPIError {
162+ ErrorBody : localVarBody ,
163+ ErrorMessage : localVarHTTPResponse .Status ,
164+ }
165+ if localVarHTTPResponse .StatusCode == 400 || localVarHTTPResponse .StatusCode == 403 || localVarHTTPResponse .StatusCode == 404 || localVarHTTPResponse .StatusCode == 429 {
166+ var v APIErrorResponse
167+ err = a .Client .Decode (& v , localVarBody , localVarHTTPResponse .Header .Get ("Content-Type" ))
168+ if err != nil {
169+ return localVarReturnValue , localVarHTTPResponse , newErr
170+ }
171+ newErr .ErrorModel = v
172+ }
173+ return localVarReturnValue , localVarHTTPResponse , newErr
174+ }
175+
176+ err = a .Client .Decode (& localVarReturnValue , localVarBody , localVarHTTPResponse .Header .Get ("Content-Type" ))
177+ if err != nil {
178+ newErr := datadog.GenericOpenAPIError {
179+ ErrorBody : localVarBody ,
180+ ErrorMessage : err .Error (),
181+ }
182+ return localVarReturnValue , localVarHTTPResponse , newErr
183+ }
184+
185+ return localVarReturnValue , localVarHTTPResponse , nil
186+ }
187+
99188// GetTestOptimizationServiceSettings Get Test Optimization service settings.
100189// Retrieve Test Optimization settings for a specific service identified by repository, service name, and environment.
101190func (a * TestOptimizationApi ) GetTestOptimizationServiceSettings (ctx _context.Context , body TestOptimizationGetServiceSettingsRequest ) (TestOptimizationServiceSettingsResponse , * _nethttp.Response , error ) {
@@ -473,6 +562,96 @@ func (a *TestOptimizationApi) UpdateFlakyTests(ctx _context.Context, body Update
473562 return localVarReturnValue , localVarHTTPResponse , nil
474563}
475564
565+ // UpdateFlakyTestsManagementPolicies Update Flaky Tests Management policies.
566+ // Partially update Flaky Tests Management repository-level policies for the given repository.
567+ // Only provided policy blocks are updated; omitted blocks are left unchanged.
568+ func (a * TestOptimizationApi ) UpdateFlakyTestsManagementPolicies (ctx _context.Context , body TestOptimizationFlakyTestsManagementPoliciesUpdateRequest ) (TestOptimizationFlakyTestsManagementPoliciesResponse , * _nethttp.Response , error ) {
569+ var (
570+ localVarHTTPMethod = _nethttp .MethodPatch
571+ localVarPostBody interface {}
572+ localVarReturnValue TestOptimizationFlakyTestsManagementPoliciesResponse
573+ )
574+
575+ operationId := "v2.UpdateFlakyTestsManagementPolicies"
576+ isOperationEnabled := a .Client .Cfg .IsUnstableOperationEnabled (operationId )
577+ if ! isOperationEnabled {
578+ return localVarReturnValue , nil , datadog.GenericOpenAPIError {ErrorMessage : _fmt .Sprintf ("Unstable operation '%s' is disabled" , operationId )}
579+ }
580+ if isOperationEnabled && a .Client .Cfg .Debug {
581+ _log .Printf ("WARNING: Using unstable operation '%s'" , operationId )
582+ }
583+
584+ localBasePath , err := a .Client .Cfg .ServerURLWithContext (ctx , "v2.TestOptimizationApi.UpdateFlakyTestsManagementPolicies" )
585+ if err != nil {
586+ return localVarReturnValue , nil , datadog.GenericOpenAPIError {ErrorMessage : err .Error ()}
587+ }
588+
589+ localVarPath := localBasePath + "/api/v2/ci/test-optimization/settings/policies"
590+
591+ localVarHeaderParams := make (map [string ]string )
592+ localVarQueryParams := _neturl.Values {}
593+ localVarFormParams := _neturl.Values {}
594+ localVarHeaderParams ["Content-Type" ] = "application/json"
595+ localVarHeaderParams ["Accept" ] = "application/json"
596+
597+ // body params
598+ localVarPostBody = & body
599+ if a .Client .Cfg .DelegatedTokenConfig != nil {
600+ err = datadog .UseDelegatedTokenAuth (ctx , & localVarHeaderParams , a .Client .Cfg .DelegatedTokenConfig )
601+ if err != nil {
602+ return localVarReturnValue , nil , err
603+ }
604+ } else {
605+ datadog .SetAuthKeys (
606+ ctx ,
607+ & localVarHeaderParams ,
608+ [2 ]string {"apiKeyAuth" , "DD-API-KEY" },
609+ [2 ]string {"appKeyAuth" , "DD-APPLICATION-KEY" },
610+ )
611+ }
612+ req , err := a .Client .PrepareRequest (ctx , localVarPath , localVarHTTPMethod , localVarPostBody , localVarHeaderParams , localVarQueryParams , localVarFormParams , nil )
613+ if err != nil {
614+ return localVarReturnValue , nil , err
615+ }
616+
617+ localVarHTTPResponse , err := a .Client .CallAPI (req )
618+ if err != nil || localVarHTTPResponse == nil {
619+ return localVarReturnValue , localVarHTTPResponse , err
620+ }
621+
622+ localVarBody , err := datadog .ReadBody (localVarHTTPResponse )
623+ if err != nil {
624+ return localVarReturnValue , localVarHTTPResponse , err
625+ }
626+
627+ if localVarHTTPResponse .StatusCode >= 300 {
628+ newErr := datadog.GenericOpenAPIError {
629+ ErrorBody : localVarBody ,
630+ ErrorMessage : localVarHTTPResponse .Status ,
631+ }
632+ if localVarHTTPResponse .StatusCode == 400 || localVarHTTPResponse .StatusCode == 403 || localVarHTTPResponse .StatusCode == 404 || localVarHTTPResponse .StatusCode == 429 {
633+ var v APIErrorResponse
634+ err = a .Client .Decode (& v , localVarBody , localVarHTTPResponse .Header .Get ("Content-Type" ))
635+ if err != nil {
636+ return localVarReturnValue , localVarHTTPResponse , newErr
637+ }
638+ newErr .ErrorModel = v
639+ }
640+ return localVarReturnValue , localVarHTTPResponse , newErr
641+ }
642+
643+ err = a .Client .Decode (& localVarReturnValue , localVarBody , localVarHTTPResponse .Header .Get ("Content-Type" ))
644+ if err != nil {
645+ newErr := datadog.GenericOpenAPIError {
646+ ErrorBody : localVarBody ,
647+ ErrorMessage : err .Error (),
648+ }
649+ return localVarReturnValue , localVarHTTPResponse , newErr
650+ }
651+
652+ return localVarReturnValue , localVarHTTPResponse , nil
653+ }
654+
476655// UpdateTestOptimizationServiceSettings Update Test Optimization service settings.
477656// Partially update Test Optimization settings for a specific service identified by repository, service name, and environment.
478657// Only provided fields are updated; null or omitted fields are left unchanged.
0 commit comments