Seeding - how to add FK to the factory?

[eliezerbaschkier]

Hi Peter!
Thank you very much for supporting this library!

Quick question: I have two entities, Countries and States, where Country is a FK of States.
How do I specify in States.Factory.ts that states.country should take one of the rows generated in Countries.Factory.ts?

I leave below my code.
Thanks for the help!

// States.Entity.ts
@Entity()
export class States extends Base {
  @Column({ nullable: false, type: 'varchar', unique: true })
  descri: string;

  @ManyToOne(() => Countries, country => country.state, { nullable: false })
  @JoinColumn()
  country: Countries;
}

// States.Factory.ts
import { setSeederFactory } from 'typeorm-extension';
import { Countries } from '../entities/Countries';
import { States } from '../entities/States';

export default setSeederFactory(States, faker => {
  const state = new States();
  state.descri = faker.address.state();
  state.country = ???????;
  return state;
});

// Countries.Entity.ts
@Entity()
export class Countries extends Base {
  @Column({ nullable: false, type: 'varchar', unique: true })
  descri: string;

  @OneToMany(() => States, state => state.country)
  state: States[];
}

// Countries.Factory.ts
import { setSeederFactory } from 'typeorm-extension';
import { Countries } from '../entities/Countries';

export default setSeederFactory(Countries, faker => {
  const country = new Countries();
  country.descri = faker.address.country();
  return country;
});

[netojose]

You can do this:

Method 1

factoryManager.set(States, (faker: Faker, meta: unknown) => {
      const state = new States()
      state.country = 99 // HERE!
      return state
})

const stateFactory = await factoryManager.get(States)
await stateFactory.save()

Method 2

// states.seeder.ts
import { DataSource } from 'typeorm'
import { Seeder, SeederFactoryManager } from 'typeorm-extension'

import { States } from 'path/to/entities'

export default class ServiceSeeder implements Seeder {
  public async run(
    dataSource: DataSource,
    factoryManager: SeederFactoryManager,
  ): Promise<void> {
    const stateFactory = await factoryManager.get(States)
    stateFactory.setMeta(1) // Here you can pass data to factory
    await stateFactory.saveMany(60)
  }
}

// state.factory.ts
import { setSeederFactory } from 'typeorm-extension'
import { State } from 'path/to/entity'

export default setSeederFactory(
  State,
  (faker, meta: number) => {
    const state = new States();
    state.descri = faker.address.state();
    state.country = meta; // HERE you can get the values passed by seeder to factory
    return state;
  },
)