import { describe, expect, it } from "vitest";

import { ORGAN_IDS } from "../constants";
import { createOrganOutlineSet, createUniformOrganColorSchemes } from "../lib/organControlState";

describe("organ control helpers", () => {
  it("creates uniform color schemes for all organs", () => {
    const schemes = createUniformOrganColorSchemes(ORGAN_IDS, "yellow");

    for (const organId of ORGAN_IDS) {
      expect(schemes[organId]).toBe("yellow");
    }
  });

  it("creates outline sets based on the enabled flag", () => {
    const enabledSet = createOrganOutlineSet(ORGAN_IDS, true);
    const disabledSet = createOrganOutlineSet(ORGAN_IDS, false);

    expect(enabledSet.size).toBe(ORGAN_IDS.length);
    for (const organId of ORGAN_IDS) {
      expect(enabledSet.has(organId)).toBe(true);
    }

    expect(disabledSet.size).toBe(0);
  });
});