@@ -21,6 +21,8 @@ import (
2121 ec2types "github.com/aws/aws-sdk-go-v2/service/ec2/types"
2222 "github.com/aws/aws-sdk-go-v2/service/inspector2"
2323 inspector2types "github.com/aws/aws-sdk-go-v2/service/inspector2/types"
24+ "github.com/aws/aws-sdk-go-v2/service/ssoadmin"
25+ ssoadmintypes "github.com/aws/aws-sdk-go-v2/service/ssoadmin/types"
2426 "github.com/aws/aws-sdk-go/aws"
2527 "github.com/aws/aws-sdk-go/aws/arn"
2628 "github.com/aws/aws-sdk-go/aws/endpoints"
@@ -29,7 +31,6 @@ import (
2931 "github.com/aws/aws-sdk-go/service/ec2"
3032 "github.com/aws/aws-sdk-go/service/iam"
3133 "github.com/aws/aws-sdk-go/service/outposts"
32- "github.com/aws/aws-sdk-go/service/ssoadmin"
3334 "github.com/aws/aws-sdk-go/service/wafv2"
3435 "github.com/hashicorp/aws-sdk-go-base/v2/awsv1shim/v2/tfawserr"
3536 "github.com/hashicorp/terraform-plugin-go/tfprotov5"
@@ -1029,31 +1030,28 @@ func PreCheckOrganizationMemberAccount(ctx context.Context, t *testing.T) {
10291030}
10301031
10311032func PreCheckSSOAdminInstances (ctx context.Context , t * testing.T ) {
1032- conn := Provider .Meta ().(* conns.AWSClient ).SSOAdminConn (ctx )
1033+ conn := Provider .Meta ().(* conns.AWSClient ).SSOAdminClient (ctx )
10331034 input := & ssoadmin.ListInstancesInput {}
1034- var instances []* ssoadmin .InstanceMetadata
1035+ var instances []ssoadmintypes .InstanceMetadata
10351036
1036- err := conn .ListInstancesPagesWithContext (ctx , input , func (page * ssoadmin.ListInstancesOutput , lastPage bool ) bool {
1037- if page == nil {
1038- return ! lastPage
1037+ paginator := ssoadmin .NewListInstancesPaginator (conn , input )
1038+ for paginator .HasMorePages () {
1039+ page , err := paginator .NextPage (ctx )
1040+ if PreCheckSkipError (err ) {
1041+ t .Skipf ("skipping tests: %s" , err )
1042+ }
1043+ if err != nil {
1044+ t .Fatalf ("listing SSO Instances: %s" , err )
10391045 }
10401046
1041- instances = append (instances , page .Instances ... )
1042-
1043- return ! lastPage
1044- })
1045-
1046- if PreCheckSkipError (err ) {
1047- t .Skipf ("skipping tests: %s" , err )
1047+ if page != nil {
1048+ instances = append (instances , page .Instances ... )
1049+ }
10481050 }
10491051
10501052 if len (instances ) == 0 {
10511053 t .Skip ("skipping tests; no SSO Instances found." )
10521054 }
1053-
1054- if err != nil {
1055- t .Fatalf ("listing SSO Instances: %s" , err )
1056- }
10571055}
10581056
10591057func PreCheckHasIAMRole (ctx context.Context , t * testing.T , roleName string ) {
0 commit comments