Sweeper: Add sweeper for certificates.

This commit is contained in:
Andrew Starr-Bochicchio 2018-11-28 18:15:53 -05:00
parent d07036cfa1
commit 5d01382b9a
1 changed files with 36 additions and 0 deletions

View File

@ -9,6 +9,7 @@ import (
"crypto/x509/pkix" "crypto/x509/pkix"
"encoding/pem" "encoding/pem"
"fmt" "fmt"
"log"
"math/big" "math/big"
"strings" "strings"
"testing" "testing"
@ -20,6 +21,41 @@ import (
"github.com/hashicorp/terraform/terraform" "github.com/hashicorp/terraform/terraform"
) )
func init() {
resource.AddTestSweepers("digitalocean_certificate", &resource.Sweeper{
Name: "digitalocean_certificate",
F: testSweepCertificate,
})
}
func testSweepCertificate(region string) error {
meta, err := sharedConfigForRegion(region)
if err != nil {
return err
}
client := meta.(*godo.Client)
opt := &godo.ListOptions{PerPage: 200}
certs, _, err := client.Certificates.List(context.Background(), opt)
if err != nil {
return err
}
for _, c := range certs {
if strings.HasPrefix(c.Name, "certificate-") {
log.Printf("Destroying certificate %s", c.Name)
if _, err := client.Certificates.Delete(context.Background(), c.ID); err != nil {
return err
}
}
}
return nil
}
func TestAccDigitalOceanCertificate_Basic(t *testing.T) { func TestAccDigitalOceanCertificate_Basic(t *testing.T) {
var cert godo.Certificate var cert godo.Certificate
rInt := acctest.RandInt() rInt := acctest.RandInt()