ZacharyZhang-NYandClaude Opus 5.5
fd79f059ca
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
60 lines
3.5 KiB
TypeScript
60 lines
3.5 KiB
TypeScript
import { randomUUID } from "node:crypto";
|
|
import { DatabaseSync } from "node:sqlite";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { describe, expect, test } from "vitest";
|
|
import { DatabaseConnection } from "../../src/main/database/connection";
|
|
import { OnboardingService } from "../../src/main/services/onboarding-service";
|
|
import { ProviderService } from "../../src/main/services/provider-service";
|
|
import { providerAdapters } from "../../src/main/services/provider-adapters";
|
|
import { completeSetup, loadOnboardingDraft, saveOnboardingDraft } from "../../src/main/services/onboarding-progress";
|
|
import { onboardingProgressSchema } from "../../src/main/database/migrations/105-onboarding-progress";
|
|
import type { OnboardingDraft } from "../../src/shared/commands";
|
|
|
|
const vault = { encryptSecret: (secret: string) => Buffer.from(secret, "utf8"), decryptSecret: (ciphertext: Buffer) => ciphertext.toString("utf8") };
|
|
|
|
const draft: OnboardingDraft = {
|
|
step: "mode", companyName: "Resume Co", companyDescription: "d", employeeName: "Dana", workspacePath: "", executionMode: "ai_ceo",
|
|
providerKey: "openai", providerBaseUrl: "", providerModel: "gpt-5.4", goalTitle: "", goalDescription: "",
|
|
};
|
|
|
|
describe("onboarding progress", () => {
|
|
test("the wizard is kept until launch, and a launched company stays unfinished until its setup is done", () => {
|
|
const connection = new DatabaseConnection(join(tmpdir(), `ac-onboarding-progress-${randomUUID()}.db`));
|
|
const db = connection.db;
|
|
expect(loadOnboardingDraft(db)).toBeNull();
|
|
saveOnboardingDraft(db, draft);
|
|
expect(loadOnboardingDraft(db)).toEqual(draft);
|
|
|
|
const providers = new ProviderService(db, providerAdapters, vault);
|
|
const onboarding = new OnboardingService({ db, saveProviderCredential: (input) => providers.saveCredential(input) });
|
|
let orgId = "";
|
|
connection.transaction(() => {
|
|
orgId = onboarding.bootstrap({
|
|
companyName: "Resume Co", companyDescription: "d", employeeName: "Dana", workspacePath: "", executionMode: "ai_ceo", autoApproveHires: true,
|
|
reviewDeliverables: false, goalTitle: "Ship.", goalDescription: "d", providerKey: "openai", providerSecret: "sk-test", providerModel: "gpt-5.4",
|
|
}).orgId;
|
|
});
|
|
expect(loadOnboardingDraft(db)).toBeNull();
|
|
expect(db.prepare("select setup_completed_at from organizations where id = ?").get(orgId)).toEqual({ setup_completed_at: null });
|
|
completeSetup(db, orgId);
|
|
const done = (db.prepare("select setup_completed_at from organizations where id = ?").get(orgId) as { setup_completed_at: string }).setup_completed_at;
|
|
expect(done).not.toBeNull();
|
|
completeSetup(db, orgId);
|
|
expect(db.prepare("select setup_completed_at from organizations where id = ?").get(orgId)).toEqual({ setup_completed_at: done });
|
|
expect(() => completeSetup(db, "org_missing")).toThrow("Organization was not found.");
|
|
});
|
|
|
|
test("companies that existed before count as set up", () => {
|
|
const db = new DatabaseSync(":memory:");
|
|
db.exec(`
|
|
create table schema_migrations(version integer, name text, applied_at text);
|
|
create table app_state(id integer primary key, updated_at text);
|
|
create table organizations(id text primary key, created_at text);
|
|
insert into organizations(id, created_at) values ('org_a', '2026-01-01T00:00:00.000Z');
|
|
`);
|
|
db.exec(onboardingProgressSchema);
|
|
expect(db.prepare("select setup_completed_at from organizations").get()).toEqual({ setup_completed_at: "2026-01-01T00:00:00.000Z" });
|
|
});
|
|
});
|