cherry-studio/tests/__mocks__/renderer/DataApiService.ts

import type { ConcreteApiPaths } from '@shared/data/api/apiSchemas'
import type {
  ApiClient,
  BatchRequest,
  BatchResponse,
  DataRequest,
  DataResponse,
  SubscriptionCallback,
  SubscriptionOptions,
  TransactionRequest
} from '@shared/data/api/apiTypes'
import { vi } from 'vitest'

/**
 * Mock DataApiService for testing
 * Provides a comprehensive mock of the DataApiService with realistic behavior
 */

// Mock response utilities
const createMockResponse = <T>(data: T, success = true): DataResponse<T> => ({
  success,
  data,
  timestamp: new Date().toISOString(),
  ...(success ? {} : { error: { code: 'MOCK_ERROR', message: 'Mock error', details: {} } })
})

const createMockError = (message: string): DataResponse<never> => ({
  success: false,
  error: {
    code: 'MOCK_ERROR',
    message,
    details: {}
  },
  timestamp: new Date().toISOString()
})

/**
 * Mock implementation of DataApiService
 */
export const createMockDataApiService = (customBehavior: Partial<ApiClient> = {}): ApiClient => {
  const mockService: ApiClient = {
    // HTTP Methods
    get: vi.fn(async (path: ConcreteApiPaths, options?: any) => {
      // Default mock behavior - return empty data based on path
      const mockData = getMockDataForPath(path, 'GET')
      return createMockResponse(mockData)
    }),

    post: vi.fn(async (path: ConcreteApiPaths, options?: any) => {
      const mockData = getMockDataForPath(path, 'POST')
      return createMockResponse(mockData)
    }),

    put: vi.fn(async (path: ConcreteApiPaths, options?: any) => {
      const mockData = getMockDataForPath(path, 'PUT')
      return createMockResponse(mockData)
    }),

    patch: vi.fn(async (path: ConcreteApiPaths, options?: any) => {
      const mockData = getMockDataForPath(path, 'PATCH')
      return createMockResponse(mockData)
    }),

    delete: vi.fn(async (path: ConcreteApiPaths, options?: any) => {
      return createMockResponse({ deleted: true })
    }),

    // Batch operations
    batch: vi.fn(async (requests: BatchRequest[]): Promise<BatchResponse> => {
      const responses = requests.map((request, index) => ({
        id: request.id || `batch_${index}`,
        success: true,
        data: getMockDataForPath(request.path as ConcreteApiPaths, request.method),
        timestamp: new Date().toISOString()
      }))

      return {
        success: true,
        responses,
        timestamp: new Date().toISOString()
      }
    }),

    // Transaction support
    transaction: vi.fn(async (operations: TransactionRequest[]): Promise<DataResponse<any[]>> => {
      const results = operations.map((op, index) => ({
        operation: op.operation,
        result: getMockDataForPath(op.path as ConcreteApiPaths, 'POST'),
        success: true
      }))

      return createMockResponse(results)
    }),

    // Subscription methods
    subscribe: vi.fn((path: ConcreteApiPaths, callback: SubscriptionCallback, options?: SubscriptionOptions) => {
      // Return a mock unsubscribe function
      return vi.fn()
    }),

    unsubscribe: vi.fn((path: ConcreteApiPaths) => {
      // Mock unsubscribe
    }),

    // Connection management
    connect: vi.fn(async () => {
      return createMockResponse({ connected: true })
    }),

    disconnect: vi.fn(async () => {
      return createMockResponse({ disconnected: true })
    }),

    // Health check
    ping: vi.fn(async () => {
      return createMockResponse({ pong: true, timestamp: new Date().toISOString() })
    }),

    // Apply custom behavior overrides
    ...customBehavior
  }

  return mockService
}

/**
 * Get mock data based on API path and method
 * Provides realistic mock responses for common API endpoints
 */
