test(ai,coding-agent): stabilize env-sensitive test cases (#4119)
This commit is contained in:
@@ -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);
|
||||||
}),
|
}),
|
||||||
|
|||||||
@@ -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();
|
||||||
|
|||||||
@@ -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 () => {
|
||||||
|
|||||||
Reference in New Issue
Block a user