import type { RestartSentinelPayload } from "../../infra/restart-sentinel.js";

import type { RespawnSupervisor } from "../../infra/supervisor-markers.js";

import type { UpdateInstallSurface, UpdateRunResult } from "../../infra/update-runner.js";

import { withEnvAsync } from "../../test-utils/env.js";

// Capture the sentinel payload written during update.run

let capturedPayload: RestartSentinelPayload | undefined;

updates: Record<string, string | undefined>,

run: () => Promise<T>,

): Promise<T> {

const previous = new Map<string, string | undefined>();

for (const key of Object.keys(updates)) {

previous.set(key, process.env[key]);

const value = updates[key];

if (value === undefined) {

delete process.env[key];

} else {

process.env[key] = value;

}

}

try {

return await run();

} finally {

for (const [key, value] of previous) {

if (value === undefined) {

delete process.env[key];

} else {

process.env[key] = value;

}

}

}

return await withEnvAsync(updates, run);

}

function mockGlobalInstallSurface() {