在 TypeOrm + NestJS 中模拟关系

Question

我在测试 NestJS 服务时遇到了困难，该服务使用了一个只有一个关系的简单实体。然而，相关实体与许多其他实体相关，而这些其他实体又与其他实体相关。我不想导入所有实体并传递给TypeOrmModule.forRoot，如果不这样做，我会收到如下错误：

Entity metadata for Wallet#customer was not found.

对于每个尚未导入的实体。是否有可能以某种方式“模拟”这种关系？

测试：

const module: TestingModule = await Test.createTestingModule({
      imports: [
        TypeOrmModule.forRoot(CONNECTION_OPTIONS),
        TypeOrmModule.forFeature([WalletEntity, WalletTransactionEntity]),
      ],
      providers: [WalletService],
    }).compile();

实体

@Entity('wallet')
export class WalletEntity extends BaseEntity {
  @Column()
  customerId: string;

  @OneToOne(() => CustomerEntity, (customer) => customer.wallet)
  customer: CustomerEntity;

  @OneToMany(() => WalletTransactionEntity, (transaction) => transaction.wallet)
  transactions: WalletTransactionEntity[];
}

Answer 1

对于单元测试，我建议不要连接到实时数据库或使用任何imports，而只是使用自定义提供程序来模拟您正在测试的服务的依赖关系。 This repository 有很多例子，包括一些用于 TypeORM 的例子。根据您的导入，一个非常基本的设置可能类似于

beforeEach(async () => {
  const modRef = await Test.createTestingModule({
    providers: [
      WalletService,
      {
        provide: getRepositoryToken(WalletEntity),
        useValue: walletRepoMock,
      },
      {
        provide: getRepositoryToken(WalletTransactionEntity),
        useValue: walletTransactionRepoMock,
      }
    ]
  }).compile();
});

现在您可以将 walletRepoMock 和 walletTransactionRepoMock 替换为您用于 Repository 的方法的模拟实现，这样就可以了