Cover workspace registry behavior

Author: slashdevcorpse

apps/codex-claw/src/server/codex-cli.test.ts

@@ -1,6 +1,18 @@
-import { describe, expect, it } from 'vitest'
+import { mkdtempSync, rmSync } from 'node:fs'
+import os from 'node:os'
+import path from 'node:path'
+import { afterEach, beforeEach, describe, expect, it } from 'vitest'
 
-import { mergeAssistantText, processCodexJsonLine } from './codex-cli'
+import {
+  createCodexWorkspace,
+  deleteCodexWorkspace,
+  getCodexPaths,
+  listCodexWorkspaces,
+  mergeAssistantText,
+  patchCodexWorkspace,
+  processCodexJsonLine,
+  resetCodexServerStateForTests,
+} from './codex-cli'
 
 describe('processCodexJsonLine', function () {
   it('parses assistant deltas before the final completed item', function () {
@@ -61,3 +73,103 @@ describe('mergeAssistantText', function () {
     expect(repeated).toBe('Hello')
   })
 })
+
+describe('codex workspace registry', function () {
+  const originalStateDir = process.env.CODEX_CLAW_STATE_DIR
+  const originalCommand = process.env.CODEX_CLI_COMMAND
+  const originalWorkdir = process.env.CODEX_CLI_WORKDIR
+  const originalSandbox = process.env.CODEX_CLI_SANDBOX
+  let tempDir = ''
+
+  beforeEach(function () {
+    tempDir = mkdtempSync(path.join(os.tmpdir(), 'codex-claw-test-'))
+    process.env.CODEX_CLAW_STATE_DIR = path.join(tempDir, 'state')
+    process.env.CODEX_CLI_COMMAND = 'codex-test'
+    process.env.CODEX_CLI_WORKDIR = tempDir
+    process.env.CODEX_CLI_SANDBOX = 'read-only'
+    resetCodexServerStateForTests()
+  })
+
+  afterEach(function () {
+    if (originalStateDir === undefined) {
+      delete process.env.CODEX_CLAW_STATE_DIR
+    } else {
+      process.env.CODEX_CLAW_STATE_DIR = originalStateDir
+    }
+    if (originalCommand === undefined) {
+      delete process.env.CODEX_CLI_COMMAND
+    } else {
+      process.env.CODEX_CLI_COMMAND = originalCommand
+    }
+    if (originalWorkdir === undefined) {
+      delete process.env.CODEX_CLI_WORKDIR
+    } else {
+      process.env.CODEX_CLI_WORKDIR = originalWorkdir
+    }
+    if (originalSandbox === undefined) {
+      delete process.env.CODEX_CLI_SANDBOX
+    } else {
+      process.env.CODEX_CLI_SANDBOX = originalSandbox
+    }
+    resetCodexServerStateForTests()
+    rmSync(tempDir, { recursive: true, force: true })
+  })
+
+  it('stores independent active workspace configuration', function () {
+    expect(getCodexPaths().workspace).toMatchObject({
+      id: 'default',
+      codexCommand: 'codex-test',
+      codexWorkdir: tempDir,
+      stateDir: path.join(tempDir, 'state'),
+    })
+
+    createCodexWorkspace({
+      name: 'Docs repo',
+      codexCommand: 'codex-next',
+      codexSandbox: 'workspace-write',
+      codexWorkdir: path.join(tempDir, 'docs'),
+      stateDir: path.join(tempDir, 'docs-state'),
+      active: true,
+    })
+
+    expect(getCodexPaths()).toMatchObject({
+      stateDir: path.join(tempDir, 'docs-state'),
+      workspace: {
+        id: 'docs-repo',
+        name: 'Docs repo',
+        codexCommand: 'codex-next',
+        codexSandbox: 'workspace-write',
+        codexWorkdir: path.join(tempDir, 'docs'),
+      },
+    })
+  })
+
+  it('renames and removes non-default workspaces', function () {
+    createCodexWorkspace({
+      name: 'Feature repo',
+      codexWorkdir: path.join(tempDir, 'feature'),
+      stateDir: path.join(tempDir, 'feature-state'),
+      active: true,
+    })
+    patchCodexWorkspace({
+      id: 'feature-repo',
+      name: 'Feature branch repo',
+    })
+
+    expect(listCodexWorkspaces().workspaces).toEqual(
+      expect.arrayContaining([
+        expect.objectContaining({
+          id: 'feature-repo',
+          name: 'Feature branch repo',
+        }),
+      ]),
+    )
+
+    deleteCodexWorkspace('feature-repo')
+
+    expect(listCodexWorkspaces()).toMatchObject({
+      activeWorkspaceId: 'default',
+      workspaces: [expect.objectContaining({ id: 'default' })],
+    })
+  })
+})