terraform-provider-greenhost/digitalocean/resource_digitalocean_conta...

99 lines
2.5 KiB
Go

package digitalocean
import (
"context"
"fmt"
"testing"
"github.com/digitalocean/godo"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"
)
func TestAccDigitalOceanContainerRegistry_Basic(t *testing.T) {
var reg godo.Registry
resource.Test(t, resource.TestCase{
PreCheck: func() { testAccPreCheck(t) },
ProviderFactories: testAccProviderFactories,
CheckDestroy: testAccCheckDigitalOceanContainerRegistryDestroy,
Steps: []resource.TestStep{
{
Config: testAccCheckDigitalOceanContainerRegistryConfig_basic,
Check: resource.ComposeTestCheckFunc(
testAccCheckDigitalOceanContainerRegistryExists("digitalocean_container_registry.foobar", &reg),
testAccCheckDigitalOceanContainerRegistryAttributes(&reg),
resource.TestCheckResourceAttr(
"digitalocean_container_registry.foobar", "name", "foobar"),
resource.TestCheckResourceAttr(
"digitalocean_container_registry.foobar", "endpoint", "registry.digitalocean.com/foobar"),
resource.TestCheckResourceAttr(
"digitalocean_container_registry.foobar", "server_url", "registry.digitalocean.com"),
),
},
},
})
}
func testAccCheckDigitalOceanContainerRegistryDestroy(s *terraform.State) error {
client := testAccProvider.Meta().(*CombinedConfig).godoClient()
for _, rs := range s.RootModule().Resources {
if rs.Type != "digitalocean_container_registry" {
continue
}
// Try to find the key
_, _, err := client.Registry.Get(context.Background())
if err == nil {
return fmt.Errorf("Container Registry still exists")
}
}
return nil
}
func testAccCheckDigitalOceanContainerRegistryAttributes(reg *godo.Registry) resource.TestCheckFunc {
return func(s *terraform.State) error {
if reg.Name != "foobar" {
return fmt.Errorf("Bad name: %s", reg.Name)
}
return nil
}
}
func testAccCheckDigitalOceanContainerRegistryExists(n string, reg *godo.Registry) resource.TestCheckFunc {
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[n]
if !ok {
return fmt.Errorf("Not found: %s", n)
}
if rs.Primary.ID == "" {
return fmt.Errorf("No Record ID is set")
}
client := testAccProvider.Meta().(*CombinedConfig).godoClient()
// Try to find the registry
foundReg, _, err := client.Registry.Get(context.Background())
if err != nil {
return err
}
*reg = *foundReg
return nil
}
}
var testAccCheckDigitalOceanContainerRegistryConfig_basic = `
resource "digitalocean_container_registry" "foobar" {
name = "foobar"
}`