dynamic test skipping

Author: kuhe

packages/credential-provider-imds/src/fromInstanceMetadata.e2e.spec.ts

@@ -1,4 +1,3 @@
-import { CredentialsProviderError } from "@smithy/property-provider";
 import { afterEach, beforeEach, describe, expect, test as it } from "vitest";
 
 import { fromInstanceMetadata, getMetadataToken } from "./fromInstanceMetadata";
@@ -24,8 +23,11 @@ describe("fromInstanceMetadata (Live EC2 E2E Tests)", () => {
     process.env = { ...originalEnv };
   });
 
-  const test = imdsAvailable ? it : it.skip;
-  test("should fetch metadata token successfully", async () => {
+  it("should fetch metadata token successfully", async (context) => {
+    if (!imdsAvailable) {
+      return context.skip();
+    }
+
     const options = {
       path: "/latest/api/token",
       method: "PUT",
@@ -40,8 +42,10 @@ describe("fromInstanceMetadata (Live EC2 E2E Tests)", () => {
     expect(token.length).toBeGreaterThan(0);
   });
 
-  it("retrieves credentials successfully", async () => {
-    if (!imdsAvailable) return;
+  it("retrieves credentials successfully", async (context) => {
+    if (!imdsAvailable) {
+      return context.skip();
+    }
 
     const provider = fromInstanceMetadata({ timeout: 1000, maxRetries: 2 });
     const credentials = await provider();
@@ -52,24 +56,20 @@ describe("fromInstanceMetadata (Live EC2 E2E Tests)", () => {
     expect(typeof credentials.secretAccessKey).toBe("string");
   });
 
-  it("retrieves credentials with account ID on allowlisted instances", async () => {
-    if (!imdsAvailable) return;
+  it("retrieves credentials with account ID on allowlisted instances", async (context) => {
+    if (!imdsAvailable) {
+      return context.skip();
+    }
 
     const provider = fromInstanceMetadata({ timeout: 1000, maxRetries: 2 });
     const credentials = await provider();
 
     if (!credentials.accountId) {
-      it.skip("account ID test skipped - not an allowlisted instance", () => {});
+      context.skip();
     }
 
     expect(credentials.accountId).toBeDefined();
     expect(typeof credentials.accountId).toBe("string");
-
-    console.log("IMDSv2 Credentials with Account ID:", {
-      accessKeyId: credentials.accessKeyId,
-      sessionToken: credentials.sessionToken?.slice(0, 10) + "...",
-      accountId: credentials.accountId,
-    });
   });
 
   it("IMDS access disabled via AWS_EC2_METADATA_DISABLED", async () => {
@@ -88,23 +88,25 @@ describe("fromInstanceMetadata (Live EC2 E2E Tests)", () => {
     await expect(provider()).rejects.toThrow();
   });
 
-  it("Uses configured profile name from env", async () => {
-    if (!imdsAvailable) return;
+  it("Uses configured profile name from env", async (context) => {
+    if (!imdsAvailable) {
+      return context.skip();
+    }
 
     const provider = fromInstanceMetadata({ timeout: 1000 });
 
     try {
       const credentials = await provider();
       expect(credentials).toHaveProperty("accessKeyId");
-      console.log("Used configured profile name from env.");
     } catch (error) {
       expect(error).toBeDefined();
-      console.log("Profile test completed (profile may not exist).");
     }
   });
 
-  it("Multiple calls return stable results", async () => {
-    if (!imdsAvailable) return;
+  it("Multiple calls return stable results", async (context) => {
+    if (!imdsAvailable) {
+      return context.skip();
+    }
 
     const provider = fromInstanceMetadata({ timeout: 1000 });
     const creds1 = await provider();
@@ -113,12 +115,12 @@ describe("fromInstanceMetadata (Live EC2 E2E Tests)", () => {
     expect(creds1.accessKeyId).toBeTruthy();
     expect(creds2.accessKeyId).toBeTruthy();
     expect(creds1.accessKeyId).toBe(creds2.accessKeyId);
-
-    console.log("Stable credentials returned across calls.");
   });
 
-  it("should timeout as expected when a request exceeds the specified duration", async () => {
-    if (!imdsAvailable) return;
+  it("should timeout as expected when a request exceeds the specified duration", async (context) => {
+    if (!imdsAvailable) {
+      return context.skip();
+    }
     const provider = fromInstanceMetadata({ timeout: 1 });
 
     await expect(provider()).rejects.toThrow(/timeout|timed out|TimeoutError/i);