test.utils.ts

// Copyright 2020-2024 SubQuery Pte Ltd authors & contributors
// SPDX-License-Identifier: GPL-3.0

// Utils for integration tests

import { DynamicModule, INestApplication } from '@nestjs/common';
import { EventEmitterModule } from '@nestjs/event-emitter';
import { ScheduleModule } from '@nestjs/schedule';
import { Test } from '@nestjs/testing';
import { DbModule, NodeConfig, registerApp } from '@subql/node-core';
import { ConfigureModule } from '../configure/configure.module';
import { SubqueryProject } from '../configure/SubqueryProject';
import { FetchModule } from '../indexer/fetch.module';
import { MetaModule } from '../meta/meta.module';

const mockInstance = async (
  cid: string,
  schemaName: string,
  disableHistorical: boolean,
  useSubscription: boolean,
  timestampField: boolean,
) => {
  const argv: Record<string, any> = {
    _: [],
    disableHistorical,
    subquery: `ipfs://${cid}`,
    dbSchema: schemaName,
    allowSchemaMigration: true,
    ipfs: 'https://unauthipfs.subquery.network/ipfs/api/v0',
    networkEndpoint: 'wss://rpc.polkadot.io/public-ws',
    timestampField,
    subscription: useSubscription,
  };
  return registerApp<SubqueryProject>(
    argv,
    SubqueryProject.create.bind(SubqueryProject),
    jest.fn(),
    '',
  );
};

async function mockRegister(
  cid: string,
  schemaName: string,
  disableHistorical: boolean,
  useSubscription: boolean,
  timestampField: boolean,
): Promise<DynamicModule> {
  const { nodeConfig, project } = await mockInstance(
    cid,
    schemaName,
    disableHistorical,
    useSubscription,
    timestampField,
  );

  return {
    module: ConfigureModule,
    providers: [
      {
        provide: NodeConfig,
        useValue: nodeConfig,
      },
      {
        provide: 'ISubqueryProject',
        useValue: project,
      },
      {
        provide: 'IProjectUpgradeService',
        useValue: project,
      },
      {
        provide: 'Null',
        useValue: null,
      },
    ],
    exports: [NodeConfig, 'ISubqueryProject', 'IProjectUpgradeService', 'Null'],
  };
}

export async function prepareApp(
  schemaName: string,
  cid: string,
  disableHistorical = false,
  useSubscription = false,
  timestampField = false,
): Promise<INestApplication> {
  const m = await Test.createTestingModule({
    imports: [
      DbModule.forRoot(),
      EventEmitterModule.forRoot(),
      mockRegister(
        cid,
        schemaName,
        disableHistorical,
        useSubscription,
        timestampField,
      ),
      ScheduleModule.forRoot(),
      FetchModule,
      MetaModule,
    ],
    controllers: [],
  }).compile();

  const app = m.createNestApplication();
  await app.init();
  return app;
}