From cf816ca71f68268bb00bfc7f38fedf00be43a761 Mon Sep 17 00:00:00 2001
From: Dinesh Katariya <thedineshk24@gmail.com>
Date: Wed, 7 Jun 2023 22:06:55 +0530
Subject: [PATCH] test : added test cases for getMostUsedCustomColors

---
 .../ColorPicker/colorPicker.test.ts           | 114 +++++++++++++++++-
 1 file changed, 113 insertions(+), 1 deletion(-)

diff --git a/src/components/ColorPicker/colorPicker.test.ts b/src/components/ColorPicker/colorPicker.test.ts
index 82098112a..06d910ee1 100644
--- a/src/components/ColorPicker/colorPicker.test.ts
+++ b/src/components/ColorPicker/colorPicker.test.ts
@@ -1,7 +1,10 @@
 import { ColorPaletteCustom } from "../../colors";
+import { ExcalidrawElement } from "../../element/types";
 import {
   getColorNameAndShadeFromColor,
   colorPickerHotkeyBindings,
+  isCustomColor,
+  getMostUsedCustomColors,
 } from "./colorPickerUtils";
 
 describe("getColorNameAndShadeFromColor", () => {
@@ -20,7 +23,6 @@ describe("getColorNameAndShadeFromColor", () => {
 
 describe("colorPickerHotkeyBindings", () => {
   it("should contain all the expected hotkey bindings", () => {
-
     const testBindingKeys = [
       ["q", "w", "e", "r", "t"],
       ["a", "s", "d", "f", "g"],
@@ -30,3 +32,113 @@ describe("colorPickerHotkeyBindings", () => {
     expect(testBindingKeys).toEqual(colorPickerHotkeyBindings);
   });
 });
+
+describe("isCustomColor", () => {
+  it("should return true for a custom color not present in the palette", () => {
+    const palette = {
+      green: "#00FF00",
+      orange: "#E07C24",
+    };
+    const color = "#FF6666";
+
+    const result = isCustomColor({ color, palette });
+
+    expect(result).toBe(true);
+  });
+
+  it("should return false for a color present in the palette", () => {
+    const palette = {
+      green: "#00FF00",
+      orange: "#E07C24",
+    };
+    const color = "#00FF00";
+
+    const result = isCustomColor({ color, palette });
+
+    expect(result).toBe(false);
+  });
+
+  it("should handle empty palette correctly", () => {
+    const palette = {};
+    const color = "#FF6666";
+
+    const result = isCustomColor({ color, palette });
+
+    expect(result).toBe(true);
+  });
+});
+
+describe("getMostUsedCustomColors", () => {
+  const elements = [
+    {
+      type: "rectangle",
+      id: "1",
+      isDeleted: false,
+      backgroundColor: "#FF0000",
+      strokeColor: "#00FF00",
+    },
+    {
+      type: "ellipse",
+      id: "2",
+      isDeleted: false,
+      backgroundColor: "#FF0000",
+      strokeColor: "#0000FF",
+    },
+    {
+      type: "rectangle",
+      id: "3",
+      isDeleted: false,
+      backgroundColor: "#00FF00",
+      strokeColor: "#00FF00",
+    },
+    {
+      type: "rectangle",
+      id: "4",
+      isDeleted: true,
+      backgroundColor: "#FFFF00",
+      strokeColor: "#FF00FF",
+    },
+  ];
+
+  const palette = {
+    red: "#FF0000",
+    green: "#00FF00",
+    blue: "#0000FF",
+  };
+
+  it("should return the most used custom colors for element background", () => {
+    const type = "elementBackground";
+
+    // @ts-ignore
+    const result = getMostUsedCustomColors(elements, type, palette);
+
+    expect(result).toEqual(["#FF0000"]);
+  });
+
+  it("should return the most used custom colors for element stroke", () => {
+    const type = "elementStroke";
+
+    // @ts-ignore
+    const result = getMostUsedCustomColors(elements, type, palette);
+
+    expect(result).toEqual(["#00FF00"]);
+  });
+
+  it("should handle empty elements correctly", () => {
+    const type = "elementBackground";
+    const emptyElements: readonly ExcalidrawElement[] = [];
+
+    const result = getMostUsedCustomColors(emptyElements, type, palette);
+
+    expect(result).toEqual([]);
+  });
+
+  it("should handle empty palette correctly", () => {
+    const type = "elementBackground";
+    const emptyPalette = {};
+    // @ts-ignore
+    const result = getMostUsedCustomColors(elements, type, emptyPalette);
+
+    expect(result).toEqual([]);
+  });
+});