test: create e2e global hook

kuhe · kuhe · commit 595eb60b9571 · 2024-10-31T00:24:38.000Z
diff --git a/clients/client-cognito-identity/test/e2e/CognitoIdentity.e2e.spec.ts b/clients/client-cognito-identity/test/e2e/CognitoIdentity.e2e.spec.ts
@@ -1,3 +1,4 @@
+import { ready } from "@aws-sdk/aws-util-test";
 import { beforeAll, describe, expect, test as it } from "vitest";
 
 import { getIntegTestResources } from "../../../../tests/e2e/get-integ-test-resources";
@@ -9,6 +10,7 @@ describe("@aws-sdk/client-cognito-identity", () => {
   let IdentityPoolId: string;
 
   beforeAll(async () => {
+    await ready();
     const integTestResourcesEnv = await getIntegTestResources();
     Object.assign(process.env, integTestResourcesEnv);
 
diff --git a/clients/client-kinesis/test/Kinesis.e2e.spec.ts b/clients/client-kinesis/test/Kinesis.e2e.spec.ts
@@ -1,8 +1,13 @@
-import { describe, expect, test as it } from "vitest";
+import { ready } from "@aws-sdk/aws-util-test";
+import { beforeAll, describe, expect, test as it } from "vitest";
 
 import { KinesisClient, ListStreamsCommand } from "../src/index";
 
 describe("@aws-sdk/client-kinesis", () => {
+  beforeAll(async () => {
+    await ready();
+  });
+
   const client = new KinesisClient({
     region: "us-west-2",
     credentials: {
diff --git a/clients/client-s3/test/e2e/S3.browser.e2e.spec.ts b/clients/client-s3/test/e2e/S3.browser.e2e.spec.ts
@@ -1,7 +1,4 @@
-/**
- * This is the integration test that make sure the client can make request cross-platform-ly
- * in NodeJS and browsers.
- */
+import { ready } from "@aws-sdk/aws-util-test";
 import type { S3, SelectObjectContentEventStream } from "@aws-sdk/client-s3";
 import { fromNodeProviderChain } from "@aws-sdk/credential-providers";
 import { FetchHttpHandler } from "@smithy/fetch-http-handler";
@@ -20,6 +17,7 @@ describe("@aws-sdk/client-s3", () => {
   let Key = `${Date.now()}`;
 
   beforeAll(async () => {
+    await ready();
     const integTestResourcesEnv = await getIntegTestResources();
     Object.assign(process.env, integTestResourcesEnv);
 
diff --git a/clients/client-s3/test/e2e/S3.e2e.spec.ts b/clients/client-s3/test/e2e/S3.e2e.spec.ts
@@ -1,5 +1,6 @@
 import "@aws-sdk/signature-v4-crt";
 
+import { ready } from "@aws-sdk/aws-util-test";
 import { S3, SelectObjectContentEventStream } from "@aws-sdk/client-s3";
 import { AwsCredentialIdentity } from "@aws-sdk/types";
 import { afterAll, afterEach, beforeAll, describe, expect, test as it } from "vitest";
@@ -17,6 +18,7 @@ describe("@aws-sdk/client-s3", () => {
   let mrapArn: string;
 
   beforeAll(async () => {
+    await ready();
     const integTestResourcesEnv = await getIntegTestResources();
     Object.assign(process.env, integTestResourcesEnv);
 
diff --git a/clients/client-transcribe-streaming/test/index.e2e.spec.ts b/clients/client-transcribe-streaming/test/index.e2e.spec.ts
@@ -1,6 +1,7 @@
+import { ready } from "@aws-sdk/aws-util-test";
 import { createReadStream } from "fs";
 import { join } from "path";
-import { afterAll, describe, expect, test as it } from "vitest";
+import { afterAll, beforeAll, describe, expect, test as it } from "vitest";
 
 import { TranscribeStreaming } from "../src/index";
 const audio = createReadStream(join(__dirname, "numbers.wav"));
@@ -13,6 +14,10 @@ describe("TranscribeStream client", () => {
       secretAccessKey: "CLIENT_TEST",
     },
   });
+
+  beforeAll(async () => {
+    await ready();
+  });
   afterAll(() => {
     client.destroy();
   });
diff --git a/lib/lib-dynamodb/src/test/lib-dynamodb.e2e.spec.ts b/lib/lib-dynamodb/src/test/lib-dynamodb.e2e.spec.ts
@@ -1,3 +1,4 @@
+import { ready } from "@aws-sdk/aws-util-test";
 import {
   BillingMode,
   CreateTableCommandOutput,
@@ -196,6 +197,7 @@ describe(
     const passError = (e: any) => e;
 
     beforeAll(async () => {
+      await ready();
       log.describe = await dynamodb
         .describeTable({
           TableName,
diff --git a/lib/lib-storage/src/lib-storage.e2e.spec.ts b/lib/lib-storage/src/lib-storage.e2e.spec.ts
@@ -1,3 +1,4 @@
+import { ready } from "@aws-sdk/aws-util-test";
 import { S3 } from "@aws-sdk/client-s3";
 import { Upload } from "@aws-sdk/lib-storage";
 import type { AwsCredentialIdentity } from "@smithy/types";
@@ -17,6 +18,7 @@ describe("@aws-sdk/lib-storage", () => {
   let credentials: AwsCredentialIdentity;
 
   beforeAll(async () => {
+    await ready();
     const integTestResourcesEnv = await getIntegTestResources();
     Object.assign(process.env, integTestResourcesEnv);
 
diff --git a/packages/ec2-metadata-service/src/MetadataService.e2e.spec.ts b/packages/ec2-metadata-service/src/MetadataService.e2e.spec.ts
@@ -1,3 +1,4 @@
+import { ready } from "@aws-sdk/aws-util-test";
 import { fromInstanceMetadata } from "@aws-sdk/credential-providers";
 import { beforeAll, describe, expect, test as it, vi } from "vitest";
 
@@ -9,6 +10,7 @@ describe("MetadataService E2E Tests", () => {
   let metadataServiceAvailable: any;
 
   beforeAll(async () => {
+    await ready();
     try {
       await provider();
       metadataServiceAvailable = true;
diff --git a/packages/middleware-sdk-s3/src/region-redirect-middleware.e2e.spec.ts b/packages/middleware-sdk-s3/src/region-redirect-middleware.e2e.spec.ts
@@ -1,3 +1,4 @@
+import { ready } from "@aws-sdk/aws-util-test";
 import { S3 } from "@aws-sdk/client-s3";
 import { GetCallerIdentityCommandOutput, STS } from "@aws-sdk/client-sts";
 import { afterAll, beforeAll, describe, expect, test as it } from "vitest";
@@ -22,6 +23,7 @@ describe("S3 Global Client Test", () => {
   const randId = alphabet[(Math.random() * alphabet.length) | 0] + alphabet[(Math.random() * alphabet.length) | 0];
 
   beforeAll(async () => {
+    await ready();
     callerID = await stsClient.getCallerIdentity({});
     bucketNames = regionConfigs.map((config) => `${callerID.Account}-${randId}-redirect-${config.region}`);
     await Promise.all(bucketNames.map((bucketName, index) => deleteBucket(s3Clients[index], bucketName)));
diff --git a/packages/middleware-sdk-s3/src/s3-expires-middleware.e2e.spec.ts b/packages/middleware-sdk-s3/src/s3-expires-middleware.e2e.spec.ts
@@ -1,3 +1,4 @@
+import { ready } from "@aws-sdk/aws-util-test";
 import { S3 } from "@aws-sdk/client-s3";
 import { GetCallerIdentityCommandOutput, STS } from "@aws-sdk/client-sts";
 import { afterAll, beforeAll, describe, expect, test as it, vi } from "vitest";
@@ -24,6 +25,7 @@ describe("S3 Expires e2e test", () => {
   const randId = alphabet[(Math.random() * alphabet.length) | 0] + alphabet[(Math.random() * alphabet.length) | 0];
 
   beforeAll(async () => {
+    await ready();
     callerID = await stsClient.getCallerIdentity({});
     Bucket = `${callerID.Account}-${randId}-s3-expires`;
     await s3.createBucket({
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,3 +1,4 @@
+import { ready } from "@aws-sdk/aws-util-test";
 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";
@@ -34,6 +35,7 @@ describe("s3 express CRUD test suite", () => {
   let readWriteDeleteRecorder: typeof recorder.calls;
 
   beforeAll(async () => {
+    await ready();
     ({ s3, controller, bucketName, recorder } = await createClientAndRecorder());
 
     await s3.createBucket({
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
@@ -1,3 +1,4 @@
+import { ready } from "@aws-sdk/aws-util-test";
 import { S3 } from "@aws-sdk/client-s3";
 import { GetCallerIdentityCommandOutput, STS } from "@aws-sdk/client-sts";
 import { afterAll, beforeAll, describe, test as it } from "vitest";
@@ -15,6 +16,7 @@ describe("S3 throw 200 exceptions", () => {
   let callerID: GetCallerIdentityCommandOutput;
 
   beforeAll(async () => {
+    await ready();
     callerID = await stsClient.getCallerIdentity({});
     Bucket = `${callerID.Account}-${randId}-s3-200s-e2e-test-empty-${config.region}-${(Date.now() / 1000) | 0}`;
 
diff --git a/packages/s3-presigned-post/src/createPresignedPost.e2e.spec.ts b/packages/s3-presigned-post/src/createPresignedPost.e2e.spec.ts
@@ -1,5 +1,4 @@
-// Run using AWS_SMOKE_TEST_REGION=[Region] AWS_SMOKE_TEST_BUCKET=[Bucket] yarn test:e2e
-// These params are established in /tests/e2e.
+import { ready } from "@aws-sdk/aws-util-test";
 import { NoSuchKey, S3 } from "@aws-sdk/client-s3";
 import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, test as it } from "vitest";
 
@@ -16,6 +15,7 @@ describe(createPresignedPost.name, () => {
   let region: string;
 
   beforeAll(async () => {
+    await ready();
     const integTestResourcesEnv = await getIntegTestResources();
     Object.assign(process.env, integTestResourcesEnv);
 
diff --git a/private/aws-util-test/package.json b/private/aws-util-test/package.json
@@ -19,6 +19,7 @@
   "dependencies": {
     "@aws-sdk/aws-protocoltests-json": "*",
     "@smithy/protocol-http": "^4.1.4",
+    "@smithy/shared-ini-file-loader": "^3.1.8",
     "@smithy/types": "^3.5.0",
     "tslib": "^2.6.2"
   },
diff --git a/private/aws-util-test/src/index.ts b/private/aws-util-test/src/index.ts
@@ -1 +1,2 @@
 export * from "./requests/test-http-handler";
+export * from "./set-test-credentials";
diff --git a/private/aws-util-test/src/set-test-credentials.ts b/private/aws-util-test/src/set-test-credentials.ts
@@ -0,0 +1,16 @@
+import { loadSharedConfigFiles } from "@smithy/shared-ini-file-loader";
+
+export const ready = async () => {
+  const ini = await loadSharedConfigFiles();
+  const profileData = ini.configFile["sdk-integ-test"] ?? ini.credentialsFile["sdk-integ-test"];
+  if (profileData) {
+    console.log("Setting AWS_PROFILE=sdk-integ-test");
+    process.env.AWS_PROFILE = "sdk-integ-test";
+    if (profileData.role_arn) {
+      console.log(`Setting AWS_CONTAINER_CREDENTIALS_RELATIVE_URI=${profileData.role_arn}`);
+      process.env.AWS_CONTAINER_CREDENTIALS_RELATIVE_URI = `/role-arn/${profileData.role_arn}`;
+    }
+  } else {
+    // console.log("AWS_PROFILE is", process.env.AWS_PROFILE);
+  }
+};
diff --git a/tests/e2e/get-integ-test-resources.js b/tests/e2e/get-integ-test-resources.js
@@ -6,22 +6,15 @@ const { S3ControlClient, ListMultiRegionAccessPointsCommand } = require("@aws-sd
 const { ensureTestStack } = require("./ensure-test-stack");
 const { deleteStaleChangesets } = require("./delete-stale-changesets");
 const { loadSharedConfigFiles } = require("@smithy/shared-ini-file-loader");
-const { createCredentialChain, fromIni, fromHttp } = require("@aws-sdk/credential-providers");
 
 exports.getIntegTestResources = async () => {
   const ini = await loadSharedConfigFiles();
   const profileData = ini.configFile["sdk-integ-test"] ?? ini.credentialsFile["sdk-integ-test"];
   if (profileData) {
-    console.log("Setting AWS_PROFILE=sdk-integ-test");
     process.env.AWS_PROFILE = "sdk-integ-test";
     if (profileData.role_arn) {
-      console.log(
-        `Setting AWS_CONTAINER_CREDENTIALS_RELATIVE_URI=${process.env.AWS_CONTAINER_CREDENTIALS_RELATIVE_URI}`
-      );
       process.env.AWS_CONTAINER_CREDENTIALS_RELATIVE_URI = `/role-arn/${profileData.role_arn}`;
     }
-  } else {
-    console.log("AWS_PROFILE is", process.env.AWS_PROFILE);
   }
 
   const region = "us-west-2";
