vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-18 10:34:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: tighten nvidia metadata assertions

Browse Source
This commit is contained in:
Shakker
2026-05-11 03:20:36 +01:00
parent 7c3f447c56
commit fbf7048b72
1 changed files with 25 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
extensions/nvidia/index.test.ts
View File

@@ -8,7 +8,7 @@ import { describe, expect, it } from "vitest";
import plugin from "./index.js";
type NvidiaManifest = {
providerAuthChoices?: Array<{ choiceId?: string; method?: string; provider?: string }>;
providerAuthChoices?: Array<Record<string, unknown>>;
};
type RegisteredModelCatalogProvider = Parameters<
ReturnType<typeof createTestPluginApi>["registerModelCatalogProvider"]
@@ -45,15 +45,21 @@ describe("nvidia provider hooks", () => {
});
expect(choice?.provider.id).toBe("nvidia");
expect(choice?.method.id).toBe("api-key");
expect(readManifest().providerAuthChoices).toEqual(
expect.arrayContaining([
expect.objectContaining({
provider: "nvidia",
method: "api-key",
choiceId: "nvidia-api-key",
}),
]),
);
expect(readManifest().providerAuthChoices).toStrictEqual([
{
provider: "nvidia",
method: "api-key",
choiceId: "nvidia-api-key",
choiceLabel: "NVIDIA API key",
groupId: "nvidia",
groupLabel: "NVIDIA",
groupHint: "Direct API key",
optionKey: "nvidiaApiKey",
cliFlag: "--nvidia-api-key",
cliOption: "--nvidia-api-key <key>",
cliDescription: "NVIDIA API key",
},
]);
});
it("keeps nvidia auth setup metadata aligned", async () => {
@@ -85,20 +91,24 @@ describe("nvidia provider hooks", () => {
it("keeps nvidia wizard setup metadata aligned", async () => {
const provider = await registerNvidiaProvider();
expect(provider.wizard?.setup).toMatchObject({
expect(provider.wizard?.setup).toStrictEqual({
choiceId: "nvidia-api-key",
choiceLabel: "NVIDIA API key",
groupId: "nvidia",
groupLabel: "NVIDIA",
groupHint: "Direct API key",
methodId: "api-key",
modelSelection: {
promptWhenAuthChoiceProvided: true,
allowKeepCurrent: false,
},
});
});
it("keeps nvidia model picker metadata aligned", async () => {
const provider = await registerNvidiaProvider();
expect(provider.wizard?.modelPicker).toMatchObject({
expect(provider.wizard?.modelPicker).toStrictEqual({
label: "NVIDIA (custom)",
hint: "Use NVIDIA-hosted open models",
methodId: "api-key",
@@ -162,9 +172,9 @@ describe("nvidia provider hooks", () => {
}),
);
expect(registeredProviders).toContain("nvidia");
expect(registeredModelCatalogProviders.map((provider) => provider.provider)).toContain(
expect(registeredProviders).toStrictEqual(["nvidia"]);
expect(registeredModelCatalogProviders.map((provider) => provider.provider)).toStrictEqual([
"nvidia",
);
]);
});
});