oh-my-opencode/src/hooks/session-recovery/storage/readers-from-sdk.test.ts

import { describe, expect, it } from "bun:test"
async function importFreshReaders() {
  const token = `${Date.now()}-${Math.random()}`
  const [{ readMessagesFromSDK, readMessages }, { readPartsFromSDK, readParts }] = await Promise.all([
    import(`./messages-reader?test=${token}`),
    import(`./parts-reader?test=${token}`),
  ])

  return { readMessagesFromSDK, readPartsFromSDK, readMessages, readParts }
}

function createMockClient(handlers: {
  messages?: (sessionID: string) => unknown[]
  message?: (sessionID: string, messageID: string) => unknown
}) {
  return {
    session: {
      messages: async (opts: { path: { id: string } }) => {
        if (handlers.messages) {
          return { data: handlers.messages(opts.path.id) }
        }
        throw new Error("not implemented")
      },
      message: async (opts: { path: { id: string; messageID: string } }) => {
        if (handlers.message) {
          return { data: handlers.message(opts.path.id, opts.path.messageID) }
        }
        throw new Error("not implemented")
      },
    },
  } as unknown
}

describe("session-recovery storage SDK readers", () => {
  it("readPartsFromSDK returns empty array when fetch fails", async () => {
    //#given a client that throws on request
    const { readPartsFromSDK } = await importFreshReaders()
    const client = createMockClient({}) as Parameters<typeof readPartsFromSDK>[0]

    //#when readPartsFromSDK is called
    const result = await readPartsFromSDK(client, "ses_test", "msg_test")

    //#then it returns empty array
    expect(result).toEqual([])
  })

  it("readPartsFromSDK returns stored parts from SDK response", async () => {
    //#given a client that returns a message with parts
    const { readPartsFromSDK } = await importFreshReaders()
    const sessionID = "ses_test"
    const messageID = "msg_test"
    const storedParts = [
      { id: "prt_1", sessionID, messageID, type: "text", text: "hello" },
    ]

    const client = createMockClient({
      message: (_sid, _mid) => ({ parts: storedParts }),
    }) as Parameters<typeof readPartsFromSDK>[0]

    //#when readPartsFromSDK is called
    const result = await readPartsFromSDK(client, sessionID, messageID)

    //#then it returns the parts
    expect(result).toEqual(storedParts)
  })

  it("readMessagesFromSDK normalizes and sorts messages", async () => {
    //#given a client that returns messages list
    const { readMessagesFromSDK } = await importFreshReaders()
    const sessionID = "ses_test"
    const client = createMockClient({
      messages: () => [
        { id: "msg_b", role: "assistant", time: { created: 2 } },
        { id: "msg_a", role: "user", time: { created: 1 } },
        { id: "msg_c" },
      ],
    }) as Parameters<typeof readMessagesFromSDK>[0]

    //#when readMessagesFromSDK is called
    const result = await readMessagesFromSDK(client, sessionID)

    //#then it returns sorted StoredMessageMeta with defaults
    expect(result).toEqual([
      { id: "msg_c", sessionID, role: "user", time: { created: 0 } },
      { id: "msg_a", sessionID, role: "user", time: { created: 1 } },
      { id: "msg_b", sessionID, role: "assistant", time: { created: 2 } },
    ])
  })

  it("readParts returns empty array for nonexistent message", async () => {
    //#given a message ID that has no stored parts
    const { readParts } = await importFreshReaders()
    //#when readParts is called
    const parts = readParts("msg_nonexistent")

    //#then it returns empty array
    expect(parts).toEqual([])
  })

  it("readMessages returns empty array for nonexistent session", async () => {
    //#given a session ID that has no stored messages
    const { readMessages } = await importFreshReaders()
    //#when readMessages is called
    const messages = readMessages("ses_nonexistent")

    //#then it returns empty array
    expect(messages).toEqual([])
  })
})