admin/Moltbot
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

refactor(test): reuse env helper in update cli tests

Browse Source
This commit is contained in:
Peter Steinberger
2026-02-21 18:24:24 +00:00
parent b2ed54f600
commit bd9d3e2f87
1 changed files with 21 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/cli/update-cli.test.ts
View File

@@ -3,7 +3,7 @@ import path from "node:path";
import { beforeEach, describe, expect, it, vi } from "vitest";
import type { OpenClawConfig, ConfigFileSnapshot } from "../config/types.openclaw.js";
import type { UpdateRunResult } from "../infra/update-runner.js";
import { captureEnv } from "../test-utils/env.js";
import { withEnvAsync } from "../test-utils/env.js";
const confirm = vi.fn();
const select = vi.fn();
@@ -604,30 +604,31 @@ describe("update-cli", () => {
});
it("updateCommand continues after doctor sub-step and clears update flag", async () => {
const envSnapshot = captureEnv(["OPENCLAW_UPDATE_IN_PROGRESS"]);
const randomSpy = vi.spyOn(Math, "random").mockReturnValue(0);
try {
delete process.env.OPENCLAW_UPDATE_IN_PROGRESS;
vi.mocked(runGatewayUpdate).mockResolvedValue(makeOkUpdateResult());
vi.mocked(runDaemonRestart).mockResolvedValue(true);
vi.mocked(doctorCommand).mockResolvedValue(undefined);
vi.mocked(defaultRuntime.log).mockClear();
await withEnvAsync({ OPENCLAW_UPDATE_IN_PROGRESS: undefined }, async () => {
vi.mocked(runGatewayUpdate).mockResolvedValue(makeOkUpdateResult());
vi.mocked(runDaemonRestart).mockResolvedValue(true);
vi.mocked(doctorCommand).mockResolvedValue(undefined);
vi.mocked(defaultRuntime.log).mockClear();
await updateCommand({});
await updateCommand({});
expect(doctorCommand).toHaveBeenCalledWith(
defaultRuntime,
expect.objectContaining({ nonInteractive: true }),
);
expect(process.env.OPENCLAW_UPDATE_IN_PROGRESS).toBeUndefined();
expect(doctorCommand).toHaveBeenCalledWith(
defaultRuntime,
expect.objectContaining({ nonInteractive: true }),
);
expect(process.env.OPENCLAW_UPDATE_IN_PROGRESS).toBeUndefined();
const logLines = vi.mocked(defaultRuntime.log).mock.calls.map((call) => String(call[0]));
expect(
logLines.some((line) => line.includes("Leveled up! New skills unlocked. You're welcome.")),
).toBe(true);
const logLines = vi.mocked(defaultRuntime.log).mock.calls.map((call) => String(call[0]));
expect(
logLines.some((line) =>
line.includes("Leveled up! New skills unlocked. You're welcome."),
),
).toBe(true);
});
} finally {
randomSpy.mockRestore();
envSnapshot.restore();
}
});
@@ -731,10 +732,8 @@ describe("update-cli", () => {
it("updateWizardCommand offers dev checkout and forwards selections", async () => {
const tempDir = createCaseDir("openclaw-update-wizard");
const envSnapshot = captureEnv(["OPENCLAW_GIT_DIR"]);
try {
await withEnvAsync({ OPENCLAW_GIT_DIR: tempDir }, async () => {
setTty(true);
process.env.OPENCLAW_GIT_DIR = tempDir;
vi.mocked(checkUpdateStatus).mockResolvedValue({
root: "/test/path",
@@ -760,8 +759,6 @@ describe("update-cli", () => {
const call = vi.mocked(runGatewayUpdate).mock.calls[0]?.[0];
expect(call?.channel).toBe("dev");
} finally {
envSnapshot.restore();
}
});
});
});