electron/spec-main/api-net-log-spec.js

151 lines
4.1 KiB
JavaScript
Raw Normal View History

2018-06-28 22:40:30 +00:00
const chai = require('chai')
const dirtyChai = require('dirty-chai')
const http = require('http')
const fs = require('fs')
const os = require('os')
const path = require('path')
const ChildProcess = require('child_process')
const {session} = require('electron')
const appPath = path.join(__dirname, 'fixtures', 'api', 'net-log')
const dumpFile = path.join(os.tmpdir(), 'net_log.json')
const dumpFileDynamic = path.join(os.tmpdir(), 'net_log_dynamic.json')
2018-09-13 16:10:51 +00:00
const { expect } = chai
2018-06-28 22:40:30 +00:00
chai.use(dirtyChai)
const isCI = global.isCI
const testNetLog = () => session.fromPartition('net-log').netLog
describe('netLog module', () => {
let server
const connections = new Set()
2018-06-28 22:40:30 +00:00
before(done => {
server = http.createServer()
server.listen(0, '127.0.0.1', () => {
server.url = `http://127.0.0.1:${server.address().port}`
done()
})
server.on('connection', (connection) => {
connections.add(connection)
connection.once('close', () => {
connections.delete(connection)
})
})
server.on('request', (request, response) => {
response.end()
})
})
2018-06-28 22:40:30 +00:00
after(done => {
for (const connection of connections) {
connection.destroy()
}
server.close(() => {
server = null
done()
})
})
beforeEach(() => {
expect(testNetLog().currentlyLogging).to.be.false()
})
afterEach(() => {
try {
if (fs.existsSync(dumpFile)) {
fs.unlinkSync(dumpFile)
}
if (fs.existsSync(dumpFileDynamic)) {
fs.unlinkSync(dumpFileDynamic)
}
} catch (e) {
// Ignore error
}
expect(testNetLog().currentlyLogging).to.be.false()
})
it('should begin and end logging to file when .startLogging() and .stopLogging() is called', async () => {
await testNetLog().startLogging(dumpFileDynamic)
expect(testNetLog().currentlyLogging).to.be.true()
expect(testNetLog().currentlyLoggingPath).to.equal(dumpFileDynamic)
await testNetLog().stopLogging()
expect(fs.existsSync(dumpFileDynamic)).to.be.true()
})
it('should throw an error when .stopLogging() is called without calling .startLogging()', async () => {
await expect(testNetLog().stopLogging()).to.be.rejectedWith('No net log in progress')
})
it('should throw an error when .startLogging() is called with an invalid argument', () => {
expect(() => testNetLog().startLogging('')).to.throw()
expect(() => testNetLog().startLogging(null)).to.throw()
expect(() => testNetLog().startLogging([])).to.throw()
})
2018-06-28 22:40:30 +00:00
it('should begin and end logging automatically when --log-net-log is passed', done => {
if (isCI && process.platform === 'linux') {
done()
return
}
const appProcess = ChildProcess.spawn(process.execPath,
[appPath], {
env: {
TEST_REQUEST_URL: server.url,
TEST_DUMP_FILE: dumpFile
}
})
appProcess.once('exit', () => {
2018-06-28 22:40:30 +00:00
expect(fs.existsSync(dumpFile)).to.be.true()
done()
})
})
it('should begin and end logging automtically when --log-net-log is passed, and behave correctly when .startLogging() and .stopLogging() is called', done => {
if (isCI && process.platform === 'linux') {
done()
return
}
const appProcess = ChildProcess.spawn(process.execPath,
[appPath], {
env: {
TEST_REQUEST_URL: server.url,
TEST_DUMP_FILE: dumpFile,
TEST_DUMP_FILE_DYNAMIC: dumpFileDynamic,
TEST_MANUAL_STOP: true
}
})
appProcess.once('exit', () => {
2018-06-28 22:40:30 +00:00
expect(fs.existsSync(dumpFile)).to.be.true()
expect(fs.existsSync(dumpFileDynamic)).to.be.true()
done()
})
})
2018-06-28 22:40:30 +00:00
it('should end logging automatically when only .startLogging() is called', done => {
if (isCI && process.platform === 'linux') {
done()
return
}
const appProcess = ChildProcess.spawn(process.execPath,
[appPath], {
env: {
TEST_REQUEST_URL: server.url,
TEST_DUMP_FILE_DYNAMIC: dumpFileDynamic
}
})
appProcess.once('close', () => {
2018-06-28 22:40:30 +00:00
expect(fs.existsSync(dumpFileDynamic)).to.be.true()
done()
})
})
})