| 1 | sinon = require "sinon"
|
| 2 | _ = require "lodash"
|
| 3 | sandbox = []
|
| 4 |
|
| 5 | register = (type, obj, method, func) ->
|
| 6 | item = null
|
| 7 |
|
| 8 | try
|
| 9 | if type == "stub" && !_.isEmpty(func)
|
| 10 | item = sinon.stub(obj, method).callsFake(func)
|
| 11 | else
|
| 12 | item = sinon[type].apply(sinon, [ obj, method, func ])
|
| 13 |
|
| 14 | sandbox.push item
|
| 15 | catch e
|
| 16 | if /already wrapped/i.test e.message
|
| 17 | item = obj[method]
|
| 18 | else throw e
|
| 19 |
|
| 20 | item
|
| 21 |
|
| 22 | stub = (obj, method, func) ->
|
| 23 | register 'stub', obj, method, func
|
| 24 |
|
| 25 | spy = (obj, method) ->
|
| 26 | register 'spy', obj, method
|
| 27 |
|
| 28 | reset = ->
|
| 29 | _.each sandbox, (stub_or_spy) ->
|
| 30 | stub_or_spy.reset()
|
| 31 |
|
| 32 | restore = ->
|
| 33 | _.each sandbox, (stub) ->
|
| 34 | stub.restore() if stub.restore
|
| 35 |
|
| 36 | module.exports =
|
| 37 | stub: stub
|
| 38 | spy: spy
|
| 39 | reset: reset
|
| 40 | restore: restore
|