resource_tc_cfs_file_system_test.go

package cfs_test

import (
	"context"
	"fmt"
	"log"
	"testing"
	"time"

	tcacctest "github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/acctest"
	tccommon "github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/common"
	localcfs "github.com/tencentcloudstack/terraform-provider-tencentcloud/tencentcloud/services/cfs"

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

func init() {
	// go test -v ./tencentcloud -sweep=ap-guangzhou -sweep-run=tencentcloud_cfs_file_system
	resource.AddTestSweepers("tencentcloud_cfs_file_system", &resource.Sweeper{
		Name: "tencentcloud_cfs_file_system",
		F: func(r string) error {
			logId := tccommon.GetLogId(tccommon.ContextNil)
			ctx := context.WithValue(context.TODO(), tccommon.LogIdKey, logId)
			cli, _ := tcacctest.SharedClientForRegion(r)
			client := cli.(tccommon.ProviderMeta).GetAPIV3Conn()

			service := localcfs.NewCfsService(client)

			fsList, err := service.DescribeFileSystem(ctx, "", "", "")
			if err != nil {
				return err
			}

			// add scanning resources
			var resources, nonKeepResources []*tccommon.ResourceInstance
			for _, v := range fsList {
				if !tccommon.CheckResourcePersist(*v.FileSystemId, *v.CreationTime) {
					nonKeepResources = append(nonKeepResources, &tccommon.ResourceInstance{
						Id:   *v.FileSystemId,
						Name: *v.FsName,
					})
				}
				resources = append(resources, &tccommon.ResourceInstance{
					Id:         *v.FileSystemId,
					Name:       *v.FsName,
					CreateTime: *v.CreationTime,
				})
			}
			tccommon.ProcessScanCloudResources(client, resources, nonKeepResources, "CreateCfsFileSystem")

			for i := range fsList {
				item := fsList[i]
				id := *item.FileSystemId
				name := *item.FsName
				created := time.Time{}
				if item.CreationTime != nil {
					if result, err := time.Parse(time.RFC3339, *item.CreationTime); err != nil {
						created = result
					}
				}
				if tcacctest.IsResourcePersist(name, &created) {
					continue
				}
				log.Printf("%s -> %s will be sweep", id, name)
				err = service.DeleteFileSystem(ctx, id)
				if err != nil {
					continue
				}
			}

			return nil
		},
	})
}

func TestAccTencentCloudCfsFileSystemResource_Basic(t *testing.T) {
	t.Parallel()
	resource.Test(t, resource.TestCase{
		PreCheck:     func() { tcacctest.AccPreCheck(t) },
		Providers:    tcacctest.AccProviders,
		CheckDestroy: testAccCheckCfsFileSystemDestroy,
		Steps: []resource.TestStep{
			{
				Config: testAccCfsFileSystem,
				Check: resource.ComposeTestCheckFunc(
					testAccCheckCfsFileSystemExists("tencentcloud_cfs_file_system.foo"),
					resource.TestCheckResourceAttr("tencentcloud_cfs_file_system.foo", "name", "test_cfs_file_system"),
					resource.TestCheckResourceAttr("tencentcloud_cfs_file_system.foo", "availability_zone", "ap-guangzhou-3"),
					resource.TestCheckResourceAttrSet("tencentcloud_cfs_file_system.foo", "access_group_id"),
					resource.TestCheckResourceAttr("tencentcloud_cfs_file_system.foo", "protocol", "NFS"),
				),
			},
			// add tag
			{
				Config: testAccCfsMasterInstance_multiTags("master"),
				Check: resource.ComposeAggregateTestCheckFunc(
					testAccCheckCfsFileSystemExists("tencentcloud_cfs_file_system.foo"),
					resource.TestCheckResourceAttr("tencentcloud_cfs_file_system.foo", "tags.role", "master"),
				),
			},
			// update tag
			{
				Config: testAccCfsMasterInstance_multiTags("master-version2"),
				Check: resource.ComposeAggregateTestCheckFunc(
					testAccCheckCfsFileSystemExists("tencentcloud_cfs_file_system.foo"),
					resource.TestCheckResourceAttr("tencentcloud_cfs_file_system.foo", "tags.role", "master-version2"),
				),
			},
			// remove tag
			{
				Config: testAccCfsFileSystem,
				Check: resource.ComposeAggregateTestCheckFunc(
					testAccCheckCfsFileSystemExists("tencentcloud_cfs_file_system.foo"),
					resource.TestCheckNoResourceAttr("tencentcloud_cfs_file_system.foo", "tags.role"),
				),
			},
		},
	})
}

func testAccCheckCfsFileSystemDestroy(s *terraform.State) error {
	logId := tccommon.GetLogId(tccommon.ContextNil)
	ctx := context.WithValue(context.TODO(), tccommon.LogIdKey, logId)
	cfsService := localcfs.NewCfsService(tcacctest.AccProvider.Meta().(tccommon.ProviderMeta).GetAPIV3Conn())
	for _, rs := range s.RootModule().Resources {
		if rs.Type != "tencentcloud_cfs_file_system" {
			continue
		}

		fileSystems, err := cfsService.DescribeFileSystem(ctx, rs.Primary.ID, "", "")
		if err != nil {
			err = resource.Retry(tccommon.ReadRetryTimeout, func() *resource.RetryError {
				fileSystems, err = cfsService.DescribeFileSystem(ctx, rs.Primary.ID, "", "")
				if err != nil {
					return tccommon.RetryError(err)
				}
				return nil
			})
		}
		if err != nil {
			return err
		}
		if len(fileSystems) > 0 {
			return fmt.Errorf("cfs file system still exist: %s", rs.Primary.ID)
		}
	}
	return nil
}

func testAccCheckCfsFileSystemExists(n 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[n]
		if !ok {
			return fmt.Errorf("cfs file system %s is not found", n)
		}
		if rs.Primary.ID == "" {
			return fmt.Errorf("cfs file system id is not set")
		}
		cfsService := localcfs.NewCfsService(tcacctest.AccProvider.Meta().(tccommon.ProviderMeta).GetAPIV3Conn())
		fileSystems, err := cfsService.DescribeFileSystem(ctx, rs.Primary.ID, "", "")
		if err != nil {
			err = resource.Retry(tccommon.ReadRetryTimeout, func() *resource.RetryError {
				fileSystems, err = cfsService.DescribeFileSystem(ctx, rs.Primary.ID, "", "")
				if err != nil {
					return tccommon.RetryError(err)
				}
				return nil
			})
		}
		if err != nil {
			return err
		}
		if len(fileSystems) < 1 {
			return fmt.Errorf("cfs file system is not found")
		}
		return nil
	}
}

