scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-08-28 05:27:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix ut

Browse source
This commit is contained in:
aheizi 2025-03-11 23:19:13 +08:00
parent 29aa464ce3
commit f534e80f78
2 changed files with 120 additions and 1465 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1507
src/core/prompts/__tests__/__snapshots__/system.test.ts.snap
View file

File diff suppressed because it is too large Load diff

78
src/core/prompts/__tests__/system.test.ts
View file

@ -225,6 +225,52 @@ describe("SYSTEM_PROMPT", () => {
true, // enableMcpServerCreation
)
// Verify basic MCP server info is included
expect(prompt).toContain("MCP SERVERS")
expect(prompt).toContain("The Model Context Protocol (MCP) enables communication")
// Verify server types are described
expect(prompt).toContain("Local (Stdio-based) servers")
expect(prompt).toContain("Remote (SSE-based) servers")
// Verify connected servers section exists
expect(prompt).toContain("Connected MCP Servers")
expect(prompt).toContain("When a server is connected")
// Verify server creation info is included
expect(prompt).toContain("Creating an MCP Server")
expect(prompt).toContain("MCP Server Types and Configuration")
// Verify configuration examples are included
expect(prompt).toContain("Local (Stdio) Server Configuration")
expect(prompt).toContain("Remote (SSE) Server Configuration")
expect(prompt).toContain("Common configuration options for both types")
// Verify example server info is included
expect(prompt).toContain("Example Local MCP Server")
expect(prompt).toContain("create-typescript-server")
})
it("should include MCP server creation info when enabled", async () => {
const mockMcpHub = createMockMcpHub()
const prompt = await SYSTEM_PROMPT(
mockContext,
"/test/path",
false,
mockMcpHub,
undefined,
undefined,
defaultModeSlug,
undefined,
undefined,
undefined,
undefined,
undefined,
experiments,
true,
)
expect(prompt).toMatchSnapshot()
})
@ -246,7 +292,10 @@ describe("SYSTEM_PROMPT", () => {
true, // enableMcpServerCreation
)
expect(prompt).toMatchSnapshot()
// Should not contain any MCP server related content since mcpHub is undefined
expect(prompt).not.toContain("MCP SERVERS")
expect(prompt).not.toContain("Connected MCP Servers")
expect(prompt).not.toContain("Creating an MCP Server")
})
it("should handle different browser viewport sizes", async () => {
@ -691,8 +740,18 @@ describe("addCustomInstructions", () => {
true, // enableMcpServerCreation
)
// Verify server creation info is included
expect(prompt).toContain("Creating an MCP Server")
expect(prompt).toMatchSnapshot()
expect(prompt).toContain("MCP Server Types and Configuration")
// Verify configuration examples are included
expect(prompt).toContain("Local (Stdio) Server Configuration")
expect(prompt).toContain("Remote (SSE) Server Configuration")
expect(prompt).toContain("Common configuration options for both types")
// Verify example server info is included
expect(prompt).toContain("Example Local MCP Server")
expect(prompt).toContain("create-typescript-server")
})
it("should exclude MCP server creation info when disabled", async () => {
@ -716,12 +775,23 @@ describe("addCustomInstructions", () => {
)
expect(prompt).not.toContain("Creating an MCP Server")
expect(prompt).toMatchSnapshot()
expect(prompt).not.toContain("MCP Server Types and Configuration")
expect(prompt).toContain("MCP SERVERS")
expect(prompt).toContain("Connected MCP Servers")
})
it("should prioritize mode-specific rules for code mode", async () => {
const instructions = await addCustomInstructions("", "", "/test/path", defaultModeSlug)
expect(instructions).toMatchSnapshot()
const rulesSection = instructions.split("\n\n").find((section) => section.startsWith("Rules:"))
expect(rulesSection).toBeDefined()
// Get the rules as separate lines and filter for header lines only
const ruleLines = rulesSection?.split("\n").filter((line) => line.startsWith("#")) ?? []
// Verify we have both rule headers
expect(ruleLines).toHaveLength(2)
expect(ruleLines[0]).toContain(`Rules from .clinerules-${defaultModeSlug}`)
expect(ruleLines[1]).toContain("Rules from .clinerules")
})
it("should prioritize mode-specific rules for ask mode", async () => {