Support transformation of waiter configuration (#389)

aws · Feb 22, 2023 · 2d477e0 · 2d477e0
1 parent b264bbc
commit 2d477e0
Show file tree

Hide file tree

Showing 24 changed files with 471 additions and 51 deletions.
diff --git a/.changeset/eighty-lemons-cheer.md b/.changeset/eighty-lemons-cheer.md
@@ -0,0 +1,5 @@
+---
+"aws-sdk-js-codemod": patch
+---
+
+Support transformation of waiter configuration
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/global-import-equals.input.ts b/src/transforms/v2-to-v3/__fixtures__/waiters/global-import-equals.input.ts
@@ -5,4 +5,13 @@ const client = new AWS.S3({ region: "REGION" });
 
 await client.waitFor("bucketNotExists", { Bucket }).promise();
 await client.createBucket({ Bucket }).promise();
-await client.waitFor("bucketExists", { Bucket }).promise();
+await client.waitFor("bucketExists", { Bucket }).promise();
+
+// Waiter configuration: delay
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2 } }).promise();
+
+// Waiter configuration: maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { maxAttempts: 5 } }).promise();
+
+// Waiter configuration: delay+maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2, maxAttempts: 5 } }).promise();
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/global-import-equals.output.ts b/src/transforms/v2-to-v3/__fixtures__/waiters/global-import-equals.output.ts
@@ -11,10 +11,36 @@ const client = new S3({ region: "REGION" });
 
 await waitUntilBucketNotExists({
   client,
-  maxWaitTime: 180
+  maxWaitTime: 200
 }, { Bucket });
 await client.createBucket({ Bucket });
 await waitUntilBucketExists({
   client,
-  maxWaitTime: 180
-}, { Bucket });
+  maxWaitTime: 200
+}, { Bucket });
+
+// Waiter configuration: delay
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 40
+}, {
+  Bucket
+});
+
+// Waiter configuration: maxAttempts
+await waitUntilBucketExists({
+  client,
+  maxWaitTime: 100
+}, {
+  Bucket
+});
+
+// Waiter configuration: delay+maxAttempts
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 20
+}, {
+  Bucket
+});
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/global-import.input.js b/src/transforms/v2-to-v3/__fixtures__/waiters/global-import.input.js
@@ -5,4 +5,13 @@ const client = new AWS.S3({ region: "REGION" });
 
 await client.waitFor("bucketNotExists", { Bucket }).promise();
 await client.createBucket({ Bucket }).promise();
-await client.waitFor("bucketExists", { Bucket }).promise();
+await client.waitFor("bucketExists", { Bucket }).promise();
+
+// Waiter configuration: delay
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2 } }).promise();
+
+// Waiter configuration: maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { maxAttempts: 5 } }).promise();
+
+// Waiter configuration: delay+maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2, maxAttempts: 5 } }).promise();
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/global-import.output.js b/src/transforms/v2-to-v3/__fixtures__/waiters/global-import.output.js
@@ -5,10 +5,36 @@ const client = new S3({ region: "REGION" });
 
 await waitUntilBucketNotExists({
   client,
-  maxWaitTime: 180
+  maxWaitTime: 200
 }, { Bucket });
 await client.createBucket({ Bucket });
 await waitUntilBucketExists({
   client,
-  maxWaitTime: 180
-}, { Bucket });
+  maxWaitTime: 200
+}, { Bucket });
+
+// Waiter configuration: delay
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 40
+}, {
+  Bucket
+});
+
+// Waiter configuration: maxAttempts
+await waitUntilBucketExists({
+  client,
+  maxWaitTime: 100
+}, {
+  Bucket
+});
+
+// Waiter configuration: delay+maxAttempts
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 20
+}, {
+  Bucket
+});
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/global-require-property.input.js b/src/transforms/v2-to-v3/__fixtures__/waiters/global-require-property.input.js
@@ -5,4 +5,13 @@ const client = new S3({ region: "REGION" });
 
 await client.waitFor("bucketNotExists", { Bucket }).promise();
 await client.createBucket({ Bucket }).promise();
-await client.waitFor("bucketExists", { Bucket }).promise();
+await client.waitFor("bucketExists", { Bucket }).promise();
+
+// Waiter configuration: delay
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2 } }).promise();
+
+// Waiter configuration: maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { maxAttempts: 5 } }).promise();
+
+// Waiter configuration: delay+maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2, maxAttempts: 5 } }).promise();
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/global-require-property.output.js b/src/transforms/v2-to-v3/__fixtures__/waiters/global-require-property.output.js
@@ -9,10 +9,36 @@ const client = new S3({ region: "REGION" });
 
 await waitUntilBucketNotExists({
   client,
-  maxWaitTime: 180
+  maxWaitTime: 200
 }, { Bucket });
 await client.createBucket({ Bucket });
 await waitUntilBucketExists({
   client,
-  maxWaitTime: 180
-}, { Bucket });
+  maxWaitTime: 200
+}, { Bucket });
+
+// Waiter configuration: delay
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 40
+}, {
+  Bucket
+});
+
+// Waiter configuration: maxAttempts
+await waitUntilBucketExists({
+  client,
+  maxWaitTime: 100
+}, {
+  Bucket
+});
+
+// Waiter configuration: delay+maxAttempts
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 20
+}, {
+  Bucket
+});
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/global-require.input.js b/src/transforms/v2-to-v3/__fixtures__/waiters/global-require.input.js
@@ -5,4 +5,13 @@ const client = new AWS.S3({ region: "REGION" });
 
 await client.waitFor("bucketNotExists", { Bucket }).promise();
 await client.createBucket({ Bucket }).promise();
