Skip to content

test(credential-provider-node): attempt 2 to migrate integ tests to vitest #7332

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from
Closed

test(credential-provider-node): attempt 2 to migrate integ tests to vitest #7332

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 0 additions & 4 deletions packages/credential-provider-node/jest.config.integ.js
View file
Open in desktop

This file was deleted.

5 changes: 3 additions & 2 deletions packages/credential-provider-node/package.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,9 @@
"build:types:downlevel": "downlevel-dts dist-types dist-types/ts3.4",
"clean": "rimraf ./dist-* && rimraf *.tsbuildinfo",
"test": "yarn g:vitest run",
"test:integration": "yarn g:jest -c jest.config.integ.js",
"test:watch": "yarn g:vitest watch"
"test:watch": "yarn g:vitest watch",
"test:integration": "yarn g:vitest run -c vitest.config.integ.mts",
"test:integration:watch": "yarn g:vitest watch -c vitest.config.integ.mts"
},
"keywords": [
"aws",
Expand Down
145 changes: 67 additions & 78 deletions packages/credential-provider-node/src/credential-provider-node.integ.spec.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,69 +6,60 @@ import type { SharedConfigInit, SourceProfileInit } from "@smithy/shared-ini-fil
import type { HttpRequest, NodeHttpHandlerOptions, ParsedIniData, SharedConfigFiles } from "@smithy/types";
import { AdaptiveRetryStrategy, StandardRetryStrategy } from "@smithy/util-retry";
import { PassThrough } from "stream";
import { afterAll, afterEach, beforeAll, beforeEach, describe, expect, test as it, vi } from "vitest";

import { defaultProvider } from "./defaultProvider";

jest.mock("fs", () => {
const actual = jest.requireActual("fs");
return {
...actual,
readFileSync(file: string, ...options: any[]) {
if (file === "token-filepath") {
return "token-contents";
}
return actual.readFileSync(file, ...options);
},
};
});
vi.mock("fs", async () => ({
readFileSync: vi.fn((file: string, ...options: any[]) => {
if (file === "token-filepath") {
return "token-contents";
}
}),
}));

let iniProfileData: ParsedIniData = null as any;
jest.mock("@smithy/shared-ini-file-loader", () => {
const actual = jest.requireActual("@smithy/shared-ini-file-loader");
return {
...actual,
async loadSsoSessionData() {
return Object.entries(iniProfileData)
.filter(([key]) => key.startsWith("sso-session."))
.reduce(
(acc, [key, value]) => ({
...acc,
[key.split("sso-session.")[1]]: value,
}),
{}
);
},
async parseKnownFiles(init: SourceProfileInit): Promise<ParsedIniData> {
return iniProfileData;
},
async loadSharedConfigFiles(init: SharedConfigInit): Promise<SharedConfigFiles> {
return {
configFile: iniProfileData,
credentialsFile: iniProfileData,
};
},
async getSSOTokenFromFile() {
return {
accessToken: "mock_sso_token",
expiresAt: "3000-01-01T00:00:00.000Z",
};
},
};
});
vi.mock("@smithy/shared-ini-file-loader", () => ({
loadSsoSessionData: vi.fn(async () => {
return Object.entries(iniProfileData)
.filter(([key]) => key.startsWith("sso-session."))
.reduce(
(acc, [key, value]) => ({
...acc,
[key.split("sso-session.")[1]]: value,
}),
{}
);
}),
parseKnownFiles: vi.fn(async (init: SourceProfileInit): Promise<ParsedIniData> => {
return iniProfileData;
}),
loadSharedConfigFiles: vi.fn(async (init: SharedConfigInit): Promise<SharedConfigFiles> => {
return {
configFile: iniProfileData,
credentialsFile: iniProfileData,
};
}),
getSSOTokenFromFile: vi.fn(async () => {
return {
accessToken: "mock_sso_token",
expiresAt: "3000-01-01T00:00:00.000Z",
};
}),
}));

const assumeRoleArns: string[] = [];

jest.mock("@smithy/node-http-handler", () => {
const actual = jest.requireActual("@smithy/node-http-handler");

vi.mock("@smithy/node-http-handler", async () => {
class MockNodeHttpHandler {
static create(instanceOrOptions?: any) {
if (typeof instanceOrOptions?.handle === "function") {
return instanceOrOptions;
}
return new MockNodeHttpHandler();
}
async handle(request: HttpRequest) {

handle = vi.fn(async (request: HttpRequest) => {
const body = new PassThrough({});

if (request.body?.includes("RoleArn=")) {
Expand Down Expand Up @@ -165,38 +156,36 @@ jest.mock("@smithy/node-http-handler", () => {
headers: {},
}),
};
}
updateHttpClientConfig(key: keyof NodeHttpHandlerOptions, value: NodeHttpHandlerOptions[typeof key]): void {}
httpHandlerConfigs(): NodeHttpHandlerOptions {
});

updateHttpClientConfig = vi.fn(
(key: keyof NodeHttpHandlerOptions, value: NodeHttpHandlerOptions[typeof key]): void => {}
);

httpHandlerConfigs = vi.fn((): NodeHttpHandlerOptions => {
return null as any;
}
});
}

return {
...actual,
NodeHttpHandler: MockNodeHttpHandler,
};
});

jest.mock("child_process", () => {
const actual = jest.requireActual("child_process");
return {
...actual,
exec(bin: string, cb: (err: unknown, data: any) => void, ...args: any[]) {
if (bin === "credential-process") {
return cb(null, {
stdout: JSON.stringify({
Version: 1,
AccessKeyId: "PROCESS_ACCESS_KEY_ID",
SecretAccessKey: "PROCESS_SECRET_ACCESS_KEY",
SessionToken: "PROCESS_SESSION_TOKEN",
}),
});
}
return actual.exec(bin, cb, ...args);
},
};
});
vi.mock("child_process", () => ({
exec: vi.fn((bin: string, cb: (err: unknown, data: any) => void, ...args: any[]) => {
if (bin === "credential-process") {
return cb(null, {
stdout: JSON.stringify({
Version: 1,
AccessKeyId: "PROCESS_ACCESS_KEY_ID",
SecretAccessKey: "PROCESS_SECRET_ACCESS_KEY",
SessionToken: "PROCESS_SESSION_TOKEN",
}),
});
}
}),
}));

describe("credential-provider-node integration test", () => {
let sts: STS = null as any;
Expand Down Expand Up @@ -271,8 +260,8 @@ describe("credential-provider-node integration test", () => {
});

afterAll(async () => {
jest.clearAllMocks();
jest.clearAllTimers();
vi.clearAllMocks();
vi.clearAllTimers();
});

describe("fromEnv", () => {
Expand Down Expand Up @@ -570,7 +559,7 @@ describe("credential-provider-node integration test", () => {
iniProfileData.credential_source_profile = {
credential_source: "EcsContainer",
};
const spy = jest.spyOn(credentialProviderHttp, "fromHttp");
const spy = vi.spyOn(credentialProviderHttp, "fromHttp");
sts = new STS({
region: "us-west-2",
credentials: defaultProvider({
Expand Down Expand Up @@ -658,7 +647,7 @@ describe("credential-provider-node integration test", () => {
role_arn: "ROLE_ARN_1",
};

const spy = jest.spyOn(credentialProviderHttp, "fromHttp");
const spy = vi.spyOn(credentialProviderHttp, "fromHttp");
sts = new STS({
region: "us-west-2",
credentials: defaultProvider({
Expand Down Expand Up @@ -710,7 +699,7 @@ describe("credential-provider-node integration test", () => {
// This scenario tests the option of having no role_arn in this step of the chain.
};

const spy = jest.spyOn(credentialProviderHttp, "fromHttp");
const spy = vi.spyOn(credentialProviderHttp, "fromHttp");
sts = new STS({
region: "us-west-2",
credentials: defaultProvider({
Expand Down Expand Up @@ -805,7 +794,7 @@ describe("credential-provider-node integration test", () => {
});
});

xit("should use instance metadata unless IMDS is disabled", async () => {
it.skip("should use instance metadata unless IMDS is disabled", async () => {
// TODO
});
});
Expand Down
Loading