test(ai,coding-agent): stabilize env-sensitive test cases (#4119)

This commit is contained in:
myu003
2026-05-03 18:55:23 +08:00
committed by Mario Zechner
parent 05abdd8fde
commit 924a1395bc
3 changed files with 61 additions and 47 deletions

View File

@@ -242,7 +242,7 @@ describe("openai-codex streaming", () => {
}; };
const result = await Promise.race([ const result = await Promise.race([
streamOpenAICodexResponses(model, context, { apiKey: token }).result(), streamOpenAICodexResponses(model, context, { apiKey: token, transport: "sse" }).result(),
new Promise<never>((_, reject) => { new Promise<never>((_, reject) => {
setTimeout(() => reject(new Error("Timed out waiting for completed SSE stream")), 1000); setTimeout(() => reject(new Error("Timed out waiting for completed SSE stream")), 1000);
}), }),
@@ -301,7 +301,7 @@ describe("openai-codex streaming", () => {
}; };
const result = await Promise.race([ const result = await Promise.race([
streamOpenAICodexResponses(model, context, { apiKey: token }).result(), streamOpenAICodexResponses(model, context, { apiKey: token, transport: "sse" }).result(),
new Promise<never>((_, reject) => { new Promise<never>((_, reject) => {
setTimeout(() => reject(new Error("Timed out waiting for incomplete SSE stream")), 1000); setTimeout(() => reject(new Error("Timed out waiting for incomplete SSE stream")), 1000);
}), }),

View File

@@ -54,6 +54,9 @@ function osc52Writes(): string[] {
beforeEach(() => { beforeEach(() => {
vi.unstubAllEnvs(); vi.unstubAllEnvs();
vi.stubEnv("SSH_CONNECTION", "");
vi.stubEnv("SSH_CLIENT", "");
vi.stubEnv("MOSH_CONNECTION", "");
stdoutWrites = []; stdoutWrites = [];
nativeResolved = false; nativeResolved = false;
mocks.clipboard.setText.mockReset(); mocks.clipboard.setText.mockReset();

View File

@@ -160,60 +160,71 @@ Content`,
}); });
it("should resolve symlinked user and project resources once", async () => { it("should resolve symlinked user and project resources once", async () => {
const sharedDir = join(tempDir, "shared-resources"); const previousHome = process.env.HOME;
const sharedExtensionsDir = join(sharedDir, "extensions"); process.env.HOME = tempDir;
const sharedSkillsDir = join(sharedDir, "skills");
const sharedPromptsDir = join(sharedDir, "prompts");
const sharedThemesDir = join(sharedDir, "themes");
mkdirSync(sharedExtensionsDir, { recursive: true });
mkdirSync(sharedSkillsDir, { recursive: true });
mkdirSync(sharedPromptsDir, { recursive: true });
mkdirSync(sharedThemesDir, { recursive: true });
writeFileSync(join(sharedExtensionsDir, "shared.ts"), "export default function() {}"); try {
mkdirSync(join(sharedSkillsDir, "shared-skill"), { recursive: true }); const sharedDir = join(tempDir, "shared-resources");
writeFileSync( const sharedExtensionsDir = join(sharedDir, "extensions");
join(sharedSkillsDir, "shared-skill", "SKILL.md"), const sharedSkillsDir = join(sharedDir, "skills");
`--- const sharedPromptsDir = join(sharedDir, "prompts");
const sharedThemesDir = join(sharedDir, "themes");
mkdirSync(sharedExtensionsDir, { recursive: true });
mkdirSync(sharedSkillsDir, { recursive: true });
mkdirSync(sharedPromptsDir, { recursive: true });
mkdirSync(sharedThemesDir, { recursive: true });
writeFileSync(join(sharedExtensionsDir, "shared.ts"), "export default function() {}");
mkdirSync(join(sharedSkillsDir, "shared-skill"), { recursive: true });
writeFileSync(
join(sharedSkillsDir, "shared-skill", "SKILL.md"),
`---
name: shared-skill name: shared-skill
description: Shared skill description: Shared skill
--- ---
Content`, Content`,
); );
writeFileSync(join(sharedPromptsDir, "shared.md"), "Shared prompt"); writeFileSync(join(sharedPromptsDir, "shared.md"), "Shared prompt");
writeFileSync(join(sharedThemesDir, "shared.json"), JSON.stringify({ name: "shared-theme" })); writeFileSync(join(sharedThemesDir, "shared.json"), JSON.stringify({ name: "shared-theme" }));
mkdirSync(join(agentDir), { recursive: true }); mkdirSync(join(agentDir), { recursive: true });
mkdirSync(join(tempDir, ".pi"), { recursive: true }); mkdirSync(join(tempDir, ".pi"), { recursive: true });
symlinkSync(sharedExtensionsDir, join(agentDir, "extensions"), "dir"); symlinkSync(sharedExtensionsDir, join(agentDir, "extensions"), "dir");
symlinkSync(sharedSkillsDir, join(agentDir, "skills"), "dir"); symlinkSync(sharedSkillsDir, join(agentDir, "skills"), "dir");
symlinkSync(sharedPromptsDir, join(agentDir, "prompts"), "dir"); symlinkSync(sharedPromptsDir, join(agentDir, "prompts"), "dir");
symlinkSync(sharedThemesDir, join(agentDir, "themes"), "dir"); symlinkSync(sharedThemesDir, join(agentDir, "themes"), "dir");
symlinkSync(sharedExtensionsDir, join(tempDir, ".pi", "extensions"), "dir"); symlinkSync(sharedExtensionsDir, join(tempDir, ".pi", "extensions"), "dir");
symlinkSync(sharedSkillsDir, join(tempDir, ".pi", "skills"), "dir"); symlinkSync(sharedSkillsDir, join(tempDir, ".pi", "skills"), "dir");
symlinkSync(sharedPromptsDir, join(tempDir, ".pi", "prompts"), "dir"); symlinkSync(sharedPromptsDir, join(tempDir, ".pi", "prompts"), "dir");
symlinkSync(sharedThemesDir, join(tempDir, ".pi", "themes"), "dir"); symlinkSync(sharedThemesDir, join(tempDir, ".pi", "themes"), "dir");
const result = await packageManager.resolve(); const result = await packageManager.resolve();
expect({ expect({
extensions: result.extensions.length, extensions: result.extensions.length,
skills: result.skills.length, skills: result.skills.length,
prompts: result.prompts.length, prompts: result.prompts.length,
themes: result.themes.length, themes: result.themes.length,
}).toEqual({ }).toEqual({
extensions: 1, extensions: 1,
skills: 1, skills: 1,
prompts: 1, prompts: 1,
themes: 1, themes: 1,
}); });
// Project auto-discovered has higher precedence than user auto-discovered, // Project auto-discovered has higher precedence than user auto-discovered,
// so the surviving entry should be scoped to project. // so the surviving entry should be scoped to project.
expect(result.extensions[0].metadata.scope).toBe("project"); expect(result.extensions[0].metadata.scope).toBe("project");
expect(result.skills[0].metadata.scope).toBe("project"); expect(result.skills[0].metadata.scope).toBe("project");
expect(result.prompts[0].metadata.scope).toBe("project"); expect(result.prompts[0].metadata.scope).toBe("project");
expect(result.themes[0].metadata.scope).toBe("project"); expect(result.themes[0].metadata.scope).toBe("project");
} finally {
if (previousHome === undefined) {
delete process.env.HOME;
} else {
process.env.HOME = previousHome;
}
}
}); });
it("should auto-discover project prompts with overrides", async () => { it("should auto-discover project prompts with overrides", async () => {