@@ -9,6 +9,7 @@ import type { InternalGetReplyOptions } from "../auto-reply/reply/get-reply.type
99import { clearConfigCache } from "../config/config.js" ;
1010import type { AgentModelConfig } from "../config/types.agents-shared.js" ;
1111import { createDeferred } from "../test-utils/deferred.js" ;
12+ import { captureEnv , setTestEnvValue } from "../test-utils/env.js" ;
1213import { GATEWAY_CLIENT_MODES , GATEWAY_CLIENT_NAMES } from "../utils/message-channel.js" ;
1314import { setMaxChatHistoryMessagesBytesForTest } from "./server-constants.js" ;
1415import type { GatewayRequestContext , RespondFn } from "./server-methods/shared-types.js" ;
@@ -1992,7 +1993,7 @@ describe("gateway server chat", () => {
19921993 await connectOk ( ws ) ;
19931994 const sessionDir = await createSessionDir ( ) ;
19941995 const sessionId = "sess-claude-cli-backfill" ;
1995- const originalHome = process . env . HOME ;
1996+ const homeEnvSnapshot = captureEnv ( [ " HOME" ] ) ;
19961997 const homeDir = path . join ( sessionDir , "home" ) ;
19971998 const cliSessionId = "5b8b202c-f6bb-4046-9475-d2f15fd07530" ;
19981999 const claudeProjectsDir = path . join ( homeDir , ".claude" , "projects" , "workspace" ) ;
@@ -2030,7 +2031,7 @@ describe("gateway server chat", () => {
20302031 ] . join ( "\n" ) ,
20312032 "utf-8" ,
20322033 ) ;
2033- process . env . HOME = homeDir ;
2034+ setTestEnvValue ( " HOME" , homeDir ) ;
20342035 try {
20352036 await writeSessionStore ( {
20362037 entries : {
@@ -2058,11 +2059,7 @@ describe("gateway server chat", () => {
20582059 expect ( assistantMessage . role ) . toBe ( "assistant" ) ;
20592060 expect ( assistantMessage . provider ) . toBe ( "claude-cli" ) ;
20602061 } finally {
2061- if ( originalHome === undefined ) {
2062- delete process . env . HOME ;
2063- } else {
2064- process . env . HOME = originalHome ;
2065- }
2062+ homeEnvSnapshot . restore ( ) ;
20662063 }
20672064 } ) ;
20682065 } ) ;
0 commit comments