index.ts

import {
  type BridgeControllerState,
  getDefaultBridgeControllerState,
  type GenericQuoteRequest,
} from '@metamask/bridge-controller';
import { initialState } from '../_mocks_/initialState';
import { mockBridgeReducerState } from '../_mocks_/bridgeReducerState';
import type { BridgeState } from '../../../../core/redux/slices/bridge';
import { RootState } from '../../../../reducers';
import { DeepPartial } from '../../../../util/test/renderWithProvider';
import { merge } from 'lodash';

// Re-export all fixtures (no heavy dependencies)
export * from './fixtures';

type BridgeControllerStateOverride = Partial<
  Omit<BridgeControllerState, 'quoteRequest'>
> & { quoteRequest?: Partial<GenericQuoteRequest> };
/**
 * Creates a complete bridge controller state by merging default state with overrides
 * @param overrides - Partial state to override default values
 * @returns Complete bridge controller state
 */
export const createBridgeControllerState = ({
  quoteRequest,
  ...overrides
}: BridgeControllerStateOverride = {}): BridgeControllerState =>
  merge(
    getDefaultBridgeControllerState(),
    overrides,
    quoteRequest
      ? {
          quoteRequest: Array.isArray(quoteRequest)
            ? quoteRequest
            : [quoteRequest],
        }
      : {},
  );

/**
 * Creates a complete test state for bridge components/hooks
 * @param overrides - Object containing optional overrides for bridge controller and reducer state
 * @returns Complete test state
 */
export const createBridgeTestState = (
  overrides: {
    bridgeControllerOverrides?: BridgeControllerStateOverride;
    bridgeReducerOverrides?: Partial<BridgeState>;
  } = {},
  state: DeepPartial<RootState> = initialState,
) => {
  const bridgeControllerState = createBridgeControllerState(
    overrides.bridgeControllerOverrides ?? {},
  );

  return {
    ...state,
    engine: {
      ...state.engine,
      backgroundState: {
        ...state.engine?.backgroundState,
        BridgeController: bridgeControllerState,
      },
    },
    bridge: {
      ...mockBridgeReducerState,
      ...(overrides.bridgeReducerOverrides ?? {}),
    },
  };
};