documents/utils/sidebar.test.ts

import { describe, expect, test, beforeAll, afterAll } from "vitest";
import { scanDir } from "../utils/sidebar";
import { resolve } from "path";
import { mkdtempSync, writeFileSync, rmdirSync } from "fs";
import { tmpdir } from "os";
import { join } from "path";

let tempDir: string;

beforeAll(() => {
  tempDir = mkdtempSync(join(tmpdir(), 'test-'));
  writeFileSync(join(tempDir, 'test1.md'), '# Test 1');
  writeFileSync(join(tempDir, 'test2.md'), '# Test 2');
  writeFileSync(join(tempDir, 'test.txt'), 'Test text file');
});

afterAll(() => {
  rmdirSync(tempDir, { recursive: true });
});

describe("scanDir", () => {
  test("should return markdown files with correct structure", () => {
    const res = scanDir(tempDir);
    expect(res).toBeInstanceOf(Array);
    expect(res.length).toBe(2);
    res.forEach(item => {
      expect(item).toHaveProperty("filename");
      expect(item).toHaveProperty("link");
      expect(item.filename).toMatch(/\.md$/);
      expect(item.link).toBe(resolve(tempDir, item.filename));
    });
  });

  test("should return an empty array if no markdown files are found", () => {
    const emptyDir = mkdtempSync(join(tmpdir(), 'empty-'));
    const res = scanDir(emptyDir);
    expect(res).toEqual([]);
    rmdirSync(emptyDir, { recursive: true });
  });
});