1352 lines
42 KiB
JavaScript
1352 lines
42 KiB
JavaScript
const assert = require('assert')
|
|
const {remote} = require('electron')
|
|
const {ipcRenderer} = require('electron')
|
|
const http = require('http')
|
|
const url = require('url')
|
|
const {net} = remote
|
|
const {session} = remote
|
|
|
|
function randomBuffer (size, start, end) {
|
|
start = start || 0
|
|
end = end || 255
|
|
let range = 1 + end - start
|
|
const buffer = Buffer.allocUnsafe(size)
|
|
for (let i = 0; i < size; ++i) {
|
|
buffer[i] = start + Math.floor(Math.random() * range)
|
|
}
|
|
return buffer
|
|
}
|
|
|
|
function randomString (length) {
|
|
let buffer = randomBuffer(length, '0'.charCodeAt(0), 'z'.charCodeAt(0))
|
|
return buffer.toString()
|
|
}
|
|
|
|
const kOneKiloByte = 1024
|
|
const kOneMegaByte = kOneKiloByte * kOneKiloByte
|
|
|
|
describe('net module', function () {
|
|
// this.timeout(0)
|
|
describe('HTTP basics', function () {
|
|
let server
|
|
beforeEach(function (done) {
|
|
server = http.createServer()
|
|
server.listen(0, '127.0.0.1', function () {
|
|
server.url = 'http://127.0.0.1:' + server.address().port
|
|
done()
|
|
})
|
|
})
|
|
|
|
afterEach(function () {
|
|
server.close(function () {
|
|
})
|
|
server = null
|
|
})
|
|
|
|
it('should be able to issue a basic GET request', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert.equal(request.method, 'GET')
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request(`${server.url}${requestUrl}`)
|
|
urlRequest.on('response', function (response) {
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should be able to issue a basic POST request', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert.equal(request.method, 'POST')
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request({
|
|
method: 'POST',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should fetch correct data in a GET request', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const bodyData = 'Hello World!'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert.equal(request.method, 'GET')
|
|
response.write(bodyData)
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request(`${server.url}${requestUrl}`)
|
|
urlRequest.on('response', function (response) {
|
|
let expectedBodyData = ''
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
expectedBodyData += chunk.toString()
|
|
})
|
|
response.on('end', function () {
|
|
assert.equal(expectedBodyData, bodyData)
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should post the correct data in a POST request', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const bodyData = 'Hello World!'
|
|
server.on('request', function (request, response) {
|
|
let postedBodyData = ''
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert.equal(request.method, 'POST')
|
|
request.on('data', function (chunk) {
|
|
postedBodyData += chunk.toString()
|
|
})
|
|
request.on('end', function () {
|
|
assert.equal(postedBodyData, bodyData)
|
|
response.end()
|
|
})
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request({
|
|
method: 'POST',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.write(bodyData)
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should support chunked encoding', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.chunkedEncoding = true
|
|
assert.equal(request.method, 'POST')
|
|
assert.equal(request.headers['transfer-encoding'], 'chunked')
|
|
assert(!request.headers['content-length'])
|
|
request.on('data', function (chunk) {
|
|
response.write(chunk)
|
|
})
|
|
request.on('end', function (chunk) {
|
|
response.end(chunk)
|
|
})
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request({
|
|
method: 'POST',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
|
|
let chunkIndex = 0
|
|
let chunkCount = 100
|
|
let sentChunks = []
|
|
let receivedChunks = []
|
|
urlRequest.on('response', function (response) {
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
receivedChunks.push(chunk)
|
|
})
|
|
response.on('end', function () {
|
|
let sentData = Buffer.concat(sentChunks)
|
|
let receivedData = Buffer.concat(receivedChunks)
|
|
assert.equal(sentData.toString(), receivedData.toString())
|
|
assert.equal(chunkIndex, chunkCount)
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.chunkedEncoding = true
|
|
while (chunkIndex < chunkCount) {
|
|
++chunkIndex
|
|
let chunk = randomBuffer(kOneKiloByte)
|
|
sentChunks.push(chunk)
|
|
assert(urlRequest.write(chunk))
|
|
}
|
|
urlRequest.end()
|
|
})
|
|
})
|
|
|
|
describe('ClientRequest API', function () {
|
|
let server
|
|
beforeEach(function (done) {
|
|
server = http.createServer()
|
|
server.listen(0, '127.0.0.1', function () {
|
|
server.url = 'http://127.0.0.1:' + server.address().port
|
|
done()
|
|
})
|
|
})
|
|
|
|
afterEach(function () {
|
|
server.close(function () {
|
|
})
|
|
server = null
|
|
session.defaultSession.webRequest.onBeforeRequest(null)
|
|
})
|
|
|
|
it('request/response objects should emit expected events', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
let bodyData = randomString(kOneMegaByte)
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.write(bodyData)
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
let requestResponseEventEmitted = false
|
|
let requestFinishEventEmitted = false
|
|
let requestCloseEventEmitted = false
|
|
let responseDataEventEmitted = false
|
|
let responseEndEventEmitted = false
|
|
|
|
function maybeDone (done) {
|
|
if (!requestCloseEventEmitted || !responseEndEventEmitted) {
|
|
return
|
|
}
|
|
|
|
assert(requestResponseEventEmitted)
|
|
assert(requestFinishEventEmitted)
|
|
assert(requestCloseEventEmitted)
|
|
assert(responseDataEventEmitted)
|
|
assert(responseEndEventEmitted)
|
|
done()
|
|
}
|
|
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
requestResponseEventEmitted = true
|
|
const statusCode = response.statusCode
|
|
assert.equal(statusCode, 200)
|
|
let buffers = []
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
buffers.push(chunk)
|
|
responseDataEventEmitted = true
|
|
})
|
|
response.on('end', function () {
|
|
let receivedBodyData = Buffer.concat(buffers)
|
|
assert(receivedBodyData.toString() === bodyData)
|
|
responseEndEventEmitted = true
|
|
maybeDone(done)
|
|
})
|
|
response.resume()
|
|
response.on('error', function (error) {
|
|
assert.ifError(error)
|
|
})
|
|
response.on('aborted', function () {
|
|
assert(false)
|
|
})
|
|
})
|
|
urlRequest.on('finish', function () {
|
|
requestFinishEventEmitted = true
|
|
})
|
|
urlRequest.on('error', function (error) {
|
|
assert.ifError(error)
|
|
})
|
|
urlRequest.on('abort', function () {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('close', function () {
|
|
requestCloseEventEmitted = true
|
|
maybeDone(done)
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should be able to set a custom HTTP request header before first write', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const customHeaderName = 'Some-Custom-Header-Name'
|
|
const customHeaderValue = 'Some-Customer-Header-Value'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert.equal(request.headers[customHeaderName.toLowerCase()],
|
|
customHeaderValue)
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
const statusCode = response.statusCode
|
|
assert.equal(statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.setHeader(customHeaderName, customHeaderValue)
|
|
assert.equal(urlRequest.getHeader(customHeaderName),
|
|
customHeaderValue)
|
|
assert.equal(urlRequest.getHeader(customHeaderName.toLowerCase()),
|
|
customHeaderValue)
|
|
urlRequest.write('')
|
|
assert.equal(urlRequest.getHeader(customHeaderName),
|
|
customHeaderValue)
|
|
assert.equal(urlRequest.getHeader(customHeaderName.toLowerCase()),
|
|
customHeaderValue)
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should not be able to set a custom HTTP request header after first write', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const customHeaderName = 'Some-Custom-Header-Name'
|
|
const customHeaderValue = 'Some-Customer-Header-Value'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert(!request.headers[customHeaderName.toLowerCase()])
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
const statusCode = response.statusCode
|
|
assert.equal(statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.write('')
|
|
assert.throws(() => {
|
|
urlRequest.setHeader(customHeaderName, customHeaderValue)
|
|
})
|
|
assert(!urlRequest.getHeader(customHeaderName))
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should be able to remove a custom HTTP request header before first write', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const customHeaderName = 'Some-Custom-Header-Name'
|
|
const customHeaderValue = 'Some-Customer-Header-Value'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert(!request.headers[customHeaderName.toLowerCase()])
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
const statusCode = response.statusCode
|
|
assert.equal(statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.setHeader(customHeaderName, customHeaderValue)
|
|
assert.equal(urlRequest.getHeader(customHeaderName),
|
|
customHeaderValue)
|
|
urlRequest.removeHeader(customHeaderName)
|
|
assert(!urlRequest.getHeader(customHeaderName))
|
|
urlRequest.write('')
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should not be able to remove a custom HTTP request header after first write', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const customHeaderName = 'Some-Custom-Header-Name'
|
|
const customHeaderValue = 'Some-Customer-Header-Value'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert.equal(request.headers[customHeaderName.toLowerCase()],
|
|
customHeaderValue)
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
const statusCode = response.statusCode
|
|
assert.equal(statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.setHeader(customHeaderName, customHeaderValue)
|
|
assert.equal(urlRequest.getHeader(customHeaderName),
|
|
customHeaderValue)
|
|
urlRequest.write('')
|
|
assert.throws(function () {
|
|
urlRequest.removeHeader(customHeaderName)
|
|
})
|
|
assert.equal(urlRequest.getHeader(customHeaderName),
|
|
customHeaderValue)
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should be able to abort an HTTP request before first write', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
server.on('request', function (request, response) {
|
|
assert(false)
|
|
})
|
|
|
|
let requestAbortEventEmitted = false
|
|
let requestCloseEventEmitted = false
|
|
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('finish', function () {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('error', function () {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('abort', function () {
|
|
requestAbortEventEmitted = true
|
|
})
|
|
urlRequest.on('close', function () {
|
|
requestCloseEventEmitted = true
|
|
assert(requestAbortEventEmitted)
|
|
assert(requestCloseEventEmitted)
|
|
done()
|
|
})
|
|
urlRequest.abort()
|
|
assert(!urlRequest.write(''))
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('it should be able to abort an HTTP request before request end', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
let requestReceivedByServer = false
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
requestReceivedByServer = true
|
|
cancelRequest()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
let requestAbortEventEmitted = false
|
|
let requestCloseEventEmitted = false
|
|
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('finish', function () {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('error', function () {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('abort', function () {
|
|
requestAbortEventEmitted = true
|
|
})
|
|
urlRequest.on('close', function () {
|
|
requestCloseEventEmitted = true
|
|
assert(requestReceivedByServer)
|
|
assert(requestAbortEventEmitted)
|
|
assert(requestCloseEventEmitted)
|
|
done()
|
|
})
|
|
|
|
urlRequest.chunkedEncoding = true
|
|
urlRequest.write(randomString(kOneKiloByte))
|
|
function cancelRequest () {
|
|
urlRequest.abort()
|
|
}
|
|
})
|
|
|
|
it('it should be able to abort an HTTP request after request end and before response', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
let requestReceivedByServer = false
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
requestReceivedByServer = true
|
|
cancelRequest()
|
|
process.nextTick(() => {
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.end()
|
|
})
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
let requestAbortEventEmitted = false
|
|
let requestFinishEventEmitted = false
|
|
let requestCloseEventEmitted = false
|
|
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('finish', function () {
|
|
requestFinishEventEmitted = true
|
|
})
|
|
urlRequest.on('error', function () {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('abort', function () {
|
|
requestAbortEventEmitted = true
|
|
})
|
|
urlRequest.on('close', function () {
|
|
requestCloseEventEmitted = true
|
|
assert(requestFinishEventEmitted)
|
|
assert(requestReceivedByServer)
|
|
assert(requestAbortEventEmitted)
|
|
assert(requestCloseEventEmitted)
|
|
done()
|
|
})
|
|
|
|
urlRequest.end(randomString(kOneKiloByte))
|
|
function cancelRequest () {
|
|
urlRequest.abort()
|
|
}
|
|
})
|
|
|
|
it('it should be able to abort an HTTP request after response start', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
let requestReceivedByServer = false
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
requestReceivedByServer = true
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.write(randomString(kOneKiloByte))
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
let requestFinishEventEmitted = false
|
|
let requestResponseEventEmitted = false
|
|
let requestAbortEventEmitted = false
|
|
let requestCloseEventEmitted = false
|
|
let responseAbortedEventEmitted = false
|
|
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
requestResponseEventEmitted = true
|
|
const statusCode = response.statusCode
|
|
assert.equal(statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
assert(false)
|
|
})
|
|
response.resume()
|
|
response.on('error', function () {
|
|
assert(false)
|
|
})
|
|
response.on('aborted', function () {
|
|
responseAbortedEventEmitted = true
|
|
})
|
|
urlRequest.abort()
|
|
})
|
|
urlRequest.on('finish', function () {
|
|
requestFinishEventEmitted = true
|
|
})
|
|
urlRequest.on('error', function () {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('abort', function () {
|
|
requestAbortEventEmitted = true
|
|
})
|
|
urlRequest.on('close', function () {
|
|
requestCloseEventEmitted = true
|
|
assert(requestFinishEventEmitted, 'request should emit "finish" event')
|
|
assert(requestReceivedByServer, 'request should be received by the server')
|
|
assert(requestResponseEventEmitted, '"response" event should be emitted')
|
|
assert(requestAbortEventEmitted, 'request should emit "abort" event')
|
|
assert(responseAbortedEventEmitted, 'response should emit "aborted" event')
|
|
assert(requestCloseEventEmitted, 'request should emit "close" event')
|
|
done()
|
|
})
|
|
urlRequest.end(randomString(kOneKiloByte))
|
|
})
|
|
|
|
it('abort event should be emitted at most once', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
let requestReceivedByServer = false
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
requestReceivedByServer = true
|
|
cancelRequest()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
let requestFinishEventEmitted = false
|
|
let requestAbortEventCount = 0
|
|
let requestCloseEventEmitted = false
|
|
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('finish', function () {
|
|
requestFinishEventEmitted = true
|
|
})
|
|
urlRequest.on('error', function () {
|
|
assert(false)
|
|
})
|
|
urlRequest.on('abort', function () {
|
|
++requestAbortEventCount
|
|
urlRequest.abort()
|
|
})
|
|
urlRequest.on('close', function () {
|
|
requestCloseEventEmitted = true
|
|
// Let all pending async events to be emitted
|
|
setTimeout(function () {
|
|
assert(requestFinishEventEmitted)
|
|
assert(requestReceivedByServer)
|
|
assert.equal(requestAbortEventCount, 1)
|
|
assert(requestCloseEventEmitted)
|
|
done()
|
|
}, 500)
|
|
})
|
|
|
|
urlRequest.end(randomString(kOneKiloByte))
|
|
function cancelRequest () {
|
|
urlRequest.abort()
|
|
urlRequest.abort()
|
|
}
|
|
})
|
|
|
|
it('Requests should be intercepted by webRequest module', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const redirectUrl = '/redirectUrl'
|
|
let requestIsRedirected = false
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert(false)
|
|
break
|
|
case redirectUrl:
|
|
requestIsRedirected = true
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
let requestIsIntercepted = false
|
|
session.defaultSession.webRequest.onBeforeRequest(
|
|
function (details, callback) {
|
|
if (details.url === `${server.url}${requestUrl}`) {
|
|
requestIsIntercepted = true
|
|
callback({
|
|
redirectURL: `${server.url}${redirectUrl}`
|
|
})
|
|
} else {
|
|
callback({
|
|
cancel: false
|
|
})
|
|
}
|
|
})
|
|
|
|
const urlRequest = net.request(`${server.url}${requestUrl}`)
|
|
|
|
urlRequest.on('response', function (response) {
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
assert(requestIsRedirected, 'The server should receive a request to the forward URL')
|
|
assert(requestIsIntercepted, 'The request should be intercepted by the webRequest module')
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should to able to create and intercept a request using a custom session object', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const redirectUrl = '/redirectUrl'
|
|
const customPartitionName = 'custom-partition'
|
|
let requestIsRedirected = false
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert(false)
|
|
break
|
|
case redirectUrl:
|
|
requestIsRedirected = true
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
session.defaultSession.webRequest.onBeforeRequest(
|
|
function (details, callback) {
|
|
assert(false, 'Request should not be intercepted by the default session')
|
|
})
|
|
|
|
let customSession = session.fromPartition(customPartitionName, {
|
|
cache: false
|
|
})
|
|
let requestIsIntercepted = false
|
|
customSession.webRequest.onBeforeRequest(
|
|
function (details, callback) {
|
|
if (details.url === `${server.url}${requestUrl}`) {
|
|
requestIsIntercepted = true
|
|
callback({
|
|
redirectURL: `${server.url}${redirectUrl}`
|
|
})
|
|
} else {
|
|
callback({
|
|
cancel: false
|
|
})
|
|
}
|
|
})
|
|
|
|
const urlRequest = net.request({
|
|
url: `${server.url}${requestUrl}`,
|
|
session: customSession
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
assert(requestIsRedirected, 'The server should receive a request to the forward URL')
|
|
assert(requestIsIntercepted, 'The request should be intercepted by the webRequest module')
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should throw if given an invalid session option', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
try {
|
|
const urlRequest = net.request({
|
|
url: `${server.url}${requestUrl}`,
|
|
session: 1
|
|
})
|
|
urlRequest
|
|
} catch (exception) {
|
|
done()
|
|
}
|
|
})
|
|
|
|
it('should to able to create and intercept a request using a custom partition name', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const redirectUrl = '/redirectUrl'
|
|
const customPartitionName = 'custom-partition'
|
|
let requestIsRedirected = false
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert(false)
|
|
break
|
|
case redirectUrl:
|
|
requestIsRedirected = true
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
session.defaultSession.webRequest.onBeforeRequest(
|
|
function (details, callback) {
|
|
assert(false, 'Request should not be intercepted by the default session')
|
|
})
|
|
|
|
let customSession = session.fromPartition(customPartitionName, {
|
|
cache: false
|
|
})
|
|
let requestIsIntercepted = false
|
|
customSession.webRequest.onBeforeRequest(
|
|
function (details, callback) {
|
|
if (details.url === `${server.url}${requestUrl}`) {
|
|
requestIsIntercepted = true
|
|
callback({
|
|
redirectURL: `${server.url}${redirectUrl}`
|
|
})
|
|
} else {
|
|
callback({
|
|
cancel: false
|
|
})
|
|
}
|
|
})
|
|
|
|
const urlRequest = net.request({
|
|
url: `${server.url}${requestUrl}`,
|
|
partition: customPartitionName
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
assert(requestIsRedirected, 'The server should receive a request to the forward URL')
|
|
assert(requestIsIntercepted, 'The request should be intercepted by the webRequest module')
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should throw if given an invalid partition option', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
try {
|
|
const urlRequest = net.request({
|
|
url: `${server.url}${requestUrl}`,
|
|
partition: 1
|
|
})
|
|
urlRequest
|
|
} catch (exception) {
|
|
done()
|
|
}
|
|
})
|
|
|
|
it('should be able to create a request with options', function (done) {
|
|
const requestUrl = '/'
|
|
const customHeaderName = 'Some-Custom-Header-Name'
|
|
const customHeaderValue = 'Some-Customer-Header-Value'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
assert.equal(request.method, 'GET')
|
|
assert.equal(request.headers[customHeaderName.toLowerCase()],
|
|
customHeaderValue)
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
const serverUrl = url.parse(server.url)
|
|
let options = {
|
|
port: serverUrl.port,
|
|
hostname: '127.0.0.1',
|
|
headers: {}
|
|
}
|
|
options.headers[customHeaderName] = customHeaderValue
|
|
const urlRequest = net.request(options)
|
|
urlRequest.on('response', function (response) {
|
|
assert.equal(response.statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should be able to pipe a readable stream into a net request', function (done) {
|
|
const nodeRequestUrl = '/nodeRequestUrl'
|
|
const netRequestUrl = '/netRequestUrl'
|
|
const bodyData = randomString(kOneMegaByte)
|
|
let netRequestReceived = false
|
|
let netRequestEnded = false
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case nodeRequestUrl:
|
|
response.write(bodyData)
|
|
response.end()
|
|
break
|
|
case netRequestUrl:
|
|
netRequestReceived = true
|
|
let receivedBodyData = ''
|
|
request.on('data', function (chunk) {
|
|
receivedBodyData += chunk.toString()
|
|
})
|
|
request.on('end', function (chunk) {
|
|
netRequestEnded = true
|
|
if (chunk) {
|
|
receivedBodyData += chunk.toString()
|
|
}
|
|
assert.equal(receivedBodyData, bodyData)
|
|
response.end()
|
|
})
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
|
|
let nodeRequest = http.request(`${server.url}${nodeRequestUrl}`)
|
|
nodeRequest.on('response', function (nodeResponse) {
|
|
const netRequest = net.request(`${server.url}${netRequestUrl}`)
|
|
netRequest.on('response', function (netResponse) {
|
|
assert.equal(netResponse.statusCode, 200)
|
|
netResponse.pause()
|
|
netResponse.on('data', function (chunk) {
|
|
})
|
|
netResponse.on('end', function () {
|
|
assert(netRequestReceived)
|
|
assert(netRequestEnded)
|
|
done()
|
|
})
|
|
netResponse.resume()
|
|
})
|
|
nodeResponse.pipe(netRequest)
|
|
})
|
|
nodeRequest.end()
|
|
})
|
|
|
|
it('should emit error event on server socket close', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
request.socket.destroy()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
let requestErrorEventEmitted = false
|
|
const urlRequest = net.request(`${server.url}${requestUrl}`)
|
|
urlRequest.on('error', function (error) {
|
|
assert(error)
|
|
requestErrorEventEmitted = true
|
|
})
|
|
urlRequest.on('close', function () {
|
|
assert(requestErrorEventEmitted)
|
|
done()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
})
|
|
describe('IncomingMessage API', function () {
|
|
let server
|
|
beforeEach(function (done) {
|
|
server = http.createServer()
|
|
server.listen(0, '127.0.0.1', function () {
|
|
server.url = 'http://127.0.0.1:' + server.address().port
|
|
done()
|
|
})
|
|
})
|
|
|
|
afterEach(function () {
|
|
server.close()
|
|
server = null
|
|
})
|
|
|
|
it('response object should implement the IncomingMessage API', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
const customHeaderName = 'Some-Custom-Header-Name'
|
|
const customHeaderValue = 'Some-Customer-Header-Value'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.setHeader(customHeaderName, customHeaderValue)
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
const statusCode = response.statusCode
|
|
assert(typeof statusCode === 'number')
|
|
assert.equal(statusCode, 200)
|
|
const statusMessage = response.statusMessage
|
|
assert(typeof statusMessage === 'string')
|
|
assert.equal(statusMessage, 'OK')
|
|
const headers = response.headers
|
|
assert(typeof headers === 'object')
|
|
assert.deepEqual(headers[customHeaderName.toLowerCase()],
|
|
[customHeaderValue])
|
|
const httpVersion = response.httpVersion
|
|
assert(typeof httpVersion === 'string')
|
|
assert(httpVersion.length > 0)
|
|
const httpVersionMajor = response.httpVersionMajor
|
|
assert(typeof httpVersionMajor === 'number')
|
|
assert(httpVersionMajor >= 1)
|
|
const httpVersionMinor = response.httpVersionMinor
|
|
assert(typeof httpVersionMinor === 'number')
|
|
assert(httpVersionMinor >= 0)
|
|
response.pause()
|
|
response.on('data', function (chunk) {
|
|
})
|
|
response.on('end', function () {
|
|
done()
|
|
})
|
|
response.resume()
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
|
|
it('should be able to pipe a net response into a writable stream', function (done) {
|
|
const nodeRequestUrl = '/nodeRequestUrl'
|
|
const netRequestUrl = '/netRequestUrl'
|
|
const bodyData = randomString(kOneMegaByte)
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case netRequestUrl:
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.write(bodyData)
|
|
response.end()
|
|
break
|
|
case nodeRequestUrl:
|
|
let receivedBodyData = ''
|
|
request.on('data', function (chunk) {
|
|
receivedBodyData += chunk.toString()
|
|
})
|
|
request.on('end', function (chunk) {
|
|
if (chunk) {
|
|
receivedBodyData += chunk.toString()
|
|
}
|
|
assert.equal(receivedBodyData, bodyData)
|
|
response.end()
|
|
})
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
ipcRenderer.once('api-net-spec-done', function () {
|
|
done()
|
|
})
|
|
// Execute below code directly within the browser context without
|
|
// using the remote module.
|
|
ipcRenderer.send('eval', `
|
|
const {net} = require('electron')
|
|
const http = require('http')
|
|
const netRequest = net.request('${server.url}${netRequestUrl}')
|
|
netRequest.on('response', function (netResponse) {
|
|
const serverUrl = url.parse('${server.url}')
|
|
const nodeOptions = {
|
|
method: 'POST',
|
|
path: '${nodeRequestUrl}',
|
|
port: serverUrl.port
|
|
}
|
|
let nodeRequest = http.request(nodeOptions)
|
|
nodeRequest.on('response', function (nodeResponse) {
|
|
nodeResponse.on('data', function (chunk) {
|
|
})
|
|
nodeResponse.on('end', function (chunk) {
|
|
event.sender.send('api-net-spec-done')
|
|
})
|
|
})
|
|
netResponse.pipe(nodeRequest)
|
|
})
|
|
netRequest.end()
|
|
`)
|
|
})
|
|
|
|
it('should not emit any event after close', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
let bodyData = randomString(kOneKiloByte)
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.write(bodyData)
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
let requestCloseEventEmitted = false
|
|
const urlRequest = net.request({
|
|
method: 'GET',
|
|
url: `${server.url}${requestUrl}`
|
|
})
|
|
urlRequest.on('response', function (response) {
|
|
assert(!requestCloseEventEmitted)
|
|
const statusCode = response.statusCode
|
|
assert.equal(statusCode, 200)
|
|
response.pause()
|
|
response.on('data', function () {
|
|
})
|
|
response.on('end', function () {
|
|
})
|
|
response.resume()
|
|
response.on('error', function () {
|
|
assert(!requestCloseEventEmitted)
|
|
})
|
|
response.on('aborted', function () {
|
|
assert(!requestCloseEventEmitted)
|
|
})
|
|
})
|
|
urlRequest.on('finish', function () {
|
|
assert(!requestCloseEventEmitted)
|
|
})
|
|
urlRequest.on('error', function () {
|
|
assert(!requestCloseEventEmitted)
|
|
})
|
|
urlRequest.on('abort', function () {
|
|
assert(!requestCloseEventEmitted)
|
|
})
|
|
urlRequest.on('close', function () {
|
|
requestCloseEventEmitted = true
|
|
// Wait so that all async events get scheduled.
|
|
setTimeout(function () {
|
|
done()
|
|
}, 100)
|
|
})
|
|
urlRequest.end()
|
|
})
|
|
})
|
|
describe('Stability and performance', function (done) {
|
|
let server
|
|
beforeEach(function (done) {
|
|
server = http.createServer()
|
|
server.listen(0, '127.0.0.1', function () {
|
|
server.url = 'http://127.0.0.1:' + server.address().port
|
|
done()
|
|
})
|
|
})
|
|
|
|
afterEach(function () {
|
|
server.close()
|
|
server = null
|
|
})
|
|
|
|
it('should free unreferenced, never-started request objects without crash', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
ipcRenderer.once('api-net-spec-done', function () {
|
|
done()
|
|
})
|
|
ipcRenderer.send('eval', `
|
|
const {net} = require('electron')
|
|
const urlRequest = net.request('${server.url}${requestUrl}')
|
|
process.nextTick(function () {
|
|
const v8Util = process.atomBinding('v8_util')
|
|
v8Util.requestGarbageCollectionForTesting()
|
|
event.sender.send('api-net-spec-done')
|
|
})
|
|
`)
|
|
})
|
|
it('should not collect on-going requests without crash', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.write(randomString(kOneKiloByte))
|
|
ipcRenderer.once('api-net-spec-resume', function () {
|
|
response.write(randomString(kOneKiloByte))
|
|
response.end()
|
|
})
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
ipcRenderer.once('api-net-spec-done', function () {
|
|
done()
|
|
})
|
|
// Execute below code directly within the browser context without
|
|
// using the remote module.
|
|
ipcRenderer.send('eval', `
|
|
const {net} = require('electron')
|
|
const urlRequest = net.request('${server.url}${requestUrl}')
|
|
urlRequest.on('response', function (response) {
|
|
response.on('data', function () {
|
|
})
|
|
response.on('end', function () {
|
|
event.sender.send('api-net-spec-done')
|
|
})
|
|
process.nextTick(function () {
|
|
// Trigger a garbage collection.
|
|
const v8Util = process.atomBinding('v8_util')
|
|
v8Util.requestGarbageCollectionForTesting()
|
|
event.sender.send('api-net-spec-resume')
|
|
})
|
|
})
|
|
urlRequest.end()
|
|
`)
|
|
})
|
|
it('should collect unreferenced, ended requests without crash', function (done) {
|
|
const requestUrl = '/requestUrl'
|
|
server.on('request', function (request, response) {
|
|
switch (request.url) {
|
|
case requestUrl:
|
|
response.statusCode = 200
|
|
response.statusMessage = 'OK'
|
|
response.end()
|
|
break
|
|
default:
|
|
assert(false)
|
|
}
|
|
})
|
|
ipcRenderer.once('api-net-spec-done', function () {
|
|
done()
|
|
})
|
|
ipcRenderer.send('eval', `
|
|
const {net} = require('electron')
|
|
const urlRequest = net.request('${server.url}${requestUrl}')
|
|
urlRequest.on('response', function (response) {
|
|
response.on('data', function () {
|
|
})
|
|
response.on('end', function () {
|
|
})
|
|
})
|
|
urlRequest.on('close', function () {
|
|
process.nextTick(function () {
|
|
const v8Util = process.atomBinding('v8_util')
|
|
v8Util.requestGarbageCollectionForTesting()
|
|
event.sender.send('api-net-spec-done')
|
|
})
|
|
})
|
|
urlRequest.end()
|
|
`)
|
|
})
|
|
})
|
|
})
|