ZacharyZhang-NYandClaude Opus 5.5
4a6641fd49
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
67 lines
4.1 KiB
TypeScript
67 lines
4.1 KiB
TypeScript
import { EventEmitter } from "node:events";
|
|
import { randomUUID } from "node:crypto";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { describe, expect, test, vi } from "vitest";
|
|
|
|
vi.mock("electron", () => ({
|
|
net: {
|
|
request: () => {
|
|
const request = new EventEmitter() as EventEmitter & { setHeader: () => void; write: () => void; end: () => void; abort: () => void };
|
|
request.setHeader = () => undefined;
|
|
request.write = () => undefined;
|
|
request.abort = () => undefined;
|
|
request.end = () => {
|
|
const response = new EventEmitter() as EventEmitter & { statusCode: number };
|
|
response.statusCode = 503;
|
|
request.emit("response", response);
|
|
response.emit("data", Buffer.from(JSON.stringify({ error: "Service Unavailable", detail: "busy" }, null, 2)));
|
|
response.emit("end");
|
|
};
|
|
return request;
|
|
},
|
|
},
|
|
}));
|
|
|
|
const { DatabaseConnection } = await import("../../src/main/database/connection");
|
|
const { OnboardingService } = await import("../../src/main/services/onboarding-service");
|
|
const { ProviderService } = await import("../../src/main/services/provider-service");
|
|
const { providerAdapters } = await import("../../src/main/services/provider-adapters");
|
|
const { IntegrationKeyStore } = await import("../../src/main/services/integration-key-store");
|
|
const { claimNextQueuedRun } = await import("../../src/main/app/task-queue-worker");
|
|
const { executeComposioTool } = await import("../../src/main/services/composio/composio-tool-bridge");
|
|
const { withRunControl } = await import("../../src/main/services/run-control");
|
|
|
|
const vault = { encryptSecret: (secret: string) => Buffer.from(secret, "utf8"), decryptSecret: (ciphertext: Buffer) => ciphertext.toString("utf8") };
|
|
|
|
describe("a run tells the user which capabilities it lacked", () => {
|
|
test("a connector whose actions cannot be listed is named on the run, as well as told to the agent", async () => {
|
|
const connection = new DatabaseConnection(join(tmpdir(), `ac-capability-${randomUUID()}.db`));
|
|
const providers = new ProviderService(connection.db, providerAdapters, vault);
|
|
const onboarding = new OnboardingService({ db: connection.db, saveProviderCredential: (input) => providers.saveCredential(input) });
|
|
let orgId = "";
|
|
let ceoId = "";
|
|
connection.transaction(() => {
|
|
const summary = onboarding.bootstrap({
|
|
companyName: "FlowPilot", companyDescription: "d", employeeName: "Yang", workspacePath: "", executionMode: "ai_ceo", autoApproveHires: true,
|
|
reviewDeliverables: false, goalTitle: "Ship.", goalDescription: "d", providerKey: "openai", providerSecret: "sk-test", providerModel: "gpt-5.4",
|
|
});
|
|
orgId = summary.orgId;
|
|
ceoId = summary.ceoAgentId;
|
|
const now = new Date().toISOString();
|
|
connection.db.prepare("insert into connector_accounts(id, org_id, toolkit_slug, toolkit_name, label, auth_config_id, status, created_at, updated_at) values ('ca_gh', ?, 'github', 'GitHub', 'main', 'ac_gh', 'ACTIVE', ?, ?)").run(orgId, now, now);
|
|
});
|
|
const keys = new IntegrationKeyStore(connection.db, vault);
|
|
keys.write("composio", "ak_capability");
|
|
const run = claimNextQueuedRun(connection)!;
|
|
const output = await withRunControl(run.id, () => executeComposioTool(connection.db, keys, orgId, ceoId, { id: "c1", name: "composio_find_actions", arguments: { query: "create pull request" } }), 60_000);
|
|
expect(output).toMatch(/^ERROR: .*503/);
|
|
const warnings = (connection.db.prepare("select capability_warnings from task_runs where id = ?").get(run.id) as { capability_warnings: string }).capability_warnings;
|
|
expect(warnings).toMatch(/^The actions of github could not be listed: .*503/);
|
|
expect(warnings.split("\n")).toHaveLength(1);
|
|
await withRunControl(run.id, () => executeComposioTool(connection.db, keys, orgId, ceoId, { id: "c2", name: "composio_find_actions", arguments: { query: "again" } }), 60_000);
|
|
expect((connection.db.prepare("select capability_warnings from task_runs where id = ?").get(run.id) as { capability_warnings: string }).capability_warnings).toBe(warnings);
|
|
connection.close();
|
|
});
|
|
});
|