From 2d1413adc8277340d49cf78e687b1f1cb7a177b9 Mon Sep 17 00:00:00 2001 From: Wainer dos Santos Moschetta Date: Fri, 31 Jan 2025 14:53:44 -0300 Subject: [PATCH] test: add function to delete AWS bucket key Delete the key from bucket that contains the raw disk image. Signed-off-by: Wainer dos Santos Moschetta --- .../test/provisioner/aws/provision_common.go | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/src/cloud-api-adaptor/test/provisioner/aws/provision_common.go b/src/cloud-api-adaptor/test/provisioner/aws/provision_common.go index a8f9f3a0d..5bbbc5fde 100644 --- a/src/cloud-api-adaptor/test/provisioner/aws/provision_common.go +++ b/src/cloud-api-adaptor/test/provisioner/aws/provision_common.go @@ -259,6 +259,11 @@ func (a *AWSProvisioner) DeleteVPC(ctx context.Context, cfg *envconf.Config) err a.Image.deregisterImage() } + if a.Bucket.Key != "" { + log.Infof("Delete key %s from bucket: %s", a.Bucket.Key, a.Bucket.Name) + a.Bucket.deleteKey() + } + return nil } @@ -964,6 +969,17 @@ func (b *S3Bucket) uploadLargeFileWithCli(filepath string) error { return nil } +func (b *S3Bucket) deleteKey() error { + if _, err := b.Client.DeleteObject(context.TODO(), &s3.DeleteObjectInput{ + Bucket: aws.String(b.Name), + Key: aws.String(b.Key), + }); err != nil { + return err + } + + return nil +} + // ConvertQcow2ToRaw Converts an qcow2 image to raw. Requires `qemu-img` installed. func ConvertQcow2ToRaw(qcow2 string, raw string) error { cmd := exec.Command("qemu-img", "convert", "-O", "raw", qcow2, raw)