vitest.setup.ts

import { vi } from 'vitest';

// Mock environment variables
process.env.NODE_ENV = 'test';
process.env.JWT_SECRET = 'test-jwt-secret';
process.env.DATABASE_URL = 'postgresql://test:test@localhost:5432/antbase_test';
process.env.REDIS_URL = 'redis://localhost:6379/1';
process.env.OPENAI_API_KEY = 'test-openai-key';

// Mock console methods in test environment
global.console = {
  ...console,
  // Keep these for test output
  log: console.log,
  warn: console.warn,
  error: console.error,
  // Mock these to reduce noise
  debug: vi.fn(),
  info: vi.fn(),
};

// Mock fetch globally
global.fetch = vi.fn();

// Mock WebSocket
global.WebSocket = vi.fn(() => ({
  addEventListener: vi.fn(),
  removeEventListener: vi.fn(),
  send: vi.fn(),
  close: vi.fn(),
  readyState: 1,
})) as any;

// Mock localStorage
const localStorageMock = (() => {
  let store: Record<string, string> = {};
  
  return {
    getItem: (key: string) => store[key] || null,
    setItem: (key: string, value: string) => {
      store[key] = value;
    },
    removeItem: (key: string) => {
      delete store[key];
    },
    clear: () => {
      store = {};
    },
    length: Object.keys(store).length,
    key: (index: number) => Object.keys(store)[index] || null,
  };
})();

Object.defineProperty(global, 'localStorage', {
  value: localStorageMock,
});

// Mock sessionStorage
Object.defineProperty(global, 'sessionStorage', {
  value: localStorageMock,
});

// Mock URL
global.URL = {
  createObjectURL: vi.fn(() => 'mocked-url'),
  revokeObjectURL: vi.fn(),
} as any;

// Mock crypto
Object.defineProperty(global, 'crypto', {
  value: {
    randomUUID: vi.fn(() => 'mocked-uuid'),
    subtle: {
      digest: vi.fn(),
      encrypt: vi.fn(),
      decrypt: vi.fn(),
    },
  },
});

// Mock ResizeObserver
global.ResizeObserver = vi.fn().mockImplementation(() => ({
  observe: vi.fn(),
  unobserve: vi.fn(),
  disconnect: vi.fn(),
}));

// Mock IntersectionObserver
global.IntersectionObserver = vi.fn().mockImplementation(() => ({
  observe: vi.fn(),
  unobserve: vi.fn(),
  disconnect: vi.fn(),
  root: null,
  rootMargin: '',
  thresholds: [],
}));

// Mock File and FileReader
global.File = vi.fn().mockImplementation((fileBits, fileName, options) => ({
  name: fileName,
  size: fileBits.reduce((acc: number, bit: any) => acc + bit.length, 0),
  type: options?.type || '',
  lastModified: Date.now(),
})) as any;

global.FileReader = vi.fn().mockImplementation(() => ({
  readAsText: vi.fn(function(this: any) {
    this.onload?.({ target: { result: 'file content' } });
  }),
  readAsDataURL: vi.fn(function(this: any) {
    this.onload?.({ target: { result: 'data:text/plain;base64,ZmlsZSBjb250ZW50' } });
  }),
  result: null,
  error: null,
  onload: null,
  onerror: null,
})) as any;

// Mock Blob
global.Blob = vi.fn().mockImplementation((content, options) => ({
  size: content?.reduce?.((acc: number, item: any) => acc + item.length, 0) || 0,
  type: options?.type || '',
  slice: vi.fn(),
  stream: vi.fn(),
  text: vi.fn().mockResolvedValue('blob content'),
})) as any;

// Mock FormData
global.FormData = vi.fn().mockImplementation(() => {
  const data = new Map();
  return {
    append: vi.fn((key, value) => data.set(key, value)),
    delete: vi.fn((key) => data.delete(key)),
    get: vi.fn((key) => data.get(key)),
    has: vi.fn((key) => data.has(key)),
    set: vi.fn((key, value) => data.set(key, value)),
    entries: vi.fn(() => data.entries()),
    keys: vi.fn(() => data.keys()),
    values: vi.fn(() => data.values()),
  };
}) as any;

// Mock URLSearchParams
global.URLSearchParams = vi.fn().mockImplementation(() => {
  const params = new Map();
  return {
    append: vi.fn((key, value) => params.set(key, value)),
    delete: vi.fn((key) => params.delete(key)),
    get: vi.fn((key) => params.get(key)),
    has: vi.fn((key) => params.has(key)),
    set: vi.fn((key, value) => params.set(key, value)),
    toString: vi.fn(() => Array.from(params.entries()).map(([k, v]) => `${k}=${v}`).join('&')),
  };
}) as any;

// Setup test database connection
beforeAll(() => {
  // This would be implemented based on your database setup
});

afterAll(() => {
  // Cleanup test database
});

// Reset all mocks between tests
afterEach(() => {
  vi.clearAllMocks();
  localStorageMock.clear();
});