36 lines
1.3 KiB
TypeScript
36 lines
1.3 KiB
TypeScript
import { describe, expect, it } from 'vitest'
|
|
import { analyzeAgentOutput } from '../../src/utils/agentOutput'
|
|
import { assessAgentCommandRisk } from '../../src/utils/agentCommandRisk'
|
|
|
|
describe('Agent command guardrail', () => {
|
|
it.each([
|
|
['df -h', 'safe'],
|
|
['journalctl -u nginx -n 100', 'safe'],
|
|
['custom-health-check --json', 'low'],
|
|
['sudo cp /tmp/a /etc/a', 'high'],
|
|
['systemctl restart nginx', 'high'],
|
|
['rm -rf /', 'critical'],
|
|
['sudo rm -r -f -- /', 'critical'],
|
|
['dd if=/dev/zero of=/dev/sda', 'critical'],
|
|
['find /tmp -delete', 'high'],
|
|
['cat /etc/os-release && systemctl restart nginx', 'high'],
|
|
])('classifies %s as %s', (command, expected) => {
|
|
expect(assessAgentCommandRisk(command)).toBe(expected)
|
|
})
|
|
})
|
|
|
|
describe('Agent analyze_output tool', () => {
|
|
it('keeps error evidence while bounding long output by line count', () => {
|
|
const lines = Array.from({ length: 160 }, (_, index) =>
|
|
index === 150 ? 'FATAL connection refused' : `normal line ${index}`
|
|
)
|
|
const result = analyzeAgentOutput(lines.join('\n'))
|
|
|
|
expect(result.totalLines).toBe(160)
|
|
expect(result.truncated).toBe(true)
|
|
expect(result.errorLines).toBe(1)
|
|
expect(result.text).toContain('FATAL connection refused')
|
|
expect(result.text).toContain('已省略 80 行')
|
|
})
|
|
})
|