jest.setup.js

import '@testing-library/jest-dom'

// Mock Next.js Request and Response
global.Request = class {
  constructor(url, options = {}) {
    this.url = url
    this.method = options.method || 'GET'
    this.headers = options.headers || new Map()
    this.json = async () => options.body ? JSON.parse(options.body) : null
  }
}

global.Response = class {
  constructor(body, options = {}) {
    this.body = body
    this.status = options.status || 200
    this.headers = options.headers || new Map()
  }

  async json() {
    return this.body
  }

  static json(body, options = {}) {
    return new global.Response(body, {
      status: options.status || 200,
      headers: options.headers || new Map()
    })
  }
}

global.NextResponse = class {
  static json(body, options = {}) {
    return new global.Response(body, {
      status: options.status || 200,
      headers: options.headers || new Map()
    })
  }
}

// Mock Next.js router
jest.mock('next/navigation', () => ({
  useRouter() {
    return {
      push: jest.fn(),
      pathname: '/',
      query: {},
    }
  },
  usePathname() {
    return '/'
  },
  useSearchParams() {
    return new URLSearchParams()
  },
}))

// Mock NextAuth
jest.mock('next-auth/react', () => ({
  useSession: jest.fn(() => ({
    data: null,
    status: 'unauthenticated',
  })),
  signIn: jest.fn(),
  signOut: jest.fn(),
  getSession: jest.fn(),
}))

// Mock localStorage
Object.defineProperty(window, 'localStorage', {
  value: {
    getItem: jest.fn(),
    setItem: jest.fn(),
    removeItem: jest.fn(),
    clear: jest.fn(),
  },
  writable: true,
})

// Mock Prisma
jest.mock('@/lib/prisma', () => ({
  prisma: {
    user: {
      findUnique: jest.fn(),
      create: jest.fn(),
      update: jest.fn(),
    },
    account: {
      findUnique: jest.fn(),
      create: jest.fn(),
    },
    session: {
      findUnique: jest.fn(),
      create: jest.fn(),
    },
    watchlist: {
      findMany: jest.fn(),
      create: jest.fn(),
      delete: jest.fn(),
    },
  },
}))

// Mock Chart.js
jest.mock('chart.js', () => ({
  Chart: {
    register: jest.fn(),
  },
}))

// Mock Recharts
jest.mock('recharts', () => ({
  LineChart: jest.fn(() => null),
  Line: jest.fn(() => null),
  XAxis: jest.fn(() => null),
  YAxis: jest.fn(() => null),
  CartesianGrid: jest.fn(() => null),
  Tooltip: jest.fn(() => null),
  Legend: jest.fn(() => null),
  ResponsiveContainer: jest.fn(({ children }) => children),
  BarChart: jest.fn(() => null),
  Bar: jest.fn(() => null),
  AreaChart: jest.fn(() => null),
  Area: jest.fn(() => null),
  ComposedChart: jest.fn(() => null),
  ReferenceLine: jest.fn(() => null),
}))

// Mock react-chartjs-2
jest.mock('react-chartjs-2', () => ({
  Line: jest.fn(() => null),
  Bar: jest.fn(() => null),
}))

// Mock financial APIs
jest.mock('yahoo-finance2', () => ({
  quote: jest.fn(),
  quoteSummary: jest.fn(),
  chart: jest.fn(),
  options: jest.fn(),
  search: jest.fn(),
}))

jest.mock('alphavantage', () => {
  const mockAlphaVantage = jest.fn()
  mockAlphaVantage.quote = jest.fn()
  mockAlphaVantage.fundamental = jest.fn()
  mockAlphaVantage.crypto = jest.fn()
  mockAlphaVantage.news_sentiment = jest.fn()
  return mockAlphaVantage
})

// Mock bcrypt
jest.mock('bcryptjs', () => ({
  compare: jest.fn().mockResolvedValue(true),
  hash: jest.fn().mockResolvedValue('hashed_password'),
}))