fix(common): Synthetic data server example start error (#4)

positive-js · Mar 19, 2020 · d4c5c02 · d4c5c02
1 parent 56a4663
commit d4c5c02
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/apps/mc-mocks-server/src/modules/vulnerabilities/vulnerability.dto.ts b/apps/mc-mocks-server/src/modules/vulnerabilities/vulnerability.dto.ts
@@ -25,7 +25,7 @@ export class VulnerabilityDto implements IVulnerabilityDto {
     name: string;
     records: RecordDto[];
 
-    id: string | number;
+    id: string;
 
     constructor(model: Partial<IVulnerabilityDto>) {
         Object.assign(this, model);

diff --git a/apps/mc-mocks-server/src/modules/vulnerabilities/vulnerability.entity.ts b/apps/mc-mocks-server/src/modules/vulnerabilities/vulnerability.entity.ts
@@ -1,7 +1,8 @@
 import { Column, Entity, JoinColumn, ManyToOne, OneToMany, PrimaryGeneratedColumn } from 'typeorm';
+import { RecordDto, VulnerabilityDto } from './vulnerability.dto';
 
 @Entity()
-export class Vulnerability {
+export class Vulnerability implements VulnerabilityDto {
     @PrimaryGeneratedColumn('uuid')
     id: string;
 
@@ -20,13 +21,13 @@ export class Vulnerability {
 }
 
 @Entity()
-export class Record {
+export class Record implements RecordDto {
     @PrimaryGeneratedColumn('uuid')
     id: string;
 
     @Column({length: 256})
     name: string;
 
     @ManyToOne((type) => Vulnerability, (vulnerability) => vulnerability.records)
-    vulnerabilityId: Vulnerability;
+    vulnerabilityId: string[];
 }