generator-aspnetpostgresql/test/test-utility.js

Yeoman generator for ASP.NET MVC Core apps with PostgreSQL support

'use strict';
var util = (function() {

  var yeoman = require('yeoman-generator');
  var path = require('path');
  var os = require('os');
  var crypto = require('crypto');
  var assert;
  var mockGen;

  function makeTempDir() {
    return path.join(os.tmpdir(), crypto.randomBytes(20).toString('hex'));
  }

  function goCreate(subgenerator, tempDir) {
    var testDirectory;
    if (tempDir) {
      // Don't clear the test directory, we need it to have previous contents.
      before(function() {
        testDirectory = yeoman.test.testDirectory;
        yeoman.test.testDirectory = function(dir, cb) {
          process.chdir(dir);
          cb();
        };
      });
      after(function() {
        yeoman.test.testDirectory = testDirectory;
      });
    }
    before(function(done) {

      assert = yeoman.assert;
      mockGen = yeoman.test;

      var ctx = mockGen.run(path.join(__dirname, '../' + subgenerator));

      if (tempDir) {
        ctx.inDir(tempDir);
      }

      ctx.on('end', done);
    });
  }

  function goCreateWithArgs(subgenerator, args, tempDir) {
    var testDirectory;
    if (tempDir) {
      // Don't clear the test directory, we need it to have previous contents.
      before(function() {
        testDirectory = yeoman.test.testDirectory;
        yeoman.test.testDirectory = function(dir, cb) {
          process.chdir(dir);
          cb();
        };
      });
      after(function() {
        yeoman.test.testDirectory = testDirectory;
      });
    }
    before(function(done) {
      assert = yeoman.assert;
      mockGen = yeoman.test;

      var ctx = mockGen.run(path.join(__dirname, '../' + subgenerator))
        .withArguments(args);

      if (tempDir) {
        ctx.inDir(tempDir);
      }

      ctx.on('end', done);
    });
  }

  function goCreateApplication(type, applicationName, tempDir) {
    before(function(done) {

      assert = yeoman.assert;
      mockGen = yeoman.test;

      var mockPrompt = {
        type: type,
        applicationName: applicationName
      };

      var ctx = mockGen.run(path.join(__dirname, '../app'))
        .withPrompts(mockPrompt);

      if (tempDir) {
        ctx.inDir(tempDir);
      }

      ctx.on('end', done);
    });

  }

  function goCreateApplicationWithOptions(type, applicationName, ui, options) {
    before(function(done) {

      assert = yeoman.assert;
      mockGen = yeoman.test;

      var mockPrompt = {
        type: type,
        applicationName: applicationName,
        ui: ui
      };

      mockGen.run(path.join(__dirname, '../app'))
        .withPrompts(mockPrompt)
        .withOptions(options)
        .on('end', done);
    });

  }

  function dirsCheck(dirs) {
    describe('Directories Creation', function() {
      for (var i = 0; i < dirs.length; i++) {
        /*jshint loopfunc: true */
        it(dirs[i] + ' created.', function() {
          assert.file(dirs[i]);
        });
      }

    });

  }

  function filesCheck(file) {

    it(file + ' created.', function() {
      assert.file(file);
    });

  }

  function dirCheck(message, dir) {
    describe('Directory Creation', function() {
      it(message, function() {
        assert.file(dir);
      });
    });
  }

  function fileCheck(message, file) {
    describe('File Creation', function() {
      it(message, function() {
        assert.file(file);
      });
    });
  }

  function fileContentCheck(file, message, content) {
    it(message, function() {
      assert.fileContent(file, content);
    });
  }

  /**
   * The opposite function: specific content cannot be found
   * in a file assertion
   * @param  {String} file
   * @param  {String} message
   * @param  {String} content
   * @return {Boolean} true if condition is met
   */
  function noFileContentCheck(file, message, content) {
    it(message, function() {
      assert.noFileContent(file, content);
    });
  }

  var methods = {
    goCreateApplication: goCreateApplication,
    goCreateApplicationWithOptions: goCreateApplicationWithOptions,
    goCreate: goCreate,
    goCreateWithArgs: goCreateWithArgs,
    fileCheck: fileCheck,
    filesCheck: filesCheck,
    dirCheck: dirCheck,
    dirsCheck: dirsCheck,
    fileContentCheck: fileContentCheck,
    noFileContentCheck: noFileContentCheck,
    makeTempDir: makeTempDir
  };

  return methods;
})();

module.exports = util;