54 lines
2.5 KiB
TypeScript
54 lines
2.5 KiB
TypeScript
import { test } from "node:test";
|
|
import assert from "node:assert/strict";
|
|
import { extractCommandPrefix } from "../../open-sse/utils/claudeCodeMetaRequests.ts";
|
|
import { extractFilepathsFromCommand } from "../../open-sse/utils/claudeCodeMetaRequests.ts";
|
|
|
|
test("extractCommandPrefix returns two-word prefix for known multi-verb tools", () => {
|
|
assert.equal(extractCommandPrefix("git commit -m 'x'"), "git commit");
|
|
assert.equal(extractCommandPrefix("npm install lodash"), "npm install");
|
|
assert.equal(extractCommandPrefix("docker build ."), "docker build");
|
|
});
|
|
|
|
test("extractCommandPrefix returns single word for simple commands", () => {
|
|
assert.equal(extractCommandPrefix("ls -la"), "ls");
|
|
assert.equal(extractCommandPrefix("cat file.txt"), "cat");
|
|
});
|
|
|
|
test("extractCommandPrefix strips leading env assignments", () => {
|
|
assert.equal(extractCommandPrefix("FOO=bar npm run build"), "npm run");
|
|
});
|
|
|
|
test("extractCommandPrefix detects command injection", () => {
|
|
assert.equal(extractCommandPrefix("ls; rm -rf /"), "command_injection_detected");
|
|
assert.equal(extractCommandPrefix("echo $(whoami)"), "command_injection_detected");
|
|
assert.equal(extractCommandPrefix("cat `id`"), "command_injection_detected");
|
|
});
|
|
|
|
test("extractFilepathsFromCommand returns read files for read commands", () => {
|
|
assert.deepEqual(extractFilepathsFromCommand("cat src/a.ts src/b.ts", ""), ["src/a.ts", "src/b.ts"]);
|
|
assert.deepEqual(extractFilepathsFromCommand("head -n5 README.md", ""), ["README.md"]);
|
|
});
|
|
|
|
test("extractFilepathsFromCommand returns empty for listing commands", () => {
|
|
assert.deepEqual(extractFilepathsFromCommand("ls -la src/", ""), []);
|
|
assert.deepEqual(extractFilepathsFromCommand("find . -name '*.ts'", ""), []);
|
|
});
|
|
|
|
test("extractFilepathsFromCommand skips the grep pattern arg", () => {
|
|
assert.deepEqual(extractFilepathsFromCommand("grep -n foo src/a.ts", ""), ["src/a.ts"]);
|
|
});
|
|
|
|
test("extractCommandPrefix stays conservative when a flag precedes the subcommand", () => {
|
|
// -C takes a path arg; we don't parse flag-args, so fall back to the head
|
|
// (never emit a wrong two-word prefix like "git -C" or "git /x").
|
|
assert.equal(extractCommandPrefix("git -C /x commit"), "git");
|
|
});
|
|
|
|
test("extractFilepathsFromCommand handles grep -e pattern then file", () => {
|
|
assert.deepEqual(extractFilepathsFromCommand("grep -e pattern a.ts", ""), ["a.ts"]);
|
|
});
|
|
|
|
test("extractFilepathsFromCommand handles grep -f patternfile then file", () => {
|
|
assert.deepEqual(extractFilepathsFromCommand("grep -f pats.txt a.ts", ""), ["a.ts"]);
|
|
});
|