diff --git a/tests/browser-nodejs/browser.js b/tests/browser-nodejs/browser.js index 5688ee6..81fc72e 100644 --- a/tests/browser-nodejs/browser.js +++ b/tests/browser-nodejs/browser.js @@ -1,3 +1,4 @@ +'use strict' /* eslint-env mocha */ const expect = require('chai').expect @@ -25,4 +26,52 @@ describe('libp2p-websockets', function () { done() }) }) + + describe('stress', () => { + it('one big write', (done) => { + const mh = multiaddr('/ip4/127.0.0.1/tcp/9090/websockets') + const conn = ws.dial(mh) + const message = new Buffer(1000000).fill('a').toString('hex') + conn.write(message) + conn.on('data', (data) => { + expect(data.toString()).to.equal(message) + conn.end() + done() + }) + }) + + it('many writes in 2 batches', (done) => { + const mh = multiaddr('/ip4/127.0.0.1/tcp/9090/websockets') + const conn = ws.dial(mh) + let expected = '' + let counter = 0 + while (++counter < 10000) { + conn.write(`${counter} `) + expected += `${counter} ` + } + + setTimeout(() => { + while (++counter < 20000) { + conn.write(`${counter} `) + expected += `${counter} ` + } + + conn.write('STOP') + }, 1000) + + let result = '' + conn.on('data', (data) => { + if (data.toString() === 'STOP') { + conn.end() + return + } + result += data.toString() + }) + + conn.on('end', () => { + expect(result).to.equal(expected) + done() + }) + }) + }) })