Blame view

node_modules/shimmer/test/unwrap.tap.js 2.66 KB
f7563de62   Palak Handa   first commit
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
  '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 }
  
  var generator = {
    inc: counter
  }
  
  test('should unwrap safely', function (t) {
    t.plan(9)
  
    t.equal(counter, generator.inc, 'basic function equality testing should work')
    t.doesNotThrow(function () { generator.inc() })
    t.equal(1, outsider, 'calls have side effects')
  
    function wrapper (original) {
      return function () {
        return original.apply(this, arguments)
      }
    }
    shimmer.wrap(generator, 'inc', wrapper)
  
    t.doesNotEqual(counter, generator.inc, 'function should be wrapped')
  
    t.doesNotThrow(function () { generator.inc() })
    t.equal(2, outsider, 'original function has still been called')
  
    shimmer.unwrap(generator, 'inc')
    t.equal(counter, generator.inc, 'basic function equality testing should work')
    t.doesNotThrow(function () { generator.inc() })
    t.equal(3, outsider, 'original function has still been called')
  })
  
  test("shouldn't throw on double unwrapping", function (t) {
    t.plan(6)
  
    t.equal(counter, generator.inc, 'basic function equality testing should work')
  
    var mock = sinon.expectation
      .create('logger')
      .withArgs('no original to unwrap to -- ' +
        'has inc already been unwrapped?')
      .once()
    shimmer({logger: mock})
  
    function wrapper (original) {
      return function () {
        return original.apply(this, arguments)
      }
    }
    shimmer.wrap(generator, 'inc', wrapper)
  
    t.doesNotEqual(counter, generator.inc, 'function should be wrapped')
  
    shimmer.unwrap(generator, 'inc')
    t.equal(counter, generator.inc, 'basic function equality testing should work')
  
    t.doesNotThrow(function () { shimmer.unwrap(generator, 'inc') },
      'should double unwrap without issue')
    t.equal(counter, generator.inc, 'function is unchanged after unwrapping')
  
    t.doesNotThrow(function () {
      mock.verify()
    }, 'logger was called with the expected message')
  })
  
  test('unwrap called with no arguments', function (t) {
    t.plan(2)
  
    var mock = sinon.expectation
      .create('logger')
      .twice()
    shimmer({logger: mock})
  
    t.doesNotThrow(function () { shimmer.unwrap() }, 'should log instead of throwing')
  
    t.doesNotThrow(function () {
      mock.verify()
    }, 'logger was called with the expected message')
  })
  
  test('unwrap called with module but no name', function (t) {
    t.plan(2)
  
    var mock = sinon.expectation
      .create('logger')
      .twice()
    shimmer({logger: mock})
  
    t.doesNotThrow(function () { shimmer.unwrap({}) }, 'should log instead of throwing')
  
    t.doesNotThrow(function () {
      mock.verify()
    }, 'logger was called with the expected message')
  })