Sweeper: Add sweeper for domains.

This commit is contained in:
Andrew Starr-Bochicchio 2018-11-28 18:03:44 -05:00
parent 402a92d63e
commit fcf0d80687
2 changed files with 38 additions and 1 deletions

View File

@ -3,6 +3,8 @@ package digitalocean
import (
"context"
"fmt"
"log"
"strings"
"testing"
"github.com/digitalocean/godo"
@ -11,6 +13,41 @@ import (
"github.com/hashicorp/terraform/terraform"
)
func init() {
resource.AddTestSweepers("digitalocean_domain", &resource.Sweeper{
Name: "digitalocean_domain",
F: testSweepDomain,
})
}
func testSweepDomain(region string) error {
meta, err := sharedConfigForRegion(region)
if err != nil {
return err
}
client := meta.(*godo.Client)
opt := &godo.ListOptions{PerPage: 200}
domains, _, err := client.Domains.List(context.Background(), opt)
if err != nil {
return err
}
for _, d := range domains {
if strings.HasPrefix(d.Name, "foobar-") {
log.Printf("Destroying domain %s", d.Name)
if _, err := client.Domains.Delete(context.Background(), d.Name); err != nil {
return err
}
}
}
return nil
}
func TestAccDigitalOceanDomain_Basic(t *testing.T) {
var domain godo.Domain
domainName := fmt.Sprintf("foobar-test-terraform-%s.com", acctest.RandString(10))

View File

@ -175,7 +175,7 @@ resource "digitalocean_droplet" "foobar" {
var testAccCheckDigitalOceanFloatingIPAssignmentConfig_createBeforeDestroy = `
resource "digitalocean_droplet" "foobar" {
image = "centos-7-x64"
name = "foobar"
name = "foo-bar"
region = "nyc3"
size = "s-1vcpu-1gb"
private_networking = false