fix: stream protocols not completing (#21759)

Co-authored-by: Jeremy Apthorp <nornagon@nornagon.net>
This commit is contained in:
trop[bot]
2020-01-14 16:09:46 +09:00
committed by Cheng Zhao
parent f47fbccf1a
commit 5a2c451c0b
2 changed files with 22 additions and 0 deletions

View File

@@ -102,6 +102,9 @@ void NodeStreamLoader::ReadMore() {
if (!ret.ToLocal(&buffer) || !node::Buffer::HasInstance(buffer)) {
readable_ = false;
is_reading_ = false;
if (ended_) {
NotifyComplete(result_);
}
return;
}

View File

@@ -392,6 +392,25 @@ describe('protocol module', () => {
const r = await ajax(protocolName + '://fake-host')
expect(r.data).to.have.lengthOf(data.length)
})
it('can handle a stream completing while writing', async () => {
function dumbPassthrough () {
return new stream.Transform({
async transform (chunk, encoding, cb) {
cb(null, chunk)
}
})
}
await registerStreamProtocol(protocolName, (request, callback) => {
callback({
statusCode: 200,
headers: { 'Content-Type': 'text/plain' },
data: getStream(1024 * 1024, Buffer.alloc(1024 * 1024 * 2)).pipe(dumbPassthrough())
})
})
const r = await ajax(protocolName + '://fake-host')
expect(r.data).to.have.lengthOf(1024 * 1024 * 2)
})
})
describe('protocol.isProtocolHandled', () => {