四好公路
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

174 lines
4.6 KiB

'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,
dec: anticounter
}
var arrow = {
in: counter,
out: anticounter
}
var nester = {
in: counter,
out: anticounter
}
test('should wrap multiple functions safely', function (t) {
t.plan(9)
t.equal(counter, generator.inc, 'basic function equality testing should work')
t.equal(anticounter, generator.dec, 'basic function equality testing should work')
t.doesNotThrow(function () { generator.inc() })
t.doesNotThrow(function () { generator.dec() })
t.equal(0, outsider, 'calls have side effects')
var count = 0
function wrapper (original) {
return function () {
count++
var returned = original.apply(this, arguments)
count++
return returned
}
}
shimmer.massWrap(generator, ['inc', 'dec'], wrapper)
t.doesNotThrow(function () { generator.inc() })
t.doesNotThrow(function () { generator.dec() })
t.equal(4, count, 'both pre and post increments should have happened')
t.equal(0, outsider, 'original function has still been called')
})
test('should wrap multiple functions on multiple modules safely', function (t) {
t.plan(15)
t.equal(counter, arrow.in, 'basic function equality testing should work')
t.equal(counter, nester.in, 'basic function equality testing should work')
t.equal(anticounter, arrow.out, 'basic function equality testing should work')
t.equal(anticounter, nester.out, 'basic function equality testing should work')
t.doesNotThrow(function () { arrow.in() })
t.doesNotThrow(function () { nester.in() })
t.doesNotThrow(function () { arrow.out() })
t.doesNotThrow(function () { nester.out() })
t.equal(0, outsider, 'calls have side effects')
var count = 0
function wrapper (original) {
return function () {
count++
var returned = original.apply(this, arguments)
count++
return returned
}
}
shimmer.massWrap([arrow, nester], ['in', 'out'], wrapper)
t.doesNotThrow(function () { arrow.in() })
t.doesNotThrow(function () { arrow.out() })
t.doesNotThrow(function () { nester.in() })
t.doesNotThrow(function () { nester.out() })
t.equal(8, count, 'both pre and post increments should have happened')
t.equal(0, outsider, 'original function has still been called')
})
test('wrap called with no arguments', function (t) {
t.plan(2)
var mock = sinon.expectation
.create('logger')
.twice()
shimmer({ logger: mock })
t.doesNotThrow(function () {
shimmer.massWrap()
}, "wrapping with no arguments doesn't throw")
t.doesNotThrow(function () {
mock.verify()
}, 'logger was called with the expected message')
})
test('wrap called with module but nothing else', function (t) {
t.plan(2)
var mock = sinon.expectation
.create('logger')
.withExactArgs('must provide one or more functions to wrap on modules')
.once()
shimmer({ logger: mock })
t.doesNotThrow(function () {
shimmer.massWrap(generator)
}, "wrapping with only 1 argument doesn't throw")
t.doesNotThrow(function () {
mock.verify()
}, 'logger was called with the expected message')
})
test('wrap called with original but no wrapper', function (t) {
t.plan(2)
var mock = sinon.expectation
.create('logger')
.twice()
shimmer({ logger: mock })
t.doesNotThrow(function () {
shimmer.massWrap(generator, ['inc'])
}, "wrapping with only original function doesn't throw")
t.doesNotThrow(function () {
mock.verify()
}, 'logger was called with the expected message')
})
test('wrap called with non-function original', function (t) {
t.plan(2)
var mock = sinon.expectation
.create('logger')
.withExactArgs('must provide one or more functions to wrap on modules')
.once()
shimmer({ logger: mock })
t.doesNotThrow(function () {
shimmer.massWrap({ orange: 'slices' }, 'orange', function () {})
}, "wrapping non-function original doesn't throw")
t.doesNotThrow(function () {
mock.verify()
}, 'logger was called with the expected message')
})
test('wrap called with non-function wrapper', function (t) {
t.plan(2)
var mock = sinon.expectation
.create('logger')
.withArgs('must provide one or more functions to wrap on modules')
.once()
shimmer({ logger: mock })
t.doesNotThrow(function () {
shimmer.massWrap({ orange: function () {} }, 'orange', 'hamchunx')
}, "wrapping with non-function wrapper doesn't throw")
t.doesNotThrow(function () {
mock.verify()
}, 'logger was called with the expected message')
})