root/fusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(FN-4772): add hybrid wiring integration and lifecycle coverage

Browse Source 
Fusion-Task-Id: FN-4772
Fusion-Task-Lineage: d296e96a-fe44-4896-928a-fa44da62b41a
This commit is contained in:
Fusion (runfusion.ai)
2026-05-16 11:17:49 -07:00
committed by gsxdsm
parent 19d1b65e3f
commit 55575dbf8c
5 changed files with 176 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
packages/cli/src/commands/__tests__/daemon.test.ts
View File

@@ -4,10 +4,16 @@ import { mkdtempSync, rmSync } from "node:fs";
import { join } from "node:path";
import { tmpdir } from "node:os";
const { mockSyncStartupModels } = vi.hoisted(() => ({
const { mockSyncStartupModels, mockShouldUseHybridExecutor, mockHybridExecutorCtor, mockHybridExecutorInitialize, mockHybridExecutorShutdown } = vi.hoisted(() => ({
mockSyncStartupModels: vi.fn().mockResolvedValue(undefined),
mockShouldUseHybridExecutor: vi.fn().mockResolvedValue({ enabled: false, reason: "single-project-local-only" }),
mockHybridExecutorInitialize: vi.fn().mockResolvedValue(undefined),
mockHybridExecutorShutdown: vi.fn().mockResolvedValue(undefined),
mockHybridExecutorCtor: vi.fn().mockImplementation(() => ({
initialize: mockHybridExecutorInitialize,
shutdown: mockHybridExecutorShutdown,
})),
}));
vi.mock("../startup-model-sync.js", () => ({
syncStartupModels: mockSyncStartupModels,
}));
@@ -596,9 +602,10 @@ vi.mock("@fusion/engine", async (importOriginal) => {
createAiPromptExecutor: mocks.createAiPromptExecutorMock,
HeartbeatMonitor: mocks.heartbeatMonitorCtor,
HeartbeatTriggerScheduler: mocks.heartbeatTriggerSchedulerCtor,
shouldUseHybridExecutor: mockShouldUseHybridExecutor,
HybridExecutor: mockHybridExecutorCtor,
});
});
vi.mock("@mariozechner/pi-coding-agent", () => ({
AuthStorage: {
create: vi.fn(() => mocks.authStorage),
@@ -889,6 +896,23 @@ describe("runDaemon", () => {
expect(listenCall.server.close).toHaveBeenCalledTimes(1);
expect(mocks.taskStores[0].close).toHaveBeenCalledTimes(1);
});
it("enables HybridExecutor with env override and shuts down before engine stop", async () => {
process.env.FUSION_HYBRID_EXECUTOR = "1";
mockShouldUseHybridExecutor.mockResolvedValue({ enabled: true, reason: "env-override" });
await runDaemon({});
expect(mockHybridExecutorCtor).toHaveBeenCalledTimes(1);
expect(mockHybridExecutorInitialize).toHaveBeenCalledTimes(1);
await triggerSignal("SIGTERM");
expect(mockHybridExecutorShutdown).toHaveBeenCalledTimes(1);
expect(mockHybridExecutorShutdown.mock.invocationCallOrder[0]).toBeLessThan(
mocks.projectEngineInstances[0].stop.mock.invocationCallOrder[0],
);
delete process.env.FUSION_HYBRID_EXECUTOR;
});
});
describe("runDaemon --token-only mode", () => {

32
packages/cli/src/commands/__tests__/serve.test.ts
View File

@@ -4,10 +4,16 @@ import { mkdtempSync, rmSync } from "node:fs";
import { join } from "node:path";
import { tmpdir } from "node:os";
const { mockSyncStartupModels } = vi.hoisted(() => ({
const { mockSyncStartupModels, mockShouldUseHybridExecutor, mockHybridExecutorCtor, mockHybridExecutorInitialize, mockHybridExecutorShutdown } = vi.hoisted(() => ({
mockSyncStartupModels: vi.fn().mockResolvedValue(undefined),
mockShouldUseHybridExecutor: vi.fn().mockResolvedValue({ enabled: false, reason: "single-project-local-only" }),
mockHybridExecutorInitialize: vi.fn().mockResolvedValue(undefined),
mockHybridExecutorShutdown: vi.fn().mockResolvedValue(undefined),
mockHybridExecutorCtor: vi.fn().mockImplementation(() => ({
initialize: mockHybridExecutorInitialize,
shutdown: mockHybridExecutorShutdown,
})),
}));
vi.mock("../startup-model-sync.js", () => ({
syncStartupModels: mockSyncStartupModels,
}));
@@ -655,9 +661,10 @@ vi.mock("@fusion/engine", async (importOriginal) => {
createAiPromptExecutor: mocks.createAiPromptExecutorMock,
HeartbeatMonitor: mocks.heartbeatMonitorCtor,
HeartbeatTriggerScheduler: mocks.heartbeatTriggerSchedulerCtor,
shouldUseHybridExecutor: mockShouldUseHybridExecutor,
HybridExecutor: mockHybridExecutorCtor,
});
});
vi.mock("@mariozechner/pi-coding-agent", () => ({
AuthStorage: {
create: vi.fn(() => mocks.authStorage),
@@ -847,6 +854,25 @@ describe("runServe", () => {
expect(mocks.taskStores[0].close).toHaveBeenCalledTimes(1);
});
it("enables HybridExecutor when env override is set and shuts it down before engine stop", async () => {
process.env.FUSION_HYBRID_EXECUTOR = "1";
mockShouldUseHybridExecutor.mockResolvedValue({ enabled: true, reason: "env-override" });
await runServe(4040, {});
expect(mockHybridExecutorCtor).toHaveBeenCalledTimes(1);
expect(mockHybridExecutorInitialize).toHaveBeenCalledTimes(1);
await triggerSignal("SIGTERM");
expect(mockHybridExecutorShutdown).toHaveBeenCalledTimes(1);
const shutdownOrder = [
mockHybridExecutorShutdown.mock.invocationCallOrder[0],
mocks.projectEngineInstances[0].stop.mock.invocationCallOrder[0],
];
expect(shutdownOrder[0]).toBeLessThan(shutdownOrder[1]);
delete process.env.FUSION_HYBRID_EXECUTOR;
});
it("listens on 127.0.0.1 by default and respects a custom host", async () => {
await runServe(3010, {});
expect(mocks.listenCalls[0]).toMatchObject({