Yaowei Zheng
76 lines
2.9 KiB
TypeScript
76 lines
2.9 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it } from "vitest";
|
|
import { getMessages, maskApiKey, resolveLanguage } from "../src/i18n.js";
|
|
|
|
describe("resolveLanguage (env PENGUIN_LANG, default en)", () => {
|
|
let prev: string | undefined;
|
|
beforeEach(() => {
|
|
prev = process.env.PENGUIN_LANG;
|
|
});
|
|
afterEach(() => {
|
|
if (prev === undefined) delete process.env.PENGUIN_LANG;
|
|
else process.env.PENGUIN_LANG = prev;
|
|
});
|
|
|
|
it("defaults to en when unset", () => {
|
|
delete process.env.PENGUIN_LANG;
|
|
expect(resolveLanguage()).toBe("en");
|
|
});
|
|
it("matches zh exactly (case-insensitive, trimmed)", () => {
|
|
process.env.PENGUIN_LANG = "zh";
|
|
expect(resolveLanguage()).toBe("zh");
|
|
process.env.PENGUIN_LANG = " ZH ";
|
|
expect(resolveLanguage()).toBe("zh");
|
|
});
|
|
it("falls back to en for non-exact zh prefixes and anything else", () => {
|
|
process.env.PENGUIN_LANG = "zh-CN"; // no longer prefix-matched -> en
|
|
expect(resolveLanguage()).toBe("en");
|
|
process.env.PENGUIN_LANG = "fr";
|
|
expect(resolveLanguage()).toBe("en");
|
|
process.env.PENGUIN_LANG = "en";
|
|
expect(resolveLanguage()).toBe("en");
|
|
});
|
|
});
|
|
|
|
describe("getMessages", () => {
|
|
it("provides zh and en runtime + help strings", () => {
|
|
expect(getMessages("zh").modelAdded("m", "m")).toContain("已添加");
|
|
expect(getMessages("en").modelAdded("m", "m")).toContain("Added");
|
|
expect(getMessages("zh").modelUpdated("m", "m")).toContain("已更新");
|
|
expect(getMessages("en").modelUpdated("m", "m")).toContain("Updated");
|
|
// Command/option descriptions are also localized.
|
|
expect(getMessages("zh").config.addDesc).toContain("模型");
|
|
expect(getMessages("en").config.addDesc).toContain("model");
|
|
expect(getMessages("en").run.desc).toContain("Task");
|
|
// config lang copy.
|
|
expect(getMessages("zh").config.langDesc).toContain("语言");
|
|
expect(getMessages("en").config.langDesc).toContain("language");
|
|
expect(getMessages("en").langSet("zh", "/x/.zshrc")).toContain("/x/.zshrc");
|
|
expect(getMessages("zh").langInvalid("fr")).toContain("fr");
|
|
});
|
|
|
|
it("header shows the version and Agent / Workspace / Model on their own lines", () => {
|
|
for (const lang of ["en", "zh"] as const) {
|
|
const lines = getMessages(lang).header("run", "1.2.3", "ag", "/ws", "mod").split("\n");
|
|
expect(lines).toHaveLength(4);
|
|
expect(lines[0]).toContain("run");
|
|
expect(lines[0]).toContain("v1.2.3");
|
|
expect(lines[1]).toContain("ag");
|
|
expect(lines[2]).toContain("/ws");
|
|
expect(lines[3]).toContain("mod");
|
|
}
|
|
});
|
|
});
|
|
|
|
describe("maskApiKey", () => {
|
|
it("masks all but the last 4 chars", () => {
|
|
expect(maskApiKey("sk-1234567890")).toBe("****7890");
|
|
});
|
|
it("fully masks short keys (≤12 chars would leak most of the secret)", () => {
|
|
expect(maskApiKey("sk-test-1234")).toBe("***");
|
|
expect(maskApiKey("short")).toBe("***");
|
|
});
|
|
it("returns - when absent", () => {
|
|
expect(maskApiKey(undefined)).toBe("-");
|
|
});
|
|
});
|