158 lines
3.9 KiB
TypeScript
158 lines
3.9 KiB
TypeScript
import { Cli } from "incur";
|
|
import { expect, it } from "vitest";
|
|
import { registerCaptureCommands } from "../../../src/cli/captureCommands.js";
|
|
import { silentLogger } from "../../../src/logger.js";
|
|
import { createEvidence } from "../../../src/domain/evidence.js";
|
|
import { ProviderCleanupError } from "../../../src/domain/providerCleanupError.js";
|
|
import { err } from "../../../src/domain/result.js";
|
|
|
|
it.each([false, true])(
|
|
"awaits cleanup retry and preserves completed output when retry failure is %s",
|
|
async (failAgain) => {
|
|
const evidence = createEvidence(
|
|
undefined,
|
|
{ id: "fixture", name: "Completed capture fixture", version: "1" },
|
|
{
|
|
operation: "inspect_packet_capture",
|
|
parameters: {},
|
|
result: { packets: [{ "frame.number": ["1"] }] },
|
|
confidence: "observed",
|
|
},
|
|
);
|
|
const original = new ProviderCleanupError(
|
|
"tshark-capture",
|
|
["original-workspace"],
|
|
{ reason: "original cleanup failure" },
|
|
{ operation: "inspect_packet_capture", partialObservation: evidence },
|
|
);
|
|
let closed = false;
|
|
const cli = Cli.create("capture-cleanup-test", { sync: false });
|
|
registerCaptureCommands(
|
|
cli,
|
|
silentLogger,
|
|
{},
|
|
{
|
|
inspect: async () => err(original),
|
|
close: async () => {
|
|
await Promise.resolve();
|
|
closed = true;
|
|
if (failAgain)
|
|
throw new ProviderCleanupError(
|
|
"tshark-capture",
|
|
["retry-workspace"],
|
|
{ reason: "retry failed" },
|
|
{ operation: "inspect_packet_capture" },
|
|
);
|
|
},
|
|
},
|
|
);
|
|
let stdout = "";
|
|
await cli.serve(
|
|
[
|
|
"inspect-packet-capture",
|
|
"recording.pcap",
|
|
"--fields",
|
|
"ip.addr",
|
|
"--format",
|
|
"json",
|
|
],
|
|
{
|
|
stdout: (text) => {
|
|
expect(closed).toBe(true);
|
|
stdout += text;
|
|
},
|
|
exit: () => {},
|
|
},
|
|
);
|
|
const output: unknown = JSON.parse(stdout);
|
|
expect(output).toMatchObject({
|
|
code: "cleanup_incomplete",
|
|
details: {
|
|
partial_observation: evidence,
|
|
resources: expect.arrayContaining(["original-workspace"]),
|
|
},
|
|
});
|
|
if (failAgain)
|
|
expect(output).toMatchObject({
|
|
details: { resources: expect.arrayContaining(["retry-workspace"]) },
|
|
});
|
|
},
|
|
);
|
|
|
|
it("rejects an oversized selector budget before the CLI invokes its service", async () => {
|
|
const cli = Cli.create("capture-selector-test", { sync: false });
|
|
let called = false;
|
|
registerCaptureCommands(
|
|
cli,
|
|
silentLogger,
|
|
{},
|
|
{
|
|
inspect: async () => {
|
|
called = true;
|
|
throw new Error("Must not inspect");
|
|
},
|
|
close: async () => {},
|
|
},
|
|
);
|
|
let stdout = "";
|
|
await cli.serve(
|
|
[
|
|
"inspect-packet-capture",
|
|
"recording.pcap",
|
|
"--fields",
|
|
"x".repeat(8192),
|
|
"--display_filter",
|
|
"界".repeat(8192),
|
|
"--format",
|
|
"json",
|
|
],
|
|
{
|
|
stdout: (text) => {
|
|
stdout += text;
|
|
},
|
|
exit: () => {},
|
|
},
|
|
);
|
|
expect(called).toBe(false);
|
|
expect(stdout).toContain("32 KiB");
|
|
});
|
|
|
|
it("rejects conflicting output options before the CLI invokes its service", async () => {
|
|
const cli = Cli.create("capture-window-test", { sync: false });
|
|
let called = false;
|
|
registerCaptureCommands(
|
|
cli,
|
|
silentLogger,
|
|
{},
|
|
{
|
|
inspect: async () => {
|
|
called = true;
|
|
throw new Error("Must not inspect");
|
|
},
|
|
close: async () => {},
|
|
},
|
|
);
|
|
let stdout = "";
|
|
await cli.serve(
|
|
[
|
|
"inspect-packet-capture",
|
|
"recording.pcap",
|
|
"--fields",
|
|
"ip.addr",
|
|
"--all_matches",
|
|
"--match_limit",
|
|
"20",
|
|
"--format",
|
|
"json",
|
|
],
|
|
{
|
|
stdout: (text) => {
|
|
stdout += text;
|
|
},
|
|
exit: () => {},
|
|
},
|
|
);
|
|
expect(called).toBe(false);
|
|
expect(stdout).toContain("mutually exclusive");
|
|
});
|