@postxl/generator 0.7.1 → 0.8.0

package/dist/generator.js

@@ -178,7 +178,7 @@ const generate = ({ models, enums, config, prismaClientPath, logger, }) => __awa
         generated.write(`/${meta.data.dataMockModuleFilePath}.ts`, (0, datamockmodule_generator_1.generateDataMockModule)({ models, meta }));
         generated.write(`/${meta.data.dataModuleFilePath}.ts`, (0, datamodule_generator_1.generateDataModule)({ models, meta }));
         generated.write(`/${meta.data.dataServiceFilePath}.ts`, (0, dataservice_generator_1.generateDataService)({ models, meta }));
-        generated.write(`/${meta.data.testDataServiceFilePath}.ts`, (0, testdataservice_generator_1.generateTestDataService)({ meta }));
+        generated.write(`/${meta.data.testDataServiceFilePath}.ts`, (0, testdataservice_generator_1.generateTestDataService)({ models, meta }));
         generated.write(`/${meta.data.repositoriesConstFilePath}.ts`, (0, repositories_generator_1.generateRepositoriesArray)({ models, meta }));
         generated.write(`/${meta.data.repositoriesIndexFilePath}.ts`, (0, repositories_generator_1.generateRepositoriesIndex)({ models, meta }));
         generated.write(`/${meta.data.stubIndexFilePath}.ts`, (0, stubs_generator_1.generateStubsIndex)({ models, meta }));

package/dist/generators/indices/testdataservice.generator.d.ts

@@ -1,7 +1,9 @@
 import { SchemaMetaData } from '../../lib/meta';
+import { Model } from '../../lib/schema/schema';
 /**
  * Generates a generic data service object that may be used on the server to access database.
  */
-export declare function generateTestDataService({ meta }: {
+export declare function generateTestDataService({ meta, models }: {
     meta: SchemaMetaData;
+    models: Model[];
 }): string;

package/dist/generators/indices/testdataservice.generator.js

@@ -24,16 +24,27 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateTestDataService = void 0;
+const meta_1 = require("../../lib/meta");
 const Types = __importStar(require("../../lib/schema/types"));
 const imports_1 = require("../../lib/imports");
 /**
  * Generates a generic data service object that may be used on the server to access database.
  */
-function generateTestDataService({ meta }) {
+function generateTestDataService({ meta, models }) {
     const imports = imports_1.ImportsGenerator.from(meta.data.testDataServiceFilePath).addImport({
         items: [Types.toVariableName('MockData')],
         from: meta.data.dataMockModuleFilePath,
     });
+    const mm = models.map((model) => ({ model, meta: (0, meta_1.getModelMetadata)({ model }) }));
+    for (const { meta } of mm) {
+        imports.addImport({
+            items: [meta.data.repositoryClassName],
+            from: meta.data.repoFilePath,
+        });
+    }
+    const constructor = mm
+        .map(({ meta }) => `private ${meta.data.dataServiceName} :${meta.data.repositoryClassName}`)
+        .join(',\n');
     return `
 		import { Injectable, Logger } from '@nestjs/common'
 
@@ -47,14 +58,19 @@ function generateTestDataService({ meta }) {
 			private logger = new Logger(TestDataService.name)
 			constructor(
 				private db: DbService,
-
 				private resetService: ResetService,
+				${constructor}
 			) {}
 
 			public async resetTestData(data: MockData) {
 				this.logger.log(\`✅ Reset test data\`)
 				await this.db.emptyDatabase()
-				return this.resetService.reset()
+
+				${mm
+        .map(({ meta }) => `await this.${meta.data.dataServiceName}.reInit(data.${meta.data.mockDataPropertyName} ?? [])`)
+        .join('\n')}
+				
+					return this.resetService.reset()
 			}
 		}
 	`;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@postxl/generator",
-  "version": "0.7.1",
+  "version": "0.8.0",
   "main": "./dist/generator.js",
   "typings": "./dist/generator.d.ts",
   "bin": {