ktg-plugin-marketplace/plugins/llm-security/tests/hooks/pre-write-pathguard.test.mjs

129 lines
5.2 KiB
JavaScript

// pre-write-pathguard.test.mjs — Tests for hooks/scripts/pre-write-pathguard.mjs
// Zero external dependencies: node:test + node:assert only.
import { describe, it } from 'node:test';
import assert from 'node:assert/strict';
import { resolve } from 'node:path';
import { runHook } from './hook-helper.mjs';
const SCRIPT = resolve(import.meta.dirname, '../../hooks/scripts/pre-write-pathguard.mjs');
function writePayload(filePath) {
return { tool_name: 'Write', tool_input: { file_path: filePath, content: 'data' } };
}
// ---------------------------------------------------------------------------
// BLOCK cases — exit code 2
// ---------------------------------------------------------------------------
describe('pre-write-pathguard — BLOCK cases', () => {
it('blocks a write to .env (environment file)', async () => {
const result = await runHook(SCRIPT, writePayload('/project/.env'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
assert.match(result.stderr, /env/);
});
it('blocks a write to .env.local (environment file variant)', async () => {
const result = await runHook(SCRIPT, writePayload('/project/.env.local'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
});
it('blocks a write to .env.production (environment file variant)', async () => {
const result = await runHook(SCRIPT, writePayload('/project/.env.production'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
});
it('blocks a write to .ssh/id_rsa (SSH directory)', async () => {
const result = await runHook(SCRIPT, writePayload('/home/user/.ssh/id_rsa'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
assert.match(result.stderr, /ssh/i);
});
it('blocks a write to .aws/credentials (AWS credentials directory)', async () => {
const result = await runHook(SCRIPT, writePayload('/home/user/.aws/credentials'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
assert.match(result.stderr, /aws/i);
});
it('blocks a write to .gnupg/private-keys-v1.d/key (GPG directory)', async () => {
const result = await runHook(SCRIPT, writePayload('/home/user/.gnupg/private-keys-v1.d/key'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
assert.match(result.stderr, /gnupg/i);
});
it('blocks a write to .npmrc (credential file)', async () => {
const result = await runHook(SCRIPT, writePayload('/home/user/.npmrc'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
});
it('blocks a write to credentials.json (credential file)', async () => {
const result = await runHook(SCRIPT, writePayload('/project/credentials.json'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
});
it('blocks a write to .claude/settings.json (Claude settings file)', async () => {
const result = await runHook(SCRIPT, writePayload('/home/user/.claude/settings.json'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
assert.match(result.stderr, /settings/i);
});
it('blocks a write to .vscode/settings.json (VS Code settings file)', async () => {
const result = await runHook(SCRIPT, writePayload('/project/.vscode/settings.json'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
});
it('blocks a write to /etc/passwd (system directory)', async () => {
const result = await runHook(SCRIPT, writePayload('/etc/passwd'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
assert.match(result.stderr, /system/i);
});
it('blocks a write to a hooks/scripts/*.mjs path (hook script protection)', async () => {
const result = await runHook(SCRIPT, writePayload('/project/hooks/scripts/my-hook.mjs'));
assert.equal(result.code, 2);
assert.match(result.stderr, /PATH GUARD/);
assert.match(result.stderr, /hooks/i);
});
});
// ---------------------------------------------------------------------------
// ALLOW cases — exit code 0
// ---------------------------------------------------------------------------
describe('pre-write-pathguard — ALLOW cases', () => {
it('allows a write to a normal source file (src/app.js)', async () => {
const result = await runHook(SCRIPT, writePayload('src/app.js'));
assert.equal(result.code, 0);
});
it('allows a write to README.md', async () => {
const result = await runHook(SCRIPT, writePayload('README.md'));
assert.equal(result.code, 0);
});
it('allows a write to settings.json at the project root (not inside .claude/ or .vscode/)', async () => {
const result = await runHook(SCRIPT, writePayload('/project/settings.json'));
assert.equal(result.code, 0);
});
it('allows a write when file_path is empty', async () => {
const result = await runHook(SCRIPT, { tool_name: 'Write', tool_input: { file_path: '', content: 'x' } });
assert.equal(result.code, 0);
});
it('exits 0 gracefully when stdin is not valid JSON', async () => {
const result = await runHook(SCRIPT, 'not json {{{');
assert.equal(result.code, 0);
});
});