From e85019f236f80f4298ee73f17a37e087c1169585 Mon Sep 17 00:00:00 2001
From: George Fu <kuhe@users.noreply.github.com>
Date: Wed, 7 Aug 2024 11:18:09 -0400
Subject: [PATCH] test(middleware-sdk-s3): add e2e test for throw-200-errors
 (#6360)

---
 .../src/throw-200-exceptions.e2e.spec.ts      | 57 +++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 packages/middleware-sdk-s3/src/throw-200-exceptions.e2e.spec.ts

diff --git a/packages/middleware-sdk-s3/src/throw-200-exceptions.e2e.spec.ts b/packages/middleware-sdk-s3/src/throw-200-exceptions.e2e.spec.ts
new file mode 100644
index 000000000000..c4e32a64bb91
--- /dev/null
+++ b/packages/middleware-sdk-s3/src/throw-200-exceptions.e2e.spec.ts
@@ -0,0 +1,57 @@
+import { S3 } from "@aws-sdk/client-s3";
+import { GetCallerIdentityCommandOutput, STS } from "@aws-sdk/client-sts";
+
+jest.setTimeout(100000);
+
+describe("S3 throw 200 exceptions", () => {
+  const config = {
+    region: "us-west-2",
+  };
+  const s3 = new S3(config);
+  const stsClient = new STS(config);
+
+  const alphabet = "abcdefghijklmnopqrstuvwxyz";
+  const randId = alphabet[(Math.random() * alphabet.length) | 0] + alphabet[(Math.random() * alphabet.length) | 0];
+  let Bucket: string;
+  let callerID: GetCallerIdentityCommandOutput;
+
+  beforeAll(async () => {
+    callerID = await stsClient.getCallerIdentity({});
+    Bucket = `${callerID.Account}-${randId}-s3-200s-e2e-test-empty-${config.region}-${(Date.now() / 1000) | 0}`;
+
+    await s3.createBucket({
+      Bucket,
+    });
+  });
+
+  afterAll(async () => {
+    const list = await s3.listObjectsV2({
+      Bucket,
+    });
+    for (const item of list.Contents ?? []) {
+      await s3.deleteObject({
+        Bucket,
+        Key: item.Key,
+      });
+    }
+
+    await s3.deleteBucket({
+      Bucket,
+    });
+    s3.destroy();
+  });
+
+  it("should split stream successfully for less than 3kb payload and greater than 3kb payload", async () => {
+    for (let i = 0; i < 10; ++i) {
+      await s3.listObjects({
+        Bucket,
+      });
+
+      await s3.putObject({
+        Bucket,
+        Key: i + "long-text-".repeat(10),
+        Body: "abcd",
+      });
+    }
+  });
+});