function getMockDataForPath(path: ConcreteApiPaths, method: string): any {
  // Parse path to determine data type
  if (path.includes('/topics')) {
    if (method === 'GET' && path.endsWith('/topics')) {
      return {
        topics: [
          { id: 'topic1', name: 'Mock Topic 1', createdAt: '2024-01-01T00:00:00Z' },
          { id: 'topic2', name: 'Mock Topic 2', createdAt: '2024-01-02T00:00:00Z' }
        ],
        total: 2
      }
    }
    if (method === 'GET' && path.match(/\/topics\/[^/]+$/)) {
      return {
        id: 'topic1',
        name: 'Mock Topic',
        messages: [],
        createdAt: '2024-01-01T00:00:00Z'
      }
    }
    if (method === 'POST' && path.endsWith('/topics')) {
      return {
        id: 'new_topic',
        name: 'New Mock Topic',
        createdAt: new Date().toISOString()
      }
    }
  }

  if (path.includes('/messages')) {
    if (method === 'GET') {
      return {
        messages: [
          { id: 'msg1', content: 'Mock message 1', role: 'user', timestamp: '2024-01-01T00:00:00Z' },
          { id: 'msg2', content: 'Mock message 2', role: 'assistant', timestamp: '2024-01-01T00:01:00Z' }
        ],
        total: 2
      }
    }
    if (method === 'POST') {
      return {
        id: 'new_message',
        content: 'New mock message',
        role: 'user',
        timestamp: new Date().toISOString()
      }
    }
  }

  if (path.includes('/preferences')) {
    if (method === 'GET') {
      return {
        preferences: {
          'ui.theme': 'light',
          'ui.language': 'en',
          'data.export.format': 'markdown'
        }
      }
    }
    if (method === 'POST' || method === 'PUT') {
      return { updated: true, timestamp: new Date().toISOString() }
    }
  }

  // Default mock data
  return {
    id: 'mock_id',
    data: 'mock_data',
    timestamp: new Date().toISOString()
  }
}

// Default mock instance
export const mockDataApiService = createMockDataApiService()

// Singleton instance mock
export const MockDataApiService = {
  DataApiService: class MockDataApiService {
    static getInstance() {
      return mockDataApiService
    }

    // Instance methods delegate to the mock
    async get(path: ConcreteApiPaths, options?: any) {
      return mockDataApiService.get(path, options)
    }

    async post(path: ConcreteApiPaths, options?: any) {
      return mockDataApiService.post(path, options)
    }

    async put(path: ConcreteApiPaths, options?: any) {
      return mockDataApiService.put(path, options)
    }

    async patch(path: ConcreteApiPaths, options?: any) {
      return mockDataApiService.patch(path, options)
    }

    async delete(path: ConcreteApiPaths, options?: any) {
      return mockDataApiService.delete(path, options)
    }

    async batch(requests: BatchRequest[]) {
      return mockDataApiService.batch(requests)
    }

    async transaction(operations: TransactionRequest[]) {
      return mockDataApiService.transaction(operations)
    }

    subscribe(path: ConcreteApiPaths, callback: SubscriptionCallback, options?: SubscriptionOptions) {
      return mockDataApiService.subscribe(path, callback, options)
    }

    unsubscribe(path: ConcreteApiPaths) {
      return mockDataApiService.unsubscribe(path)
    }

    async connect() {
      return mockDataApiService.connect()
    }

    async disconnect() {
      return mockDataApiService.disconnect()
    }

    async ping() {
      return mockDataApiService.ping()
    }
  },
  dataApiService: mockDataApiService
}

/**
 * Utility functions for testing
 */
export const MockDataApiUtils = {
  /**
   * Reset all mock function call counts and implementations
   */
  resetMocks: () => {
    Object.values(mockDataApiService).forEach(method => {
      if (vi.isMockFunction(method)) {
        method.mockClear()
      }
    })
  },

  /**
   * Set custom response for a specific path and method
   */
  setCustomResponse: (path: ConcreteApiPaths, method: string, response: any) => {
    const methodFn = mockDataApiService[method.toLowerCase() as keyof ApiClient] as any
    if (vi.isMockFunction(methodFn)) {
      methodFn.mockImplementation(async (requestPath: string, options?: any) => {
        if (requestPath === path) {
          return createMockResponse(response)
        }
        // Fall back to default behavior
        return createMockResponse(getMockDataForPath(requestPath as ConcreteApiPaths, method))
      })
    }
  },

  /**
   * Set error response for a specific path and method
   */
  setErrorResponse: (path: ConcreteApiPaths, method: string, errorMessage: string) => {
    const methodFn = mockDataApiService[method.toLowerCase() as keyof ApiClient] as any
    if (vi.isMockFunction(methodFn)) {
      methodFn.mockImplementation(async (requestPath: string, options?: any) => {
        if (requestPath === path) {
          return createMockError(errorMessage)
        }
        // Fall back to default behavior
        return createMockResponse(getMockDataForPath(requestPath as ConcreteApiPaths, method))
      })
    }
  },

  /**
   * Get call count for a specific method
   */
  getCallCount: (method: keyof ApiClient): number => {
    const methodFn = mockDataApiService[method] as any
    return vi.isMockFunction(methodFn) ? methodFn.mock.calls.length : 0
  },

  /**
   * Get calls for a specific method
   */
  getCalls: (method: keyof ApiClient): any[] => {
    const methodFn = mockDataApiService[method] as any
    return vi.isMockFunction(methodFn) ? methodFn.mock.calls : []
  }
}