tests: add context and rekor-api tests (#57)

* tas: migrate tests

Signed-off-by: Carlos Feria <[email protected]>

* fix: lint and add 1 test

Signed-off-by: Carlos Feria <[email protected]>

* tests: add context and rekor-api tests

Signed-off-by: Carlos Feria <[email protected]>

* tests: add context and rekor-api tests

Signed-off-by: Carlos Feria <[email protected]>

---------

Signed-off-by: Carlos Feria <[email protected]>

Author: carlosthe19916

client/src/app/api/context.test.tsx

@@ -0,0 +1,43 @@
+/// <reference types="@vitest/browser/context" />
+
+import { beforeAll, describe, expect, it, vi } from "vitest";
+
+// Mock the ENV module before any imports
+vi.mock("@app/env", () => ({
+  default: {
+    NEXT_PUBLIC_REKOR_DEFAULT_DOMAIN: "https://example.com",
+  },
+}));
+
+import { fireEvent, render, screen } from "@testing-library/react";
+import { RekorClientProvider, useRekorBaseUrl, useRekorClient } from "./context";
+
+const TestConsumerComponent = () => {
+  useRekorClient();
+  const [baseUrl, setBaseUrl] = useRekorBaseUrl();
+
+  return (
+    <div>
+      <button onClick={() => setBaseUrl("https://new.example.com")}>Change Base URL</button>
+      <p>Base URL: {baseUrl}</p>
+    </div>
+  );
+};
+
+describe("RekorClientContext", () => {
+  beforeAll(() => vi.clearAllMocks());
+
+  it("provides a RekorClient instance and manages base URL", () => {
+    render(
+      <RekorClientProvider>
+        <TestConsumerComponent />
+      </RekorClientProvider>
+    );
+
+    expect(screen.getByText(/Base URL: https:\/\/example.com/)).toBeInTheDocument();
+
+    fireEvent.click(screen.getByText(/Change Base URL/));
+
+    expect(screen.getByText(/Base URL: https:\/\/new.example.com/)).toBeInTheDocument();
+  });
+});

client/src/app/api/context.tsx

@@ -1,3 +1,4 @@
+import ENV from "@app/env";
 import {
   createContext,
   type FunctionComponent,
@@ -15,6 +16,7 @@ export interface RekorClientContext {
   setBaseUrl: (_base: string | undefined) => void;
 }
 
+/* eslint-disable react-refresh/only-export-components */
 export const RekorClientContext = createContext<RekorClientContext | undefined>(undefined);
 
 interface RekorClientProviderProps {
@@ -29,8 +31,8 @@ export const RekorClientProvider: FunctionComponent<PropsWithChildren<RekorClien
 
   useEffect(() => {
     if (baseUrl === undefined) {
-      if (process.env.NEXT_PUBLIC_REKOR_DEFAULT_DOMAIN) {
-        setBaseUrl(process.env.NEXT_PUBLIC_REKOR_DEFAULT_DOMAIN);
+      if (ENV.NEXT_PUBLIC_REKOR_DEFAULT_DOMAIN) {
+        setBaseUrl(ENV.NEXT_PUBLIC_REKOR_DEFAULT_DOMAIN);
       } else {
         setBaseUrl("https://rekor.sigstore.dev");
       }

client/src/app/api/rekor_api.test.ts

@@ -0,0 +1,39 @@
+/// <reference types="vitest/globals" />
+
+import { renderHook } from "@testing-library/react";
+import { vi, type Mock } from "vitest";
+import { useRekorSearch } from "./rekor-api";
+import { useRekorClient } from "./context";
+
+vi.mock("./context", () => ({
+  useRekorClient: vi.fn(),
+}));
+
+Object.defineProperty(global.self, "crypto", {
+  value: {
+    subtle: {
+      digest: vi.fn().mockImplementation(() => {
+        const hashBuffer = new ArrayBuffer(32);
+        const hashArray = new Uint8Array(hashBuffer);
+        hashArray.fill(0);
+        return hashBuffer;
+      }),
+    },
+  },
+});
+
+describe("useRekorSearch", () => {
+  it("searches by logIndex", async () => {
+    const mockGetLogEntryByIndex = vi.fn().mockResolvedValue(0);
+
+    (useRekorClient as Mock).mockReturnValue({
+      entries: { getLogEntryByIndex: mockGetLogEntryByIndex },
+    });
+
+    const { result } = renderHook(() => useRekorSearch());
+
+    await result.current({ attribute: "logIndex", query: 123 });
+
+    expect(mockGetLogEntryByIndex).toHaveBeenCalledWith({ logIndex: 123 });
+  });
+});

common/src/environment.ts

@@ -44,6 +44,9 @@ export interface ConsoleEnvType {
   /** Target URL for the UI server's `/api` proxy */
   CONSOLE_API_URL?: string;
 
+  /** Rekor Search */
+  NEXT_PUBLIC_REKOR_DEFAULT_DOMAIN?: string;
+
   /** Location of branding files (relative paths computed from the project source root) */
   BRANDING?: string;
 }
@@ -72,6 +75,7 @@ export const buildConsoleEnv = ({
 
   UI_INGRESS_PROXY_BODY_SIZE = "500m",
   CONSOLE_API_URL,
+  NEXT_PUBLIC_REKOR_DEFAULT_DOMAIN,
   BRANDING,
 }: Partial<ConsoleEnvType> = {}): ConsoleEnvType => ({
   NODE_ENV,
@@ -88,6 +92,7 @@ export const buildConsoleEnv = ({
 
   UI_INGRESS_PROXY_BODY_SIZE,
   CONSOLE_API_URL,
+  NEXT_PUBLIC_REKOR_DEFAULT_DOMAIN,
   BRANDING,
 });