@testim/testim-cli/workers/WorkerAppium.js

"use strict";

const {CLI_MODE} = require('../commons/constants');
const BaseWorker = require('./BaseWorker');
const {timeoutMessages} = require('../commons/constants');
const logger = require('../commons/logger').getLogger("worker-selenium");
const Promise = require('bluebird');
const exec = Promise.promisify(require('child_process').exec);
const reporter = require("../reports/reporter");
const MobileTestPlayer = require('../player/mobile/mobileTestPlayer');
const { manifestVersion } = require('../commons/getSessionPlayerRequire');

class WorkerAppium extends BaseWorker {
    initPlayer(testRunHandler) {
        const { projectData } = this.options;
        return new MobileTestPlayer(this.id, testRunHandler.getRunParams(), projectData.type);
    }

    getBrowserOnce(testRunHandler, customExtensionLocalLocation, player) {
        reporter.onGetSession(this.id, this.testName, testRunHandler.getRunMode());

        player.clearSessionTabs();
        const {driver} = player;

        //use fake browser name
        return this.getGridSlot("chrome", testRunHandler)
            .then(gridInfo => driver.init(this.options, gridInfo, testRunHandler.getNativeAppData(), this.testName, this.testResultId))
            .then(() => player.addTab())
            .catch(err => {
                logger.error(`failed to get device/app (message=${err.message})`, {err: err});
                throw err;
            });
    }

    runTestCleanup() {
        const { mode } = this.options;

        if(mode === CLI_MODE.DEVICE_FARM_APPIUM) {
            return exec("adb shell pm clear $DEVICEFARM_APP_NAME");
        }
        return Promise.resolve();
    }

    runTestOnce(testRunHandler, player) {
        const version = manifestVersion || "runner";
        let that = this;

        reporter.onWaitToTestComplete(this.id, this.isCodeMode);

        function runSeleniumTest() {
            return new Promise(resolve => player.sessionPlayer.playByTestId(that.testId, that.executionId, that.testResultId, that.baseUrl, that.userData, version, resolve, false, that.overrideTestConfigId, that.branch))
                .timeout(that.testRunTimeout, timeoutMessages.TEST_COMPLETE_TIMEOUT_MSG)
                .catch(Promise.TimeoutError, err => {
                    player.sessionPlayer.stopPlayingOnTestTimeout && player.sessionPlayer.stopPlayingOnTestTimeout();
                    throw err;
                })
                .then(testResult => {
                    testResult.stepsResults = null;
                    return Promise.resolve(testResult);
                });
        }

        return super.runTestOnce(testRunHandler, player)
            .then(() => runSeleniumTest())
            .catch(err => {
                logger.error("failed to run test once", {err: err});
                return Promise.reject(err);
            });
    }
}

module.exports = WorkerAppium;