@tyvm/knowhow 0.0.148 → 0.0.149

package/ts_build/tests/unit/auth/registerIdentity.test.js

@@ -0,0 +1,86 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs_1 = __importDefault(require("fs"));
+const os_1 = __importDefault(require("os"));
+const path_1 = __importDefault(require("path"));
+const registerPublicKey = jest.fn();
+const exchangePublicKeyJwt = jest.fn();
+const storeJwt = jest.fn();
+const getOrCreatePublicKey = jest.fn();
+const getConfig = jest.fn();
+const updateConfig = jest.fn();
+jest.mock("../../../src/auth/keyAuth", () => ({ registerPublicKey, exchangePublicKeyJwt, storeJwt }));
+jest.mock("../../../src/auth/keyManager", () => ({ getOrCreatePublicKey }));
+jest.mock("../../../src/config", () => ({ getConfig, updateConfig }));
+const registerIdentity_1 = require("../../../src/auth/registerIdentity");
+function makeJwt(payload) {
+    return `${Buffer.from("{}").toString("base64url")}.${Buffer.from(JSON.stringify(payload)).toString("base64url")}.sig`;
+}
+describe("registerIdentity", () => {
+    const originalCwd = process.cwd();
+    let workDir;
+    const sourceJwt = makeJwt({ org: "org-test" });
+    const publicKeyBase64 = Buffer.alloc(32, 5).toString("base64");
+    beforeEach(() => {
+        jest.clearAllMocks();
+        jest.spyOn(console, "log").mockImplementation(() => undefined);
+        workDir = fs_1.default.mkdtempSync(path_1.default.join(os_1.default.tmpdir(), "knowhow-register-identity-"));
+        fs_1.default.mkdirSync(path_1.default.join(workDir, ".knowhow"), { recursive: true });
+        fs_1.default.writeFileSync(path_1.default.join(workDir, ".knowhow", ".jwt"), sourceJwt);
+        process.chdir(workDir);
+        getOrCreatePublicKey.mockReturnValue({
+            publicKeyBase64,
+            fingerprint: "SHA256:test",
+            privateKeyPath: path_1.default.join(workDir, ".knowhow", "keys", "id_ed25519"),
+        });
+        registerPublicKey.mockResolvedValue(undefined);
+        exchangePublicKeyJwt.mockResolvedValue("identity.jwt.token");
+        getConfig.mockResolvedValue({ modelProviders: [] });
+        updateConfig.mockResolvedValue(undefined);
+    });
+    afterEach(() => {
+        process.chdir(originalCwd);
+        fs_1.default.rmSync(workDir, { recursive: true, force: true });
+        jest.restoreAllMocks();
+    });
+    it("registers from the project JWT and stores the exchanged JWT", async () => {
+        await (0, registerIdentity_1.registerIdentity)({ apiUrl: "https://api.example" });
+        expect(registerPublicKey).toHaveBeenCalledWith(sourceJwt, publicKeyBase64, "https://api.example");
+        expect(exchangePublicKeyJwt).toHaveBeenCalledWith("org-test", "https://api.example", undefined);
+        expect(storeJwt).toHaveBeenCalledWith("identity.jwt.token");
+        expect(updateConfig).toHaveBeenCalledWith(expect.objectContaining({
+            orgId: "org-test",
+            modelProviders: [{ provider: "knowhow" }],
+            cliIdentityPath: expect.stringContaining("id_ed25519"),
+        }));
+    });
+    it("uses an explicitly selected identity", async () => {
+        await (0, registerIdentity_1.registerIdentity)({ identityPath: "/keys/id", apiUrl: "https://api.example" });
+        expect(getOrCreatePublicKey).toHaveBeenCalledWith("/keys/id");
+        expect(exchangePublicKeyJwt).toHaveBeenCalledWith("org-test", "https://api.example", "/keys/id");
+    });
+    it("requires a nonempty project JWT", async () => {
+        fs_1.default.writeFileSync(path_1.default.join(workDir, ".knowhow", ".jwt"), " ");
+        await expect((0, registerIdentity_1.registerIdentity)()).rejects.toThrow("JWT file is empty");
+        expect(registerPublicKey).not.toHaveBeenCalled();
+    });
+    it("does not replace the JWT when registration fails", async () => {
+        registerPublicKey.mockRejectedValue(new Error("registration failed"));
+        await expect((0, registerIdentity_1.registerIdentity)()).rejects.toThrow("registration failed");
+        expect(storeJwt).not.toHaveBeenCalled();
+    });
+});
+describe("JWT organization extraction", () => {
+    it("decodes payload and accepts canonical org", () => {
+        expect((0, registerIdentity_1.decodeJwtPayload)(makeJwt({ org: "one" }))).toMatchObject({ org: "one" });
+        expect((0, registerIdentity_1.extractOrgIdFromJwt)(makeJwt({ org: "one" }))).toBe("one");
+    });
+    it("rejects malformed and organization-less JWTs", () => {
+        expect(() => (0, registerIdentity_1.decodeJwtPayload)("invalid")).toThrow("Invalid JWT format");
+        expect(() => (0, registerIdentity_1.extractOrgIdFromJwt)(makeJwt({ sub: "user" }))).toThrow("organization claim");
+    });
+});
+//# sourceMappingURL=registerIdentity.test.js.map

