mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 09:01:14 +00:00
fix(coding-agent): add offline startup mode and network timeouts (#1631)
This commit is contained in:
parent
f129ac93c5
commit
757d36a41b
7 changed files with 147 additions and 8 deletions
|
|
@ -1,7 +1,7 @@
|
|||
import { mkdirSync, rmSync, writeFileSync } from "node:fs";
|
||||
import { tmpdir } from "node:os";
|
||||
import { join, relative } from "node:path";
|
||||
import { afterEach, beforeEach, describe, expect, it } from "vitest";
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import { DefaultPackageManager, type ProgressEvent, type ResolvedResource } from "../src/core/package-manager.js";
|
||||
import { SettingsManager } from "../src/core/settings-manager.js";
|
||||
|
||||
|
|
@ -17,8 +17,11 @@ describe("DefaultPackageManager", () => {
|
|||
let agentDir: string;
|
||||
let settingsManager: SettingsManager;
|
||||
let packageManager: DefaultPackageManager;
|
||||
let previousOfflineEnv: string | undefined;
|
||||
|
||||
beforeEach(() => {
|
||||
previousOfflineEnv = process.env.PI_OFFLINE;
|
||||
delete process.env.PI_OFFLINE;
|
||||
tempDir = join(tmpdir(), `pm-test-${Date.now()}-${Math.random().toString(36).slice(2)}`);
|
||||
mkdirSync(tempDir, { recursive: true });
|
||||
agentDir = join(tempDir, "agent");
|
||||
|
|
@ -33,16 +36,25 @@ describe("DefaultPackageManager", () => {
|
|||
});
|
||||
|
||||
afterEach(() => {
|
||||
if (previousOfflineEnv === undefined) {
|
||||
delete process.env.PI_OFFLINE;
|
||||
} else {
|
||||
process.env.PI_OFFLINE = previousOfflineEnv;
|
||||
}
|
||||
vi.restoreAllMocks();
|
||||
vi.unstubAllGlobals();
|
||||
rmSync(tempDir, { recursive: true, force: true });
|
||||
});
|
||||
|
||||
describe("resolve", () => {
|
||||
it("should return empty paths when no sources configured", async () => {
|
||||
it("should return no package-sourced paths when no sources configured", async () => {
|
||||
const result = await packageManager.resolve();
|
||||
expect(result.extensions).toEqual([]);
|
||||
expect(result.skills).toEqual([]);
|
||||
expect(result.prompts).toEqual([]);
|
||||
expect(result.themes).toEqual([]);
|
||||
expect(result.skills.every((r) => r.metadata.source === "auto" && r.metadata.origin === "top-level")).toBe(
|
||||
true,
|
||||
);
|
||||
});
|
||||
|
||||
it("should resolve local extension paths from settings", async () => {
|
||||
|
|
@ -1153,4 +1165,66 @@ export default function(api) { api.registerTool({ name: "test", description: "te
|
|||
expect(result.extensions.filter((r) => r.enabled).length).toBe(1);
|
||||
});
|
||||
});
|
||||
|
||||
describe("offline mode and network timeouts", () => {
|
||||
it("should skip installing missing package sources when offline", async () => {
|
||||
process.env.PI_OFFLINE = "1";
|
||||
settingsManager.setProjectPackages(["npm:missing-package", "git:github.com/example/missing-repo"]);
|
||||
|
||||
const installParsedSourceSpy = vi.spyOn(packageManager as any, "installParsedSource");
|
||||
|
||||
const result = await packageManager.resolve();
|
||||
const allResources = [...result.extensions, ...result.skills, ...result.prompts, ...result.themes];
|
||||
expect(allResources.some((r) => r.metadata.origin === "package")).toBe(false);
|
||||
expect(installParsedSourceSpy).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("should skip refreshing temporary git sources when offline", async () => {
|
||||
process.env.PI_OFFLINE = "1";
|
||||
const gitSource = "git:github.com/example/repo";
|
||||
const parsedGitSource = (packageManager as any).parseSource(gitSource);
|
||||
const installedPath = (packageManager as any).getGitInstallPath(parsedGitSource, "temporary") as string;
|
||||
|
||||
mkdirSync(join(installedPath, "extensions"), { recursive: true });
|
||||
writeFileSync(join(installedPath, "extensions", "index.ts"), "export default function() {};");
|
||||
|
||||
const refreshTemporaryGitSourceSpy = vi.spyOn(packageManager as any, "refreshTemporaryGitSource");
|
||||
|
||||
const result = await packageManager.resolveExtensionSources([gitSource], { temporary: true });
|
||||
expect(result.extensions.some((r) => r.path.endsWith("extensions/index.ts") && r.enabled)).toBe(true);
|
||||
expect(refreshTemporaryGitSourceSpy).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("should not call fetch in npmNeedsUpdate when offline", async () => {
|
||||
process.env.PI_OFFLINE = "1";
|
||||
const installedPath = join(tempDir, "installed-package");
|
||||
mkdirSync(installedPath, { recursive: true });
|
||||
writeFileSync(join(installedPath, "package.json"), JSON.stringify({ version: "1.0.0" }));
|
||||
|
||||
const fetchSpy = vi.spyOn(globalThis, "fetch");
|
||||
|
||||
const needsUpdate = await (packageManager as any).npmNeedsUpdate(
|
||||
{ type: "npm", spec: "example", name: "example", pinned: false },
|
||||
installedPath,
|
||||
);
|
||||
|
||||
expect(needsUpdate).toBe(false);
|
||||
expect(fetchSpy).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("should pass an AbortSignal timeout when fetching npm latest version", async () => {
|
||||
const fetchMock = vi.fn().mockResolvedValue({
|
||||
ok: true,
|
||||
json: async () => ({ version: "1.2.3" }),
|
||||
});
|
||||
vi.stubGlobal("fetch", fetchMock);
|
||||
|
||||
const latest = await (packageManager as any).getLatestNpmVersion("example");
|
||||
expect(latest).toBe("1.2.3");
|
||||
expect(fetchMock).toHaveBeenCalledTimes(1);
|
||||
|
||||
const [, options] = fetchMock.mock.calls[0] as [string, RequestInit | undefined];
|
||||
expect(options?.signal).toBeDefined();
|
||||
});
|
||||
});
|
||||
});
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue