axios-miniprogram/scripts/test.utils.ts

import { test } from 'vitest';
import {
  PLAIN_METHODS,
  WITH_PARAMS_METHODS,
  WITH_DATA_METHODS,
} from '@/constants/methods';
import { AxiosAdapterRequestConfig } from '@/adpater/createAdapter';

export function asyncNext() {
  return Promise.resolve().then;
}

export function asyncTimeout(delay = 0) {
  return new Promise((resolve) => setTimeout(resolve, delay));
}

export function captureError<T = any>(fn: () => void) {
  try {
    fn();
    throw new Error('without Error');
  } catch (err) {
    return err as T;
  }
}

export function noop() {
  return;
}

export function mockResponse(
  status = 200,
  headers: AnyObject = {},
  data: AnyObject = {},
) {
  return {
    status,
    headers,
    data,
  };
}

export interface MockAdapterOptions {
  headers?: AnyObject;
  data?: AnyObject;
  delay?: number;
  before?: (config: AxiosAdapterRequestConfig) => void;
  after?: () => void;
}

function mockAdapterBase(
  type: 'success' | 'error' | 'fail' = 'success',
  options: MockAdapterOptions = {},
) {
  const {
    headers = {},
    data = {
      result: null,
    },
    delay = 0,
    before,
    after,
  } = options;

  return (config: AxiosAdapterRequestConfig) => {
    let canceled = false;

    before?.(config);

    setTimeout(() => {
      if (!canceled) {
        switch (type) {
          case 'success':
            config.success(mockResponse(200, headers, data));
            break;
          case 'error':
            config.success(mockResponse(500, headers, data));
            break;
          case 'fail':
            config.fail(mockResponse(400, headers, data));
            break;
        }

        after?.();
      }
    }, delay);

    return {
      abort() {
        canceled = true;
      },
    };
  };
}

export function mockAdapter(options: MockAdapterOptions = {}) {
  return mockAdapterBase('success', options);
}

export function mockAdapterError(options: MockAdapterOptions = {}) {
  return mockAdapterBase('error', options);
}

export function mockAdapterFail(options: MockAdapterOptions = {}) {
  return mockAdapterBase('fail', options);
}

export const methods = [
  ...PLAIN_METHODS,
  ...WITH_PARAMS_METHODS,
  ...WITH_DATA_METHODS,
];

export const testEachMethods = test.each(methods);

export function eachMethods(cb: (k: (typeof methods)[number]) => void) {
  methods.forEach(cb);
}