Function createTestServiceBroker

server/test/utils.ts:18–80

(
  serviceCls: typeof TcService,
  options?: TestServiceBrokerOptions
)

 * @param serviceCls 微服务类
 */
export function createTestServiceBroker<T extends TcService = TcService>(
  serviceCls: typeof TcService,
  options?: TestServiceBrokerOptions
): {
  broker: TcBroker;
  contextCallMock: jest.Mock;
  service: MockedService<T>;
  insertTestData: <E, R extends E = E>(
    entity: E
  ) => Promise<DocumentType<R & { _id: string }>>;
} {
  const broker = new TcBroker({ logger: false });
  const service = broker.createService(serviceCls) as MockedService<T>;
  const testDataStack = [];
  const contextCallMock = jest.fn(options?.contextCallMockFn);

  broker.ContextFactory = class extends broker.ContextFactory {
    call = contextCallMock as any;
  } as any;

  // Mock
  service.roomcastNotify = jest.fn();
  service.cleanActionCache = jest.fn();

  beforeAll(async () => {
    await broker.start();
  });
  afterAll(async () => {
    await Promise.all(
      testDataStack.map((item) => {
        if (typeof service.adapter !== 'object') {
          throw new Error('无法调用 insertTestData');
        }

        return service.adapter.removeById(item._id);
      })
    )
      .then(() => {
        console.log(`已清理 ${testDataStack.length} 条测试数据`);
      })
      .catch((err) => {
        console.error('测试数据清理失败:', err);
      });

    await broker.stop();
  });

  const insertTestData = async (entity: any) => {
    if (typeof service.adapter !== 'object') {
      throw new Error('无法调用 insertTestData');
    }
    const doc = await service.adapter.insert(entity);
    testDataStack.push(doc);
    return doc;
  };

  return {
    broker,