Add sweeper for VPCs. (#942)

* Add sweeper for VPCs.

* Update sweep.go

digitalocean/sweep/sweep_test.go

@@ -16,6 +16,7 @@ import (
 	_ "github.com/digitalocean/terraform-provider-digitalocean/digitalocean/snapshot"
 	_ "github.com/digitalocean/terraform-provider-digitalocean/digitalocean/sshkey"
 	_ "github.com/digitalocean/terraform-provider-digitalocean/digitalocean/volume"
+	_ "github.com/digitalocean/terraform-provider-digitalocean/digitalocean/vpc"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
 )

digitalocean/vpc/datasource_vpc_test.go

@@ -101,7 +101,7 @@ resource "digitalocean_vpc" "foobar" {
 const testAccCheckDataSourceDigitalOceanVPCConfig_RegionDefault = `
 // Create Droplet to ensure default VPC exists
 resource "digitalocean_droplet" "foo" {
-  image              = "ubuntu-18-04-x64"
+  image              = "ubuntu-22-04-x64"
   name               = "%s"
   region             = "nyc3"
   size               = "s-1vcpu-1gb"

digitalocean/vpc/resource_vpc.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"log"
+	"net/http"
 	"time"
 
 	"github.com/digitalocean/godo"
@@ -169,7 +170,7 @@ func resourceDigitalOceanVPCDelete(ctx context.Context, d *schema.ResourceData,
 		resp, err := client.VPCs.Delete(context.Background(), vpcID)
 		if err != nil {
 			// Retry if VPC still contains member resources to prevent race condition
-			if resp.StatusCode == 403 {
+			if resp.StatusCode == http.StatusForbidden {
 				return resource.RetryableError(err)
 			} else {
 				return resource.NonRetryableError(fmt.Errorf("Error deleting VPC: %s", err))

digitalocean/vpc/sweep.go

@@ -0,0 +1,54 @@
+package vpc
+
+import (
+	"context"
+	"log"
+	"net/http"
+	"strings"
+
+	"github.com/digitalocean/godo"
+	"github.com/digitalocean/terraform-provider-digitalocean/digitalocean/config"
+	"github.com/digitalocean/terraform-provider-digitalocean/digitalocean/sweep"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+)
+
+func init() {
+	resource.AddTestSweepers("digitalocean_vpc", &resource.Sweeper{
+		Name: "digitalocean_vpc",
+		F:    sweepVPC,
+		Dependencies: []string{
+			"digitalocean_droplet",
+		},
+	})
+}
+
+func sweepVPC(region string) error {
+	meta, err := sweep.SharedConfigForRegion(region)
+	if err != nil {
+		return err
+	}
+
+	client := meta.(*config.CombinedConfig).GodoClient()
+
+	opt := &godo.ListOptions{PerPage: 200}
+	vpcs, _, err := client.VPCs.List(context.Background(), opt)
+	if err != nil {
+		return err
+	}
+
+	for _, v := range vpcs {
+		if strings.HasPrefix(v.Name, sweep.TestNamePrefix) {
+			log.Printf("[DEBUG] Destroying VPC %s", v.Name)
+			resp, err := client.VPCs.Delete(context.Background(), v.ID)
+			if err != nil {
+				if resp.StatusCode == http.StatusForbidden {
+					log.Printf("[DEBUG] Skipping VPC %s; still contains resources", v.Name)
+				} else {
+					return err
+				}
+			}
+		}
+	}
+
+	return nil
+}