Skip to content

Commit

Permalink
Merge pull request #22649 from hashicorp/f-s3-bucket-request-payment-…
Browse files Browse the repository at this point in the history
…config-resource

r/s3_bucket_request_payment_configuration: new resource
  • Loading branch information
anGie44 authored Feb 3, 2022
2 parents 53a887a + 8a9587f commit a0434a3
Show file tree
Hide file tree
Showing 5 changed files with 434 additions and 0 deletions.
3 changes: 3 additions & 0 deletions .changelog/22649.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
```release-note:new-resource
aws_s3_bucket_request_payment_configuration
```
1 change: 1 addition & 0 deletions internal/provider/provider.go
Original file line number Diff line number Diff line change
Expand Up @@ -1631,6 +1631,7 @@ func Provider() *schema.Provider {
"aws_s3_bucket_policy": s3.ResourceBucketPolicy(),
"aws_s3_bucket_public_access_block": s3.ResourceBucketPublicAccessBlock(),
"aws_s3_bucket_replication_configuration": s3.ResourceBucketReplicationConfiguration(),
"aws_s3_bucket_request_payment_configuration": s3.ResourceBucketRequestPaymentConfiguration(),
"aws_s3_bucket_versioning": s3.ResourceBucketVersioning(),
"aws_s3_bucket_website_configuration": s3.ResourceBucketWebsiteConfiguration(),
"aws_s3_object": s3.ResourceObject(),
Expand Down
175 changes: 175 additions & 0 deletions internal/service/s3/bucket_request_payment_configuration.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,175 @@
package s3

import (
"context"
"fmt"
"log"

"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/service/s3"
"github.com/hashicorp/aws-sdk-go-base/v2/awsv1shim/v2/tfawserr"
"github.com/hashicorp/terraform-plugin-sdk/v2/diag"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/validation"
"github.com/hashicorp/terraform-provider-aws/internal/conns"
"github.com/hashicorp/terraform-provider-aws/internal/verify"
)

func ResourceBucketRequestPaymentConfiguration() *schema.Resource {
return &schema.Resource{
CreateContext: resourceBucketRequestPaymentConfigurationCreate,
ReadContext: resourceBucketRequestPaymentConfigurationRead,
UpdateContext: resourceBucketRequestPaymentConfigurationUpdate,
DeleteContext: resourceBucketRequestPaymentConfigurationDelete,
Importer: &schema.ResourceImporter{
StateContext: schema.ImportStatePassthroughContext,
},

Schema: map[string]*schema.Schema{
"bucket": {
Type: schema.TypeString,
Required: true,
ForceNew: true,
ValidateFunc: validation.StringLenBetween(1, 63),
},
"expected_bucket_owner": {
Type: schema.TypeString,
Optional: true,
ForceNew: true,
ValidateFunc: verify.ValidAccountID,
},
"payer": {
Type: schema.TypeString,
Required: true,
ValidateFunc: validation.StringInSlice(s3.Payer_Values(), false),
},
},
}
}

func resourceBucketRequestPaymentConfigurationCreate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
conn := meta.(*conns.AWSClient).S3Conn

bucket := d.Get("bucket").(string)
expectedBucketOwner := d.Get("expected_bucket_owner").(string)

input := &s3.PutBucketRequestPaymentInput{
Bucket: aws.String(bucket),
RequestPaymentConfiguration: &s3.RequestPaymentConfiguration{
Payer: aws.String(d.Get("payer").(string)),
},
}

if expectedBucketOwner != "" {
input.ExpectedBucketOwner = aws.String(expectedBucketOwner)
}

_, err := verify.RetryOnAWSCode(s3.ErrCodeNoSuchBucket, func() (interface{}, error) {
return conn.PutBucketRequestPaymentWithContext(ctx, input)
})

if err != nil {
return diag.FromErr(fmt.Errorf("error creating S3 bucket (%s) request payment configuration: %w", bucket, err))
}

d.SetId(CreateResourceID(bucket, expectedBucketOwner))

return resourceBucketRequestPaymentConfigurationRead(ctx, d, meta)
}

func resourceBucketRequestPaymentConfigurationRead(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
conn := meta.(*conns.AWSClient).S3Conn

bucket, expectedBucketOwner, err := ParseResourceID(d.Id())
if err != nil {
return diag.FromErr(err)
}

input := &s3.GetBucketRequestPaymentInput{
Bucket: aws.String(bucket),
}

if expectedBucketOwner != "" {
input.ExpectedBucketOwner = aws.String(expectedBucketOwner)
}

output, err := conn.GetBucketRequestPaymentWithContext(ctx, input)

if !d.IsNewResource() && tfawserr.ErrCodeEquals(err, s3.ErrCodeNoSuchBucket) {
log.Printf("[WARN] S3 Bucket Request Payment Configuration (%s) not found, removing from state", d.Id())
d.SetId("")
return nil
}

if output == nil {
return diag.FromErr(fmt.Errorf("error reading S3 bucket request payment configuration (%s): empty output", d.Id()))
}

d.Set("bucket", bucket)
d.Set("expected_bucket_owner", expectedBucketOwner)
d.Set("payer", output.Payer)

return nil
}

