From ba195a74dd094787940502a76346fa99630603d2 Mon Sep 17 00:00:00 2001
From: kterada0509 <kterada.0509sg@gmail.com>
Date: Tue, 12 Feb 2019 20:11:31 +0900
Subject: [PATCH] Add tests for aws_media_store_container_cors_policy resource

---
 ..._media_store_container_cors_policy_test.go | 182 ++++++++++++++++++
 1 file changed, 182 insertions(+)
 create mode 100644 aws/resource_aws_media_store_container_cors_policy_test.go

diff --git a/aws/resource_aws_media_store_container_cors_policy_test.go b/aws/resource_aws_media_store_container_cors_policy_test.go
new file mode 100644
index 00000000000..07866d2647b
--- /dev/null
+++ b/aws/resource_aws_media_store_container_cors_policy_test.go
@@ -0,0 +1,182 @@
+package aws
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/service/mediastore"
+	"github.com/hashicorp/terraform/helper/acctest"
+	"github.com/hashicorp/terraform/helper/resource"
+	"github.com/hashicorp/terraform/terraform"
+)
+
+func TestAccAWSMediaStoreContainerCorsPolicy_basic(t *testing.T) {
+	resourceName := "aws_media_store_container_cors_policy.test"
+	rName := acctest.RandString(5)
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckAwsMediaStoreContainerCorsPolicyDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccMediaStoreContainerCorsPolicyConfig(rName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckAwsMediaStoreContainerCorsPolicyExists(resourceName),
+					resource.TestCheckResourceAttrPair(
+						resourceName, "container_name",
+						"aws_media_store_container.test", "name"),
+				),
+			},
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}
+
+func TestAccAWSMediaStoreContainerCorsPolicy_optional(t *testing.T) {
+	resourceName := "aws_media_store_container_cors_policy.test"
+	rName := acctest.RandString(5)
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckAwsMediaStoreContainerCorsPolicyDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccMediaStoreContainerCorsPolicyConfig_optional(rName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckAwsMediaStoreContainerCorsPolicyExists(resourceName),
+				),
+			},
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}
+
+func TestAccAWSMediaStoreContainerCorsPolicy_update(t *testing.T) {
+	resourceName := "aws_media_store_container_cors_policy.test"
+	rName := acctest.RandString(5)
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckAwsMediaStoreContainerCorsPolicyDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccMediaStoreContainerCorsPolicyConfig(rName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckAwsMediaStoreContainerCorsPolicyExists(resourceName),
+				),
+			},
+			{
+				Config: testAccMediaStoreContainerCorsPolicyConfig_update(rName),
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckAwsMediaStoreContainerCorsPolicyExists(resourceName),
+				),
+			},
+			{
+				ResourceName:      resourceName,
+				ImportState:       true,
+				ImportStateVerify: true,
+			},
+		},
+	})
+}
+
+func testAccCheckAwsMediaStoreContainerCorsPolicyDestroy(s *terraform.State) error {
+	conn := testAccProvider.Meta().(*AWSClient).mediastoreconn
+
+	for _, rs := range s.RootModule().Resources {
+		if rs.Type != "aws_media_store_container_cors_policy" {
+			continue
+		}
+
+		input := &mediastore.GetCorsPolicyInput{
+			ContainerName: aws.String(rs.Primary.ID),
+		}
+
+		_, err := conn.GetCorsPolicy(input)
+		if err != nil {
+			if isAWSErr(err, mediastore.ErrCodeCorsPolicyNotFoundException, "") || isAWSErr(err, mediastore.ErrCodeContainerNotFoundException, "") {
+				return nil
+			}
+			return err
+		}
+		return fmt.Errorf("Media Store Container Cors Policy(%s) is stil exists.", rs.Primary.ID)
+	}
+	return nil
+}
+
+func testAccCheckAwsMediaStoreContainerCorsPolicyExists(name string) resource.TestCheckFunc {
+	return func(s *terraform.State) error {
+		rs, ok := s.RootModule().Resources[name]
+		if !ok {
+			return fmt.Errorf("Not found: %s", name)
+		}
+
+		conn := testAccProvider.Meta().(*AWSClient).mediastoreconn
+
+		input := &mediastore.GetCorsPolicyInput{
+			ContainerName: aws.String(rs.Primary.ID),
+		}
+
+		_, err := conn.GetCorsPolicy(input)
+
+		return err
+	}
+}
+
+func testAccMediaStoreContainerCorsPolicyConfig(rName string) string {
+	return fmt.Sprintf(`
+%s
+
+resource "aws_media_store_container_cors_policy" "test" {
+	container_name = "${aws_media_store_container.test.name}"
+
+	cors_policy = {
+		allowed_headers = ["*"]
+		allowed_methods = ["GET"]
+		allowed_origins = ["*"]
+	}
+}`, testAccMediaStoreContainerConfig(rName))
+}
+
+func testAccMediaStoreContainerCorsPolicyConfig_optional(rName string) string {
+	return fmt.Sprintf(`
+%s
+
+resource "aws_media_store_container_cors_policy" "test" {
+	container_name = "${aws_media_store_container.test.name}"
+
+	cors_policy = {
+		allowed_headers = ["*"]
+		allowed_methods = ["GET", "HEAD"]
+		allowed_origins = ["http://aaa.example.com"]
+		expose_headers 	= ["*"]
+		max_age_seconds = 0
+	}
+}`, testAccMediaStoreContainerConfig(rName))
+}
+
+func testAccMediaStoreContainerCorsPolicyConfig_update(rName string) string {
+	return fmt.Sprintf(`
+%s
+
+resource "aws_media_store_container_cors_policy" "test" {
+	container_name = "${aws_media_store_container.test.name}"
+
+	cors_policy = {
+		allowed_headers = ["Access-Control-Request-Headers"]
+		allowed_methods = ["PUT", "GET"]
+		allowed_origins = ["http://aaa.example.com", "http://bbb.example.com"]
+		expose_headers 	= ["XMLHttpRequest"]
+		max_age_seconds = 3600
+	}
+}`, testAccMediaStoreContainerConfig(rName))
+}