@ciscospark/jsdoctrinetest/dist/transform.js

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _deleteProperty = require('babel-runtime/core-js/reflect/delete-property');

var _deleteProperty2 = _interopRequireDefault(_deleteProperty);

exports.default = transform;

var _parse = require('./parse');

var _parse2 = _interopRequireDefault(_parse);

var _babelTemplate = require('babel-template');

var _babelTemplate2 = _interopRequireDefault(_babelTemplate);

var _babelTraverse = require('babel-traverse');

var _babelTraverse2 = _interopRequireDefault(_babelTraverse);

var _mochaTemplate = require('./mocha-template');

var _mochaTemplate2 = _interopRequireDefault(_mochaTemplate);

var _literal = require('./assertions/literal');

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

var assertionStatementTemplate = (0, _babelTemplate2.default)('\n  return Promise.resolve(ORIGINAL)\n    .then(function(result) {\n      ASSERTIONS\n    })\n');

/**
 * @param {Object} options
 * @param {ast} options.comment
 * @param {string} options.name
 * @param {string} options.filename
 * @param {string} options.type
 * @returns {ast}
 */
/*!
 * Copyright (c) 2015-2017 Cisco Systems, Inc. See LICENSE file.
 */

function transform(_ref) {
  var comment = _ref.comment,
      name = _ref.name,
      filename = _ref.filename,
      type = _ref.type;

  var ast = (0, _parse2.default)(comment);

  (0, _babelTraverse2.default)(ast, {
    enter: function enter(path) {
      if (path.node.trailingComments) {
        var assertions = path.node.trailingComments.map(makeAsserter);
        if (assertions.length) {
          (0, _deleteProperty2.default)(path.node, 'trailingComments');

          path.replaceWith(assertionStatementTemplate({
            ORIGINAL: path.node,
            ASSERTIONS: assertions
          }));
        }
      }
    }
  });

  return (0, _mochaTemplate2.default)({
    testCase: ast,
    name: name,
    filename: filename,
    type: type
  });
}

/**
 * Takes a trailing comment from an example block and changes it to an assertion
 * @param {CommentBlock} comment
 * @returns {ast}
 */
function makeAsserter(comment) {
  if ((0, _literal.test)(comment.value)) {
    return (0, _literal.build)(comment.value);
  }

  return null;
}
//# sourceMappingURL=transform.js.map