@probelabs/probe
Version:
Node.js wrapper for the probe code search tool
315 lines (241 loc) โข 7.95 kB
Markdown
# ACP Test Utilities
This directory contains comprehensive testing tools for Agent Client Protocol (ACP) implementations. Since the official ACP specification doesn't provide dedicated test tooling, we've created our own utilities to ensure protocol compliance and facilitate development.
## ๐งช Test Tools Overview
### 1. ACP Tester (`acp-tester.js`)
**Protocol compliance testing tool**
A comprehensive test suite that validates ACP implementations against the protocol specification.
**Features:**
- Protocol initialization and capability testing
- Session management verification
- Error handling validation
- Performance benchmarking
- JSON-RPC 2.0 compliance checking
**Usage:**
```bash
node acp-tester.js --agent-command "node my-agent.js --acp"
node acp-tester.js --agent-command "python agent.py" --timeout 60 --verbose
```
### 2. Mock ACP Client (`mock-acp-client.js`)
**Interactive client simulator**
Simulates a code editor or IDE communicating with ACP agents, useful for manual testing and development.
**Features:**
- Interactive chat sessions
- Session mode management
- Tool call progress monitoring
- Notification handling
- Demo conversation flows
**Usage:**
```bash
node mock-acp-client.js --agent "node agent.js --acp"
node mock-acp-client.js --agent "my-agent-binary" --interactive
```
## ๐ง Test Suite Details
### ACP Tester Test Cases
#### 1. Protocol Initialize
- Validates JSON-RPC 2.0 format
- Checks protocol version negotiation
- Verifies server information structure
- Confirms capability advertisement
#### 2. Session Management
- Tests session creation with unique IDs
- Validates session loading and persistence
- Checks session metadata (timestamps, history)
#### 3. Error Handling
- Unknown method rejection (-32601)
- Invalid parameter handling (-32602)
- Session not found errors
- Timeout behavior
#### 4. Performance Benchmark
- Measures response times
- Tests concurrent requests
- Evaluates throughput
- Reports performance metrics
### Mock Client Features
#### Interactive Mode
```bash
> hello agent
> mode planning
> find all functions in this project
> status
> quit
```
#### Demo Mode
Runs predefined conversation flow:
1. Agent introduction
2. File listing request
3. Code search queries
4. Entry point analysis
## ๐ Example Test Output
### ACP Tester Results
```
๐ ACP Protocol Compliance Test Suite
๐ ========================================
โ
Protocol Initialize (9ms)
โ
Session Management (4ms)
โ
Error Handling (1ms)
โ
Performance Benchmark (2ms)
๐ Test Results Summary
๐ ====================
๐ Total time: 1024ms
๐ Passed: 4
๐ Failed: 0
๐ Total: 4
โ
Agent is ACP compliant! ๐
```
### Mock Client Session
```
[14:30:15] ๐ Starting agent: node agent.js --acp
[14:30:16] โ
Connected to my-agent v1.0.0
[14:30:16] ๐ ๏ธ Available tools: search, query, extract
[14:30:16] ๐ Creating new session...
[14:30:16] โ
Session created: abc-123-def
[14:30:17] ๐ฌ Sending: "Hello! Can you tell me about yourself?"
[14:30:18] ๐ค Agent response:
โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ
I'm an AI agent that helps with code analysis and search...
โโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโโ
```
## ๐ฏ Testing Your ACP Implementation
### Step 1: Basic Compliance
```bash
node acp-tester.js --agent-command "your-agent-command"
```
This will run the core compliance tests and report any issues.
### Step 2: Interactive Testing
```bash
node mock-acp-client.js --agent "your-agent-command" --interactive
```
Test your agent manually with realistic interactions.
### Step 3: Performance Testing
```bash
node acp-tester.js --agent-command "your-agent-command" --verbose
```
Get detailed performance metrics and timing information.
## ๐๏ธ Test Development
### Adding New Tests to ACP Tester
```javascript
async testCustomFeature() {
return this.runTest('Custom Feature Test', async () => {
// Your test logic here
const result = await this.sendRequest('customMethod', { param: 'value' });
// Validate result
if (!result.expected) {
throw new Error('Custom feature not working');
}
return { customMetric: result.value };
});
}
```
### Extending Mock Client
```javascript
handleCustomNotification(notification) {
if (notification.method === 'customEvent') {
this.log(`๐ฏ Custom event: ${notification.params.data}`);
}
}
```
## ๐ ๏ธ Utilities API
### ACPTester Class
**Methods:**
- `runAllTests()` - Execute complete test suite
- `runTest(name, testFn)` - Run individual test
- `sendRequest(method, params)` - Send JSON-RPC request
- `validateInitializeResponse(result)` - Validate init response
**Options:**
- `agentCommand` - Command to start agent
- `timeout` - Request timeout (ms)
- `verbose` - Enable detailed output
### MockACPClient Class
**Methods:**
- `initialize()` - Initialize ACP connection
- `createSession()` - Create new session
- `sendPrompt(message)` - Send user message
- `setSessionMode(mode)` - Change session mode
- `runInteractiveSession()` - Start interactive mode
**Options:**
- `agentCommand` - Command to start agent
- `interactive` - Enable interactive mode
- `debug` - Show debug information
- `timeout` - Request timeout (ms)
## ๐จ Customization
### Custom Test Configuration
```javascript
const tester = new ACPTester({
agentCommand: 'python my_agent.py',
timeout: 60000, // 60 seconds
verbose: true
});
// Add custom validation
tester.validateCustomResponse = (result) => {
// Your custom validation logic
};
await tester.runAllTests();
```
### Client Behavior Customization
```javascript
const client = new MockACPClient({
agentCommand: './my-agent-binary',
interactive: false,
debug: true
});
// Custom prompt handling
client.handleCustomPrompt = async (prompt) => {
// Your custom logic
};
await client.run();
```
## ๐ Integration with CI/CD
### GitHub Actions Example
```yaml
name: ACP Compliance Test
on: [push, pull_request]
jobs:
test-acp:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/setup-node@v3
with:
node-version: '18'
- name: Build agent
run: npm install && npm run build
- name: Test ACP compliance
run: |
node test-utils/acp-tester.js \
--agent-command "node dist/agent.js --acp" \
--timeout 30
```
### Docker Testing
```dockerfile
FROM node:18-alpine
COPY . /app
WORKDIR /app
RUN npm install
RUN npm run build
# Test ACP compliance
RUN node test-utils/acp-tester.js \
--agent-command "node dist/agent.js --acp"
```
## ๐ Reference Implementations
The test utilities work with these ACP implementations:
**Official Examples:**
- Zed Industries TypeScript agent/client examples
- Reference Rust implementations
**Tested Implementations:**
- Probe Agent (this repository)
- Custom Python agents
- Go-based agents
## ๐ค Contributing Test Tools
To add new test utilities:
1. Create new test file in this directory
2. Follow the established patterns (ACPTester/MockACPClient)
3. Add comprehensive documentation
4. Include usage examples
5. Test against multiple implementations
### Test Tool Guidelines
- **Comprehensive**: Cover all protocol aspects
- **Robust**: Handle edge cases and errors gracefully
- **Informative**: Provide clear success/failure messages
- **Configurable**: Allow customization for different needs
- **Well-documented**: Include usage examples and API docs
These test utilities ensure ACP implementations are reliable, compliant, and ready for production use with code editors and development tools.