MailUI4Agents/plugins/dsh-mail-bridge/test/workspace.test.mjs

/**
 * 工作目录解析的回归测试。
 *
 * 这是「dsh 指定工作目录完全失效，所有对话都落在未分组下」那次故障的直接回归：
 * 插件曾无视寻址里的 path 位，每封邮件自己拼一个 ~/.dsh/mail-sessions/mail-<uuid>，
 * 而 DSH 按 cwd 分组，于是所有邮件会话既不属于任何项目、彼此也不同组。
 *
 *   node --test test/
 */

import { test } from 'node:test';
import assert from 'node:assert/strict';
import { mkdtempSync, rmSync, writeFileSync } from 'node:fs';
import { tmpdir, homedir } from 'node:os';
import { join } from 'node:path';
import { resolveWorkspaceCwd, ensureCwd } from '../lib/workspace.js';

const fallbackOf = key => join(homedir(), '.dsh', 'mail-sessions', key);

test('存在的绝对路径直接用作 cwd', () => {
  const dir = mkdtempSync(join(tmpdir(), 'ws-test-'));
  try {
    const got = resolveWorkspaceCwd(dir, 'mail-1');
    assert.equal(got.cwd, dir);
    assert.equal(got.grouped, true);
  } finally {
    rmSync(dir, { recursive: true, force: true });
  }
});

test('不变量：同一 path 的多封邮件得到同一个 cwd（这才能同组）', () => {
  const dir = mkdtempSync(join(tmpdir(), 'ws-test-'));
  try {
    const a = resolveWorkspaceCwd(dir, 'mail-aaa');
    const b = resolveWorkspaceCwd(dir, 'mail-bbb');
    assert.equal(a.cwd, b.cwd, 'fallbackKey 不同却应得到同一个 cwd');
  } finally {
    rmSync(dir, { recursive: true, force: true });
  }
});

test('path 为空时回退到兜底目录', () => {
  const got = resolveWorkspaceCwd('', 'mail-2');
  assert.equal(got.cwd, fallbackOf('mail-2'));
  assert.equal(got.grouped, false);
});

test('path 缺失/非字符串时回退', () => {
  for (const v of [undefined, null, 42, {}]) {
    const got = resolveWorkspaceCwd(v, 'mail-3');
    assert.equal(got.grouped, false);
    assert.equal(got.cwd, fallbackOf('mail-3'));
  }
});

test('不变量：不存在的目录不创建，回退到兜底', () => {
  // 一个笔误（/home/porgram/x）不该在磁盘上落下真目录 ——
  // Agent 会在里面一无所获地干活，比明确回退更难排查。
  const got = resolveWorkspaceCwd('/nonexistent/path/xyz-should-not-exist', 'mail-4');
  assert.equal(got.grouped, false);
  assert.equal(got.cwd, fallbackOf('mail-4'));
});

test('不变量：相对路径被拒绝', () => {
  // cwd 的相对基准是 harness 进程的启动目录，systemd 下通常是 /，
  // 那是个与邮件语义完全无关的量。
  for (const rel of ['relative/path', './x', '../y', 'src']) {
    const got = resolveWorkspaceCwd(rel, 'mail-5');
    assert.equal(got.grouped, false, `${rel} 不该被当作工作目录`);
  }
});

test('指向文件而非目录时回退', () => {
  const dir = mkdtempSync(join(tmpdir(), 'ws-test-'));
  const file = join(dir, 'a-file');
  writeFileSync(file, 'x');
  try {
    const got = resolveWorkspaceCwd(file, 'mail-6');
    assert.equal(got.grouped, false);
  } finally {
    rmSync(dir, { recursive: true, force: true });
  }
});

test('两端空白被修掉', () => {
  const dir = mkdtempSync(join(tmpdir(), 'ws-test-'));
  try {
    const got = resolveWorkspaceCwd(`  ${dir}  `, 'mail-7');
    assert.equal(got.cwd, dir);
    assert.equal(got.grouped, true);
  } finally {
    rmSync(dir, { recursive: true, force: true });
  }
});

test('ensureCwd 只建兜底目录，不碰寻址指定的目录', () => {
  const base = mkdtempSync(join(tmpdir(), 'ws-ensure-'));
  try {
    const target = join(base, 'made-by-ensure');
    ensureCwd(target, false);
    // 建出来了
    const got = resolveWorkspaceCwd(target, 'x');
    assert.equal(got.grouped, true, 'ensureCwd 应已创建该目录');

    // grouped=true 时不该创建（那种目录本来就存在）
    const never = join(base, 'should-not-exist');
    ensureCwd(never, true);
    assert.equal(resolveWorkspaceCwd(never, 'x').grouped, false);
  } finally {
    rmSync(base, { recursive: true, force: true });
  }
});