From 5e607ae1eb0e249ee3b3a75e691292dcfea6a3b7 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sat, 21 Feb 2026 18:42:27 +0000
Subject: [PATCH] refactor(test): snapshot deprecated auth profile env in e2e

---
 ...or-auth.deprecated-cli-profiles.e2e.test.ts | 18 ++++--------------
 1 file changed, 4 insertions(+), 14 deletions(-)

diff --git a/src/commands/doctor-auth.deprecated-cli-profiles.e2e.test.ts b/src/commands/doctor-auth.deprecated-cli-profiles.e2e.test.ts
index bf3e59c2d..d6436d702 100644
--- a/src/commands/doctor-auth.deprecated-cli-profiles.e2e.test.ts
+++ b/src/commands/doctor-auth.deprecated-cli-profiles.e2e.test.ts
@@ -3,11 +3,11 @@ import os from "node:os";
 import path from "node:path";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import type { OpenClawConfig } from "../config/config.js";
+import { captureEnv } from "../test-utils/env.js";
 import { maybeRemoveDeprecatedCliAuthProfiles } from "./doctor-auth.js";
 import type { DoctorPrompter } from "./doctor-prompter.js";
 
-let originalAgentDir: string | undefined;
-let originalPiAgentDir: string | undefined;
+let envSnapshot: ReturnType<typeof captureEnv>;
 let tempAgentDir: string | undefined;
 
 function makePrompter(confirmValue: boolean): DoctorPrompter {
@@ -23,24 +23,14 @@ function makePrompter(confirmValue: boolean): DoctorPrompter {
 }
 
 beforeEach(() => {
-  originalAgentDir = process.env.OPENCLAW_AGENT_DIR;
-  originalPiAgentDir = process.env.PI_CODING_AGENT_DIR;
+  envSnapshot = captureEnv(["OPENCLAW_AGENT_DIR", "PI_CODING_AGENT_DIR"]);
   tempAgentDir = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-auth-"));
   process.env.OPENCLAW_AGENT_DIR = tempAgentDir;
   process.env.PI_CODING_AGENT_DIR = tempAgentDir;
 });
 
 afterEach(() => {
-  if (originalAgentDir === undefined) {
-    delete process.env.OPENCLAW_AGENT_DIR;
-  } else {
-    process.env.OPENCLAW_AGENT_DIR = originalAgentDir;
-  }
-  if (originalPiAgentDir === undefined) {
-    delete process.env.PI_CODING_AGENT_DIR;
-  } else {
-    process.env.PI_CODING_AGENT_DIR = originalPiAgentDir;
-  }
+  envSnapshot.restore();
   if (tempAgentDir) {
     fs.rmSync(tempAgentDir, { recursive: true, force: true });
     tempAgentDir = undefined;