clean-tostring/index.js

Clean function `toString` from code coverage mess. It can't be ideal, but it's useful in testing process.

/*!
 * clean-tostring <https://github.com/tunnckoCore/clean-tostring>
 *
 * Copyright (c) 2016 Charlike Mike Reagent <@tunnckoCore> (http://www.tunnckocore.tk)
 * Released under the MIT license.
 */

'use strict'

var cleanupCoverageCode = require('cleanup-coverage-code')

/**
 * > Clean coverage code of `fn.toString()`, but be aware of that
 * it remains minified.
 *
 * **Example**
 *
 * ```js
 * var cleanup = require('clean-tostring')
 * var isIstanbul = process.env.running_under_istanbul
 *
 * function fixture (a) {
 *   if (typeof a === 'string') {
 *     return a
 *   }
 *   return false
 * }
 *
 * if (isIstanbul) {
 *   console.log(cleanup(fixture))
 *   // => 'function fixture(a) {if(typeof a===\'string\'){return a;}else{}return false;}'
 * } else {
 *   console.log(cleanup(fixture))
 *   // => 'function fixture(a) {\n    if (typeof a === \'string\') {\n      return a\n    }\n    return false\n  }'
 * }
 * ```
 *
 * @param  {Function} `fn`
 * @return {String}
 * @api public
 */

module.exports = function cleanTostring (fn) {
  if (typeof fn !== 'function') {
    throw new TypeError('clean-tostring: expect `fn` be function')
  }
  return cleanupCoverageCode(fn.toString()).replace('\)\{', ') {')
}