mirror of
https://github.com/electron/electron.git
synced 2026-04-10 03:01:51 -04:00
fix: stream protocols not completing (#21759)
Co-authored-by: Jeremy Apthorp <nornagon@nornagon.net>
This commit is contained in:
@@ -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;
|
||||
}
|
||||
|
||||
|
||||
@@ -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', () => {
|
||||
|
||||
Reference in New Issue
Block a user