fix(plugin): harden kubernetes fast upload edges

This commit is contained in:
Dotta
2026-05-13 13:51:12 -05:00
committed by Chris Farhood
parent 713fb6eb4e
commit 40e8638aa3
4 changed files with 46 additions and 1 deletions
@@ -27,6 +27,17 @@ describe("execInPod", () => {
expect(result).toEqual({ exitCode: 0, timedOut: false, stdout: "ok\n", stderr: "" });
});
it("finishes when Kubernetes reports status without ending output streams", async () => {
execMock.mockImplementation((_namespace, _pod, _container, _command, stdout, _stderr, _stdin, _tty, statusCallback) => {
stdout.write("ok\n");
statusCallback({ status: "Success" });
return Promise.resolve(new EventEmitter());
});
const result = await execInPod({} as never, "ns", "pod-1", "agent", ["echo", "ok"]);
expect(result).toEqual({ exitCode: 0, timedOut: false, stdout: "ok\n", stderr: "" });
});
it("returns an execution failure if the websocket closes before a status frame", async () => {
const ws = new EventEmitter();
execMock.mockResolvedValue(ws);
@@ -75,6 +75,25 @@ describe("FastUploadInterceptor", () => {
expect(interceptor.pendingCount).toBe(0);
});
it("fails fast when init repeats for an in-progress upload", () => {
const interceptor = new FastUploadInterceptor();
const target = "/workspace/file.bin";
const initCommand =
`mkdir -p '/workspace' && rm -f '${target}.paperclip-upload.b64' && : > '${target}.paperclip-upload.b64'`;
expect(interceptor.decide(initCommand)).toMatchObject({ action: "ack" });
expect(
interceptor.decide(`printf '%s' 'aGVsbG8=' >> '${target}.paperclip-upload.b64'`),
).toMatchObject({ action: "ack" });
const decision = interceptor.decide(initCommand);
expect(decision).toMatchObject({
action: "error",
message: expect.stringContaining("Fast upload already in progress"),
});
expect(interceptor.pendingCount).toBe(0);
});
it("clears buffered uploads on reset", () => {
const interceptor = new FastUploadInterceptor();
const target = "/workspace/file.bin";