MailUI4Agents/plugins/pi-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, mailSessionFallback } from '../lib/workspace.js';

// 兜底目录现在由调用方给（各平台不同）。DSH 用 mailSessionFallback，
// opencode 用插件启动时的 directory。
const fallbackOf = key => mailSessionFallback(key);

test('存在的绝对路径直接用作 cwd', () => {
  const dir = mkdtempSync(join(tmpdir(), 'ws-test-'));
  try {
    const got = resolveWorkspaceCwd(dir, fallbackOf('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, fallbackOf('mail-aaa'));
    const b = resolveWorkspaceCwd(dir, fallbackOf('mail-bbb'));
    assert.equal(a.cwd, b.cwd, 'fallbackKey 不同却应得到同一个 cwd');
  } finally {
    rmSync(dir, { recursive: true, force: true });
  }
});

test('path 为空时回退到兜底目录', () => {
  const got = resolveWorkspaceCwd('', fallbackOf('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, fallbackOf('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', fallbackOf('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, fallbackOf('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, fallbackOf('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}  `, fallbackOf('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, '');
    assert.equal(got.grouped, true, 'ensureCwd 应已创建该目录');

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

test('兜底为空串时返回空 cwd（交给平台自己决定）', () => {
  // opencode 没配 directory 时就是这种情况：session.create 不带 query.directory，
  // 由平台按自己的默认规则选目录。比硬塞一个我们猜的路径好。
  const got = resolveWorkspaceCwd('', '');
  assert.equal(got.cwd, '');
  assert.equal(got.grouped, false);
});

test('mailSessionFallback 同一 key 稳定、不同 key 不同', () => {
  assert.equal(mailSessionFallback('a'), mailSessionFallback('a'));
  assert.notEqual(mailSessionFallback('a'), mailSessionFallback('b'));
  assert.match(mailSessionFallback('a'), /mail-sessions/);
});