oh-my-opencode/src/hooks/directory-readme-injector/injector.test.ts

import { afterEach, beforeEach, describe, expect, it, mock } from "bun:test"
import { mkdirSync, rmSync, writeFileSync } from "node:fs"
import { tmpdir } from "node:os"
import { join } from "node:path"

const findReadmeMdUpMock = mock((_: { startDir: string; rootDir: string }) => [] as string[])
const resolveFilePathMock = mock((_: string, path: string) => path)
const loadInjectedPathsMock = mock((_: string) => new Set<string>())
const saveInjectedPathsMock = mock((_: string, __: Set<string>) => {})

describe("processFilePathForReadmeInjection", () => {
  let testRoot = ""

  beforeEach(() => {
    findReadmeMdUpMock.mockClear()
    resolveFilePathMock.mockClear()
    loadInjectedPathsMock.mockClear()
    saveInjectedPathsMock.mockClear()

    testRoot = join(
      tmpdir(),
      `directory-readme-injector-${Date.now()}-${Math.random().toString(16).slice(2)}`
    )
    mkdirSync(testRoot, { recursive: true })
  })

  afterEach(() => {
    mock.restore()
    rmSync(testRoot, { recursive: true, force: true })
  })

  it("does not save when all discovered paths are already cached", async () => {
    //#given
    const sessionID = "session-1"
    const repoRoot = join(testRoot, "repo")
    const readmePath = join(repoRoot, "src", "README.md")
    const cachedDirectory = join(repoRoot, "src")
    mkdirSync(join(repoRoot, "src"), { recursive: true })
    writeFileSync(readmePath, "# README")

    loadInjectedPathsMock.mockReturnValueOnce(new Set([cachedDirectory]))
    findReadmeMdUpMock.mockReturnValueOnce([readmePath])

    const truncator = {
      truncate: mock(async () => ({ result: "trimmed", truncated: false })),
    }

    mock.module("./finder", () => ({
      findReadmeMdUp: findReadmeMdUpMock,
      resolveFilePath: resolveFilePathMock,
    }))
    mock.module("./storage", () => ({
      loadInjectedPaths: loadInjectedPathsMock,
      saveInjectedPaths: saveInjectedPathsMock,
    }))

    const { processFilePathForReadmeInjection } = await import("./injector")

    //#when
    await processFilePathForReadmeInjection({
      ctx: { directory: repoRoot } as never,
      truncator: truncator as never,
      sessionCaches: new Map(),
      filePath: join(repoRoot, "src", "file.ts"),
      sessionID,
      output: { title: "Result", output: "", metadata: {} },
    })

    //#then
    expect(saveInjectedPathsMock).not.toHaveBeenCalled()
  })

  it("saves when a new path is injected", async () => {
    //#given
    const sessionID = "session-2"
    const repoRoot = join(testRoot, "repo")
    const readmePath = join(repoRoot, "src", "README.md")
    const injectedDirectory = join(repoRoot, "src")
    mkdirSync(join(repoRoot, "src"), { recursive: true })
    writeFileSync(readmePath, "# README")

    loadInjectedPathsMock.mockReturnValueOnce(new Set())
    findReadmeMdUpMock.mockReturnValueOnce([readmePath])

    const truncator = {
      truncate: mock(async () => ({ result: "trimmed", truncated: false })),
    }

    mock.module("./finder", () => ({
      findReadmeMdUp: findReadmeMdUpMock,
      resolveFilePath: resolveFilePathMock,
    }))
    mock.module("./storage", () => ({
      loadInjectedPaths: loadInjectedPathsMock,
      saveInjectedPaths: saveInjectedPathsMock,
    }))

    const { processFilePathForReadmeInjection } = await import("./injector")

    //#when
    await processFilePathForReadmeInjection({
      ctx: { directory: repoRoot } as never,
      truncator: truncator as never,
      sessionCaches: new Map(),
      filePath: join(repoRoot, "src", "file.ts"),
      sessionID,
      output: { title: "Result", output: "", metadata: {} },
    })

    //#then
    expect(saveInjectedPathsMock).toHaveBeenCalledTimes(1)
    const saveCall = saveInjectedPathsMock.mock.calls[0]
    expect(saveCall[0]).toBe(sessionID)
    expect((saveCall[1] as Set<string>).has(injectedDirectory)).toBe(true)
  })

  it("saves once when cached and new paths are mixed", async () => {
    //#given
    const sessionID = "session-3"
    const repoRoot = join(testRoot, "repo")
    const cachedReadmePath = join(repoRoot, "already-cached", "README.md")
    const newReadmePath = join(repoRoot, "new-dir", "README.md")
    mkdirSync(join(repoRoot, "already-cached"), { recursive: true })
    mkdirSync(join(repoRoot, "new-dir"), { recursive: true })
    writeFileSync(cachedReadmePath, "# README")
    writeFileSync(newReadmePath, "# README")

    loadInjectedPathsMock.mockReturnValueOnce(new Set([join(repoRoot, "already-cached")]))
    findReadmeMdUpMock.mockReturnValueOnce([cachedReadmePath, newReadmePath])

    const truncator = {
      truncate: mock(async () => ({ result: "trimmed", truncated: false })),
    }

    mock.module("./finder", () => ({
      findReadmeMdUp: findReadmeMdUpMock,
      resolveFilePath: resolveFilePathMock,
    }))
    mock.module("./storage", () => ({
      loadInjectedPaths: loadInjectedPathsMock,
      saveInjectedPaths: saveInjectedPathsMock,
    }))

    const { processFilePathForReadmeInjection } = await import("./injector")

    //#when
    await processFilePathForReadmeInjection({
      ctx: { directory: repoRoot } as never,
      truncator: truncator as never,
      sessionCaches: new Map(),
      filePath: join(repoRoot, "new-dir", "file.ts"),
      sessionID,
      output: { title: "Result", output: "", metadata: {} },
    })

    //#then
    expect(saveInjectedPathsMock).toHaveBeenCalledTimes(1)
    const saveCall = saveInjectedPathsMock.mock.calls[0]
    expect((saveCall[1] as Set<string>).has(join(repoRoot, "new-dir"))).toBe(true)
  })
})