serverless/test/utils/s3.js

'use strict';

const awsRequest = require('@serverless/test/aws-request');
const S3Service = require('aws-sdk').S3;

async function createBucket(bucket) {
  return awsRequest(S3Service, 'createBucket', { Bucket: bucket });
}

async function createAndRemoveInBucket(bucket, opts = {}) {
  const prefix = opts.prefix || '';
  const suffix = opts.suffix || '';
  const fileName = opts.fileName || 'object';

  const params = {
    Bucket: bucket,
    Key: `${prefix}${fileName}${suffix}`,
    Body: 'hello world',
  };

  return awsRequest(S3Service, 'putObject', params).then(() => {
    delete params.Body;
    return awsRequest(S3Service, 'deleteObject', params);
  });
}

async function emptyBucket(bucket) {
  return awsRequest(S3Service, 'listObjects', { Bucket: bucket }).then((data) => {
    const items = data.Contents;
    const numItems = items.length;
    if (numItems) {
      const keys = items.map((item) => Object.assign({}, { Key: item.Key }));
      return awsRequest(S3Service, 'deleteObjects', {
        Bucket: bucket,
        Delete: {
          Objects: keys,
        },
      });
    }
    return null;
  });
}

async function deleteBucket(bucket) {
  return emptyBucket(bucket).then(() => awsRequest(S3Service, 'deleteBucket', { Bucket: bucket }));
}

module.exports = {
  createBucket,
  createAndRemoveInBucket,
  emptyBucket,
  deleteBucket,
};