massUnwrap.tap.js
4.05 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
'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
}
test('should unwrap safely', function (t) {
t.plan(18)
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.equal(1, outsider, 'calls have side effects')
t.doesNotThrow(function () { generator.dec() })
t.equal(0, outsider, 'calls have side effects')
function wrapper (original) {
return function () {
return original.apply(this, arguments)
}
}
shimmer.massWrap(generator, ['inc', 'dec'], wrapper)
t.doesNotEqual(counter, generator.inc, 'function should be wrapped')
t.doesNotEqual(anticounter, generator.dec, 'function should be wrapped')
t.doesNotThrow(function () { generator.inc() })
t.equal(1, outsider, 'original function has still been called')
t.doesNotThrow(function () { generator.dec() })
t.equal(0, outsider, 'original function has still been called')
shimmer.massUnwrap(generator, ['inc', 'dec'])
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.equal(1, outsider, 'original function has still been called')
t.doesNotThrow(function () { generator.dec() })
t.equal(0, outsider, 'original function has still been called')
})
test("shouldn't throw on double unwrapping", function (t) {
t.plan(10)
t.equal(counter, generator.inc, 'basic function equality testing should work')
t.equal(anticounter, generator.dec, 'basic function equality testing should work')
var mock = sinon.stub()
shimmer({ logger: mock })
function wrapper (original) {
return function () {
return original.apply(this, arguments)
}
}
shimmer.wrap(generator, 'inc', wrapper)
shimmer.wrap(generator, 'dec', wrapper)
t.doesNotEqual(counter, generator.inc, 'function should be wrapped')
t.doesNotEqual(anticounter, generator.dec, 'function should be wrapped')
shimmer.massUnwrap(generator, ['inc', 'dec'])
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 () { shimmer.massUnwrap(generator, ['inc', 'dec']) },
'should double unwrap without issue')
t.equal(counter, generator.inc, 'function is unchanged after unwrapping')
t.equal(anticounter, generator.dec, 'function is unchanged after unwrapping')
t.doesNotThrow(function () {
sinon.assert.calledWith(mock, 'no original to unwrap to -- ' +
'has inc already been unwrapped?')
sinon.assert.calledWith(mock, 'no original to unwrap to -- ' +
'has dec already been unwrapped?')
sinon.assert.calledTwice(mock)
}, 'logger was called with the expected message')
})
test('massUnwrap called with no arguments', function (t) {
t.plan(2)
var mock = sinon.expectation
.create('logger')
.twice()
shimmer({ logger: mock })
t.doesNotThrow(function () { shimmer.massUnwrap() }, 'should log instead of throwing')
t.doesNotThrow(function () {
mock.verify()
}, 'logger was called with the expected message')
})
test('massUnwrap 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 unwrap on modules')
.once()
shimmer({ logger: mock })
t.doesNotThrow(function () {
shimmer.massUnwrap(generator)
}, "wrapping with only 1 argument doesn't throw")
t.doesNotThrow(function () {
mock.verify()
}, 'logger was called with the expected message')
})