baoyu-skills/packages/baoyu-fetch/src/__tests__/profile.test.ts

import { afterEach, describe, expect, test } from "bun:test";
import fs from "node:fs";
import path from "node:path";
import os from "node:os";
import { ensureChromeProfileDir, resolveChromeProfileDir } from "../browser/profile";

const originalProfile = process.env.BAOYU_CHROME_PROFILE_DIR;

afterEach(() => {
  if (originalProfile === undefined) {
    delete process.env.BAOYU_CHROME_PROFILE_DIR;
  } else {
    process.env.BAOYU_CHROME_PROFILE_DIR = originalProfile;
  }
});

describe("resolveChromeProfileDir", () => {
  test("uses BAOYU_CHROME_PROFILE_DIR when set", () => {
    process.env.BAOYU_CHROME_PROFILE_DIR = "/tmp/baoyu-profile";
    expect(resolveChromeProfileDir()).toBe("/tmp/baoyu-profile");
  });

  test("falls back to shared baoyu-skills profile path", () => {
    delete process.env.BAOYU_CHROME_PROFILE_DIR;
    const resolved = resolveChromeProfileDir();
    if (process.platform === "darwin") {
      expect(resolved).toBe(path.join(os.homedir(), "Library", "Application Support", "baoyu-skills", "chrome-profile"));
    } else if (process.platform === "win32") {
      expect(resolved.endsWith(path.join("baoyu-skills", "chrome-profile"))).toBe(true);
    } else {
      expect(resolved.endsWith(path.join("baoyu-skills", "chrome-profile"))).toBe(true);
    }
  });
});

describe("ensureChromeProfileDir", () => {
  test("creates the profile directory when missing", () => {
    const tempRoot = fs.mkdtempSync(path.join(os.tmpdir(), "baoyu-fetch-profile-"));
    const profileDir = path.join(tempRoot, "nested", "chrome-profile");

    try {
      expect(fs.existsSync(profileDir)).toBe(false);
      expect(ensureChromeProfileDir(profileDir)).toBe(profileDir);
      expect(fs.statSync(profileDir).isDirectory()).toBe(true);
    } finally {
      fs.rmSync(tempRoot, { force: true, recursive: true });
    }
  });
});