const testAccCfsFileSystem = DefaultCfsAccessGroup + `
resource "tencentcloud_vpc" "vpc" {
  name       = "test-cfs-vpc"
  cidr_block = "10.2.0.0/16"
}

resource "tencentcloud_subnet" "subnet" {
  vpc_id            = tencentcloud_vpc.vpc.id
  name              = "test-cfs-subnet"
  cidr_block        = "10.2.11.0/24"
  availability_zone = "ap-guangzhou-3"
}

resource "tencentcloud_cfs_file_system" "foo" {
  name = "test_cfs_file_system"
  availability_zone = "ap-guangzhou-3"
  access_group_id = local.cfs_access_group_id
  protocol = "NFS"
  vpc_id = tencentcloud_vpc.vpc.id
  subnet_id = tencentcloud_subnet.subnet.id
  storage_type = "SD"
}
`

func testAccCfsMasterInstance_multiTags(value string) string {
	return fmt.Sprintf(
		`
%s
resource "tencentcloud_vpc" "vpc" {
  name       = "test-cfs-vpc"
  cidr_block = "10.2.0.0/16"
}

resource "tencentcloud_subnet" "subnet" {
  vpc_id            = tencentcloud_vpc.vpc.id
  name              = "test-cfs-subnet"
  cidr_block        = "10.2.11.0/24"
  availability_zone = "ap-guangzhou-3"
}


resource "tencentcloud_cfs_file_system" "foo" {
  name = "test_cfs_file_system"
  availability_zone = "ap-guangzhou-3"
  access_group_id = local.cfs_access_group_id
  protocol = "NFS"
  vpc_id = tencentcloud_vpc.vpc.id
  subnet_id = tencentcloud_subnet.subnet.id
  storage_type = "SD"
  
  tags = {
	  test = "test-tf"
	  role = "%s"
  }
}
`, DefaultCfsAccessGroup, value)
}