Skip to content

Commit

Permalink
fix: using adb logcat options compatible with API level 19
Browse files Browse the repository at this point in the history
  • Loading branch information
@noomorph
noomorph committed Sep 20, 2018
1 parent 8576b4c commit 3302a69
Show file tree
Hide file tree
Showing 5 changed files with 101 additions and 95 deletions.
110 changes: 54 additions & 56 deletions detox/src/devices/android/ADB.js
View file
Original file line number Diff line number Diff line change
@@ -1,18 +1,19 @@
const _ = require('lodash');
const path = require('path');
const {execWithRetriesAndLogs, spawnAndLog} = require('../../utils/exec');
const pipeCommands = require('../../utils/pipeCommands');
const {escape} = require('../../utils/pipeCommands');
const EmulatorTelnet = require('./EmulatorTelnet');
const Environment = require('../../utils/environment');

class ADB {
constructor() {
this._cachedApiLevels = new Map();
this.adbBin = path.join(Environment.getAndroidSDKPath(), 'platform-tools', 'adb');
}

async devices() {
const output = (await this.adbCmd('', 'devices')).stdout;
return await this.parseAdbDevicesConsoleOutput(output);
return this.parseAdbDevicesConsoleOutput(output);
}

async unlockScreen(deviceId) {
Expand All @@ -31,8 +32,7 @@ class ADB {
}

async _getPowerStatus(deviceId) {
const grep = pipeCommands.search.regexp;
const stdout = await this.shell(deviceId, `dumpsys power | ${grep('^[ ]*m[UW].*=')}`);
const stdout = await this.shell(deviceId, `dumpsys power | grep "^[ ]*m[UW].*="`);

return stdout
.split('\n')
Expand Down Expand Up @@ -89,7 +89,7 @@ class ADB {
}

async now(deviceId) {
return this.shell(deviceId, `date "+\\"%Y-%m-%d %T.000\\""`);
return this.shell(deviceId, `date +"%Y-%m-%d %T.000"`);
}

async install(deviceId, apkPath) {
Expand All @@ -110,23 +110,18 @@ class ADB {
}

async pidof(deviceId, bundleId) {
const bundleIdRegex = pipeCommands.escape.inQuotedRegexp(bundleId) + '[ ]*$';
const grep = pipeCommands.search.regexp;
const bundleIdRegex = escape.inQuotedRegexp(bundleId) + '$';

const processes = await this.shell(deviceId, `ps | ${grep(bundleIdRegex)}`).catch(() => '');
const processes = await this.shell(deviceId, `ps | grep "${bundleIdRegex}"`).catch(() => '');
if (!processes) {
return NaN;
}

return parseInt(processes.split(' ').filter(Boolean)[1], 10);
}

async shell(deviceId, cmd, options) {
return (await this.adbCmd(deviceId, `shell ${cmd}`, options)).stdout.trim();
}

async getFileSize(deviceId, filename) {
const { stdout, stderr } = await this.adbCmd(deviceId, 'shell wc -c ' + filename).catch(e => e);
const { stdout, stderr } = await this.adbCmd(deviceId, 'shell du ' + filename).catch(e => e);

if (stderr.includes('No such file or directory')) {
return -1;
Expand All @@ -136,7 +131,7 @@ class ADB {
}

async isFileOpen(deviceId, filename) {
const openedByProcesses = await this.shell(deviceId, 'lsof ' + filename);
const openedByProcesses = await this.shell(deviceId, `lsof | grep -e "${escape.inQuotedString(filename)}" || true`);
return openedByProcesses.length > 0;
}

Expand All @@ -150,12 +145,14 @@ class ADB {
}

async apiLevel(deviceId) {
const lvl = await this.shell(deviceId, `getprop ro.build.version.sdk`);
return Number(lvl);
const lvl = Number(await this.shell(deviceId, `getprop ro.build.version.sdk`));
this._cachedApiLevels.set(deviceId, lvl);

return lvl;
}

async screencap(deviceId, path) {
return this.adbCmd(deviceId, `shell screencap ${path}`);
await this.shell(deviceId, `screencap ${path}`);
}

/***
Expand Down Expand Up @@ -185,37 +182,57 @@ class ADB {
/***
* @returns ChildProcessPromise
*/
logcat(deviceId, { expression, file, pid, time }) {
const logcatArgs = [];

if (expression) {
logcatArgs.push('-e');
logcatArgs.push(expression);
}

if (file) {
logcatArgs.push('-f');
logcatArgs.push(file);
logcat(deviceId, { file, pid, time }) {
let shellCommand = 'logcat -v brief';

// HACK: cannot make this function async, otherwise ChildProcessPromise.childProcess field will get lost,
// and this will break interruptProcess() call for any logcat promise.
const apiLevel = this._cachedApiLevels.get(deviceId);
if (time && apiLevel >= 21) {
shellCommand += ` -T "${time}"`;
}

if (pid > 0) {
logcatArgs.push(`--pid=${pid}`);
}
const __pid = String(pid).padStart(5);
shellCommand += ` | grep "(${__pid}):"`;
}

if (time) {
logcatArgs.push('-T');
logcatArgs.push(time);
if (file) {
shellCommand += ` >> ${file}`;
}

return this.spawn(deviceId, ['logcat', ...logcatArgs]);
return this.spawn(deviceId, ['shell', shellCommand]);
}

async pull(deviceId, src, dst = '') {
return this.adbCmd(deviceId, `pull "${src}" "${dst}"`);
await this.adbCmd(deviceId, `pull "${src}" "${dst}"`);
}

async rm(deviceId, path, force = false) {
return this.adbCmd(deviceId, `shell rm ${force ? '-f' : ''} "${path}"`);
await this.shell(deviceId, `rm ${force ? '-f' : ''} "${path}"`);
}

async listInstrumentation(deviceId) {
return this.shell(deviceId, 'pm list instrumentation');
}

async getInstrumentationRunner(deviceId, bundleId) {
const instrumentationRunners = await this.listInstrumentation(deviceId);
const instrumentationRunner = this._instrumentationRunnerForBundleId(instrumentationRunners, bundleId);
if (instrumentationRunner === 'undefined') {
throw new Error(`No instrumentation runner found on device ${deviceId} for package ${bundleId}`);
}

return instrumentationRunner;
}

_instrumentationRunnerForBundleId(instrumentationRunners, bundleId) {
const runnerForBundleRegEx = new RegExp(`^instrumentation:(.*) \\(target=${bundleId.replace(new RegExp('\\.', 'g'), "\\.")}\\)$`, 'gm');
return _.get(runnerForBundleRegEx.exec(instrumentationRunners), [1], 'undefined');
}

async shell(deviceId, cmd, options) {
return (await this.adbCmd(deviceId, `shell "${escape.inQuotedString(cmd)}"`, options)).stdout.trim();
}

async adbCmd(deviceId, params, options) {
Expand All @@ -224,7 +241,7 @@ class ADB {
const retries = _.get(options, 'retries', 1);
_.unset(options, 'retries');

return await execWithRetriesAndLogs(cmd, options, undefined, retries);
return execWithRetriesAndLogs(cmd, options, undefined, retries);
}

/***
Expand All @@ -234,25 +251,6 @@ class ADB {
const serial = deviceId ? ['-s', deviceId] : [];
return spawnAndLog(this.adbBin, [...serial, ...params]);
}

async listInstrumentation(deviceId) {
return await this.shell(deviceId, 'pm list instrumentation');
}

instrumentationRunnerForBundleId(instrumentationRunners, bundleId) {
const runnerForBundleRegEx = new RegExp(`^instrumentation:(.*) \\(target=${bundleId.replace(new RegExp('\\.', 'g'), "\\.")}\\)$`, 'gm');
return _.get(runnerForBundleRegEx.exec(instrumentationRunners), [1], 'undefined');
}

async getInstrumentationRunner(deviceId, bundleId) {
const instrumentationRunners = await this.listInstrumentation(deviceId);
const instrumentationRunner = this.instrumentationRunnerForBundleId(instrumentationRunners, bundleId);
if (instrumentationRunner === 'undefined') {
throw new Error(`No instrumentation runner found on device ${deviceId} for package ${bundleId}`);
}

return instrumentationRunner;
}
}

module.exports = ADB;
30 changes: 12 additions & 18 deletions detox/src/devices/android/ADB.test.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -46,20 +46,22 @@ describe('ADB', () => {
});

it(`pidof (success)`, async () => {
adb.shell = async () => `u0_a19 2199 1701 3554600 70264 0 0 s com.google.android.ext.services `;
jest.spyOn(adb, 'shell').mockImplementation(async () =>
`u0_a19 2199 1701 3554600 70264 0 0 s com.google.android.ext.services `);

expect(await adb.pidof('', 'com.google.android.ext.services')).toBe(2199);
});

it(`pidof (failure)`, async () => {
adb.shell = async () => ``;
jest.spyOn(adb, 'shell').mockImplementation(async () => '');
expect(await adb.pidof('', 'com.google.android.ext.services')).toBe(NaN);
});

describe('unlockScreen', () => {
const deviceId = 'mockEmulator';

async function unlockScreenWithPowerStatus(mWakefulness, mUserActivityTimeoutOverrideFromWindowManager) {
adb.shell = jest.fn().mockReturnValue(`
jest.spyOn(adb, 'shell').mockImplementation(async () => `
mWakefulness=${mWakefulness}
mWakefulnessChanging=false
mWakeLockSummary=0x0
Expand Down Expand Up @@ -116,11 +118,11 @@ describe('ADB', () => {

it(`listInstrumentation passes the right deviceId`, async () => {
const deviceId = 'aDeviceId';
const spyShell = jest.spyOn(adb, 'shell');
jest.spyOn(adb, 'shell');

await adb.listInstrumentation(deviceId);

expect(spyShell).toBeCalledWith(deviceId, expect.any(String));
expect(adb.shell).toBeCalledWith(deviceId, 'pm list instrumentation');
});

it(`Parse 'adb device' output`, async () => {
Expand All @@ -144,18 +146,7 @@ describe('ADB', () => {
expect(actual).toEqual(parsedDevices);
});

it(`getInstrumentationRunner passes the right deviceId`, async () => {
const deviceId = 'aDeviceId';
const spyRunnerForBundle = jest.spyOn(adb, 'instrumentationRunnerForBundleId');
spyRunnerForBundle.mockReturnValue('');
const spyShell = jest.spyOn(adb, 'shell');

await adb.getInstrumentationRunner(deviceId, 'com.whatever.package');

expect(spyShell).toBeCalledWith(deviceId, expect.any(String));
});

it(`instrumentationRunnerForBundleId parses the correct runner for the package`, async () => {
it(`getInstrumentationRunner parses the correct runner for the package`, async () => {
const expectedRunner = "com.example.android.apis/.app.LocalSampleInstrumentation";
const expectedPackage = "com.example.android.apis";
const instrumentationRunnersShellOutput =
Expand All @@ -164,8 +155,11 @@ describe('ADB', () => {
`instrumentation:${expectedRunner} (target=${expectedPackage})\n` +
"instrumentation:org.chromium.webview_shell/.WebViewLayoutTestRunner (target=org.chromium.webview_shell)\n";

const result = await adb.instrumentationRunnerForBundleId(instrumentationRunnersShellOutput, expectedPackage);
jest.spyOn(adb, 'shell').mockImplementation(async () => instrumentationRunnersShellOutput);

const result = await adb.getInstrumentationRunner('aDeviceId', expectedPackage);

expect(adb.shell).toBeCalledWith('aDeviceId', 'pm list instrumentation');
expect(result).toEqual(expectedRunner);
});
});
Expand Down
16 changes: 10 additions & 6 deletions detox/src/devices/android/EmulatorTelnet.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,13 @@ const Telnet = require('telnet-client');
const path = require('path');
const os = require('os');
const fs = require('fs-extra');
const log = require('../../utils/logger').child({ __filename });

class EmulatorTelnet {
constructor() {
this.connection = new Telnet();
this.connection.on('timeout', () => log.error({ event: 'TELNET_TIMEOUT' }));
this.connection.on('error', (err) => log.error({ event: 'TELNET_ERROR', err }));
}

async connect(port) {
Expand All @@ -20,6 +23,7 @@ class EmulatorTelnet {
stripShellPrompt: true
};

log.trace({ event: 'TELNET_CONNECTING' }, `port: ${port}, host: ${params.host}`);
await this.connection.connect(params);
const auth = await fs.readFile(path.join(os.homedir(), '.emulator_console_auth_token'), 'utf8');
await this.exec(`auth ${auth}`);
Expand All @@ -36,18 +40,18 @@ class EmulatorTelnet {
this.connection.shell((error, stream) => {
stream.write(`${command}\n`);
stream.on('data', (data) => {
const result = data.toString();
if (result.includes('\n')) {
resolve(result);
const result = data.toString();
if (result.includes('\n')) {
resolve(result);
}
}
}
);
});
});
}

async avdName() {
return await this.exec('avd name');
return this.exec('avd name');
}

async kill() {
Expand All @@ -61,7 +65,7 @@ class EmulatorTelnet {
}

async rotate() {
return await this.shell('rotate');
return this.shell('rotate');
}
}

Expand Down
8 changes: 8 additions & 0 deletions detox/src/utils/exec.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,14 @@ async function execWithRetriesAndLogs(bin, options, statusLogs, retries = 10, in
// log.error(`${_operationCounter}: stderr:`, result.stderr);
//}

if (typeof result.stdout === 'string') {
result.stdout = result.stdout.replace(/\r\n/g, '\n');
}

if (typeof result.stderr === 'string') {
result.stderr = result.stderr.replace(/\r\n/g, '\n');
}

return result;
}

Expand Down
Loading

0 comments on commit 3302a69

Please sign in to comment.