-
Notifications
You must be signed in to change notification settings - Fork 9.1k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #22649 from hashicorp/f-s3-bucket-request-payment-…
…config-resource r/s3_bucket_request_payment_configuration: new resource
- Loading branch information
Showing
5 changed files
with
434 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,3 @@ | ||
```release-note:new-resource | ||
aws_s3_bucket_request_payment_configuration | ||
``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
175 changes: 175 additions & 0 deletions
175
internal/service/s3/bucket_request_payment_configuration.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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
205
internal/service/s3/bucket_request_payment_configuration_test.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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) | ||
} |
Oops, something went wrong.