fix types

Author: hirsch88

src/database/seeds/CreatePets.ts

@@ -10,8 +10,8 @@ export class CreatePets implements Seed {
         const em = connection.createEntityManager();
 
         await times(10, async (n) => {
-            const pet = await factory<Pet, undefined>(Pet as any)().seed();
-            const user = await factory<User, undefined>(User as any)().make();
+            const pet = await factory(Pet)().seed();
+            const user = await factory(User)().make();
             user.pets = [pet];
             await em.save(user);
         });

src/lib/seed/EntityFactory.ts

@@ -1,5 +1,5 @@
 import * as Faker from 'faker';
-import { Connection } from 'typeorm/connection/Connection';
+import { Connection, ObjectType } from 'typeorm';
 
 import { FactoryFunction } from './types';
 import { isPromiseLike } from './utils';
@@ -13,7 +13,7 @@ export class EntityFactory<Entity, Settings> {
 
     constructor(
         public name: string,
-        public entity: Entity,
+        public entity: ObjectType<Entity>,
         private factory: FactoryFunction<Entity, Settings>,
         private settings?: Settings
     ) { }
@@ -54,7 +54,7 @@ export class EntityFactory<Entity, Settings> {
             const em = connection.createEntityManager();
             try {
                 const entity = await this.make();
-                return await em.save<Entity>(this.entity, entity);
+                return await em.save<Entity>(entity);
             } catch (error) {
                 throw new Error('Could not save entity');
             }

src/lib/seed/index.ts

@@ -1,6 +1,7 @@
 import 'reflect-metadata';
 import { Connection, ObjectType } from 'typeorm';
 
+import { Factory } from '../seeds/Factory';
 import { EntityFactory } from './EntityFactory';
 import { EntityFactoryDefinition, FactoryFunction, SeedConstructor } from './types';
 import { getNameOfClass } from './utils';
@@ -46,7 +47,7 @@ export const define = <Entity, Settings>(entity: ObjectType<Entity>, factoryFn:
 /**
  * Gets a defined entity factory and pass the settigns along to the entity factory function
  */
-export const factory = <Entity, Settings>(entity: Entity) => (settings?: Settings) => {
+export const factory: Factory = <Entity, Settings>(entity: ObjectType<Entity>) => (settings?: Settings) => {
     const name = getNameOfClass(entity);
     const entityFactoryObject = (global as any).seeder.entityFactories.get(name);
     return new EntityFactory<Entity, Settings>(