-await client.waitFor("bucketExists", { Bucket }).promise();
+await client.waitFor("bucketExists", { Bucket }).promise();
+
+// Waiter configuration: delay
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2 } }).promise();
+
+// Waiter configuration: maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { maxAttempts: 5 } }).promise();
+
+// Waiter configuration: delay+maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2, maxAttempts: 5 } }).promise();
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/global-require.output.js b/src/transforms/v2-to-v3/__fixtures__/waiters/global-require.output.js
@@ -9,10 +9,36 @@ const client = new S3({ region: "REGION" });
 
 await waitUntilBucketNotExists({
   client,
-  maxWaitTime: 180
+  maxWaitTime: 200
 }, { Bucket });
 await client.createBucket({ Bucket });
 await waitUntilBucketExists({
   client,
-  maxWaitTime: 180
-}, { Bucket });
+  maxWaitTime: 200
+}, { Bucket });
+
+// Waiter configuration: delay
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 40
+}, {
+  Bucket
+});
+
+// Waiter configuration: maxAttempts
+await waitUntilBucketExists({
+  client,
+  maxWaitTime: 100
+}, {
+  Bucket
+});
+
+// Waiter configuration: delay+maxAttempts
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 20
+}, {
+  Bucket
+});
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/service-import-deep.input.js b/src/transforms/v2-to-v3/__fixtures__/waiters/service-import-deep.input.js
@@ -5,4 +5,13 @@ const client = new S3({ region: "REGION" });
 
 await client.waitFor("bucketNotExists", { Bucket }).promise();
 await client.createBucket({ Bucket }).promise();
-await client.waitFor("bucketExists", { Bucket }).promise();
+await client.waitFor("bucketExists", { Bucket }).promise();
+
+// Waiter configuration: delay
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2 } }).promise();
+
+// Waiter configuration: maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { maxAttempts: 5 } }).promise();
+
+// Waiter configuration: delay+maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2, maxAttempts: 5 } }).promise();
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/service-import-deep.output.js b/src/transforms/v2-to-v3/__fixtures__/waiters/service-import-deep.output.js
@@ -5,10 +5,36 @@ const client = new S3({ region: "REGION" });
 
 await waitUntilBucketNotExists({
   client,
-  maxWaitTime: 180
+  maxWaitTime: 200
 }, { Bucket });
 await client.createBucket({ Bucket });
 await waitUntilBucketExists({
   client,
-  maxWaitTime: 180
-}, { Bucket });
+  maxWaitTime: 200
+}, { Bucket });
+
+// Waiter configuration: delay
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 40
+}, {
+  Bucket
+});
+
+// Waiter configuration: maxAttempts
+await waitUntilBucketExists({
+  client,
+  maxWaitTime: 100
+}, {
+  Bucket
+});
+
+// Waiter configuration: delay+maxAttempts
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 20
+}, {
+  Bucket
+});
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/service-import-equals.input.ts b/src/transforms/v2-to-v3/__fixtures__/waiters/service-import-equals.input.ts
@@ -5,4 +5,13 @@ const client = new S3({ region: "REGION" });
 
 await client.waitFor("bucketNotExists", { Bucket }).promise();
 await client.createBucket({ Bucket }).promise();
-await client.waitFor("bucketExists", { Bucket }).promise();
+await client.waitFor("bucketExists", { Bucket }).promise();
+
+// Waiter configuration: delay
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2 } }).promise();
+
+// Waiter configuration: maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { maxAttempts: 5 } }).promise();
+
+// Waiter configuration: delay+maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2, maxAttempts: 5 } }).promise();
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/service-import-equals.output.ts b/src/transforms/v2-to-v3/__fixtures__/waiters/service-import-equals.output.ts
@@ -11,10 +11,36 @@ const client = new S3({ region: "REGION" });
 
 await waitUntilBucketNotExists({
   client,
-  maxWaitTime: 180
+  maxWaitTime: 200
 }, { Bucket });
 await client.createBucket({ Bucket });
 await waitUntilBucketExists({
   client,
-  maxWaitTime: 180
-}, { Bucket });
+  maxWaitTime: 200
+}, { Bucket });
+
+// Waiter configuration: delay
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 40
+}, {
+  Bucket
+});
+
+// Waiter configuration: maxAttempts
+await waitUntilBucketExists({
+  client,
+  maxWaitTime: 100
+}, {
+  Bucket
+});
+
+// Waiter configuration: delay+maxAttempts
+await waitUntilBucketExists({
+  client,
+  minDelay: 2,
+  maxWaitTime: 20
+}, {
+  Bucket
+});
diff --git a/src/transforms/v2-to-v3/__fixtures__/waiters/service-import.input.js b/src/transforms/v2-to-v3/__fixtures__/waiters/service-import.input.js
@@ -5,4 +5,13 @@ const client = new S3({ region: "REGION" });
 
 await client.waitFor("bucketNotExists", { Bucket }).promise();
 await client.createBucket({ Bucket }).promise();
-await client.waitFor("bucketExists", { Bucket }).promise();
+await client.waitFor("bucketExists", { Bucket }).promise();
+
+// Waiter configuration: delay
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2 } }).promise();
+
+// Waiter configuration: maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { maxAttempts: 5 } }).promise();
+
+// Waiter configuration: delay+maxAttempts
+await client.waitFor("bucketExists", { Bucket, $waiter: { delay: 2, maxAttempts: 5 } }).promise();