resource_tc_teo_acceleration_domain_test.go

package teo_test

import (
	"context"
	"fmt"
	"strings"
	"testing"

	tcacctest "github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/acctest"
	tccommon "github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/common"
	svcteo "github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/services/teo"

	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
	"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"
)

func TestAccTencentCloudTeoAccelerationDomainResource_basic(t *testing.T) {

	resource.Test(t, resource.TestCase{
		PreCheck:     func() { tcacctest.AccPreCheckCommon(t, tcacctest.ACCOUNT_TYPE_PRIVATE) },
		Providers:    tcacctest.AccProviders,
		CheckDestroy: testAccCheckTeoAccelerationDomainDestroy,
		Steps: []resource.TestStep{
			{
				Config: testAccTeoAccelerationDomain,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckTeoAccelerationDomainExists("tencentcloud_teo_acceleration_domain.acceleration_domain"),
					resource.TestCheckResourceAttrSet("tencentcloud_teo_acceleration_domain.acceleration_domain", "id"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "domain_name", "test.tf-teo.xyz"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "origin_info.#", "1"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "origin_info.0.origin", "150.109.8.1"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "origin_info.0.origin_type", "IP_DOMAIN"),
					resource.TestCheckResourceAttrSet("tencentcloud_teo_acceleration_domain.acceleration_domain", "cname"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "origin_protocol", "FOLLOW"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "http_origin_port", "80"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "https_origin_port", "443"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "ipv6_status", "follow"),
				),
			},
			{
				ResourceName:      "tencentcloud_teo_acceleration_domain.acceleration_domain",
				ImportState:       true,
				ImportStateVerify: true,
			},
			{
				Config: testAccTeoAccelerationDomainUp,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckTeoAccelerationDomainExists("tencentcloud_teo_acceleration_domain.acceleration_domain"),
					resource.TestCheckResourceAttrSet("tencentcloud_teo_acceleration_domain.acceleration_domain", "id"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "domain_name", "test.tf-teo.xyz"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "origin_info.#", "1"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "origin_info.0.origin", "150.109.8.1"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "origin_info.0.origin_type", "IP_DOMAIN"),
					resource.TestCheckResourceAttrSet("tencentcloud_teo_acceleration_domain.acceleration_domain", "cname"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "origin_protocol", "HTTP"),
					resource.TestCheckResourceAttr("tencentcloud_teo_acceleration_domain.acceleration_domain", "http_origin_port", "81"),
				),
			},
		},
	})
}

func testAccCheckTeoAccelerationDomainDestroy(s *terraform.State) error {
	logId := tccommon.GetLogId(tccommon.ContextNil)
	ctx := context.WithValue(context.TODO(), tccommon.LogIdKey, logId)
	service := svcteo.NewTeoService(tcacctest.AccProvider.Meta().(tccommon.ProviderMeta).GetAPIV3Conn())
	for _, rs := range s.RootModule().Resources {
		if rs.Type != "tencentcloud_teo_acceleration_domain" {
			continue
		}

		idSplit := strings.Split(rs.Primary.ID, tccommon.FILED_SP)
		if len(idSplit) != 2 {
			return fmt.Errorf("id is broken,%s", rs.Primary.ID)
		}
		zoneId := idSplit[0]
		domainName := idSplit[1]

		agents, err := service.DescribeTeoAccelerationDomainById(ctx, zoneId, domainName)
		if agents != nil {
			return fmt.Errorf("AccelerationDomain %s still exists", rs.Primary.ID)
		}
		if err != nil {
			return err
		}
	}
	return nil
}

func testAccCheckTeoAccelerationDomainExists(r string) resource.TestCheckFunc {
	return func(s *terraform.State) error {
		logId := tccommon.GetLogId(tccommon.ContextNil)
		ctx := context.WithValue(context.TODO(), tccommon.LogIdKey, logId)

		rs, ok := s.RootModule().Resources[r]
		if !ok {
			return fmt.Errorf("resource %s is not found", r)
		}

		idSplit := strings.Split(rs.Primary.ID, tccommon.FILED_SP)
		if len(idSplit) != 2 {
			return fmt.Errorf("id is broken,%s", rs.Primary.ID)
		}
		zoneId := idSplit[0]
		domainName := idSplit[1]

		service := svcteo.NewTeoService(tcacctest.AccProvider.Meta().(tccommon.ProviderMeta).GetAPIV3Conn())
		agents, err := service.DescribeTeoAccelerationDomainById(ctx, zoneId, domainName)
		if agents == nil {
			return fmt.Errorf("AccelerationDomain %s is not found", rs.Primary.ID)
		}
		if err != nil {
			return err
		}

		return nil
	}
}

const testAccTeoAccelerationDomain = testAccTeoZone + `

resource "tencentcloud_teo_ownership_verify" "ownership_verify" {
  domain = var.zone_name

  depends_on = [tencentcloud_teo_zone.basic]
}

resource "tencentcloud_teo_acceleration_domain" "acceleration_domain" {
    zone_id     = tencentcloud_teo_zone.basic.id
    domain_name = "test.tf-teo.xyz"

    origin_info {
        origin      = "150.109.8.1"
        origin_type = "IP_DOMAIN"
    }

	depends_on = [tencentcloud_teo_ownership_verify.ownership_verify]
}

`

const testAccTeoAccelerationDomainUp = testAccTeoZone + `

resource "tencentcloud_teo_acceleration_domain" "acceleration_domain" {
    zone_id     = tencentcloud_teo_zone.basic.id
    domain_name = "test.tf-teo.xyz"

    origin_info {
        origin      = "150.109.8.1"
        origin_type = "IP_DOMAIN"
    }
  	origin_protocol = "HTTP"
  	http_origin_port = 81
}

`