assistant-ui--assistant-ui/packages/cli/test/utils/logger.test.ts

import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
import { logger } from "../../src/lib/utils/logger";

describe("logger utilities", () => {
  let consoleLogSpy: ReturnType<typeof vi.spyOn>;

  beforeEach(() => {
    consoleLogSpy = vi.spyOn(console, "log").mockImplementation(() => {});
  });

  afterEach(() => {
    consoleLogSpy.mockRestore();
  });

  it("should log info messages in blue", () => {
    logger.info("Test info message");

    expect(consoleLogSpy).toHaveBeenCalledOnce();
    const message = consoleLogSpy.mock.calls[0]?.[0] as string;
    expect(message).toContain("Test info message");
  });

  it("should log success messages in green with checkmark", () => {
    logger.success("Test success message");

    expect(consoleLogSpy).toHaveBeenCalledOnce();
    const message = consoleLogSpy.mock.calls[0]?.[0] as string;
    expect(message).toContain("✓");
    expect(message).toContain("Test success message");
  });

  it("should log error messages in red with cross", () => {
    const consoleErrorSpy = vi
      .spyOn(console, "error")
      .mockImplementation(() => {});

    logger.error("Test error message");

    expect(consoleErrorSpy).toHaveBeenCalledOnce();
    const message = consoleErrorSpy.mock.calls[0]?.[0] as string;
    expect(message).toContain("✗");
    expect(message).toContain("Test error message");

    consoleErrorSpy.mockRestore();
  });

  it("should log warning messages in yellow with warning icon", () => {
    logger.warn("Test warning message");

    expect(consoleLogSpy).toHaveBeenCalledOnce();
    const message = consoleLogSpy.mock.calls[0]?.[0] as string;
    expect(message).toContain("⚠");
    expect(message).toContain("Test warning message");
  });

  it("should log step messages in cyan with arrow", () => {
    logger.step("Test step message");

    expect(consoleLogSpy).toHaveBeenCalledOnce();
    const message = consoleLogSpy.mock.calls[0]?.[0] as string;
    expect(message).toContain("→");
    expect(message).toContain("Test step message");
  });

  it("should log empty line for break", () => {
    logger.break();

    expect(consoleLogSpy).toHaveBeenCalledWith("");
  });
});