Function withMockCurrentSession

src/utils/planModeV2.test.ts:55–76

(
  session: ResolvedAuthSession,
  fn: () => T,
)

}

function withMockCurrentSession<T>(
  session: ResolvedAuthSession,
  fn: () => T,
): T {
  const runtime = getAuthRuntime()
  const originalGetCurrentSession = runtime.getCurrentSession.bind(runtime)
  ;(
    runtime as {
      getCurrentSession: typeof runtime.getCurrentSession
    }
  ).getCurrentSession = () => session

  try {
    return fn()
  } finally {
    ;(
      runtime as {
        getCurrentSession: typeof runtime.getCurrentSession
      }
    ).getCurrentSession = originalGetCurrentSession
  }
}

beforeEach(() => {
  delete process.env.CLAUDE_CODE_PLAN_V2_AGENT_COUNT