test(theme): add tests for getThemeModeScript function

Author: SutuSebastian

packages/ui/src/theme/mode-script.test.tsx

@@ -1,6 +1,6 @@
 import { render } from "@testing-library/react";
 import { describe, expect, it } from "vitest";
-import { ThemeModeScript } from "./mode-script";
+import { getThemeModeScript, ThemeModeScript } from "./mode-script";
 
 describe("ThemeModeScript", () => {
   it("should render with default props", () => {
@@ -47,3 +47,25 @@ describe("ThemeModeScript", () => {
     expect(script?.innerHTML).toContain('const className = "dark"');
   });
 });
+
+describe("getThemeModeScript", () => {
+  it("should return the correct script for prefix and version 3", () => {
+    const script = getThemeModeScript({ prefix: "custom-", version: 3 });
+    expect(script).toContain('const className = "custom-dark"');
+  });
+
+  it("should return the correct script for prefix and version 4", () => {
+    const script = getThemeModeScript({ prefix: "custom-", version: 4 });
+    expect(script).toContain('const className = "dark"');
+  });
+
+  it("should return the correct script for version 3", () => {
+    const script = getThemeModeScript({ version: 3 });
+    expect(script).toContain('const className = "dark"');
+  });
+
+  it("should return the correct script for version 4", () => {
+    const script = getThemeModeScript({ version: 4 });
+    expect(script).toContain('const className = "dark"');
+  });
+});