const fs = require('fs'); const path = require('path'); const { exec, spawn } = require('child_process'); const yargs = require('yargs/yargs') const { hideBin } = require('yargs/helpers') const logger = { warn: msg => console.error(`WARN: ${msg}`), err: msg => console.error(`ERR: ${msg}`), info: msg => console.log(`INFO: ${msg}`) } function panic(errMsg) { logger.err(errMsg); process.exit(-1); } function checkTestEnv() { const argv = yargs(hideBin(process.argv)) .usage('Usage: $0 [options] ') .example('$0 --network ethermint', 'run all tests using ethermint network') .example('$0 --network ethermint test1 test2', 'run only test1 and test2 using ethermint network') .help('h').alias('h', 'help') .describe('network', 'set which network to use: ganache|ethermint') .boolean('verbose-log').describe('verbose-log', 'print ethermintd output, default false') .argv; if (!fs.existsSync(path.join(__dirname, './node_modules'))) { panic('node_modules not existed. Please run `yarn install` before running tests.'); } const runConfig = {}; // Check test network if (!argv.network) { runConfig.network = 'ganache'; } else { if (argv.network !== 'ethermint' && argv.network !== 'ganache') { panic('network is invalid. Must be ganache or ethermint'); } else { runConfig.network = argv.network; } } // only test runConfig.onlyTest = argv['_']; runConfig.verboseLog = !!argv['verbose-log']; logger.info(`Running on network: ${runConfig.network}`); return runConfig; } function loadTests() { const validTests = []; fs.readdirSync(path.join(__dirname, 'suites')).forEach(dirname => { const dirStat = fs.statSync(path.join(__dirname, 'suites', dirname)); if (!dirStat.isDirectory) { logger.warn(`${dirname} is not a directory. Skip this test suite.`); return; } const needFiles = ['package.json', 'test']; for (const f of needFiles) { if (!fs.existsSync(path.join(__dirname, 'suites', dirname, f))) { logger.warn(`${dirname} does not contains file/dir: ${f}. Skip this test suite.`); return; } } // test package.json try { const testManifest = JSON.parse(fs.readFileSync(path.join(__dirname, 'suites', dirname, 'package.json'), 'utf-8')) const needScripts = ['test-ganache', 'test-ethermint']; for (const s of needScripts) { if (Object.keys(testManifest['scripts']).indexOf(s) === -1) { logger.warn(`${dirname} does not have test script: \`${s}\`. Skip this test suite.`); return; } } } catch (error) { logger.warn(`${dirname} test package.json load failed. Skip this test suite.`); logger.err(error); return; } validTests.push(dirname); }) return validTests; } function performTestSuite({ testName, network }) { const cmd = network === 'ganache' ? 'test-ganache' : 'test-ethermint'; return new Promise((resolve, reject) => { const testProc = spawn('yarn', [cmd], { cwd: path.join(__dirname, 'suites', testName) }); testProc.stdout.pipe(process.stdout); testProc.stderr.pipe(process.stderr); testProc.on('close', code => { if (code === 0) { console.log("end"); resolve(); } else { reject(new Error(`Test: ${testName} exited with error code ${code}`)); } }); }); } async function performTests({ allTests, runConfig }) { if (allTests.length === 0) { panic('No tests are found or all invalid!'); } if (runConfig.onlyTest.length === 0) { logger.info('Start all tests:'); } else { allTests = allTests.filter(t => runConfig.onlyTest.indexOf(t) !== -1); logger.info(`Only run tests: (${allTests.join(', ')})`); } for (const currentTestName of allTests) { logger.info(`Start test: ${currentTestName}`); await performTestSuite({ testName: currentTestName, network: runConfig.network }); } logger.info(`${allTests.length} test suites passed!`); } function setupNetwork({ runConfig, timeout }) { if (runConfig.network !== 'ethermint') { // no need to start ganache. Truffle will start it return; } // Spawn the ethermint process const spawnPromise = new Promise((resolve, reject) => { const ethermintdProc = spawn('./init-test-node.sh', { cwd: __dirname, stdio: ['ignore', runConfig.verboseLog ? 'pipe' : 'ignore', 'pipe'], }); logger.info(`Starting Ethermintd process... timeout: ${timeout}ms`); if (runConfig.verboseLog) { ethermintdProc.stdout.pipe(process.stdout); } ethermintdProc.stderr.on('data', d => { const oLine = d.toString(); if (runConfig.verboseLog) { process.stdout.write(oLine); } if (oLine.indexOf('Starting EVM RPC server') !== -1) { logger.info('Ethermintd started'); resolve(ethermintdProc); } }); }); const timeoutPromise = new Promise((resolve, reject) => { setTimeout(() => reject(new Error('Start ethermintd timeout!')), timeout); }); return Promise.race([spawnPromise, timeoutPromise]); } async function main() { const runConfig = checkTestEnv(); const allTests = loadTests(runConfig); const proc = await setupNetwork({ runConfig, timeout: 50000 }); await performTests({ allTests, runConfig }); if (proc) { proc.kill(); } process.exit(0); } main();