package/ts_build/tests/unit/auth/registerIdentity.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registerIdentity.test.js","sourceRoot":"","sources":["../../../../tests/unit/auth/registerIdentity.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAoB;AACpB,4CAAoB;AACpB,gDAAwB;AAExB,MAAM,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AACpC,MAAM,oBAAoB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AAC3B,MAAM,oBAAoB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AACvC,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AAC5B,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;AAE/B,IAAI,CAAC,IAAI,CAAC,2BAA2B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,oBAAoB,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;AACtG,IAAI,CAAC,IAAI,CAAC,8BAA8B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC;AAC5E,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;AAEtE,yEAA6G;AAE7G,SAAS,OAAO,CAAC,OAAgC;IAC/C,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,MAAM,CAAC;AACxH,CAAC;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAClC,IAAI,OAAe,CAAC;IACpB,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,CAAC;IAC/C,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAE/D,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QAC/D,OAAO,GAAG,YAAE,CAAC,WAAW,CAAC,cAAI,CAAC,IAAI,CAAC,YAAE,CAAC,MAAM,EAAE,EAAE,4BAA4B,CAAC,CAAC,CAAC;QAC/E,YAAE,CAAC,SAAS,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAClE,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC;QACpE,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACvB,oBAAoB,CAAC,eAAe,CAAC;YACnC,eAAe;YACf,WAAW,EAAE,aAAa;YAC1B,cAAc,EAAE,cAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,YAAY,CAAC;SACrE,CAAC,CAAC;QACH,iBAAiB,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC/C,oBAAoB,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,CAAC;QAC7D,SAAS,CAAC,iBAAiB,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC;QACpD,YAAY,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC3B,YAAE,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,IAAA,mCAAgB,EAAC,EAAE,MAAM,EAAE,qBAAqB,EAAE,CAAC,CAAC;QAC1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,SAAS,EAAE,eAAe,EAAE,qBAAqB,CAAC,CAAC;QAClG,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,qBAAqB,EAAE,SAAS,CAAC,CAAC;QAChG,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,oBAAoB,CAAC,CAAC;QAC5D,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAChE,KAAK,EAAE,UAAU;YACjB,cAAc,EAAE,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC;YACzC,eAAe,EAAE,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC;SACvD,CAAC,CAAC,CAAC;IACN,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,IAAA,mCAAgB,EAAC,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,EAAE,qBAAqB,EAAE,CAAC,CAAC;QACpF,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAAC,UAAU,EAAE,qBAAqB,EAAE,UAAU,CAAC,CAAC;IACnG,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,YAAE,CAAC,aAAa,CAAC,cAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,EAAE,GAAG,CAAC,CAAC;QAC9D,MAAM,MAAM,CAAC,IAAA,mCAAgB,GAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QACtE,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,iBAAiB,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;QACtE,MAAM,MAAM,CAAC,IAAA,mCAAgB,GAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC;QACxE,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,CAAC,IAAA,mCAAgB,EAAC,OAAO,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;QAChF,MAAM,CAAC,IAAA,sCAAmB,EAAC,OAAO,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,mCAAgB,EAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;QACxE,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,sCAAmB,EAAC,OAAO,CAAC,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;IAC5F,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}