Skip to content

Commit

Permalink
[datadog_security_monitoring_rules_test] Add pagination to datasource…
Browse files Browse the repository at this point in the history
… test (#2114)

* add pagination to security monitoring rules test

* record cassettes
  • Loading branch information
nkzou authored Sep 20, 2023
1 parent a8af4a3 commit d788a6c
Show file tree
Hide file tree
Showing 3 changed files with 1,571 additions and 267 deletions.
Original file line number Diff line number Diff line change
@@ -1 +1 @@
2023-03-17T18:50:36.472398-04:00
2023-09-20T11:19:41.958037-04:00
1,814 changes: 1,553 additions & 261 deletions datadog/tests/cassettes/TestAccDatadogSecurityMonitoringRuleDatasource.yaml

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
Expand Up @@ -206,13 +206,25 @@ func getAllSecurityMonitoringRules(accProvider func() (*schema.Provider, error))
auth := providerConf.Auth
apiInstances := providerConf.DatadogApiInstances

rulesResponse, _, err := apiInstances.GetSecurityMonitoringApiV2().ListSecurityMonitoringRules(auth,
*datadogV2.NewListSecurityMonitoringRulesOptionalParameters().WithPageSize(1000))
if err != nil {
return err
pageSize := int64(1000)
pageNumber := int64(0)
remaining := int64(1)

var rules []datadogV2.SecurityMonitoringRuleResponse
for remaining > int64(0) {
rulesResponse, _, err := apiInstances.GetSecurityMonitoringApiV2().ListSecurityMonitoringRules(auth,
*datadogV2.NewListSecurityMonitoringRulesOptionalParameters().WithPageSize(pageSize).WithPageNumber(pageNumber))
if err != nil {
return err
}

rules = append(rules, rulesResponse.GetData()...)

remaining = rulesResponse.Meta.Page.GetTotalCount() - pageSize*(pageNumber+1)
pageNumber++
}

allRules = &rulesResponse.Data
allRules = &rules
return nil
}

Expand Down

0 comments on commit d788a6c

Please sign in to comment.