/ var / www / html / events / node_modules / shimmer / test / [ HOME SHELL ]

Name	Size	Permission
init.tap.js	1.11 KB	-rw-rw-r--
massUnwrap.tap.js	3.93 KB	-rw-rw-r--
massWrap.tap.js	4.63 KB	-rw-rw-r--
unwrap.tap.js	2.66 KB	-rw-rw-r--
wrap.tap.js	3.87 KB	-rw-rw-r--

Code Editor : wrap.tap.js

'use strict'

var tap = require('tap')
var test = tap.test
var sinon = require('sinon')
var shimmer = require('../index.js')

var outsider = 0
function counter () { return ++outsider }
function anticounter () { return --outsider }

var generator = {
  inc: counter
}
Object.defineProperty(generator, 'dec', {
  value: anticounter,
  writable: true,
  configurable: true,
  enumerable: false
})

test('should wrap safely', function (t) {
  t.plan(12)

t.equal(counter, generator.inc, 'method is mapped to function')
  t.doesNotThrow(function () { generator.inc() }, 'original function works')
  t.equal(1, outsider, 'calls have side effects')

var count = 0
  function wrapper (original, name) {
    t.equal(name, 'inc')
    return function () {
      count++
      var returned = original.apply(this, arguments)
      count++
      return returned
    }
  }
  shimmer.wrap(generator, 'inc', wrapper)

t.ok(generator.inc.__wrapped, "function tells us it's wrapped")
  t.equal(generator.inc.__original, counter, 'original function is available')
  t.doesNotThrow(function () { generator.inc() }, 'wrapping works')
  t.equal(2, count, 'both pre and post increments should have happened')
  t.equal(2, outsider, 'original function has still been called')
  t.ok(generator.propertyIsEnumerable('inc'),
    'wrapped enumerable property is still enumerable')
  t.equal(Object.keys(generator.inc).length, 0,
    'wrapped object has no additional properties')

shimmer.wrap(generator, 'dec', function (original) {
    return function () {
      return original.apply(this, arguments)
    }
  })

t.ok(!generator.propertyIsEnumerable('dec'),
    'wrapped unenumerable property is still unenumerable')
})

test('wrap called with no arguments', function (t) {
  t.plan(2)

var mock = sinon.expectation
    .create('logger')
    .withExactArgs('no original function undefined to wrap')
    .once()
  shimmer({ logger: mock })

t.doesNotThrow(function () {
    shimmer.wrap()
  }, "wrapping with no arguments doesn't throw")