undeexcepturi/tests/features/single-table-inheritance/GH2723.test.ts

TypeScript ORM for Node.js based on Data Mapper, Unit of Work and Identity Map patterns. Supports MongoDB, MySQL, PostgreSQL and SQLite databases as well as usage with vanilla JavaScript.

import { Collection, Entity, EntityRef, ManyToOne, MikroORM, OneToMany, PrimaryKey, PrimaryKeyProp } from '@mikro-orm/core';
import { SqliteDriver } from '@mikro-orm/sqlite';

@Entity()
export class Cat {

  [PrimaryKeyProp]?: ['name', 'user'];

  @PrimaryKey()
  name!: string;

  @ManyToOne(() => User, { primary: true, deleteRule: 'CASCADE', ref: true })
  user!: EntityRef<User>;

}

@Entity()
export class User {

  @PrimaryKey()
  id!: string;

  @OneToMany(() => Cat, c => c.user, { eager: true, orphanRemoval: true })
  cats = new Collection<Cat>(this);

}

describe('GH 2723', () => {

  let orm: MikroORM;

  beforeAll(async () => {
    orm = await MikroORM.init({
      dbName: ':memory:',
      driver: SqliteDriver,
      entities: [Cat, User],
    });
    await orm.schema.createSchema();
  });

  afterAll(() => orm.close(true));

  test('FK as PK and orphan removal/cascading', async () => {
    const user = orm.em.create(User, { id: 'TestUser' }, { persist: true });
    orm.em.create(Cat, { name: 'TestCat', user }, { persist: true });
    await orm.em.flush();
    orm.em.clear();

    const u = await orm.em.findOneOrFail(User, { id: 'TestUser' });
    await orm.em.remove(u).flush();

    const users = await orm.em.count(User, {});
    const cats = await orm.em.count(User, {});
    expect(users + cats).toBe(0);
  });

});