2019-08-05 00:47:42 +00:00
import { expect } from 'chai'
import * as http from 'http'
import * as fs from 'fs'
import * as os from 'os'
import * as path from 'path'
import * as ChildProcess from 'child_process'
import { session , net } from 'electron'
2019-11-01 20:37:02 +00:00
import { Socket , AddressInfo } from 'net'
2019-10-30 23:38:21 +00:00
import { ifit } from './spec-helpers'
import { emittedOnce } from './events-helpers'
2019-08-05 00:47:42 +00:00
2018-06-19 01:45:58 +00:00
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' )
2019-07-24 23:01:08 +00:00
const testNetLog = ( ) = > session . fromPartition ( 'net-log' ) . netLog
2018-06-19 01:45:58 +00:00
describe ( 'netLog module' , ( ) = > {
2019-08-05 00:47:42 +00:00
let server : http.Server
let serverUrl : string
const connections : Set < Socket > = new Set ( )
2018-06-19 01:45:58 +00:00
2018-06-28 22:40:30 +00:00
before ( done = > {
2018-06-19 01:45:58 +00:00
server = http . createServer ( )
server . listen ( 0 , '127.0.0.1' , ( ) = > {
2019-08-05 00:47:42 +00:00
serverUrl = ` http://127.0.0.1: ${ ( server . address ( ) as AddressInfo ) . port } `
2018-06-19 01:45:58 +00:00
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 = > {
2018-06-19 01:45:58 +00:00
for ( const connection of connections ) {
connection . destroy ( )
}
server . close ( ( ) = > {
2019-08-05 00:47:42 +00:00
server = null as any
2018-06-19 01:45:58 +00:00
done ( )
} )
} )
2019-05-23 22:31:38 +00:00
beforeEach ( ( ) = > {
2019-08-05 00:47:42 +00:00
expect ( testNetLog ( ) . currentlyLogging ) . to . be . false ( 'currently logging' )
2019-05-23 22:31:38 +00:00
} )
2018-06-19 01:45:58 +00:00
afterEach ( ( ) = > {
try {
2018-10-04 18:08:56 +00:00
if ( fs . existsSync ( dumpFile ) ) {
fs . unlinkSync ( dumpFile )
}
if ( fs . existsSync ( dumpFileDynamic ) ) {
fs . unlinkSync ( dumpFileDynamic )
}
2018-06-19 01:45:58 +00:00
} catch ( e ) {
// Ignore error
}
2019-08-05 00:47:42 +00:00
expect ( testNetLog ( ) . currentlyLogging ) . to . be . false ( 'currently logging' )
2018-06-19 01:45:58 +00:00
} )
2019-02-19 10:48:27 +00:00
it ( 'should begin and end logging to file when .startLogging() and .stopLogging() is called' , async ( ) = > {
2019-07-24 23:01:08 +00:00
await testNetLog ( ) . startLogging ( dumpFileDynamic )
2019-02-19 10:48:27 +00:00
2019-08-05 00:47:42 +00:00
expect ( testNetLog ( ) . currentlyLogging ) . to . be . true ( 'currently logging' )
2019-05-23 22:31:38 +00:00
2020-03-18 23:46:05 +00:00
await testNetLog ( ) . stopLogging ( )
2019-02-19 10:48:27 +00:00
2019-08-05 00:47:42 +00:00
expect ( fs . existsSync ( dumpFileDynamic ) ) . to . be . true ( 'currently logging' )
2019-02-19 10:48:27 +00:00
} )
2019-05-23 22:31:38 +00:00
it ( 'should throw an error when .stopLogging() is called without calling .startLogging()' , async ( ) = > {
2019-07-24 23:01:08 +00:00
await expect ( testNetLog ( ) . stopLogging ( ) ) . to . be . rejectedWith ( 'No net log in progress' )
2019-05-23 22:31:38 +00:00
} )
2019-02-19 10:48:27 +00:00
2019-05-23 22:31:38 +00:00
it ( 'should throw an error when .startLogging() is called with an invalid argument' , ( ) = > {
2019-07-24 23:01:08 +00:00
expect ( ( ) = > testNetLog ( ) . startLogging ( '' ) ) . to . throw ( )
2019-08-05 00:47:42 +00:00
expect ( ( ) = > testNetLog ( ) . startLogging ( null as any ) ) . to . throw ( )
expect ( ( ) = > testNetLog ( ) . startLogging ( [ ] as any ) ) . to . throw ( )
2019-11-01 20:37:02 +00:00
expect ( ( ) = > testNetLog ( ) . startLogging ( 'aoeu' , { captureMode : 'aoeu' as any } ) ) . to . throw ( )
expect ( ( ) = > testNetLog ( ) . startLogging ( 'aoeu' , { maxFileSize : null as any } ) ) . to . throw ( )
2019-07-25 23:06:39 +00:00
} )
it ( 'should include cookies when requested' , async ( ) = > {
2019-11-01 20:37:02 +00:00
await testNetLog ( ) . startLogging ( dumpFileDynamic , { captureMode : 'includeSensitive' } )
2019-07-25 23:06:39 +00:00
const unique = require ( 'uuid' ) . v4 ( )
await new Promise ( ( resolve ) = > {
2019-08-05 00:47:42 +00:00
const req = net . request ( serverUrl )
2019-07-25 23:06:39 +00:00
req . setHeader ( 'Cookie' , ` foo= ${ unique } ` )
req . on ( 'response' , ( response ) = > {
2019-11-01 20:37:02 +00:00
response . on ( 'data' , ( ) = > { } ) // https://github.com/electron/electron/issues/19214
2019-07-25 23:06:39 +00:00
response . on ( 'end' , ( ) = > resolve ( ) )
} )
req . end ( )
} )
await testNetLog ( ) . stopLogging ( )
expect ( fs . existsSync ( dumpFileDynamic ) ) . to . be . true ( 'dump file exists' )
const dump = fs . readFileSync ( dumpFileDynamic , 'utf8' )
expect ( dump ) . to . contain ( ` foo= ${ unique } ` )
} )
it ( 'should include socket bytes when requested' , async ( ) = > {
2019-11-01 20:37:02 +00:00
await testNetLog ( ) . startLogging ( dumpFileDynamic , { captureMode : 'everything' } )
2019-07-25 23:06:39 +00:00
const unique = require ( 'uuid' ) . v4 ( )
await new Promise ( ( resolve ) = > {
2019-11-01 20:37:02 +00:00
const req = net . request ( { method : 'POST' , url : serverUrl } )
2019-07-25 23:06:39 +00:00
req . on ( 'response' , ( response ) = > {
2019-11-01 20:37:02 +00:00
response . on ( 'data' , ( ) = > { } ) // https://github.com/electron/electron/issues/19214
2019-07-25 23:06:39 +00:00
response . on ( 'end' , ( ) = > resolve ( ) )
} )
req . end ( Buffer . from ( unique ) )
} )
await testNetLog ( ) . stopLogging ( )
expect ( fs . existsSync ( dumpFileDynamic ) ) . to . be . true ( 'dump file exists' )
const dump = fs . readFileSync ( dumpFileDynamic , 'utf8' )
2019-08-05 00:47:42 +00:00
expect ( JSON . parse ( dump ) . events . some ( ( x : any ) = > x . params && x . params . bytes && Buffer . from ( x . params . bytes , 'base64' ) . includes ( unique ) ) ) . to . be . true ( 'uuid present in dump' )
2019-02-19 10:48:27 +00:00
} )
2019-10-30 23:38:21 +00:00
ifit ( process . platform !== 'linux' ) ( 'should begin and end logging automatically when --log-net-log is passed' , async ( ) = > {
2019-05-23 22:31:38 +00:00
const appProcess = ChildProcess . spawn ( process . execPath ,
2018-10-04 18:08:56 +00:00
[ appPath ] , {
2018-06-19 01:45:58 +00:00
env : {
2019-08-05 00:47:42 +00:00
TEST_REQUEST_URL : serverUrl ,
2018-10-04 18:08:56 +00:00
TEST_DUMP_FILE : dumpFile
2018-06-19 01:45:58 +00:00
}
} )
2019-10-30 23:38:21 +00:00
await emittedOnce ( appProcess , 'exit' )
expect ( fs . existsSync ( dumpFile ) ) . to . be . true ( 'dump file exists' )
2018-06-19 01:45:58 +00:00
} )
2019-10-30 23:38:21 +00:00
ifit ( process . platform !== 'linux' ) ( 'should begin and end logging automtically when --log-net-log is passed, and behave correctly when .startLogging() and .stopLogging() is called' , async ( ) = > {
2019-05-23 22:31:38 +00:00
const appProcess = ChildProcess . spawn ( process . execPath ,
2018-10-04 18:08:56 +00:00
[ appPath ] , {
2018-06-19 01:45:58 +00:00
env : {
2019-08-05 00:47:42 +00:00
TEST_REQUEST_URL : serverUrl ,
2018-10-04 18:08:56 +00:00
TEST_DUMP_FILE : dumpFile ,
TEST_DUMP_FILE_DYNAMIC : dumpFileDynamic ,
2019-08-05 00:47:42 +00:00
TEST_MANUAL_STOP : 'true'
2018-06-19 01:45:58 +00:00
}
} )
2019-10-30 23:38:21 +00:00
await emittedOnce ( appProcess , 'exit' )
expect ( fs . existsSync ( dumpFile ) ) . to . be . true ( 'dump file exists' )
expect ( fs . existsSync ( dumpFileDynamic ) ) . to . be . true ( 'dynamic dump file exists' )
2018-06-19 01:45:58 +00:00
} )
2019-10-30 23:38:21 +00:00
ifit ( process . platform !== 'linux' ) ( 'should end logging automatically when only .startLogging() is called' , async ( ) = > {
2019-05-23 22:31:38 +00:00
const appProcess = ChildProcess . spawn ( process . execPath ,
2018-06-19 01:45:58 +00:00
[ appPath ] , {
env : {
2019-08-05 00:47:42 +00:00
TEST_REQUEST_URL : serverUrl ,
2018-10-04 18:08:56 +00:00
TEST_DUMP_FILE_DYNAMIC : dumpFileDynamic
2018-06-19 01:45:58 +00:00
}
} )
2020-01-27 01:29:50 +00:00
await emittedOnce ( appProcess , 'exit' )
2019-10-30 23:38:21 +00:00
expect ( fs . existsSync ( dumpFileDynamic ) ) . to . be . true ( 'dynamic dump file exists' )
2018-06-19 01:45:58 +00:00
} )
} )