oh-my-opencode/src/shared/session-cursor.test.ts

import { beforeEach, describe, expect, it } from "bun:test"
import { consumeNewMessages, resetMessageCursor } from "./session-cursor"

describe("consumeNewMessages", () => {
  const sessionID = "session-123"

  const buildMessage = (id: string, created: number) => ({
    info: { id, time: { created } },
  })

  beforeEach(() => {
    resetMessageCursor(sessionID)
  })

  it("returns all messages on first read and none on repeat", () => {
    // given
    const messages = [buildMessage("m1", 1), buildMessage("m2", 2)]

    // when
    const first = consumeNewMessages(sessionID, messages)
    const second = consumeNewMessages(sessionID, messages)

    // then
    expect(first).toEqual(messages)
    expect(second).toEqual([])
  })

  it("returns only new messages after cursor advances", () => {
    // given
    const messages = [buildMessage("m1", 1), buildMessage("m2", 2)]
    consumeNewMessages(sessionID, messages)
    const extended = [...messages, buildMessage("m3", 3)]

    // when
    const next = consumeNewMessages(sessionID, extended)

    // then
    expect(next).toEqual([extended[2]])
  })

  it("resets when message history shrinks", () => {
    // given
    const messages = [buildMessage("m1", 1), buildMessage("m2", 2)]
    consumeNewMessages(sessionID, messages)
    const shorter = [buildMessage("n1", 1)]

    // when
    const next = consumeNewMessages(sessionID, shorter)

    // then
    expect(next).toEqual(shorter)
  })

  it("returns all messages when last key is missing", () => {
    // given
    const messages = [buildMessage("m1", 1), buildMessage("m2", 2)]
    consumeNewMessages(sessionID, messages)
    const replaced = [buildMessage("n1", 1), buildMessage("n2", 2)]

    // when
    const next = consumeNewMessages(sessionID, replaced)

    // then
    expect(next).toEqual(replaced)
  })
})