59 lines
1.2 KiB
JavaScript
59 lines
1.2 KiB
JavaScript
|
var inject = require('..')
|
||
|
|
var test = require('tape')
|
||
|
|
var Stream = require('stream')
|
||
|
|
var Emitter = require('events').EventEmitter
|
||
|
|
var noop = function () {}
|
||
|
|
|
||
|
|
test('connection', function (t) {
|
||
|
|
t.plan(4) // reconnect and connect
|
||
|
|
|
||
|
|
var ee = new Emitter
|
||
|
|
|
||
|
|
var reconnect = inject(function () {
|
||
|
|
var s = new Stream
|
||
|
|
ee.on('start server', function () {
|
||
|
|
s.emit('connect')
|
||
|
|
s.emit('close')
|
||
|
|
})
|
||
|
|
s.end = noop
|
||
|
|
return s
|
||
|
|
})
|
||
|
|
|
||
|
|
var reconnector = reconnect({initialDelay: 10}, function (stream) {
|
||
|
|
reconnector.reconnect = false
|
||
|
|
reconnector.disconnect()
|
||
|
|
})
|
||
|
|
|
||
|
|
var connected = false
|
||
|
|
var onConnect = function () {
|
||
|
|
connected = true
|
||
|
|
t.ok(reconnector.connected)
|
||
|
|
}
|
||
|
|
|
||
|
|
var onReconnect = function (n) {
|
||
|
|
t.notOk(connected)
|
||
|
|
t.notOk(reconnector.connected)
|
||
|
|
t.ok(reconnector.reconnect)
|
||
|
|
if (n < 4) process.nextTick(function () {
|
||
|
|
ee.emit('start server')
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
reconnector
|
||
|
|
.on('connect', onConnect)
|
||
|
|
.on('reconnect', onReconnect)
|
||
|
|
.connect()
|
||
|
|
})
|
||
|
|
|
||
|
|
test('this arg', function (t) {
|
||
|
|
var thisArg
|
||
|
|
var reconnect = inject(function () {
|
||
|
|
thisArg = this
|
||
|
|
return new Stream
|
||
|
|
})
|
||
|
|
var reconnector = reconnect()
|
||
|
|
reconnector.connect()
|
||
|
|
t.equal(thisArg, reconnector)
|
||
|
|
t.end()
|
||
|
|
})
|