providers/aws: Check that S3 website sets docs

builtin/providers/aws/resource_aws_s3_bucket_test.go

@@ -14,7 +14,6 @@ import (
 )
 
 func TestAccAWSS3Bucket(t *testing.T) {
-
  resource.Test(t, resource.TestCase{
  PreCheck: func() { testAccPreCheck(t) },
  Providers: testAccProviders,
@@ -26,10 +25,22 @@ func TestAccAWSS3Bucket(t *testing.T) {
  testAccCheckAWSS3BucketExists("aws_s3_bucket.bucket"),
  ),
  },
+ },
+ })
+}
+
+func TestAccAWSS3BucketWebsite(t *testing.T) {
+ resource.Test(t, resource.TestCase{
+ PreCheck: func() { testAccPreCheck(t) },
+ Providers: testAccProviders,
+ CheckDestroy: testAccCheckAWSS3BucketDestroy,
+ Steps: []resource.TestStep{
  resource.TestStep{
  Config: testAccAWSS3BucketWebsiteConfig,
  Check: resource.ComposeTestCheckFunc(
  testAccCheckAWSS3BucketExists("aws_s3_bucket.website"),
+ testAccCheckAWSS3BucketWebsite(
+ "aws_s3_bucket.website", "index.html", "error.html"),
  ),
  },
  },
@@ -76,6 +87,31 @@ func testAccCheckAWSS3BucketExists(n string) resource.TestCheckFunc {
  }
 }
 
+func testAccCheckAWSS3BucketWebsite(n string, indexDoc string, errorDoc string) resource.TestCheckFunc {
+ return func(s *terraform.State) error {
+ rs, _ := s.RootModule().Resources[n]
+ conn := testAccProvider.Meta().(*AWSClient).s3conn
+
+ out, err := conn.GetBucketWebsite(&s3.GetBucketWebsiteInput{
+ Bucket: aws.String(rs.Primary.ID),
+ })
+
+ if err != nil {
+ return fmt.Errorf("S3BucketWebsite error: %v", err)
+ }
+
+ if *out.IndexDocument.Suffix != indexDoc {
+ return fmt.Errorf("bad: %s", out.IndexDocument)
+ }
+
+ if *out.ErrorDocument.Key != errorDoc {
+ return fmt.Errorf("bad: %s", out.ErrorDocument)
+ }
+
+ return nil
+ }
+}
+
 // These need a bit of randoness as the name can only be used once globally
 // within AWS
 var testAccAWSS3BucketConfig = fmt.Sprintf(`
@@ -92,6 +128,7 @@ resource "aws_s3_bucket" "website" {
 
  website {
  index_document = "index.html"
+ error_document = "error.html"
  }
 }
 `, rand.New(rand.NewSource(time.Now().UnixNano())).Int())