import { render, screen } from "@testing-library/react";
import { describe, expect, it } from "vitest";
import { CompanyStatusBadge, SeverityBadge } from "@/components/ui/badge";
import { SEVERITY_LEVELS } from "@/lib/types";

describe("SeverityBadge", () => {
  it.each(SEVERITY_LEVELS)("renders the %s severity label", (severity) => {
    render(<SeverityBadge severity={severity} />);
    expect(screen.getByText(severity)).toBeInTheDocument();
  });
});

describe("CompanyStatusBadge", () => {
  it("renders active status", () => {
    render(<CompanyStatusBadge status="active" />);
    expect(screen.getByText("active")).toBeInTheDocument();
  });

  it("renders paused status", () => {
    render(<CompanyStatusBadge status="paused" />);
    expect(screen.getByText("paused")).toBeInTheDocument();
  });
});