func resourceBucketRequestPaymentConfigurationUpdate(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
conn := meta.(*conns.AWSClient).S3Conn

bucket, expectedBucketOwner, err := ParseResourceID(d.Id())
if err != nil {
return diag.FromErr(err)
}

input := &s3.PutBucketRequestPaymentInput{
Bucket: aws.String(bucket),
RequestPaymentConfiguration: &s3.RequestPaymentConfiguration{
Payer: aws.String(d.Get("payer").(string)),
},
}

if expectedBucketOwner != "" {
input.ExpectedBucketOwner = aws.String(expectedBucketOwner)
}

_, err = conn.PutBucketRequestPaymentWithContext(ctx, input)

if err != nil {
return diag.FromErr(fmt.Errorf("error updating S3 bucket request payment configuration (%s): %w", d.Id(), err))
}

return resourceBucketRequestPaymentConfigurationRead(ctx, d, meta)
}

func resourceBucketRequestPaymentConfigurationDelete(ctx context.Context, d *schema.ResourceData, meta interface{}) diag.Diagnostics {
conn := meta.(*conns.AWSClient).S3Conn

bucket, expectedBucketOwner, err := ParseResourceID(d.Id())
if err != nil {
return diag.FromErr(err)
}

input := &s3.PutBucketRequestPaymentInput{
Bucket: aws.String(bucket),
RequestPaymentConfiguration: &s3.RequestPaymentConfiguration{
// To remove a configuration, it is equivalent to disabling
// "Requester Pays" in the console; thus, we reset "Payer" back to "BucketOwner"
Payer: aws.String(s3.PayerBucketOwner),
},
}

if expectedBucketOwner != "" {
input.ExpectedBucketOwner = aws.String(expectedBucketOwner)
}

_, err = conn.PutBucketRequestPaymentWithContext(ctx, input)

if tfawserr.ErrCodeEquals(err, s3.ErrCodeNoSuchBucket) {
return nil
}

if err != nil {
return diag.FromErr(fmt.Errorf("error deleting S3 bucket request payment configuration (%s): %w", d.Id(), err))
}

return nil
}
205 changes: 205 additions & 0 deletions internal/service/s3/bucket_request_payment_configuration_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,205 @@
package s3_test

import (
"fmt"
"testing"

"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/service/s3"
"github.com/hashicorp/aws-sdk-go-base/v2/awsv1shim/v2/tfawserr"
sdkacctest "github.com/hashicorp/terraform-plugin-sdk/v2/helper/acctest"
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"
"github.com/hashicorp/terraform-provider-aws/internal/acctest"
"github.com/hashicorp/terraform-provider-aws/internal/conns"
tfs3 "github.com/hashicorp/terraform-provider-aws/internal/service/s3"
)

func TestAccS3BucketRequestPaymentConfiguration_Basic_BucketOwner(t *testing.T) {
rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
resourceName := "aws_s3_bucket_request_payment_configuration.test"

resource.ParallelTest(t, resource.TestCase{
PreCheck: func() { acctest.PreCheck(t) },
ErrorCheck: acctest.ErrorCheck(t, s3.EndpointsID),
ProviderFactories: acctest.ProviderFactories,
CheckDestroy: testAccCheckBucketRequestPaymentConfigurationDestroy,
Steps: []resource.TestStep{
{
Config: testAccBucketRequestPaymentConfigurationBasicConfig(rName, s3.PayerBucketOwner),
Check: resource.ComposeTestCheckFunc(
testAccCheckBucketRequestPaymentConfigurationExists(resourceName),
resource.TestCheckResourceAttrPair(resourceName, "bucket", "aws_s3_bucket.test", "id"),
resource.TestCheckResourceAttr(resourceName, "payer", s3.PayerBucketOwner),
),
},
{
ResourceName: resourceName,
ImportState: true,
ImportStateVerify: true,
},
},
})
}

