/* Copyright 2026 Marimo. All rights reserved. */

import { describe, expect, it } from "vitest";
import { ChartSchema } from "../schemas";

describe("ChartSchema", () => {
  it("keeps a known axis column type", () => {
    const result = ChartSchema.parse({
      general: { xColumn: { field: "count", type: "integer" } },
    });

    expect(result.general?.xColumn?.type).toBe("integer");
  });

  it("normalizes an unrecognized axis column type", () => {
    const result = ChartSchema.parse({
      general: { xColumn: { field: "geom", type: "bogus_type" } },
    });

    expect(result.general?.xColumn?.type).toBe("unknown");
  });

  it("normalizes an unrecognized tooltip field type", () => {
    const result = ChartSchema.parse({
      tooltips: {
        auto: false,
        fields: [{ field: "geom", type: "bogus_type" }],
      },
    });

    expect(result.tooltips?.fields).toEqual([
      { field: "geom", type: "unknown" },
    ]);
  });

  it("keeps geometry axis and tooltip field types", () => {
    const result = ChartSchema.parse({
      general: { xColumn: { field: "geom", type: "geometry" } },
      tooltips: {
        auto: false,
        fields: [{ field: "geom", type: "geometry" }],
      },
    });

    expect(result.general?.xColumn?.type).toBe("geometry");
    expect(result.tooltips?.fields).toEqual([
      { field: "geom", type: "geometry" },
    ]);
  });
});