can1357
f2137becb8
- Fixed help rendering so `--help` no longer triggers unrelated command loaders. - Fixed startup span logging to emit markers only with PI_DEBUG_STARTUP set. - Fixed logger startup trace behavior for `:start`, `:done`, and `:fail` phases. - Fixed prompt template processing with cached raw-template compilation and safer formatting. - Optimized symbol and tag parsing in prompt templates via manual parsers.
55 lines
1.8 KiB
TypeScript
55 lines
1.8 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "bun:test";
|
|
import type { SSHHost } from "@oh-my-pi/pi-coding-agent/capability/ssh";
|
|
import type { SourceMeta } from "@oh-my-pi/pi-coding-agent/capability/types";
|
|
import * as discovery from "@oh-my-pi/pi-coding-agent/discovery";
|
|
import type { ToolSession } from "@oh-my-pi/pi-coding-agent/tools";
|
|
import { loadSshTool } from "@oh-my-pi/pi-coding-agent/tools";
|
|
|
|
const SOURCE: SourceMeta = {
|
|
provider: "test",
|
|
providerName: "Test",
|
|
path: "/dev/null",
|
|
level: "user",
|
|
};
|
|
|
|
// Unique names so no persisted host-info cache file can exist for them.
|
|
const RUN_ID = `${Date.now()}-${process.pid}`;
|
|
const HOST_A: SSHHost = { name: `a-omp-test-${RUN_ID}`, host: "alpha.example.com", _source: SOURCE };
|
|
const HOST_B: SSHHost = { name: `b-omp-test-${RUN_ID}`, host: "beta.example.com", _source: SOURCE };
|
|
|
|
function mockHosts(hosts: SSHHost[]): void {
|
|
vi.spyOn(discovery, "loadCapability").mockResolvedValue({
|
|
items: hosts,
|
|
all: hosts,
|
|
warnings: [],
|
|
providers: ["test"],
|
|
});
|
|
}
|
|
|
|
function createSession(): ToolSession {
|
|
return { cwd: "/tmp" } as unknown as ToolSession;
|
|
}
|
|
|
|
describe("loadSshTool description", () => {
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
it("returns null when no hosts are configured", async () => {
|
|
mockHosts([]);
|
|
expect(await loadSshTool(createSession())).toBeNull();
|
|
});
|
|
|
|
it("renders uncached hosts with the detecting placeholder, sorted by name, without probing", async () => {
|
|
mockHosts([HOST_B, HOST_A]);
|
|
const tool = await loadSshTool(createSession());
|
|
expect(tool).not.toBeNull();
|
|
expect(tool?.description.startsWith("Runs commands on remote hosts.")).toBe(true);
|
|
expect(
|
|
tool?.description.endsWith(
|
|
`\n\nAvailable hosts:\n- ${HOST_A.name} (${HOST_A.host}) | detecting...\n- ${HOST_B.name} (${HOST_B.host}) | detecting...`,
|
|
),
|
|
).toBe(true);
|
|
});
|
|
});
|