From 9ad2731358dee14a27c4d86426e8124ff2a7f758 Mon Sep 17 00:00:00 2001
From: "GitStart-Cal.com" <121884634+gitstart-calcom@users.noreply.github.com>
Date: Thu, 20 Jul 2023 06:11:53 +0100
Subject: [PATCH] test: add unit tests for `packages/ui/components/badge`
 (CALCOM-613) (#10255)

Co-authored-by: gitstart-calcom <gitstart@users.noreply.github.com>
---
 packages/ui/components/badge/Badge.tsx      |  6 +-
 packages/ui/components/badge/badge.test.tsx | 92 +++++++++++++++++++++
 2 files changed, 95 insertions(+), 3 deletions(-)
 create mode 100644 packages/ui/components/badge/badge.test.tsx

diff --git a/packages/ui/components/badge/Badge.tsx b/packages/ui/components/badge/Badge.tsx
index f90331dcc9..6134a8793b 100644
--- a/packages/ui/components/badge/Badge.tsx
+++ b/packages/ui/components/badge/Badge.tsx
@@ -6,7 +6,7 @@ import { GoPrimitiveDot } from "react-icons/go";
 import classNames from "@calcom/lib/classNames";
 import type { SVGComponent } from "@calcom/types/SVGComponent";
 
-const badgeStyles = cva("font-medium inline-flex items-center justify-center rounded gap-x-1", {
+export const badgeStyles = cva("font-medium inline-flex items-center justify-center rounded gap-x-1", {
   variants: {
     variant: {
       default: "bg-attention text-attention",
@@ -68,8 +68,8 @@ export const Badge = function Badge(props: BadgeProps) {
 
   const Children = () => (
     <>
-      {withDot ? <GoPrimitiveDot className="h-3 w-3 stroke-[3px]" /> : null}
-      {StartIcon ? <StartIcon className="h-3 w-3 stroke-[3px]" /> : null}
+      {withDot ? <GoPrimitiveDot data-testid="go-primitive-dot" className="h-3 w-3 stroke-[3px]" /> : null}
+      {StartIcon ? <StartIcon data-testid="start-icon" className="h-3 w-3 stroke-[3px]" /> : null}
       {children}
     </>
   );
diff --git a/packages/ui/components/badge/badge.test.tsx b/packages/ui/components/badge/badge.test.tsx
new file mode 100644
index 0000000000..df08606d5f
--- /dev/null
+++ b/packages/ui/components/badge/badge.test.tsx
@@ -0,0 +1,92 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+
+/* eslint-disable playwright/missing-playwright-await */
+import { render, screen, fireEvent } from "@testing-library/react";
+import { vi } from "vitest";
+
+import { Badge, badgeStyles } from "./Badge";
+
+describe("Tests for Badge component", () => {
+  const variants = [
+    "default",
+    "warning",
+    "orange",
+    "success",
+    "green",
+    "gray",
+    "blue",
+    "red",
+    "error",
+    "grayWithoutHover",
+  ];
+
+  const sizes = ["sm", "md", "lg"];
+  const children = "Test Badge";
+
+  test.each(variants)("Should apply variant class", (variant) => {
+    render(<Badge variant={variant as any}>{children}</Badge>);
+    const badgeClass = screen.getByText(children).className;
+    const badgeComponentClass = badgeStyles({ variant: variant as any });
+    expect(badgeClass).toEqual(badgeComponentClass);
+  });
+
+  test.each(sizes)("Should apply size class", (size) => {
+    render(<Badge size={size as any}>{children}</Badge>);
+    const badgeClass = screen.getByText(children).className;
+    const badgeComponentClass = badgeStyles({ size: size as any });
+    expect(badgeClass).toEqual(badgeComponentClass);
+  });
+
+  test("Should render without errors", () => {
+    render(<Badge>{children}</Badge>);
+    expect(screen.getByText(children)).toBeInTheDocument();
+  });
+
+  test("Should render WithDot if the prop is true and shouldn't render if is false", () => {
+    const { rerender } = render(<Badge withDot>{children}</Badge>);
+    expect(screen.getByTestId("go-primitive-dot")).toBeInTheDocument();
+
+    rerender(<Badge>{children}</Badge>);
+    expect(screen.queryByTestId("go-primitive-dot")).not.toBeInTheDocument();
+  });
+
+  test("Should render with a startIcon when startIcon prop is provided shouldn't render if is false", () => {
+    const startIcon = () => <svg data-testid="start-icon" />;
+    const { rerender } = render(<Badge startIcon={startIcon}>{children}</Badge>);
+    expect(screen.getByTestId("start-icon")).toBeInTheDocument();
+
+    rerender(<Badge>{children}</Badge>);
+    expect(screen.queryByTestId("start-icon")).not.toBeInTheDocument();
+  });
+
+  test("Should render as a button when onClick prop is provided and shouldn't if is not", () => {
+    const handleClick = vi.fn();
+    const { rerender } = render(<Badge onClick={handleClick}>{children}</Badge>);
+    const badge = screen.getByText(children);
+    expect(badge.tagName).toBe("BUTTON");
+    fireEvent.click(badge);
+    expect(handleClick).toHaveBeenCalledTimes(1);
+
+    rerender(<Badge>{children}</Badge>);
+    const updateBadge = screen.getByText(children);
+    expect(updateBadge.tagName).not.toBe("BUTTON");
+  });
+
+  test("Should render as a div when onClick prop is not provided", () => {
+    render(<Badge>{children}</Badge>);
+    const badge = screen.getByText(children);
+    expect(badge.tagName).toBe("DIV");
+  });
+
+  test("Should render children when provided", () => {
+    const { getByText } = render(
+      <Badge>
+        <span>Child element 1</span>
+        <span>Child element 2</span>
+      </Badge>
+    );
+
+    expect(getByText("Child element 1")).toBeInTheDocument();
+    expect(getByText("Child element 2")).toBeInTheDocument();
+  });
+});