@cumulus/aws-client 11.0.0 → 11.1.1

package/S3ObjectStore.js

@@ -1,33 +1,11 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
-const querystring = __importStar(require("querystring"));
 const url_1 = require("url");
-const AWS = __importStar(require("aws-sdk"));
+const uuid_1 = require("uuid");
+const client_s3_1 = require("@aws-sdk/client-s3");
+const s3_request_presigner_1 = require("@aws-sdk/s3-request-presigner");
 const logger_1 = __importDefault(require("@cumulus/logger"));
 const S3_1 = require("./S3");
 const client_1 = __importDefault(require("./client"));
@@ -38,8 +16,35 @@ const log = new logger_1.default({ sender: '@cumulus/aws-client/S3ObjectStore' }
  *
  */
 class S3ObjectStore {
-    constructor() {
-        this.s3 = (0, client_1.default)(AWS.S3, '2006-03-01', { signatureVersion: 'v4' })();
+    constructor(config) {
+        this.s3 = (0, client_1.default)(client_s3_1.S3, '2006-03-01', {
+            signatureVersion: 'v4',
+        })(config);
+        this.queryParams = {};
+        this.middlewareName = `customQueryParams${(0, uuid_1.v4)()}`;
+    }
+    getQueryParams() {
+        return this.queryParams;
+    }
+    setQueryParams(queryParams) {
+        this.queryParams = queryParams;
+    }
+    async getS3SignedUrlWithCustomQueryParams(command) {
+        this.s3.middlewareStack.addRelativeTo((next) => (args) => {
+            const { request } = args;
+            request.query = {
+                ...this.getQueryParams(),
+                ...request.query,
+            };
+            return next(args);
+        }, {
+            name: this.middlewareName,
+            relation: 'before',
+            toMiddleware: 'presignInterceptMiddleware',
+        });
+        const signedUrl = await (0, s3_request_presigner_1.getSignedUrl)(this.s3, command);
+        this.s3.middlewareStack.remove(this.middlewareName);
+        return signedUrl;
     }
     /**
      * Returns an HTTPS URL that can be used to perform a GET on the given object
@@ -51,7 +56,7 @@ class S3ObjectStore {
      * @returns {Promise<string>} a signed URL
      * @throws TypeError - if the URL is not a recognized protocol or cannot be parsed
      */
-    async signGetObject(objectUrl, options = {}, queryParams) {
+    async signGetObject(objectUrl, options = {}, queryParams = {}) {
         log.info(`Executing signGetObject with objectUrl: ${objectUrl}, options: ${JSON.stringify(options)}, queryParams: ${JSON.stringify(queryParams)}`);
         const url = new url_1.URL(objectUrl);
         if (url.protocol.toLowerCase() !== 's3:') {
@@ -59,14 +64,11 @@ class S3ObjectStore {
         }
         const { Bucket, Key } = (0, S3_1.parseS3Uri)(objectUrl);
         await (0, S3_1.headObject)(Bucket, Key);
-        const req = this.s3.getObject({ Bucket, Key, ...options });
-        if (queryParams && req.on) {
-            (req.on('build', () => { req.httpRequest.path += `${options ? '&' : '?'}${querystring.stringify(queryParams)}`; }));
-        }
-        // TypeScript doesn't recognize that req has a presign method.  It does.
-        const result = await req.presign();
-        log.debug(`Signed GetObject request URL: ${result}`);
-        return result;
+        const command = new client_s3_1.GetObjectCommand({ Bucket, Key, ...options });
+        this.setQueryParams(queryParams);
+        const signedUrl = await this.getS3SignedUrlWithCustomQueryParams(command);
+        log.debug(`Signed GetObject request URL: ${signedUrl}`);
+        return signedUrl;
     }
     /**
      * Returns an HTTPS URL that can be used to perform a HEAD on the given object
@@ -85,14 +87,11 @@ class S3ObjectStore {
             throw new TypeError(`Invalid S3 URL: ${objectUrl}`);
         }
         const { Bucket, Key } = (0, S3_1.parseS3Uri)(objectUrl);
-        const req = this.s3.headObject({ Bucket, Key, ...options });
-        if (queryParams && req.on) {
-            (req.on('build', () => { req.httpRequest.path += `?${querystring.stringify(queryParams)}`; }));
-        }
-        // TypeScript doesn't recognize that req has a presign method.  It does.
-        const result = await req.presign();
-        log.debug(`Signed HeadObject request URL: ${result}`);
-        return result;
+        const command = new client_s3_1.HeadObjectCommand({ Bucket, Key, ...options });
+        this.setQueryParams(queryParams);
+        const signedUrl = await this.getS3SignedUrlWithCustomQueryParams(command);
+        log.debug(`Signed HeadObject request URL: ${signedUrl}`);
+        return signedUrl;
     }
 }
 module.exports = S3ObjectStore;

package/client.d.ts

@@ -1,4 +1,4 @@
-import AWS from 'aws-sdk';
+import { AWSClientTypes } from './types';
 /**
  * Return a function which, when called, will return an AWS service object
  *
@@ -14,6 +14,6 @@ import AWS from 'aws-sdk';
  *
  * @private
  */
-declare const awsClient: <T extends AWS.Service | AWS.DynamoDB.DocumentClient>(Service: new (params: object) => T, version?: string | undefined, serviceOptions?: object | undefined) => (options?: object | undefined) => T;
+declare const awsClient: <T extends AWSClientTypes>(Service: new (params: object) => T, version?: string | undefined, serviceOptions?: object | undefined) => (params?: object | undefined) => T;
 export = awsClient;
 //# sourceMappingURL=client.d.ts.map

package/client.js

@@ -3,21 +3,21 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 const aws_sdk_1 = __importDefault(require("aws-sdk"));
+const mem_1 = __importDefault(require("mem"));
 const test_utils_1 = require("./test-utils");
-const noop = () => { }; // eslint-disable-line lodash/prefer-noop
+const utils_1 = require("./utils");
 const getRegion = () => process.env.AWS_DEFAULT_REGION || process.env.AWS_REGION || 'us-east-1';
-// Workaround upload hangs. See: https://github.com/andrewrk/node-s3-client/issues/74
-// @ts-ignore - AWS.util is not part of the public API and may break
-aws_sdk_1.default.util.update(aws_sdk_1.default.S3.prototype, { addExpect100Continue: noop });
 aws_sdk_1.default.config.setPromisesDependency(Promise);
-const memoize = (fn) => {
-    let memo;
-    return (options) => {
-        if (!memo)
-            memo = fn(options);
-        return memo;
-    };
+const buildServiceClient = (Service, options) => {
+    if ((0, test_utils_1.inTestMode)()) {
+        return new Service((0, test_utils_1.getLocalstackAwsClientOptions)(Service, options));
+    }
+    return new Service(options);
 };
+const getMemoizedClient = (0, mem_1.default)(buildServiceClient, {
+    cacheKey: (arguments_) => `${(0, utils_1.getServiceIdentifer)(arguments_[0])}${JSON.stringify(arguments_[1])}`,
+});
+const getServiceClient = (Service, options = {}) => (overrides) => getMemoizedClient(Service, Object.assign(options, overrides));
 /**
  * Return a function which, when called, will return an AWS service object
  *
@@ -44,11 +44,10 @@ const awsClient = (Service, version, serviceOptions) => {
         // @ts-ignore - serviceIdentifier is not part of the public API and may break at any time
         if (aws_sdk_1.default.DynamoDB.DocumentClient.serviceIdentifier === undefined) {
             // @ts-ignore - serviceIdentifier is not part of the public API and may break at any time
-            aws_sdk_1.default.DynamoDB.DocumentClient.serviceIdentifier = 'dynamodb';
+            aws_sdk_1.default.DynamoDB.DocumentClient.serviceIdentifier = 'dynamodbclient';
         }
-        return memoize((o) => (0, test_utils_1.testAwsClient)(Service, Object.assign(options, o)));
     }
-    return memoize((o) => new Service(Object.assign(options, o)));
+    return getServiceClient(Service, options);
 };
 module.exports = awsClient;
 //# sourceMappingURL=client.js.map

package/lib/S3MultipartUploads.d.ts

@@ -1,3 +1,4 @@
+import { AbortMultipartUploadRequest, CompleteMultipartUploadRequest, CreateMultipartUploadRequest, UploadPartCopyRequest } from '@aws-sdk/client-s3';
 export interface CompleteMultipartUploadOutput extends AWS.S3.CompleteMultipartUploadOutput {
     ETag: string;
 }
@@ -23,9 +24,9 @@ declare type Chunk = {
  * @returns {Promise<Array<Chunk>>} - array of chunks
  */
 export declare const createMultipartChunks: (objectSize: number, chunkSize?: number) => Chunk[];
-export declare const createMultipartUpload: (params: AWS.S3.CreateMultipartUploadRequest) => Promise<import("aws-sdk/lib/request").PromiseResult<import("aws-sdk/clients/s3").CreateMultipartUploadOutput, import("aws-sdk").AWSError>>;
-export declare const completeMultipartUpload: (params: AWS.S3.CompleteMultipartUploadRequest) => Promise<CompleteMultipartUploadOutput>;
-export declare const abortMultipartUpload: (params: AWS.S3.AbortMultipartUploadRequest) => Promise<import("aws-sdk/lib/request").PromiseResult<import("aws-sdk/clients/s3").AbortMultipartUploadOutput, import("aws-sdk").AWSError>>;
-export declare const uploadPartCopy: (params: AWS.S3.UploadPartCopyRequest) => Promise<import("aws-sdk/lib/request").PromiseResult<import("aws-sdk/clients/s3").UploadPartCopyOutput, import("aws-sdk").AWSError>>;
+export declare const createMultipartUpload: (params: CreateMultipartUploadRequest) => Promise<import("@aws-sdk/client-s3").CreateMultipartUploadCommandOutput>;
+export declare const completeMultipartUpload: (params: CompleteMultipartUploadRequest) => Promise<CompleteMultipartUploadOutput>;
+export declare const abortMultipartUpload: (params: AbortMultipartUploadRequest) => Promise<import("@aws-sdk/client-s3").AbortMultipartUploadCommandOutput>;
+export declare const uploadPartCopy: (params: UploadPartCopyRequest) => Promise<import("@aws-sdk/client-s3").UploadPartCopyCommandOutput>;
 export {};
 //# sourceMappingURL=S3MultipartUploads.d.ts.map

package/lib/S3MultipartUploads.js

@@ -31,15 +31,15 @@ const createMultipartChunks = (objectSize, chunkSize = 250 * MB) => (0, range_1.
     end: Math.min(start + chunkSize, objectSize) - 1,
 }));
 exports.createMultipartChunks = createMultipartChunks;
-const createMultipartUpload = async (params) => await (0, services_1.s3)().createMultipartUpload(params).promise();
+const createMultipartUpload = async (params) => await (0, services_1.s3)().createMultipartUpload(params);
 exports.createMultipartUpload = createMultipartUpload;
 const completeMultipartUpload = async (params) => {
-    const result = await (0, services_1.s3)().completeMultipartUpload(params).promise();
+    const result = await (0, services_1.s3)().completeMultipartUpload(params);
     return result;
 };
 exports.completeMultipartUpload = completeMultipartUpload;
-const abortMultipartUpload = async (params) => await (0, services_1.s3)().abortMultipartUpload(params).promise();
+const abortMultipartUpload = async (params) => await (0, services_1.s3)().abortMultipartUpload(params);
 exports.abortMultipartUpload = abortMultipartUpload;
-const uploadPartCopy = async (params) => await (0, services_1.s3)().uploadPartCopy(params).promise();
+const uploadPartCopy = async (params) => await (0, services_1.s3)().uploadPartCopy(params);
 exports.uploadPartCopy = uploadPartCopy;
 //# sourceMappingURL=S3MultipartUploads.js.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cumulus/aws-client",
-  "version": "11.0.0",
+  "version": "11.1.1",
   "description": "Utilities for working with AWS",
   "keywords": [
     "GIBS",
@@ -43,17 +43,30 @@
   "author": "Cumulus Authors",
   "license": "Apache-2.0",
   "dependencies": {
-    "@cumulus/checksum": "11.0.0",
-    "@cumulus/errors": "11.0.0",
-    "@cumulus/logger": "11.0.0",
-    "aws-sdk": "^2.814.0",
+    "@aws-sdk/client-dynamodb": "^3.53.0",
+    "@aws-sdk/client-dynamodb-streams": "^3.53.0",
+    "@aws-sdk/client-s3": "^3.53.0",
+    "@aws-sdk/lib-dynamodb": "^3.53.0",
+    "@aws-sdk/lib-storage": "^3.53.0",
+    "@aws-sdk/s3-request-presigner": "^3.53.0",
+    "@aws-sdk/signature-v4-crt": "^3.53.0",
+    "@aws-sdk/types": "^3.53.0",
+    "@cumulus/checksum": "11.1.1",
+    "@cumulus/errors": "11.1.1",
+    "@cumulus/logger": "11.1.1",
+    "aws-sdk": "^2.585.0",
     "jsonpath-plus": "^1.1.0",
     "lodash": "~4.17.21",
+    "mem": "^8.0.2",
     "p-map": "^1.2.0",
     "p-retry": "^4.2.0",
     "p-timeout": "^4.1.0",
     "p-wait-for": "^3.2.0",
-    "pump": "^3.0.0"
+    "pump": "^3.0.0",
+    "uuid": "^8.2.0"
   },
-  "gitHead": "e922ad12fd94affa6cd60a97a184a465a756c50b"
+  "devDependencies": {
+    "@types/uuid": "^8.0.0"
+  },
+  "gitHead": "07fe682da23821434372759fb1948b081da429b5"
 }

package/services.d.ts

@@ -1,23 +1,27 @@
+import { DynamoDB } from '@aws-sdk/client-dynamodb';
+import { DynamoDBStreamsClient } from '@aws-sdk/client-dynamodb-streams';
+import { S3 } from '@aws-sdk/client-s3';
+import { DynamoDBDocument, TranslateConfig } from '@aws-sdk/lib-dynamodb';
 import * as AWS from 'aws-sdk';
-export declare const apigateway: (options?: object | undefined) => AWS.APIGateway;
-export declare const ecs: (options?: object | undefined) => AWS.ECS;
-export declare const ec2: (options?: object | undefined) => AWS.EC2;
-export declare const s3: (options?: object | undefined) => AWS.S3;
-export declare const kinesis: (options?: object | undefined) => AWS.Kinesis;
-export declare const lambda: (options?: object | undefined) => AWS.Lambda;
-export declare const sqs: (options?: object | undefined) => AWS.SQS;
-export declare const cloudwatchevents: (options?: object | undefined) => AWS.CloudWatchEvents;
-export declare const cloudwatchlogs: (options?: object | undefined) => AWS.CloudWatchLogs;
-export declare const cloudwatch: (options?: object | undefined) => AWS.CloudWatch;
-export declare const dynamodb: (options?: object | undefined) => AWS.DynamoDB;
-export declare const dynamodbstreams: (options?: object | undefined) => AWS.DynamoDBStreams;
-export declare const dynamodbDocClient: (options?: object | undefined) => AWS.DynamoDB.DocumentClient;
-export declare const sfn: (options?: object | undefined) => AWS.StepFunctions;
-export declare const cf: (options?: object | undefined) => AWS.CloudFormation;
-export declare const sns: (options?: object | undefined) => AWS.SNS;
-export declare const secretsManager: (options?: object | undefined) => AWS.SecretsManager;
-export declare const systemsManager: (options?: object | undefined) => AWS.SSM;
-export declare const kms: (options?: object | undefined) => AWS.KMS;
-export declare const es: (options?: object | undefined) => AWS.ES;
-export declare const sts: (options?: object | undefined) => AWS.STS;
+export declare const apigateway: (params?: object | undefined) => AWS.APIGateway;
+export declare const ecs: (params?: object | undefined) => AWS.ECS;
+export declare const ec2: (params?: object | undefined) => AWS.EC2;
+export declare const s3: (params?: object | undefined) => S3;
+export declare const kinesis: (params?: object | undefined) => AWS.Kinesis;
+export declare const lambda: (params?: object | undefined) => AWS.Lambda;
+export declare const sqs: (params?: object | undefined) => AWS.SQS;
+export declare const cloudwatchevents: (params?: object | undefined) => AWS.CloudWatchEvents;
+export declare const cloudwatchlogs: (params?: object | undefined) => AWS.CloudWatchLogs;
+export declare const cloudwatch: (params?: object | undefined) => AWS.CloudWatch;
+export declare const dynamodb: (params?: object | undefined) => DynamoDB;
+export declare const dynamodbstreams: (params?: object | undefined) => DynamoDBStreamsClient;
+export declare const dynamodbDocClient: (docClientOptions?: TranslateConfig | undefined, dynamoOptions?: object | undefined) => DynamoDBDocument;
+export declare const sfn: (params?: object | undefined) => AWS.StepFunctions;
+export declare const cf: (params?: object | undefined) => AWS.CloudFormation;
+export declare const sns: (params?: object | undefined) => AWS.SNS;
+export declare const secretsManager: (params?: object | undefined) => AWS.SecretsManager;
+export declare const systemsManager: (params?: object | undefined) => AWS.SSM;
+export declare const kms: (params?: object | undefined) => AWS.KMS;
+export declare const es: (params?: object | undefined) => AWS.ES;
+export declare const sts: (params?: object | undefined) => AWS.STS;
 //# sourceMappingURL=services.d.ts.map

package/services.js

@@ -22,29 +22,37 @@ var __importStar = (this && this.__importStar) || function (mod) {
     __setModuleDefault(result, mod);
     return result;
 };
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.sts = exports.es = exports.kms = exports.systemsManager = exports.secretsManager = exports.sns = exports.cf = exports.sfn = exports.dynamodbDocClient = exports.dynamodbstreams = exports.dynamodb = exports.cloudwatch = exports.cloudwatchlogs = exports.cloudwatchevents = exports.sqs = exports.lambda = exports.kinesis = exports.s3 = exports.ec2 = exports.ecs = exports.apigateway = void 0;
+const client_dynamodb_1 = require("@aws-sdk/client-dynamodb");
+const client_dynamodb_streams_1 = require("@aws-sdk/client-dynamodb-streams");
+const client_s3_1 = require("@aws-sdk/client-s3");
+const lib_dynamodb_1 = require("@aws-sdk/lib-dynamodb");
 const AWS = __importStar(require("aws-sdk"));
-const awsClient = require("./client");
-exports.apigateway = awsClient(AWS.APIGateway, '2015-07-09');
-exports.ecs = awsClient(AWS.ECS, '2014-11-13');
-exports.ec2 = awsClient(AWS.EC2, '2016-11-15');
-exports.s3 = awsClient(AWS.S3, '2006-03-01');
-exports.kinesis = awsClient(AWS.Kinesis, '2013-12-02');
-exports.lambda = awsClient(AWS.Lambda, '2015-03-31');
-exports.sqs = awsClient(AWS.SQS, '2012-11-05');
-exports.cloudwatchevents = awsClient(AWS.CloudWatchEvents, '2014-02-03');
-exports.cloudwatchlogs = awsClient(AWS.CloudWatchLogs, '2014-03-28');
-exports.cloudwatch = awsClient(AWS.CloudWatch, '2010-08-01');
-exports.dynamodb = awsClient(AWS.DynamoDB, '2012-08-10');
-exports.dynamodbstreams = awsClient(AWS.DynamoDBStreams, '2012-08-10');
-exports.dynamodbDocClient = awsClient(AWS.DynamoDB.DocumentClient, '2012-08-10');
-exports.sfn = awsClient(AWS.StepFunctions, '2016-11-23');
-exports.cf = awsClient(AWS.CloudFormation, '2010-05-15');
-exports.sns = awsClient(AWS.SNS, '2010-03-31');
-exports.secretsManager = awsClient(AWS.SecretsManager, '2017-10-17');
-exports.systemsManager = awsClient(AWS.SSM, '2017-10-17');
-exports.kms = awsClient(AWS.KMS, '2014-11-01');
-exports.es = awsClient(AWS.ES, '2015-01-01');
-exports.sts = awsClient(AWS.STS, '2011-06-15');
+const client_1 = __importDefault(require("./client"));
+exports.apigateway = (0, client_1.default)(AWS.APIGateway, '2015-07-09');
+exports.ecs = (0, client_1.default)(AWS.ECS, '2014-11-13');
+exports.ec2 = (0, client_1.default)(AWS.EC2, '2016-11-15');
+exports.s3 = (0, client_1.default)(client_s3_1.S3, '2006-03-01');
+exports.kinesis = (0, client_1.default)(AWS.Kinesis, '2013-12-02');
+exports.lambda = (0, client_1.default)(AWS.Lambda, '2015-03-31');
+exports.sqs = (0, client_1.default)(AWS.SQS, '2012-11-05');
+exports.cloudwatchevents = (0, client_1.default)(AWS.CloudWatchEvents, '2014-02-03');
+exports.cloudwatchlogs = (0, client_1.default)(AWS.CloudWatchLogs, '2014-03-28');
+exports.cloudwatch = (0, client_1.default)(AWS.CloudWatch, '2010-08-01');
+exports.dynamodb = (0, client_1.default)(client_dynamodb_1.DynamoDB, '2012-08-10');
+exports.dynamodbstreams = (0, client_1.default)(client_dynamodb_streams_1.DynamoDBStreamsClient, '2012-08-10');
+const dynamodbDocClient = (docClientOptions, dynamoOptions) => lib_dynamodb_1.DynamoDBDocument.from((0, client_1.default)(client_dynamodb_1.DynamoDB, '2012-08-10')(dynamoOptions), docClientOptions);
+exports.dynamodbDocClient = dynamodbDocClient;
+exports.sfn = (0, client_1.default)(AWS.StepFunctions, '2016-11-23');
+exports.cf = (0, client_1.default)(AWS.CloudFormation, '2010-05-15');
+exports.sns = (0, client_1.default)(AWS.SNS, '2010-03-31');
+exports.secretsManager = (0, client_1.default)(AWS.SecretsManager, '2017-10-17');
+exports.systemsManager = (0, client_1.default)(AWS.SSM, '2017-10-17');
+exports.kms = (0, client_1.default)(AWS.KMS, '2014-11-01');
+exports.es = (0, client_1.default)(AWS.ES, '2015-01-01');
+exports.sts = (0, client_1.default)(AWS.STS, '2011-06-15');
 //# sourceMappingURL=services.js.map

package/test-utils.d.ts

@@ -1,4 +1,4 @@
-import * as AWS from 'aws-sdk';
+import { AWSClientTypes } from './types';
 export declare const inTestMode: () => boolean;
 declare const localStackPorts: {
     stepfunctions: number;
@@ -7,7 +7,11 @@ declare const localStackPorts: {
     cloudwatch: number;
     cloudwatchevents: number;
     cloudwatchlogs: number;
-    dynamodb: number;
+    DynamoDB: number;
+    DynamoDBClient: number;
+    DynamoDBStreamsClient: number;
+    ec2: number;
+    ecs: number;
     es: number;
     firehose: number;
     iam: number;
@@ -16,7 +20,7 @@ declare const localStackPorts: {
     lambda: number;
     redshift: number;
     route53: number;
-    s3: number;
+    S3: number;
     secretsmanager: number;
     ses: number;
     sns: number;
@@ -33,6 +37,22 @@ declare const localStackPorts: {
  * @private
  */
 export declare function getLocalstackEndpoint(identifier: keyof typeof localStackPorts): string;
+/**
+ * Create an AWS service object that talks to LocalStack.
+ *
+ * This function expects that the LOCALSTACK_HOST environment variable will be set.
+ *
+ * @param {Function} Service - an AWS service object constructor function
+ * @param {Object} options - options to pass to the service object constructor function
+ * @returns {Object} an AWS service object
+ *
+ * @private
+ */
+export declare function localStackAwsClientOptions<T>(Service: new (params: object) => T, options?: {
+    credentials?: object;
+}): {
+    [key: string]: unknown;
+};
 /**
  * Create an AWS service object that does not actually talk to AWS.
  *
@@ -42,7 +62,9 @@ export declare function getLocalstackEndpoint(identifier: keyof typeof localStac
  *
  * @private
  */
-export declare function testAwsClient<T extends AWS.Service | AWS.DynamoDB.DocumentClient>(Service: new (params: object) => T, options: object): T;
+export declare function getLocalstackAwsClientOptions<T extends AWSClientTypes>(Service: new (params: object) => T, options?: {
+    credentials?: object;
+}): object;
 /**
  * Return a function that throws a ThrottlingException the first time it is called, then returns as
  * normal any other times.

package/test-utils.js

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.throttleOnce = exports.testAwsClient = exports.getLocalstackEndpoint = exports.inTestMode = void 0;
+exports.throttleOnce = exports.getLocalstackAwsClientOptions = exports.localStackAwsClientOptions = exports.getLocalstackEndpoint = exports.inTestMode = void 0;
 const errors_1 = require("@cumulus/errors");
+const utils_1 = require("./utils");
 const inTestMode = () => process.env.NODE_ENV === 'test';
 exports.inTestMode = inTestMode;
 // From https://github.com/localstack/localstack/blob/master/README.md
@@ -12,7 +13,11 @@ const localStackPorts = {
     cloudwatch: 4566,
     cloudwatchevents: 4566,
     cloudwatchlogs: 4566,
-    dynamodb: 4566,
+    DynamoDB: 4566,
+    DynamoDBClient: 4566,
+    DynamoDBStreamsClient: 4566,
+    ec2: 4566,
+    ecs: 4566,
     es: 4566,
     firehose: 4566,
     iam: 4566,
@@ -21,7 +26,7 @@ const localStackPorts = {
     lambda: 4566,
     redshift: 4566,
     route53: 4566,
-    s3: 4566,
+    S3: 4566,
     secretsmanager: 4566,
     ses: 4566,
     sns: 4566,
@@ -68,23 +73,26 @@ exports.getLocalstackEndpoint = getLocalstackEndpoint;
  *
  * @private
  */
-function localStackAwsClient(Service, options) {
+function localStackAwsClientOptions(Service, options = {}) {
     if (!process.env.LOCALSTACK_HOST) {
         throw new Error('The LOCALSTACK_HOST environment variable is not set.');
     }
-    // @ts-ignore
-    const serviceIdentifier = Service.serviceIdentifier;
+    const serviceIdentifier = (0, utils_1.getServiceIdentifer)(Service);
     const localStackOptions = {
-        ...options,
-        accessKeyId: 'my-access-key-id',
-        secretAccessKey: 'my-secret-access-key',
         region: 'us-east-1',
         endpoint: getLocalstackEndpoint(serviceIdentifier),
+        ...options,
+        credentials: {
+            accessKeyId: 'my-access-key-id',
+            secretAccessKey: 'my-secret-access-key',
+            ...options.credentials,
+        },
     };
-    if (serviceIdentifier === 's3')
-        localStackOptions.s3ForcePathStyle = true;
-    return new Service(localStackOptions);
+    if (serviceIdentifier.toLowerCase() === 's3')
+        localStackOptions.forcePathStyle = true;
+    return localStackOptions;
 }
+exports.localStackAwsClientOptions = localStackAwsClientOptions;
 /**
  * Create an AWS service object that does not actually talk to AWS.
  *
@@ -94,15 +102,14 @@ function localStackAwsClient(Service, options) {
  *
  * @private
  */
-function testAwsClient(Service, options) {
-    // @ts-ignore
-    const serviceIdentifier = Service.serviceIdentifier;
+function getLocalstackAwsClientOptions(Service, options = {}) {
+    const serviceIdentifier = (0, utils_1.getServiceIdentifer)(Service);
     if (localstackSupportedService(serviceIdentifier)) {
-        return localStackAwsClient(Service, options);
+        return localStackAwsClientOptions(Service, options);
     }
     return {};
 }
-exports.testAwsClient = testAwsClient;
+exports.getLocalstackAwsClientOptions = getLocalstackAwsClientOptions;
 /**
  * Return a function that throws a ThrottlingException the first time it is called, then returns as
  * normal any other times.

package/types.d.ts

@@ -0,0 +1,6 @@
+import * as AWS from 'aws-sdk';
+import { DynamoDBStreamsClient } from '@aws-sdk/client-dynamodb-streams';
+import { DynamoDB, DynamoDBClient } from '@aws-sdk/client-dynamodb';
+import { S3 } from '@aws-sdk/client-s3';
+export declare type AWSClientTypes = DynamoDB | DynamoDBClient | DynamoDBStreamsClient | S3 | AWS.Service | AWS.DynamoDB.DocumentClient;
+//# sourceMappingURL=types.d.ts.map

package/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/utils.d.ts

@@ -1,4 +1,5 @@
 import pRetry from 'p-retry';
+export declare const getServiceIdentifer: (service: any) => any;
 export declare const setErrorStack: (error: Error, newStack: string) => void;
 /**
  * Wrap a function and provide a better stack trace

package/utils.js

@@ -3,9 +3,11 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.retryOnThrottlingException = exports.improveStackTrace = exports.setErrorStack = void 0;
+exports.retryOnThrottlingException = exports.improveStackTrace = exports.setErrorStack = exports.getServiceIdentifer = void 0;
 const p_retry_1 = __importDefault(require("p-retry"));
 const errors_1 = require("@cumulus/errors");
+const getServiceIdentifer = (service) => service.serviceIdentifier || service.name;
+exports.getServiceIdentifer = getServiceIdentifer;
 // Replace the stack of an error
 // Note: This mutates the error that was passed in.
 const setErrorStack = (error, newStack) => {