test: repository schematic

Author: monica.lopez-gris

packages/schematics/src/lib/repository/repository.factory_spec.ts

@@ -0,0 +1,66 @@
+import { SchematicTestRunner } from '@angular-devkit/schematics/testing';
+import * as path from 'path';
+
+describe('Repository Factory', () => {
+  const runner: SchematicTestRunner = new SchematicTestRunner('.', path.join(process.cwd(), 'src/collection.json'));
+  it('should manage name only', () => {
+    const app: object = {
+      name: '',
+    };
+    const options: object = {
+      name: 'project',
+    };
+    runner.runSchematicAsync('application', app).subscribe(tree => {
+      runner.runSchematicAsync('repository', options, tree).subscribe(tree => {
+        const files: string[] = tree.files;
+        expect(files.find(filename => filename === '/src/app/model/entities/project.entity.ts')).toBeDefined();
+        expect(files.find(filename => filename === '/src/app/repositories/project.repository.ts')).toBeDefined();
+        expect(tree.readContent('/src/app/model/entities/project.entity.ts')).toEqual(
+          "import { Entity } from 'typeorm';\n" +
+            "import { BaseEntity } from '../../../shared/model/entities/base-entity.entity';\n" +
+            '\n' +
+            '@Entity()\n' +
+            'export class Project extends BaseEntity {}\n',
+        );
+        expect(tree.readContent('/src/app/repositories/project.repository.ts')).toEqual(
+          "import { Project } from '../model/entities/project.entity';\n" +
+            "import { Repository, EntityRepository } from 'typeorm';\n" +
+            '\n' +
+            '@EntityRepository(Project)\n' +
+            'export class ProjectRepository extends Repository<Project> {}\n',
+        );
+      });
+    });
+  });
+  it('should manage name and path', () => {
+    const app: object = {
+      name: 'app',
+    };
+    const options: object = {
+      name: 'project',
+      path: 'app',
+    };
+    runner.runSchematicAsync('application', app).subscribe(tree => {
+      runner.runSchematicAsync('repository', options, tree).subscribe(tree => {
+        const files: string[] = tree.files;
+        expect(files.find(filename => filename === '/app/src/app/model/entities/project.entity.ts')).toBeDefined();
+        expect(files.find(filename => filename === '/app/src/app/repositories/project.repository.ts')).toBeDefined();
+      });
+    });
+  });
+  it('should manage name to dasherize', () => {
+    const app: object = {
+      name: '',
+    };
+    const options: object = {
+      name: 'fooBar',
+    };
+    runner.runSchematicAsync('application', app).subscribe(tree => {
+      runner.runSchematicAsync('repository', options, tree).subscribe(tree => {
+        const files: string[] = tree.files;
+        expect(files.find(filename => filename === '/src/app/model/entities/foo-bar.entity.ts')).toBeDefined();
+        expect(files.find(filename => filename === '/src/app/repositories/foo-bar.repository.ts')).toBeDefined();
+      });
+    });
+  });
+});

packages/schematics/src/lib/security/security.factory_spec.ts

@@ -1,17 +1,65 @@
-import { Tree } from '@angular-devkit/schematics';
 import { SchematicTestRunner } from '@angular-devkit/schematics/testing';
 import * as path from 'path';
 
-const collectionPath = path.join(__dirname, '../collection.json');
+describe('Security Factory', () => {
+  const runner: SchematicTestRunner = new SchematicTestRunner('.', path.join(process.cwd(), 'src/collection.json'));
+  it('should work', () => {
+    const app: object = {
+      name: '',
+    };
+    const options: object = {
+      path: '',
+    };
+    runner.runSchematicAsync('application', app).subscribe(tree => {
+      runner.runSchematicAsync('security', options, tree).subscribe(tree => {
+        const files: string[] = tree.files;
+        expect(files.find(filename => filename === '/src/main.ts')).toBeDefined();
+        expect(files.find(filename => filename === '/package.json')).toBeDefined();
 
-describe('schematics', () => {
-  it('works', done => {
-    const runner = new SchematicTestRunner('schematics', collectionPath);
-    const obsTree = runner.runSchematicAsync('schematics', {}, Tree.empty());
-
-    obsTree.subscribe(tree => {
-      expect(tree.files).toEqual([]);
-      done();
+        expect(tree.readContent('/package.json')).toContain('@types/helmet');
+        expect(tree.readContent('/src/main.ts')).toEqual(
+          "import { NestFactory } from '@nestjs/core';\n" +
+            "import { AppModule } from './app/app.module';\n" +
+            "import { WinstonLogger } from './app/shared/logger/winston.logger';\n" +
+            "import { ValidationPipe } from '@nestjs/common';\n" +
+            "import * as helmet from 'helmet';\n" +
+            '\n' +
+            'async function bootstrap(): Promise<void> {\n' +
+            '  const app = await NestFactory.create(AppModule, { logger: new WinstonLogger() });\n' +
+            '  app.useGlobalPipes(\n' +
+            '    new ValidationPipe({\n' +
+            '      transform: true,\n' +
+            '    }),\n' +
+            '  );\n' +
+            "  app.setGlobalPrefix('v1');\n" +
+            '  app.use(helmet());\n' +
+            '  app.enableCors({\n' +
+            "    origin: '*',\n" +
+            '    credentials: true,\n' +
+            "    exposedHeaders: 'Authorization',\n" +
+            "    allowedHeaders: 'authorization, content-type',\n" +
+            '  });\n' +
+            '  await app.listen(3000);\n' +
+            '}\n' +
+            'bootstrap();\n',
+        );
+      });
+    });
+  });
+  it('should manage path', () => {
+    const app: object = {
+      name: 'foo',
+    };
+    const options: object = {
+      path: 'foo',
+    };
+    runner.runSchematicAsync('application', app).subscribe(tree => {
+      runner.runSchematicAsync('security', options, tree).subscribe(tree => {
+        const files: string[] = tree.files;
+        expect(files.find(filename => filename === '/foo/src/main.ts')).toBeDefined();
+        expect(files.find(filename => filename === '/foo/package.json')).toBeDefined();
+        expect(tree.readContent('/foo/package.json')).toContain('@types/helmet');
+      });
     });
   });
 });