@assistant-ui/react/src/primitives/thread/topAnchor/topAnchorTurn.test.ts

Open-source TypeScript/React library for building production-grade AI chat experiences

import { describe, expect, it } from "vitest";
import {
  getActiveTopAnchorAnchorId,
  getActiveTopAnchorTargetId,
  getActiveTopAnchorTurn,
  isTopAnchorTurnValid,
} from "./topAnchorTurn";

describe("topAnchorTurn", () => {
  it("does not activate history-loaded messages", () => {
    const messages = [
      { id: "user-1", role: "user" },
      { id: "assistant-1", role: "assistant" },
    ];

    expect(getActiveTopAnchorTurn({ isRunning: false, messages })).toBe(null);
  });

  it("activates the live user/assistant pair while running", () => {
    const messages = [
      { id: "assistant-1", role: "assistant" },
      { id: "user-2", role: "user" },
      { id: "assistant-2", role: "assistant" },
    ];

    expect(getActiveTopAnchorTurn({ isRunning: true, messages })).toEqual({
      anchorId: "user-2",
      targetId: "assistant-2",
    });
    expect(getActiveTopAnchorAnchorId({ isRunning: true, messages })).toBe(
      "user-2",
    );
    expect(getActiveTopAnchorTargetId({ isRunning: true, messages })).toBe(
      "assistant-2",
    );
  });

  it("ignores running states without a trailing user/assistant pair", () => {
    const messages = [
      { id: "user-1", role: "user" },
      { id: "assistant-1", role: "assistant" },
      { id: "user-2", role: "user" },
    ];

    expect(getActiveTopAnchorTurn({ isRunning: true, messages })).toBe(null);
  });

  it("keeps a completed turn valid while it remains the trailing turn", () => {
    const turn = { anchorId: "user-2", targetId: "assistant-2" };

    expect(
      isTopAnchorTurnValid(turn, [
        { id: "assistant-1", role: "assistant" },
        { id: "user-2", role: "user" },
        { id: "assistant-2", role: "assistant" },
      ]),
    ).toBe(true);
  });

  it("keeps a turn valid across the between-turns gap with pending user messages", () => {
    const turn = { anchorId: "user-2", targetId: "assistant-2" };

    expect(
      isTopAnchorTurnValid(turn, [
        { id: "user-2", role: "user" },
        { id: "assistant-2", role: "assistant" },
        { id: "user-3", role: "user" },
        { id: "user-4", role: "user" },
      ]),
    ).toBe(true);
  });

  it.each([
    [
      "thread switch",
      [
        { id: "other-user", role: "user" },
        { id: "other-assistant", role: "assistant" },
      ],
    ],
    ["new chat", []],
    ["last-turn deletion", [{ id: "older-assistant", role: "assistant" }]],
    [
      "branch swap",
      [
        { id: "branch-user", role: "user" },
        { id: "branch-assistant", role: "assistant" },
      ],
    ],
    [
      "an externally synced completed turn",
      [
        { id: "user-2", role: "user" },
        { id: "assistant-2", role: "assistant" },
        { id: "user-3", role: "user" },
        { id: "assistant-3", role: "assistant" },
      ],
    ],
    [
      "a resync separating the pair",
      [
        { id: "user-2", role: "user" },
        { id: "inserted-assistant", role: "assistant" },
        { id: "assistant-2", role: "assistant" },
      ],
    ],
  ])("invalidates a stored turn after %s", (_transition, messages) => {
    expect(
      isTopAnchorTurnValid(
        { anchorId: "user-2", targetId: "assistant-2" },
        messages,
      ),
    ).toBe(false);
  });
});