vitest.setup.ts

import { TextDecoder, TextEncoder } from "node:util";
import * as testingLibraryMatchers from "@testing-library/jest-dom/matchers";
import { expect, vi } from "vitest";
import "@testing-library/jest-dom/vitest";
import failOnConsole from "vitest-fail-on-console";

expect.extend(testingLibraryMatchers);

// Polyfill TextEncoder/TextDecoder for jsdom environment
global.TextEncoder = TextEncoder;
// @ts-expect-error - TextDecoder types are compatible
global.TextDecoder = TextDecoder;

// Fail tests that log errors or warnings to console
failOnConsole({
  shouldFailOnDebug: false,
  shouldFailOnError: true,
  shouldFailOnInfo: false,
  shouldFailOnLog: false,
  shouldFailOnWarn: true,
});

// Global mocks used across test files
vi.mock("next/headers", () => ({
  headers: vi.fn(() => Promise.resolve(new Headers())),
}));

vi.mock("next/navigation", () => ({
  redirect: vi.fn(),
  useRouter: vi.fn(() => ({
    push: vi.fn(),
    replace: vi.fn(),
    refresh: vi.fn(),
    back: vi.fn(),
    forward: vi.fn(),
    prefetch: vi.fn(),
  })),
}));

// Global auth server mock with default authenticated session
// Uses importActual to preserve real exports for unit tests
// Individual tests can override getSession/getAccessToken return values if needed
vi.mock("@/lib/auth/auth", async (importOriginal) => {
  const actual = await importOriginal<typeof import("@/lib/auth/auth")>();
  return {
    ...actual,
    auth: {
      ...actual.auth,
      api: {
        ...actual.auth.api,
        getSession: vi.fn(() =>
          Promise.resolve({
            user: {
              id: "mock-user-id",
              email: "test@example.com",
              name: "Test User",
            },
          }),
        ),
        getAccessToken: vi.fn(() =>
          Promise.resolve({ accessToken: "mock-test-token" }),
        ),
      },
    },
  };
});

// Common UI/runtime mocks
vi.mock("next/image", () => ({
  default: () => null,
}));

vi.mock("sonner", () => ({
  Toaster: () => null,
  toast: {
    error: vi.fn(),
    success: vi.fn(),
    info: vi.fn(),
    warning: vi.fn(),
  },
}));

export const mockSetTheme = vi.fn();
vi.mock("next-themes", () => ({
  useTheme: () => ({
    theme: "system",
    setTheme: mockSetTheme,
  }),
  ThemeProvider: ({ children }: { children: React.ReactNode }) => children,
}));

// Auth client baseline mock; individual tests can customize return values
vi.mock("@/lib/auth/auth-client", () => ({
  authClient: {
    signIn: {
      oauth2: vi.fn(),
    },
    signOut: vi.fn().mockResolvedValue({ data: null, error: null }),
  },
  signIn: {
    oauth2: vi.fn(),
  },
  signOut: vi.fn().mockResolvedValue({ data: null, error: null }),
  useSession: vi.fn(),
}));

import { cleanup } from "@testing-library/react";
// Reset mocks between test cases globally
import { afterEach } from "vitest";

afterEach(() => {
  // Clear calls/instances, but keep hoisted module mock implementations intact
  vi.clearAllMocks();
  // Clean up DOM after each test
  cleanup();
});