fix: add more tests

Signed-off-by: Muhammad Aaqil <[email protected]>

Author: aaqilniz

packages/cli/lib/base-generator.js

@@ -461,7 +461,7 @@ module.exports = class BaseGenerator extends Generator {
         );
       }
     }
-    // process.exit(0);
+    process.exit(0);
   }
 
   /**

packages/cli/snapshots/integration/generators/discover.integration.snapshots.js

@@ -66,6 +66,92 @@ import {
   get,
   getModelSchemaRef,
 } from '@loopback/rest';
+import {
+  Appointment,
+  Doctor,
+} from '../models';
+import {AppointmentRepository} from '../repositories';
+
+export class AppointmentDoctorController {
+  constructor(
+    @repository(AppointmentRepository)
+    public appointmentRepository: AppointmentRepository,
+  ) { }
+
+  @get('/appointments/{id}/doctor', {
+    responses: {
+      '200': {
+        description: 'Doctor belonging to Appointment',
+        content: {
+          'application/json': {
+            schema: getModelSchemaRef(Doctor),
+          },
+        },
+      },
+    },
+  })
+  async getDoctor(
+    @param.path.number('id') id: typeof Appointment.prototype.id,
+  ): Promise<Doctor> {
+    return this.appointmentRepository.doctor(id);
+  }
+}
+
+`;
+
+
+exports[`lb4 discover integration model discovery discovers models with --relations 2`] = `
+import {
+  repository,
+} from '@loopback/repository';
+import {
+  param,
+  get,
+  getModelSchemaRef,
+} from '@loopback/rest';
+import {
+  Appointment,
+  Patient,
+} from '../models';
+import {AppointmentRepository} from '../repositories';
+
+export class AppointmentPatientController {
+  constructor(
+    @repository(AppointmentRepository)
+    public appointmentRepository: AppointmentRepository,
+  ) { }
+
+  @get('/appointments/{id}/patient', {
+    responses: {
+      '200': {
+        description: 'Patient belonging to Appointment',
+        content: {
+          'application/json': {
+            schema: getModelSchemaRef(Patient),
+          },
+        },
+      },
+    },
+  })
+  async getPatient(
+    @param.path.number('id') id: typeof Appointment.prototype.id,
+  ): Promise<Patient> {
+    return this.appointmentRepository.patient(id);
+  }
+}
+
+`;
+
+
+exports[`lb4 discover integration model discovery discovers models with --relations 3`] = `
+import {
+  repository,
+} from '@loopback/repository';
+import {
+  param,
+  get,
+  getModelSchemaRef,
+} from '@loopback/rest';
 import {
   Doctor,
 } from '../models';
@@ -99,7 +185,7 @@ export class DoctorDoctorController {
 `;
 
 
-exports[`lb4 discover integration model discovery discovers models with --relations 2`] = `
+exports[`lb4 discover integration model discovery discovers models with --relations 4`] = `
 import {Entity, model, property, belongsTo} from '@loopback/repository';
 
 @model()
@@ -205,6 +291,92 @@ import {
   get,
   getModelSchemaRef,
 } from '@loopback/rest';
+import {
+  Appointment,
+  Doctor,
+} from '../models';
+import {AppointmentRepository} from '../repositories';
+
+export class AppointmentDoctorController {
+  constructor(
+    @repository(AppointmentRepository)
+    public appointmentRepository: AppointmentRepository,
+  ) { }
+
+  @get('/appointments/{id}/doctor', {
+    responses: {
+      '200': {
+        description: 'Doctor belonging to Appointment',
+        content: {
+          'application/json': {
+            schema: getModelSchemaRef(Doctor),
+          },
+        },
+      },
+    },
+  })
+  async getDoctor(
+    @param.path.number('id') id: typeof Appointment.prototype.id,
+  ): Promise<Doctor> {
+    return this.appointmentRepository.doctor(id);
+  }
+}
+
+`;
+
+
+exports[`lb4 discover integration model discovery generate relations with --relations 2`] = `
+import {
+  repository,
+} from '@loopback/repository';
+import {
+  param,
+  get,
+  getModelSchemaRef,
+} from '@loopback/rest';
+import {
+  Appointment,
+  Patient,
+} from '../models';
+import {AppointmentRepository} from '../repositories';
+
+export class AppointmentPatientController {
+  constructor(
+    @repository(AppointmentRepository)
+    public appointmentRepository: AppointmentRepository,
+  ) { }
+
+  @get('/appointments/{id}/patient', {
+    responses: {
+      '200': {
+        description: 'Patient belonging to Appointment',
+        content: {
+          'application/json': {
+            schema: getModelSchemaRef(Patient),
+          },
+        },
+      },
+    },
+  })
+  async getPatient(
+    @param.path.number('id') id: typeof Appointment.prototype.id,
+  ): Promise<Patient> {
+    return this.appointmentRepository.patient(id);
+  }
+}
+
+`;
+
+
+exports[`lb4 discover integration model discovery generate relations with --relations 3`] = `
+import {
+  repository,
+} from '@loopback/repository';
+import {
+  param,
+  get,
+  getModelSchemaRef,
+} from '@loopback/rest';
 import {
   Doctor,
 } from '../models';
@@ -238,7 +410,7 @@ export class DoctorDoctorController {
 `;
 
 
-exports[`lb4 discover integration model discovery generate relations with --relations 2`] = `
+exports[`lb4 discover integration model discovery generate relations with --relations 4`] = `
 import {Entity, model, property, belongsTo} from '@loopback/repository';
 import {Doctor} from './doctor.model';
 import {Patient} from './patient.model';

packages/cli/test/integration/generators/discover.integration.js

@@ -254,6 +254,8 @@ describe('lb4 discover integration', () => {
       assert.file(appointmentDoctorController);
       assert.file(appointmentPatientController);
       assert.file(doctorDoctorController);
+      expectFileToMatchSnapshot(appointmentDoctorController);
+      expectFileToMatchSnapshot(appointmentPatientController);
       expectFileToMatchSnapshot(doctorDoctorController);
       assert.file(appointmentModel);
       expectFileToMatchSnapshot(appointmentModel);
@@ -274,6 +276,8 @@ describe('lb4 discover integration', () => {
       assert.file(appointmentDoctorController);
       assert.file(appointmentPatientController);
       assert.file(doctorDoctorController);
+      expectFileToMatchSnapshot(appointmentDoctorController);
+      expectFileToMatchSnapshot(appointmentPatientController);
       expectFileToMatchSnapshot(doctorDoctorController);
       assert.file(doctorModel);
       expectFileToMatchSnapshot(doctorModel);