yetangitu/OpenSTF
1
0
Fork 0
mirror of https://github.com/openstf/stf synced 2025-10-06 03:50:04 +02:00
Code Issues Releases Wiki Activity

Add a varint-delimiting stream as well.

Browse source
This commit is contained in:
Simo Kinnunen 2014-03-31 15:59:02 +09:00
parent 1c611232f5
commit 514fc3554d
2 changed files with 113 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

119
test/wire/messagestream.js
View file

@ -3,52 +3,91 @@ var chai = require('chai')
chai.use(require('sinon-chai'))
var expect = chai.expect
var MessageStream = require('../../lib/wire/messagestream')
var ms = require('../../lib/wire/messagestream')
describe('MessageStream', function() {
it('should emit complete varint32-delimited chunks', function() {
var ms = new MessageStream()
var spy = sinon.spy()
ms.on('data', spy)
ms.write(new Buffer([1, 0x61, 2, 0x62, 0x63]))
expect(spy).to.have.been.calledTwice
expect(spy.firstCall.args).to.eql([new Buffer([0x61])])
expect(spy.secondCall.args).to.eql([new Buffer([0x62, 0x63])])
describe('DelimitedStream', function() {
it('should emit complete varint-delimited chunks', function() {
var ds = new ms.DelimitedStream()
var spy = sinon.spy()
ds.on('data', spy)
ds.write(new Buffer([1, 0x61, 2, 0x62, 0x63]))
expect(spy).to.have.been.calledTwice
expect(spy.firstCall.args).to.eql([new Buffer([0x61])])
expect(spy.secondCall.args).to.eql([new Buffer([0x62, 0x63])])
})
it('should wait for more data', function() {
var ds = new ms.DelimitedStream()
var spy = sinon.spy()
ds.on('data', spy)
ds.write(new Buffer([1]))
expect(spy).to.not.have.been.called
ds.write(new Buffer([0x66]))
expect(spy).to.have.been.calledOnce
expect(spy.firstCall.args).to.eql([new Buffer([0x66])])
})
it('should read varint32 properly', function() {
var ds = new ms.DelimitedStream()
var spy = sinon.spy()
ds.on('data', spy)
ds.write(new Buffer([172, 2])) // 300
var data = new Buffer(300)
data.fill(0)
ds.write(data)
expect(spy).to.have.been.calledOnce
expect(spy.firstCall.args).to.eql([data])
})
it('should emit "end"', function(done) {
var ds = new ms.DelimitedStream()
var spy = sinon.spy()
ds.on('data', sinon.spy())
ds.on('end', spy)
ds.write(new Buffer([1]))
ds.end()
setImmediate(function() {
expect(spy).to.have.been.called
done()
})
})
})
it('should wait for more data', function() {
var ms = new MessageStream()
var spy = sinon.spy()
ms.on('data', spy)
ms.write(new Buffer([1]))
expect(spy).to.not.have.been.called
ms.write(new Buffer([0x66]))
expect(spy).to.have.been.calledOnce
expect(spy.firstCall.args).to.eql([new Buffer([0x66])])
})
describe('DelimitingStream', function() {
it('should add delimiter chunks to stream', function() {
var ds = new ms.DelimitingStream()
var spy = sinon.spy()
ds.on('data', spy)
ds.write(new Buffer([0x66, 0x6f, 0x6f]))
expect(spy).to.have.been.calledTwice
expect(spy.firstCall.args).to.eql([new Buffer([0x03])])
expect(spy.secondCall.args).to.eql([new Buffer([0x66, 0x6f, 0x6f])])
})
it('should read varint32 properly', function() {
var ms = new MessageStream
var spy = sinon.spy()
ms.on('data', spy)
ms.write(new Buffer([172, 2])) // 300
var data = new Buffer(300)
data.fill(0)
ms.write(data)
expect(spy).to.have.been.calledOnce
expect(spy.firstCall.args).to.eql([data])
})
it('should write proper varints', function() {
var ds = new ms.DelimitingStream()
var spy = sinon.spy()
ds.on('data', spy)
var data = new Buffer(300)
data.fill(0)
ds.write(data)
expect(spy).to.have.been.calledTwice
expect(spy.firstCall.args).to.eql([new Buffer([172, 2])])
expect(spy.secondCall.args).to.eql([data])
})
it('should emit "end"', function(done) {
var ms = new MessageStream()
var spy = sinon.spy()
ms.on('data', sinon.spy())
ms.on('end', spy)
ms.write(new Buffer([1]))
ms.end()
setImmediate(function() {
expect(spy).to.have.been.called
done()
it('should emit "end"', function(done) {
var ds = new ms.DelimitingStream()
var spy = sinon.spy()
ds.on('data', sinon.spy())
ds.on('end', spy)
ds.write(new Buffer([1]))
ds.end()
setImmediate(function() {
expect(spy).to.have.been.called
done()
})
})
})
})