Skip to content

Commit

Permalink
fix(organization_project): added sweeper
Browse files Browse the repository at this point in the history
  • Loading branch information
vmyroslav committed Feb 25, 2025
1 parent d389170 commit faa6723
Showing 1 changed file with 46 additions and 0 deletions.
46 changes: 46 additions & 0 deletions internal/sdkprovider/service/organization/sweep.go
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,11 @@ const defaultPrefix = "test-acc"
func init() {
ctx := context.Background()

sweep.AddTestSweepers("aiven_organization_project", &resource.Sweeper{
Name: "aiven_organization_project",
F: sweepOrganizationProjects(ctx),
})

sweep.AddTestSweepers("aiven_organization", &resource.Sweeper{
Name: "aiven_organization",
F: sweepOrganizations(ctx),
Expand Down Expand Up @@ -78,6 +83,47 @@ func sweepOrganizations(ctx context.Context) func(string) error {
}
}

func sweepOrganizationProjects(ctx context.Context) func(string) error {
return func(_ string) error {
client, err := sweep.SharedGenClient()
if err != nil {
return err
}

organizations, err := client.AccountList(ctx)
if common.IsCritical(err) {
return fmt.Errorf("error retrieving a list of organizations: %w", err)
}

if organizations == nil {
return nil
}

for _, organization := range organizations {
if !strings.HasPrefix(organization.AccountName, "test-acc") {
continue
}

projects, err := client.OrganizationProjectsList(ctx, organization.OrganizationId)
if common.IsCritical(err) {
return fmt.Errorf("error retrieving a list of projects: %w", err)
}

for _, project := range projects.Projects {
if !strings.HasPrefix(project.ProjectId, "test-acc") {
continue
}

if err = client.OrganizationProjectsDelete(ctx, organization.OrganizationId, project.ProjectId); common.IsCritical(err) {
return fmt.Errorf("error deleting project %s: %w", project.ProjectId, err)
}
}
}

return nil
}
}

func sweepOrganizationApplicationUsers(ctx context.Context) func(string) error {
return func(id string) error {
client, err := sweep.SharedClient()
Expand Down

0 comments on commit faa6723

Please sign in to comment.