fix(cli): updateModuleImports method

adrien2p · Apr 18, 2022 · f4db805 · f4db805
1 parent 1d1dbb4
commit f4db805
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/cli/commands/generate-component.ts b/src/cli/commands/generate-component.ts
@@ -162,7 +162,7 @@ export function updateModuleImports(fullDestinationPath: string): void {
 
 			const moduleContent = readFileSync(resolvedModulePath).toString();
 
-			const shouldUpdateModuleImport = !moduleContent.match(`${componentClassName}`);
+			const shouldUpdateModuleImport = !moduleContent.match(`${parse(file.name).name}`);
 			if (!shouldUpdateModuleImport) continue;
 
 			const isComponentInSubDirectory =

diff --git a/src/cli/tests/commands/generate-component.spec.ts b/src/cli/tests/commands/generate-component.spec.ts
@@ -127,7 +127,7 @@ describe('GenerateComponent', () => {
 	it('should generate a module that includes other generated components automatically', () => {
 		generateComponent(componentName, { middleware: true, service: true, path });
 		generateComponent(componentName, { module: true, path });
-		generateComponent(componentName, { repository: true, router: true, path });
+		generateComponent(componentName, { repository: true, router: true, entity: true, path });
 
 		const subDirectory = 'subDirectory';
 		generateComponent(componentName, { validator: true, path: path + '/' + subDirectory });
@@ -142,11 +142,12 @@ describe('GenerateComponent', () => {
 			import { TestValidator } from './${subDirectory}/test.validator';
 			import { TestRouter } from './test.router';
 			import { TestRepository } from './test.repository';
+			import { Test } from './test.entity';
 			import { TestService } from './test.service';
 			import { TestMiddleware } from './test.middleware';
 			
 			@Module({
-				imports: [TestMiddleware, TestService, TestRepository, TestRouter, TestValidator]
+				imports: [TestMiddleware, TestService, Test, TestRepository, TestRouter, TestValidator]
 			})
 			export class TestModule {}
 		`)