793 строки
28 KiB
JavaScript
793 строки
28 KiB
JavaScript
const assert = require('assert')
|
|
const http = require('http')
|
|
const https = require('https')
|
|
const path = require('path')
|
|
const fs = require('fs')
|
|
const send = require('send')
|
|
const auth = require('basic-auth')
|
|
const {closeWindow} = require('./window-helpers')
|
|
|
|
const {ipcRenderer, remote} = require('electron')
|
|
const {ipcMain, session, BrowserWindow, net} = remote
|
|
|
|
/* The whole session API doesn't use standard callbacks */
|
|
/* eslint-disable standard/no-callback-literal */
|
|
|
|
describe('session module', () => {
|
|
let fixtures = path.resolve(__dirname, 'fixtures')
|
|
let w = null
|
|
let webview = null
|
|
const url = 'http://127.0.0.1'
|
|
|
|
beforeEach(() => {
|
|
w = new BrowserWindow({
|
|
show: false,
|
|
width: 400,
|
|
height: 400
|
|
})
|
|
})
|
|
|
|
afterEach(() => {
|
|
if (webview != null) {
|
|
if (!document.body.contains(webview)) {
|
|
document.body.appendChild(webview)
|
|
}
|
|
webview.remove()
|
|
}
|
|
|
|
return closeWindow(w).then(() => { w = null })
|
|
})
|
|
|
|
describe('session.defaultSession', () => {
|
|
it('returns the default session', () => {
|
|
assert.equal(session.defaultSession, session.fromPartition(''))
|
|
})
|
|
})
|
|
|
|
describe('session.fromPartition(partition, options)', () => {
|
|
it('returns existing session with same partition', () => {
|
|
assert.equal(session.fromPartition('test'), session.fromPartition('test'))
|
|
})
|
|
|
|
it('created session is ref-counted', () => {
|
|
const partition = 'test2'
|
|
const userAgent = 'test-agent'
|
|
const ses1 = session.fromPartition(partition)
|
|
ses1.setUserAgent(userAgent)
|
|
assert.equal(ses1.getUserAgent(), userAgent)
|
|
ses1.destroy()
|
|
const ses2 = session.fromPartition(partition)
|
|
assert.notEqual(ses2.getUserAgent(), userAgent)
|
|
})
|
|
})
|
|
|
|
describe('ses.cookies', () => {
|
|
it('should get cookies', (done) => {
|
|
const server = http.createServer((req, res) => {
|
|
res.setHeader('Set-Cookie', ['0=0'])
|
|
res.end('finished')
|
|
server.close()
|
|
})
|
|
server.listen(0, '127.0.0.1', () => {
|
|
const port = server.address().port
|
|
w.loadURL(`${url}:${port}`)
|
|
w.webContents.on('did-finish-load', () => {
|
|
w.webContents.session.cookies.get({url}, (error, list) => {
|
|
if (error) return done(error)
|
|
for (let i = 0; i < list.length; i++) {
|
|
const cookie = list[i]
|
|
if (cookie.name === '0') {
|
|
if (cookie.value === '0') {
|
|
return done()
|
|
} else {
|
|
return done(`cookie value is ${cookie.value} while expecting 0`)
|
|
}
|
|
}
|
|
}
|
|
done('Can\'t find cookie')
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
it('calls back with an error when setting a cookie with missing required fields', (done) => {
|
|
session.defaultSession.cookies.set({
|
|
url: '',
|
|
name: '1',
|
|
value: '1'
|
|
}, (error) => {
|
|
assert.equal(error.message, 'Setting cookie failed')
|
|
done()
|
|
})
|
|
})
|
|
|
|
it('should over-write the existent cookie', (done) => {
|
|
session.defaultSession.cookies.set({
|
|
url,
|
|
name: '1',
|
|
value: '1'
|
|
}, (error) => {
|
|
if (error) return done(error)
|
|
session.defaultSession.cookies.get({url}, (error, list) => {
|
|
if (error) return done(error)
|
|
for (let i = 0; i < list.length; i++) {
|
|
const cookie = list[i]
|
|
if (cookie.name === '1') {
|
|
if (cookie.value === '1') {
|
|
return done()
|
|
} else {
|
|
return done(`cookie value is ${cookie.value} while expecting 1`)
|
|
}
|
|
}
|
|
}
|
|
done('Can\'t find cookie')
|
|
})
|
|
})
|
|
})
|
|
|
|
it('should remove cookies', (done) => {
|
|
session.defaultSession.cookies.set({
|
|
url: url,
|
|
name: '2',
|
|
value: '2'
|
|
}, (error) => {
|
|
if (error) return done(error)
|
|
session.defaultSession.cookies.remove(url, '2', () => {
|
|
session.defaultSession.cookies.get({url}, (error, list) => {
|
|
if (error) return done(error)
|
|
for (let i = 0; i < list.length; i++) {
|
|
const cookie = list[i]
|
|
if (cookie.name === '2') return done('Cookie not deleted')
|
|
}
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
it('should set cookie for standard scheme', (done) => {
|
|
const standardScheme = remote.getGlobal('standardScheme')
|
|
const origin = standardScheme + '://fake-host'
|
|
session.defaultSession.cookies.set({
|
|
url: origin,
|
|
name: 'custom',
|
|
value: '1'
|
|
}, (error) => {
|
|
if (error) return done(error)
|
|
session.defaultSession.cookies.get({url: origin}, (error, list) => {
|
|
if (error) return done(error)
|
|
assert.equal(list.length, 1)
|
|
assert.equal(list[0].name, 'custom')
|
|
assert.equal(list[0].value, '1')
|
|
assert.equal(list[0].domain, 'fake-host')
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
|
|
it('emits a changed event when a cookie is added or removed', (done) => {
|
|
const {cookies} = session.fromPartition('cookies-changed')
|
|
|
|
cookies.once('changed', (event, cookie, cause, removed) => {
|
|
assert.equal(cookie.name, 'foo')
|
|
assert.equal(cookie.value, 'bar')
|
|
assert.equal(cause, 'explicit')
|
|
assert.equal(removed, false)
|
|
|
|
cookies.once('changed', (event, cookie, cause, removed) => {
|
|
assert.equal(cookie.name, 'foo')
|
|
assert.equal(cookie.value, 'bar')
|
|
assert.equal(cause, 'explicit')
|
|
assert.equal(removed, true)
|
|
done()
|
|
})
|
|
|
|
cookies.remove(url, 'foo', (error) => {
|
|
if (error) return done(error)
|
|
})
|
|
})
|
|
|
|
cookies.set({
|
|
url: url,
|
|
name: 'foo',
|
|
value: 'bar'
|
|
}, (error) => {
|
|
if (error) return done(error)
|
|
})
|
|
})
|
|
|
|
describe('ses.cookies.flushStore(callback)', () => {
|
|
it('flushes the cookies to disk and invokes the callback when done', (done) => {
|
|
session.defaultSession.cookies.set({
|
|
url: url,
|
|
name: 'foo',
|
|
value: 'bar'
|
|
}, (error) => {
|
|
if (error) return done(error)
|
|
session.defaultSession.cookies.flushStore(() => {
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('ses.clearStorageData(options)', () => {
|
|
fixtures = path.resolve(__dirname, 'fixtures')
|
|
it('clears localstorage data', (done) => {
|
|
ipcMain.on('count', (event, count) => {
|
|
ipcMain.removeAllListeners('count')
|
|
assert.equal(count, 0)
|
|
done()
|
|
})
|
|
w.loadURL('file://' + path.join(fixtures, 'api', 'localstorage.html'))
|
|
w.webContents.on('did-finish-load', () => {
|
|
const options = {
|
|
origin: 'file://',
|
|
storages: ['localstorage'],
|
|
quotas: ['persistent']
|
|
}
|
|
w.webContents.session.clearStorageData(options, () => {
|
|
w.webContents.send('getcount')
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('will-download event', () => {
|
|
beforeEach(() => {
|
|
if (w != null) w.destroy()
|
|
w = new BrowserWindow({
|
|
show: false,
|
|
width: 400,
|
|
height: 400
|
|
})
|
|
})
|
|
|
|
it('can cancel default download behavior', (done) => {
|
|
const mockFile = Buffer.alloc(1024)
|
|
const contentDisposition = 'inline; filename="mockFile.txt"'
|
|
const downloadServer = http.createServer((req, res) => {
|
|
res.writeHead(200, {
|
|
'Content-Length': mockFile.length,
|
|
'Content-Type': 'application/plain',
|
|
'Content-Disposition': contentDisposition
|
|
})
|
|
res.end(mockFile)
|
|
downloadServer.close()
|
|
})
|
|
|
|
downloadServer.listen(0, '127.0.0.1', () => {
|
|
const port = downloadServer.address().port
|
|
const url = `http://127.0.0.1:${port}/`
|
|
|
|
ipcRenderer.sendSync('set-download-option', false, true)
|
|
w.loadURL(url)
|
|
ipcRenderer.once('download-error', (event, downloadUrl, filename, error) => {
|
|
assert.equal(downloadUrl, url)
|
|
assert.equal(filename, 'mockFile.txt')
|
|
assert.equal(error, 'Object has been destroyed')
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('DownloadItem', () => {
|
|
const mockPDF = Buffer.alloc(1024 * 1024 * 5)
|
|
const protocolName = 'custom-dl'
|
|
let contentDisposition = 'inline; filename="mock.pdf"'
|
|
const downloadFilePath = path.join(fixtures, 'mock.pdf')
|
|
const downloadServer = http.createServer((req, res) => {
|
|
if (req.url === '/?testFilename') contentDisposition = 'inline'
|
|
res.writeHead(200, {
|
|
'Content-Length': mockPDF.length,
|
|
'Content-Type': 'application/pdf',
|
|
'Content-Disposition': contentDisposition
|
|
})
|
|
res.end(mockPDF)
|
|
downloadServer.close()
|
|
})
|
|
const assertDownload = (event, state, url, mimeType,
|
|
receivedBytes, totalBytes, disposition,
|
|
filename, port, savePath, isCustom) => {
|
|
assert.equal(state, 'completed')
|
|
assert.equal(filename, 'mock.pdf')
|
|
assert.equal(savePath, path.join(__dirname, 'fixtures', 'mock.pdf'))
|
|
if (isCustom) {
|
|
assert.equal(url, `${protocolName}://item`)
|
|
} else {
|
|
assert.equal(url, `http://127.0.0.1:${port}/`)
|
|
}
|
|
assert.equal(mimeType, 'application/pdf')
|
|
assert.equal(receivedBytes, mockPDF.length)
|
|
assert.equal(totalBytes, mockPDF.length)
|
|
assert.equal(disposition, contentDisposition)
|
|
assert(fs.existsSync(downloadFilePath))
|
|
fs.unlinkSync(downloadFilePath)
|
|
}
|
|
|
|
it('can download using WebContents.downloadURL', (done) => {
|
|
downloadServer.listen(0, '127.0.0.1', () => {
|
|
const port = downloadServer.address().port
|
|
ipcRenderer.sendSync('set-download-option', false, false)
|
|
w.webContents.downloadURL(`${url}:${port}`)
|
|
ipcRenderer.once('download-done', (event, state, url,
|
|
mimeType, receivedBytes,
|
|
totalBytes, disposition,
|
|
filename, savePath) => {
|
|
assertDownload(event, state, url, mimeType, receivedBytes,
|
|
totalBytes, disposition, filename, port, savePath)
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
|
|
it('can download from custom protocols using WebContents.downloadURL', (done) => {
|
|
const protocol = session.defaultSession.protocol
|
|
downloadServer.listen(0, '127.0.0.1', () => {
|
|
const port = downloadServer.address().port
|
|
const handler = (ignoredError, callback) => {
|
|
callback({url: `${url}:${port}`})
|
|
}
|
|
protocol.registerHttpProtocol(protocolName, handler, (error) => {
|
|
if (error) return done(error)
|
|
ipcRenderer.sendSync('set-download-option', false, false)
|
|
w.webContents.downloadURL(`${protocolName}://item`)
|
|
ipcRenderer.once('download-done', (event, state, url,
|
|
mimeType, receivedBytes,
|
|
totalBytes, disposition,
|
|
filename, savePath) => {
|
|
assertDownload(event, state, url, mimeType, receivedBytes,
|
|
totalBytes, disposition, filename, port, savePath,
|
|
true)
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
it('can download using WebView.downloadURL', (done) => {
|
|
downloadServer.listen(0, '127.0.0.1', () => {
|
|
const port = downloadServer.address().port
|
|
ipcRenderer.sendSync('set-download-option', false, false)
|
|
webview = new WebView()
|
|
webview.src = `file://${fixtures}/api/blank.html`
|
|
webview.addEventListener('did-finish-load', () => {
|
|
webview.downloadURL(`${url}:${port}/`)
|
|
})
|
|
ipcRenderer.once('download-done', (event, state, url,
|
|
mimeType, receivedBytes,
|
|
totalBytes, disposition,
|
|
filename, savePath) => {
|
|
assertDownload(event, state, url, mimeType, receivedBytes,
|
|
totalBytes, disposition, filename, port, savePath)
|
|
document.body.removeChild(webview)
|
|
done()
|
|
})
|
|
document.body.appendChild(webview)
|
|
})
|
|
})
|
|
|
|
it('can cancel download', (done) => {
|
|
downloadServer.listen(0, '127.0.0.1', () => {
|
|
const port = downloadServer.address().port
|
|
ipcRenderer.sendSync('set-download-option', true, false)
|
|
w.webContents.downloadURL(`${url}:${port}/`)
|
|
ipcRenderer.once('download-done', (event, state, url,
|
|
mimeType, receivedBytes,
|
|
totalBytes, disposition,
|
|
filename) => {
|
|
assert.equal(state, 'cancelled')
|
|
assert.equal(filename, 'mock.pdf')
|
|
assert.equal(mimeType, 'application/pdf')
|
|
assert.equal(receivedBytes, 0)
|
|
assert.equal(totalBytes, mockPDF.length)
|
|
assert.equal(disposition, contentDisposition)
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
|
|
it('can generate a default filename', function (done) {
|
|
if (process.env.APPVEYOR === 'True') {
|
|
// FIXME(alexeykuzmin): Skip the test.
|
|
// this.skip()
|
|
return done()
|
|
}
|
|
|
|
downloadServer.listen(0, '127.0.0.1', () => {
|
|
const port = downloadServer.address().port
|
|
ipcRenderer.sendSync('set-download-option', true, false)
|
|
w.webContents.downloadURL(`${url}:${port}/?testFilename`)
|
|
ipcRenderer.once('download-done', (event, state, url,
|
|
mimeType, receivedBytes,
|
|
totalBytes, disposition,
|
|
filename) => {
|
|
assert.equal(state, 'cancelled')
|
|
assert.equal(filename, 'download.pdf')
|
|
assert.equal(mimeType, 'application/pdf')
|
|
assert.equal(receivedBytes, 0)
|
|
assert.equal(totalBytes, mockPDF.length)
|
|
assert.equal(disposition, contentDisposition)
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('when a save path is specified and the URL is unavailable', () => {
|
|
it('does not display a save dialog and reports the done state as interrupted', (done) => {
|
|
ipcRenderer.sendSync('set-download-option', false, false)
|
|
ipcRenderer.once('download-done', (event, state) => {
|
|
assert.equal(state, 'interrupted')
|
|
done()
|
|
})
|
|
w.webContents.downloadURL(`file://${path.join(__dirname, 'does-not-exist.txt')}`)
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('ses.protocol', () => {
|
|
const partitionName = 'temp'
|
|
const protocolName = 'sp'
|
|
const partitionProtocol = session.fromPartition(partitionName).protocol
|
|
const protocol = session.defaultSession.protocol
|
|
const handler = (ignoredError, callback) => {
|
|
callback({data: 'test', mimeType: 'text/html'})
|
|
}
|
|
|
|
beforeEach((done) => {
|
|
if (w != null) w.destroy()
|
|
w = new BrowserWindow({
|
|
show: false,
|
|
webPreferences: {
|
|
partition: partitionName
|
|
}
|
|
})
|
|
partitionProtocol.registerStringProtocol(protocolName, handler, (error) => {
|
|
done(error != null ? error : undefined)
|
|
})
|
|
})
|
|
|
|
afterEach((done) => {
|
|
partitionProtocol.unregisterProtocol(protocolName, () => done())
|
|
})
|
|
|
|
it('does not affect defaultSession', (done) => {
|
|
protocol.isProtocolHandled(protocolName, (result) => {
|
|
assert.equal(result, false)
|
|
partitionProtocol.isProtocolHandled(protocolName, (result) => {
|
|
assert.equal(result, true)
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
|
|
xit('handles requests from partition', (done) => {
|
|
w.webContents.on('did-finish-load', () => done())
|
|
w.loadURL(`${protocolName}://fake-host`)
|
|
})
|
|
})
|
|
|
|
describe('ses.setProxy(options, callback)', () => {
|
|
it('allows configuring proxy settings', (done) => {
|
|
const config = {proxyRules: 'http=myproxy:80'}
|
|
session.defaultSession.setProxy(config, () => {
|
|
session.defaultSession.resolveProxy('http://localhost', (proxy) => {
|
|
assert.equal(proxy, 'PROXY myproxy:80')
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
|
|
it('allows bypassing proxy settings', (done) => {
|
|
const config = {
|
|
proxyRules: 'http=myproxy:80',
|
|
proxyBypassRules: '<local>'
|
|
}
|
|
session.defaultSession.setProxy(config, () => {
|
|
session.defaultSession.resolveProxy('http://localhost', (proxy) => {
|
|
assert.equal(proxy, 'DIRECT')
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('ses.getBlobData(identifier, callback)', () => {
|
|
it('returns blob data for uuid', (done) => {
|
|
const scheme = 'temp'
|
|
const protocol = session.defaultSession.protocol
|
|
const url = `${scheme}://host`
|
|
before(() => {
|
|
if (w != null) w.destroy()
|
|
w = new BrowserWindow({show: false})
|
|
})
|
|
|
|
after((done) => {
|
|
protocol.unregisterProtocol(scheme, () => {
|
|
closeWindow(w).then(() => {
|
|
w = null
|
|
done()
|
|
})
|
|
})
|
|
})
|
|
|
|
const postData = JSON.stringify({
|
|
type: 'blob',
|
|
value: 'hello'
|
|
})
|
|
const content = `<html>
|
|
<script>
|
|
const {webFrame} = require('electron')
|
|
webFrame.registerURLSchemeAsPrivileged('${scheme}')
|
|
let fd = new FormData();
|
|
fd.append('file', new Blob(['${postData}'], {type:'application/json'}));
|
|
fetch('${url}', {method:'POST', body: fd });
|
|
</script>
|
|
</html>`
|
|
|
|
protocol.registerStringProtocol(scheme, (request, callback) => {
|
|
if (request.method === 'GET') {
|
|
callback({data: content, mimeType: 'text/html'})
|
|
} else if (request.method === 'POST') {
|
|
let uuid = request.uploadData[1].blobUUID
|
|
assert(uuid)
|
|
session.defaultSession.getBlobData(uuid, (result) => {
|
|
assert.equal(result.toString(), postData)
|
|
done()
|
|
})
|
|
}
|
|
}, (error) => {
|
|
if (error) return done(error)
|
|
w.loadURL(url)
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('ses.setCertificateVerifyProc(callback)', () => {
|
|
let server = null
|
|
|
|
beforeEach((done) => {
|
|
const certPath = path.join(__dirname, 'fixtures', 'certificates')
|
|
const options = {
|
|
key: fs.readFileSync(path.join(certPath, 'server.key')),
|
|
cert: fs.readFileSync(path.join(certPath, 'server.pem')),
|
|
ca: [
|
|
fs.readFileSync(path.join(certPath, 'rootCA.pem')),
|
|
fs.readFileSync(path.join(certPath, 'intermediateCA.pem'))
|
|
],
|
|
requestCert: true,
|
|
rejectUnauthorized: false
|
|
}
|
|
|
|
server = https.createServer(options, (req, res) => {
|
|
res.writeHead(200)
|
|
res.end('<title>hello</title>')
|
|
})
|
|
server.listen(0, '127.0.0.1', done)
|
|
})
|
|
|
|
afterEach(() => {
|
|
session.defaultSession.setCertificateVerifyProc(null)
|
|
server.close()
|
|
})
|
|
|
|
it('accepts the request when the callback is called with 0', (done) => {
|
|
session.defaultSession.setCertificateVerifyProc(({hostname, certificate, verificationResult, errorCode}, callback) => {
|
|
assert(['net::ERR_CERT_AUTHORITY_INVALID', 'net::ERR_CERT_COMMON_NAME_INVALID'].includes(verificationResult), verificationResult)
|
|
assert([-202, -200].includes(errorCode), errorCode)
|
|
callback(0)
|
|
})
|
|
|
|
w.webContents.once('did-finish-load', () => {
|
|
assert.equal(w.webContents.getTitle(), 'hello')
|
|
done()
|
|
})
|
|
w.loadURL(`https://127.0.0.1:${server.address().port}`)
|
|
})
|
|
|
|
describe('deprecated function signature', () => {
|
|
it('supports accepting the request', (done) => {
|
|
session.defaultSession.setCertificateVerifyProc(({hostname, certificate}, callback) => {
|
|
assert.equal(hostname, '127.0.0.1')
|
|
callback(true)
|
|
})
|
|
|
|
w.webContents.once('did-finish-load', () => {
|
|
assert.equal(w.webContents.getTitle(), 'hello')
|
|
done()
|
|
})
|
|
w.loadURL(`https://127.0.0.1:${server.address().port}`)
|
|
})
|
|
|
|
it('supports rejecting the request', (done) => {
|
|
session.defaultSession.setCertificateVerifyProc(({hostname, certificate}, callback) => {
|
|
assert.equal(hostname, '127.0.0.1')
|
|
callback(false)
|
|
})
|
|
|
|
const url = `https://127.0.0.1:${server.address().port}`
|
|
w.webContents.once('did-finish-load', () => {
|
|
assert.equal(w.webContents.getTitle(), url)
|
|
done()
|
|
})
|
|
w.loadURL(url)
|
|
})
|
|
})
|
|
|
|
it('rejects the request when the callback is called with -2', (done) => {
|
|
session.defaultSession.setCertificateVerifyProc(({hostname, certificate, verificationResult}, callback) => {
|
|
assert.equal(hostname, '127.0.0.1')
|
|
assert.equal(certificate.issuerName, 'Intermediate CA')
|
|
assert.equal(certificate.subjectName, 'localhost')
|
|
assert.equal(certificate.issuer.commonName, 'Intermediate CA')
|
|
assert.equal(certificate.subject.commonName, 'localhost')
|
|
assert.equal(certificate.issuerCert.issuer.commonName, 'Root CA')
|
|
assert.equal(certificate.issuerCert.subject.commonName, 'Intermediate CA')
|
|
assert.equal(certificate.issuerCert.issuerCert.issuer.commonName, 'Root CA')
|
|
assert.equal(certificate.issuerCert.issuerCert.subject.commonName, 'Root CA')
|
|
assert.equal(certificate.issuerCert.issuerCert.issuerCert, undefined)
|
|
assert(['net::ERR_CERT_AUTHORITY_INVALID', 'net::ERR_CERT_COMMON_NAME_INVALID'].includes(verificationResult), verificationResult)
|
|
callback(-2)
|
|
})
|
|
|
|
const url = `https://127.0.0.1:${server.address().port}`
|
|
w.webContents.once('did-finish-load', () => {
|
|
assert.equal(w.webContents.getTitle(), url)
|
|
done()
|
|
})
|
|
w.loadURL(url)
|
|
})
|
|
})
|
|
|
|
describe('ses.createInterruptedDownload(options)', () => {
|
|
it('can create an interrupted download item', (done) => {
|
|
ipcRenderer.sendSync('set-download-option', true, false)
|
|
const filePath = path.join(__dirname, 'fixtures', 'mock.pdf')
|
|
const options = {
|
|
path: filePath,
|
|
urlChain: ['http://127.0.0.1/'],
|
|
mimeType: 'application/pdf',
|
|
offset: 0,
|
|
length: 5242880
|
|
}
|
|
w.webContents.session.createInterruptedDownload(options)
|
|
ipcRenderer.once('download-created', (event, state, urlChain,
|
|
mimeType, receivedBytes,
|
|
totalBytes, filename,
|
|
savePath) => {
|
|
assert.equal(state, 'interrupted')
|
|
assert.deepEqual(urlChain, ['http://127.0.0.1/'])
|
|
assert.equal(mimeType, 'application/pdf')
|
|
assert.equal(receivedBytes, 0)
|
|
assert.equal(totalBytes, 5242880)
|
|
assert.equal(savePath, filePath)
|
|
done()
|
|
})
|
|
})
|
|
|
|
it('can be resumed', (done) => {
|
|
const fixtures = path.join(__dirname, 'fixtures')
|
|
const downloadFilePath = path.join(fixtures, 'logo.png')
|
|
const rangeServer = http.createServer((req, res) => {
|
|
let options = { root: fixtures }
|
|
send(req, req.url, options)
|
|
.on('error', (error) => { done(error) }).pipe(res)
|
|
})
|
|
ipcRenderer.sendSync('set-download-option', true, false, downloadFilePath)
|
|
rangeServer.listen(0, '127.0.0.1', () => {
|
|
const port = rangeServer.address().port
|
|
const downloadUrl = `http://127.0.0.1:${port}/assets/logo.png`
|
|
const callback = (event, state, url, mimeType,
|
|
receivedBytes, totalBytes, disposition,
|
|
filename, savePath, urlChain,
|
|
lastModifiedTime, eTag) => {
|
|
if (state === 'cancelled') {
|
|
const options = {
|
|
path: savePath,
|
|
urlChain: urlChain,
|
|
mimeType: mimeType,
|
|
offset: receivedBytes,
|
|
length: totalBytes,
|
|
lastModified: lastModifiedTime,
|
|
eTag: eTag
|
|
}
|
|
ipcRenderer.sendSync('set-download-option', false, false, downloadFilePath)
|
|
w.webContents.session.createInterruptedDownload(options)
|
|
} else {
|
|
assert.equal(state, 'completed')
|
|
assert.equal(filename, 'logo.png')
|
|
assert.equal(savePath, downloadFilePath)
|
|
assert.equal(url, downloadUrl)
|
|
assert.equal(mimeType, 'image/png')
|
|
assert.equal(receivedBytes, 14022)
|
|
assert.equal(totalBytes, 14022)
|
|
assert(fs.existsSync(downloadFilePath))
|
|
fs.unlinkSync(downloadFilePath)
|
|
rangeServer.close()
|
|
ipcRenderer.removeListener('download-done', callback)
|
|
done()
|
|
}
|
|
}
|
|
ipcRenderer.on('download-done', callback)
|
|
w.webContents.downloadURL(downloadUrl)
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('ses.clearAuthCache(options[, callback])', () => {
|
|
it('can clear http auth info from cache', (done) => {
|
|
const ses = session.fromPartition('auth-cache')
|
|
const server = http.createServer((req, res) => {
|
|
const credentials = auth(req)
|
|
if (!credentials || credentials.name !== 'test' || credentials.pass !== 'test') {
|
|
res.statusCode = 401
|
|
res.setHeader('WWW-Authenticate', 'Basic realm="Restricted"')
|
|
res.end()
|
|
} else {
|
|
res.end('authenticated')
|
|
}
|
|
})
|
|
server.listen(0, '127.0.0.1', () => {
|
|
const port = server.address().port
|
|
function issueLoginRequest (attempt = 1) {
|
|
if (attempt > 2) {
|
|
server.close()
|
|
return done()
|
|
}
|
|
const request = net.request({
|
|
url: `http://127.0.0.1:${port}`,
|
|
session: ses
|
|
})
|
|
request.on('login', (info, callback) => {
|
|
attempt += 1
|
|
assert.equal(info.scheme, 'basic')
|
|
assert.equal(info.realm, 'Restricted')
|
|
callback('test', 'test')
|
|
})
|
|
request.on('response', (response) => {
|
|
let data = ''
|
|
response.pause()
|
|
response.on('data', (chunk) => {
|
|
data += chunk
|
|
})
|
|
response.on('end', () => {
|
|
assert.equal(data, 'authenticated')
|
|
ses.clearAuthCache({type: 'password'}, () => {
|
|
issueLoginRequest(attempt)
|
|
})
|
|
})
|
|
response.on('error', (error) => { done(error) })
|
|
response.resume()
|
|
})
|
|
// Internal api to bypass cache for testing.
|
|
request.urlRequest._setLoadFlags(1 << 1)
|
|
request.end()
|
|
}
|
|
issueLoginRequest()
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('ses.setPermissionRequestHandler(handler)', () => {
|
|
it('cancels any pending requests when cleared', (done) => {
|
|
const ses = session.fromPartition('permissionTest')
|
|
ses.setPermissionRequestHandler(() => {
|
|
ses.setPermissionRequestHandler(null)
|
|
})
|
|
|
|
webview = new WebView()
|
|
webview.addEventListener('ipc-message', (e) => {
|
|
assert.equal(e.channel, 'message')
|
|
assert.deepEqual(e.args, ['SecurityError'])
|
|
done()
|
|
})
|
|
webview.src = `file://${fixtures}/pages/permissions/midi-sysex.html`
|
|
webview.partition = 'permissionTest'
|
|
webview.setAttribute('nodeintegration', 'on')
|
|
document.body.appendChild(webview)
|
|
})
|
|
})
|
|
})
|