@rxdi/schematics/lib/gateway/gateway.factory.test.ts

import {
  SchematicTestRunner,
  UnitTestTree,
} from '@angular-devkit/schematics/testing';
import * as path from 'path';
import { GatewayOptions } from './gateway.schema';

describe('Gateway Factory', () => {
  const runner: SchematicTestRunner = new SchematicTestRunner(
    '.',
    path.join(process.cwd(), 'src/collection.json'),
  );
  it('should manage name only', () => {
    const options: GatewayOptions = {
      name: 'foo',
      flat: false,
    };
    const tree: UnitTestTree = runner.runSchematic('gateway', options);
    const files: string[] = tree.files;
    expect(
      files.find(filename => filename === '/foo/foo.gateway.ts'),
    ).not.toBeUndefined();
    expect(tree.readContent('/foo/foo.gateway.ts')).toEqual(
      "import { SubscribeMessage, WebSocketGateway } from '@nestjs/websockets';\n" +
        '\n' +
        '@WebSocketGateway()\n' +
        'export class FooGateway {\n' +
        "  @SubscribeMessage('message')\n" +
        '  handleMessage(client: any, payload: any): string {\n' +
        "    return 'Hello world!';\n" +
        '  }\n' +
        '}\n',
    );
  });
  it('should manage name as a path', () => {
    const options: GatewayOptions = {
      name: 'bar/foo',
      flat: false,
    };
    const tree: UnitTestTree = runner.runSchematic('gateway', options);
    const files: string[] = tree.files;
    expect(
      files.find(filename => filename === '/bar/foo/foo.gateway.ts'),
    ).not.toBeUndefined();
    expect(tree.readContent('/bar/foo/foo.gateway.ts')).toEqual(
      "import { SubscribeMessage, WebSocketGateway } from '@nestjs/websockets';\n" +
        '\n' +
        '@WebSocketGateway()\n' +
        'export class FooGateway {\n' +
        "  @SubscribeMessage('message')\n" +
        '  handleMessage(client: any, payload: any): string {\n' +
        "    return 'Hello world!';\n" +
        '  }\n' +
        '}\n',
    );
  });
  it('should manage name and path', () => {
    const options: GatewayOptions = {
      name: 'foo',
      path: 'baz',
      flat: false,
    };
    const tree: UnitTestTree = runner.runSchematic('gateway', options);
    const files: string[] = tree.files;
    expect(
      files.find(filename => filename === '/baz/foo/foo.gateway.ts'),
    ).not.toBeUndefined();
    expect(tree.readContent('/baz/foo/foo.gateway.ts')).toEqual(
      "import { SubscribeMessage, WebSocketGateway } from '@nestjs/websockets';\n" +
        '\n' +
        '@WebSocketGateway()\n' +
        'export class FooGateway {\n' +
        "  @SubscribeMessage('message')\n" +
        '  handleMessage(client: any, payload: any): string {\n' +
        "    return 'Hello world!';\n" +
        '  }\n' +
        '}\n',
    );
  });
  it('should manage name to dasherize', () => {
    const options: GatewayOptions = {
      name: 'fooBar',
      flat: false,
    };
    const tree: UnitTestTree = runner.runSchematic('gateway', options);
    const files: string[] = tree.files;
    expect(
      files.find(filename => filename === '/foo-bar/foo-bar.gateway.ts'),
    ).not.toBeUndefined();
    expect(tree.readContent('/foo-bar/foo-bar.gateway.ts')).toEqual(
      "import { SubscribeMessage, WebSocketGateway } from '@nestjs/websockets';\n" +
        '\n' +
        '@WebSocketGateway()\n' +
        'export class FooBarGateway {\n' +
        "  @SubscribeMessage('message')\n" +
        '  handleMessage(client: any, payload: any): string {\n' +
        "    return 'Hello world!';\n" +
        '  }\n' +
        '}\n',
    );
  });
  it('should manage path to dasherize', () => {
    const options: GatewayOptions = {
      name: 'barBaz/foo',
      flat: false,
    };
    const tree: UnitTestTree = runner.runSchematic('gateway', options);
    const files: string[] = tree.files;
    expect(
      files.find(filename => filename === '/bar-baz/foo/foo.gateway.ts'),
    ).not.toBeUndefined();
    expect(tree.readContent('/bar-baz/foo/foo.gateway.ts')).toEqual(
      "import { SubscribeMessage, WebSocketGateway } from '@nestjs/websockets';\n" +
        '\n' +
        '@WebSocketGateway()\n' +
        'export class FooGateway {\n' +
        "  @SubscribeMessage('message')\n" +
        '  handleMessage(client: any, payload: any): string {\n' +
        "    return 'Hello world!';\n" +
        '  }\n' +
        '}\n',
    );
  });
  it('should manage javascript file', () => {
    const options: GatewayOptions = {
      name: 'foo',
      language: 'js',
      flat: false,
    };
    const tree: UnitTestTree = runner.runSchematic('gateway', options);
    const files: string[] = tree.files;
    expect(
      files.find(filename => filename === '/foo/foo.gateway.js'),
    ).not.toBeUndefined();
    expect(tree.readContent('/foo/foo.gateway.js')).toEqual(
      "import { SubscribeMessage, WebSocketGateway } from '@nestjs/websockets';\n" +
        '\n' +
        '@WebSocketGateway()\n' +
        'export class FooGateway {\n' +
        "  @SubscribeMessage('message')\n" +
        '  handleMessage(client, payload) {\n' +
        "    return 'Hello world!';\n" +
        '  }\n' +
        '}\n',
    );
  });
});