func TestAccS3BucketRequestPaymentConfiguration_Basic_Requester(t *testing.T) {
rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
resourceName := "aws_s3_bucket_request_payment_configuration.test"

resource.ParallelTest(t, resource.TestCase{
PreCheck: func() { acctest.PreCheck(t) },
ErrorCheck: acctest.ErrorCheck(t, s3.EndpointsID),
ProviderFactories: acctest.ProviderFactories,
CheckDestroy: testAccCheckBucketRequestPaymentConfigurationDestroy,
Steps: []resource.TestStep{
{
Config: testAccBucketRequestPaymentConfigurationBasicConfig(rName, s3.PayerRequester),
Check: resource.ComposeTestCheckFunc(
testAccCheckBucketRequestPaymentConfigurationExists(resourceName),
resource.TestCheckResourceAttrPair(resourceName, "bucket", "aws_s3_bucket.test", "id"),
resource.TestCheckResourceAttr(resourceName, "payer", s3.PayerRequester),
),
},
{
ResourceName: resourceName,
ImportState: true,
ImportStateVerify: true,
},
},
})
}

func TestAccS3BucketRequestPaymentConfiguration_update(t *testing.T) {
rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
resourceName := "aws_s3_bucket_request_payment_configuration.test"

resource.ParallelTest(t, resource.TestCase{
PreCheck: func() { acctest.PreCheck(t) },
ErrorCheck: acctest.ErrorCheck(t, s3.EndpointsID),
ProviderFactories: acctest.ProviderFactories,
CheckDestroy: testAccCheckBucketRequestPaymentConfigurationDestroy,
Steps: []resource.TestStep{
{
Config: testAccBucketRequestPaymentConfigurationBasicConfig(rName, s3.PayerRequester),
Check: resource.ComposeTestCheckFunc(
testAccCheckBucketRequestPaymentConfigurationExists(resourceName),
),
},
{
Config: testAccBucketRequestPaymentConfigurationBasicConfig(rName, s3.PayerBucketOwner),
Check: resource.ComposeTestCheckFunc(
testAccCheckBucketRequestPaymentConfigurationExists(resourceName),
),
},
{
ResourceName: resourceName,
ImportState: true,
ImportStateVerify: true,
},
{
Config: testAccBucketRequestPaymentConfigurationBasicConfig(rName, s3.PayerRequester),
Check: resource.ComposeTestCheckFunc(
testAccCheckBucketRequestPaymentConfigurationExists(resourceName),
),
},
},
})
}

func testAccCheckBucketRequestPaymentConfigurationDestroy(s *terraform.State) error {
conn := acctest.Provider.Meta().(*conns.AWSClient).S3Conn

for _, rs := range s.RootModule().Resources {
if rs.Type != "aws_s3_bucket_request_payment_configuration" {
continue
}

bucket, expectedBucketOwner, err := tfs3.ParseResourceID(rs.Primary.ID)
if err != nil {
return err
}

input := &s3.GetBucketRequestPaymentInput{
Bucket: aws.String(bucket),
}

if expectedBucketOwner != "" {
input.ExpectedBucketOwner = aws.String(expectedBucketOwner)
}

output, err := conn.GetBucketRequestPayment(input)

if tfawserr.ErrCodeEquals(err, s3.ErrCodeNoSuchBucket) {
continue
}

if err != nil {
return fmt.Errorf("error getting S3 bucket request payment configuration (%s): %w", rs.Primary.ID, err)
}

if output != nil && aws.StringValue(output.Payer) != s3.PayerBucketOwner {
return fmt.Errorf("S3 bucket request payment configuration (%s) still exists", rs.Primary.ID)
}
}

return nil
}

func testAccCheckBucketRequestPaymentConfigurationExists(resourceName string) resource.TestCheckFunc {
return func(s *terraform.State) error {
rs, ok := s.RootModule().Resources[resourceName]
if !ok {
return fmt.Errorf("Not found: %s", resourceName)
}

if rs.Primary.ID == "" {
return fmt.Errorf("Resource (%s) ID not set", resourceName)
}

conn := acctest.Provider.Meta().(*conns.AWSClient).S3Conn

bucket, expectedBucketOwner, err := tfs3.ParseResourceID(rs.Primary.ID)
if err != nil {
return err
}

input := &s3.GetBucketRequestPaymentInput{
Bucket: aws.String(bucket),
}

if expectedBucketOwner != "" {
input.ExpectedBucketOwner = aws.String(expectedBucketOwner)
}

output, err := conn.GetBucketRequestPayment(input)

if err != nil {
return fmt.Errorf("error getting S3 bucket request payment configuration (%s): %w", rs.Primary.ID, err)
}

if output == nil {
return fmt.Errorf("S3 Bucket request payment configuration (%s) not found", rs.Primary.ID)
}

return nil
}
}

func testAccBucketRequestPaymentConfigurationBasicConfig(rName, payer string) string {
return fmt.Sprintf(`
resource "aws_s3_bucket" "test" {
bucket = %[1]q
lifecycle {
ignore_changes = [
request_payer
]
}
}
resource "aws_s3_bucket_request_payment_configuration" "test" {
bucket = aws_s3_bucket.test.id
payer = %[2]q
}
`, rName, payer)
}
Loading

0 comments on commit a0434a3

Please sign in to comment.