Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

r/s3_bucket_request_payment_configuration: new resource #22649

Merged
merged 4 commits into from
Feb 3, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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_object": s3.ResourceObject(),
"aws_s3_object_copy": s3.ResourceObjectCopy(),
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