summaryrefslogtreecommitdiff
path: root/packages/server/src/core/prompts.test.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/server/src/core/prompts.test.ts')
-rw-r--r--packages/server/src/core/prompts.test.ts106
1 files changed, 0 insertions, 106 deletions
diff --git a/packages/server/src/core/prompts.test.ts b/packages/server/src/core/prompts.test.ts
deleted file mode 100644
index 49502f92..00000000
--- a/packages/server/src/core/prompts.test.ts
+++ /dev/null
@@ -1,106 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { getCoreSystemPrompt } from './prompts.js'; // Adjust import path
-import * as process from 'node:process';
-
-// Mock tool names if they are dynamically generated or complex
-vi.mock('../tools/ls', () => ({ LSTool: { Name: 'list_directory' } }));
-vi.mock('../tools/edit', () => ({ EditTool: { Name: 'replace' } }));
-vi.mock('../tools/glob', () => ({ GlobTool: { Name: 'glob' } }));
-vi.mock('../tools/grep', () => ({ GrepTool: { Name: 'search_file_content' } }));
-vi.mock('../tools/read-file', () => ({ ReadFileTool: { Name: 'read_file' } }));
-vi.mock('../tools/read-many-files', () => ({
- ReadManyFilesTool: { Name: 'read_many_files' },
-}));
-vi.mock('../tools/shell', () => ({
- ShellTool: { Name: 'execute_bash_command' },
-}));
-vi.mock('../tools/write-file', () => ({
- WriteFileTool: { Name: 'write_file' },
-}));
-
-describe('Core System Prompt (prompts.ts)', () => {
- // Store original env vars that we modify
- let originalSandboxEnv: string | undefined;
-
- beforeEach(() => {
- // Store original value before each test
- originalSandboxEnv = process.env.SANDBOX;
- });
-
- afterEach(() => {
- // Restore original value after each test
- if (originalSandboxEnv === undefined) {
- delete process.env.SANDBOX;
- } else {
- process.env.SANDBOX = originalSandboxEnv;
- }
- });
-
- it('should return the base prompt when no userMemory is provided', () => {
- delete process.env.SANDBOX; // Ensure default state for snapshot
- const prompt = getCoreSystemPrompt();
- expect(prompt).not.toContain('---\n\n'); // Separator should not be present
- expect(prompt).toContain('You are an interactive CLI agent'); // Check for core content
- expect(prompt).toMatchSnapshot(); // Use snapshot for base prompt structure
- });
-
- it('should return the base prompt when userMemory is empty string', () => {
- delete process.env.SANDBOX;
- const prompt = getCoreSystemPrompt('');
- expect(prompt).not.toContain('---\n\n');
- expect(prompt).toContain('You are an interactive CLI agent');
- expect(prompt).toMatchSnapshot();
- });
-
- it('should return the base prompt when userMemory is whitespace only', () => {
- delete process.env.SANDBOX;
- const prompt = getCoreSystemPrompt(' \n \t ');
- expect(prompt).not.toContain('---\n\n');
- expect(prompt).toContain('You are an interactive CLI agent');
- expect(prompt).toMatchSnapshot();
- });
-
- it('should append userMemory with separator when provided', () => {
- delete process.env.SANDBOX;
- const memory = 'This is custom user memory.\nBe extra polite.';
- const expectedSuffix = `\n\n---\n\n${memory}`;
- const prompt = getCoreSystemPrompt(memory);
-
- expect(prompt.endsWith(expectedSuffix)).toBe(true);
- expect(prompt).toContain('You are an interactive CLI agent'); // Ensure base prompt follows
- expect(prompt).toMatchSnapshot(); // Snapshot the combined prompt
- });
-
- it('should include sandbox-specific instructions when SANDBOX env var is set', () => {
- process.env.SANDBOX = 'true'; // Generic sandbox value
- const prompt = getCoreSystemPrompt();
- expect(prompt).toContain('# Sandbox');
- expect(prompt).not.toContain('# MacOS Seatbelt');
- expect(prompt).not.toContain('# Outside of Sandbox');
- expect(prompt).toMatchSnapshot();
- });
-
- it('should include seatbelt-specific instructions when SANDBOX env var is "sandbox-exec"', () => {
- process.env.SANDBOX = 'sandbox-exec';
- const prompt = getCoreSystemPrompt();
- expect(prompt).toContain('# MacOS Seatbelt');
- expect(prompt).not.toContain('# Sandbox');
- expect(prompt).not.toContain('# Outside of Sandbox');
- expect(prompt).toMatchSnapshot();
- });
-
- it('should include non-sandbox instructions when SANDBOX env var is not set', () => {
- delete process.env.SANDBOX; // Ensure it's not set
- const prompt = getCoreSystemPrompt();
- expect(prompt).toContain('# Outside of Sandbox');
- expect(prompt).not.toContain('# Sandbox');
- expect(prompt).not.toContain('# MacOS Seatbelt');
- expect(prompt).toMatchSnapshot();
- });
-});