fix: class and controller tests

nestjs · Jun 19, 2022 · 5902134 · 5902134
1 parent c96a604
commit 5902134
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/src/lib/class/class.factory.test.ts b/src/lib/class/class.factory.test.ts
@@ -132,7 +132,7 @@ describe('Class Factory', () => {
   });
   it('should create a spec file', async () => {
     const options: ClassOptions = {
-      name: 'foo.entity',
+      name: 'foo',
       spec: true,
       flat: true,
     };
@@ -142,12 +142,12 @@ describe('Class Factory', () => {
     const files: string[] = tree.files;
 
     expect(
-      files.find((filename) => filename === '/foo.entity.spec.ts'),
+      files.find((filename) => filename === '/foo.spec.ts'),
     ).not.toBeUndefined();
   });
   it('should create a spec file with custom file suffix', async () => {
     const options: ClassOptions = {
-      name: 'foo.entity',
+      name: 'foo',
       spec: true,
       specFileSuffix: 'test',
       flat: true,
@@ -158,10 +158,10 @@ describe('Class Factory', () => {
     const files: string[] = tree.files;
 
     expect(
-      files.find((filename) => filename === '/foo.entity.spec.ts'),
+      files.find((filename) => filename === '/foo.spec.ts'),
     ).toBeUndefined();
     expect(
-      files.find((filename) => filename === '/foo.entity.test.ts'),
-    ).not.toBeUndefined();
+      files.find((filename) => filename === '/foo.test.ts'),
+    ).toBeDefined();
   });
 });
diff --git a/src/lib/controller/controller.factory.test.ts b/src/lib/controller/controller.factory.test.ts
@@ -232,12 +232,12 @@ describe('Controller Factory', () => {
   });
   it('should create a spec file', async () => {
     const options: ControllerOptions = {
-      name: 'foo.controller',
+      name: 'foo',
       spec: true,
       flat: true,
     };
     const tree: UnitTestTree = await runner
-      .runSchematicAsync('class', options)
+      .runSchematicAsync('controller', options)
       .toPromise();
     const files: string[] = tree.files;
 
@@ -247,13 +247,13 @@ describe('Controller Factory', () => {
   });
   it('should create a spec file with custom file suffix', async () => {
     const options: ControllerOptions = {
-      name: 'foo.controller',
+      name: 'foo',
       spec: true,
       specFileSuffix: 'test',
       flat: true,
     };
     const tree: UnitTestTree = await runner
-      .runSchematicAsync('class', options)
+      .runSchematicAsync('controller', options)
       .toPromise();
     const files: string[] = tree.files;