test(middleware-sdk-s3): add waiter for s3 express e2e test (#6051)

aws · May 2, 2024 · 003de3f · 003de3f
1 parent 190768a
commit 003de3f
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/packages/middleware-sdk-s3/src/s3-express/middleware-s3-express.e2e.spec.ts b/packages/middleware-sdk-s3/src/s3-express/middleware-s3-express.e2e.spec.ts
@@ -1,4 +1,4 @@
-import { GetObjectCommand, PutObjectCommand, S3 } from "@aws-sdk/client-s3";
+import { GetObjectCommand, PutObjectCommand, S3, waitUntilBucketExists } from "@aws-sdk/client-s3";
 import { STS } from "@aws-sdk/client-sts";
 import { getSignedUrl } from "@aws-sdk/s3-request-presigner";
 import http from "http";
@@ -51,6 +51,10 @@ describe("s3 express CRUD test suite", () => {
         },
       },
     });
+
+    // Wait for the bucket to exist
+    await waitUntilBucketExists({ client: s3, maxWaitTime: 120 }, { Bucket: bucketName });
+
     createRecorder = JSON.parse(JSON.stringify(recorder.calls));
     reset();
 
@@ -103,14 +107,17 @@ describe("s3 express CRUD test suite", () => {
       "CreateBucketCommand (normal)": {
         [bucketName]: 1,
       },
+      "HeadBucketCommand (s3 express)": {
+        [bucketName]: 1,
+      },
+      "CreateSessionCommand (normal)": {
+        [bucketName]: 1,
+      },
     });
   });
 
   it("can read/write/delete from a bucket", () => {
     expect(readWriteDeleteRecorder).toEqual({
-      "CreateSessionCommand (normal)": {
-        [bucketName]: 1,
-      },
       "PutObjectCommand (s3 express)": {
         [bucketName]: SCALE,
       },