@xenon-device-management/xenon 2.13.2 → 2.14.0

package/lib/test/unit/phone-network/phone-network-restore.spec.js

@@ -0,0 +1,246 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+require("reflect-metadata");
+const chai_1 = require("chai");
+const sinon_1 = __importDefault(require("sinon"));
+const net_1 = __importDefault(require("net"));
+const typedi_1 = require("typedi");
+const PhoneNetworkRestore_1 = require("../../../src/services/network/PhoneNetworkRestore");
+const PhoneNetworkLedger_1 = require("../../../src/services/network/PhoneNetworkLedger");
+const NetworkConditioningService_1 = require("../../../src/services/NetworkConditioningService");
+const InterceptorService_1 = require("../../../src/services/InterceptorService");
+const PortAllocator_1 = require("../../../src/services/PortAllocator");
+const prisma_1 = require("../../../src/prisma");
+const container_registration_1 = require("../../helpers/container-registration");
+const scratch_database_1 = require("../../helpers/scratch-database");
+const fake_adb_1 = require("../../helpers/fake-adb");
+const DAY = 24 * 60 * 60 * 1000;
+function deferred() {
+    let resolve;
+    const promise = new Promise((r) => (resolve = r));
+    return { promise, resolve };
+}
+/** Stand-ins for the two services that change a phone's network for a session. */
+function fakeServices(live) {
+    const net = {
+        has: (id) => live.net.has(id),
+        sessionIds: () => [...live.net],
+        reset: sinon_1.default.stub().callsFake((id) => __awaiter(this, void 0, void 0, function* () { return void live.net.delete(id); })),
+    };
+    const icp = {
+        isActive: (id) => live.icp.has(id),
+        sessionIds: () => [...live.icp],
+        stop: sinon_1.default.stub().callsFake((id) => __awaiter(this, void 0, void 0, function* () { return void live.icp.delete(id); })),
+    };
+    return { net, icp };
+}
+describe('PhoneNetworkRestore', function () {
+    this.timeout(60000);
+    (0, scratch_database_1.useScratchDatabase)();
+    let adb;
+    (0, fake_adb_1.useFakeAdb)(() => adb);
+    let restoreRegs;
+    let live;
+    let fakes;
+    let ledger;
+    let range;
+    beforeEach(() => __awaiter(this, void 0, void 0, function* () {
+        adb = (0, fake_adb_1.fakeAdb)();
+        restoreRegs = (0, container_registration_1.saveRegistrations)(NetworkConditioningService_1.NetworkConditioningService, InterceptorService_1.InterceptorService, PortAllocator_1.PortAllocator, PhoneNetworkLedger_1.PhoneNetworkLedger);
+        live = { net: new Set(), icp: new Set() };
+        fakes = fakeServices(live);
+        typedi_1.Container.set(NetworkConditioningService_1.NetworkConditioningService, fakes.net);
+        typedi_1.Container.set(InterceptorService_1.InterceptorService, fakes.icp);
+        range = [11100, 11199];
+        typedi_1.Container.set(PortAllocator_1.PortAllocator, { rangeOf: () => range });
+        ledger = new PhoneNetworkLedger_1.PhoneNetworkLedger();
+        typedi_1.Container.set(PhoneNetworkLedger_1.PhoneNetworkLedger, ledger);
+        for (const entry of yield ledger.list())
+            yield ledger.forget(entry.sessionId);
+    }));
+    afterEach(() => {
+        sinon_1.default.restore();
+        restoreRegs();
+    });
+    describe('restoreSession', () => {
+        it('resets the profile and stops the interceptor of a session', () => __awaiter(this, void 0, void 0, function* () {
+            live.net.add('s1');
+            live.icp.add('s1');
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreSession('s1', 'test');
+            (0, chai_1.expect)(fakes.net.reset.calledOnceWith('s1')).to.equal(true);
+            (0, chai_1.expect)(fakes.icp.stop.calledOnceWith('s1')).to.equal(true);
+        }));
+        it('does nothing for a session that changed nothing, or no session', () => __awaiter(this, void 0, void 0, function* () {
+            const restore = new PhoneNetworkRestore_1.PhoneNetworkRestore();
+            yield restore.restoreSession('s-plain', 'test');
+            yield restore.restoreSession(undefined, 'test');
+            yield restore.restoreSession(null, 'test');
+            (0, chai_1.expect)(fakes.net.reset.called).to.equal(false);
+            (0, chai_1.expect)(fakes.icp.stop.called).to.equal(false);
+        }));
+        it('runs once when two endings race, and both wait until it is done', () => __awaiter(this, void 0, void 0, function* () {
+            live.net.add('s1');
+            const gate = deferred();
+            fakes.net.reset.callsFake((id) => __awaiter(this, void 0, void 0, function* () {
+                yield gate.promise;
+                live.net.delete(id);
+            }));
+            const restore = new PhoneNetworkRestore_1.PhoneNetworkRestore();
+            let firstDone = false;
+            let secondDone = false;
+            const first = restore.restoreSession('s1', 'deleted').then(() => (firstDone = true));
+            const second = restore.restoreSession('s1', 'idle').then(() => (secondDone = true));
+            yield new Promise((r) => setImmediate(r));
+            (0, chai_1.expect)(firstDone || secondDone, 'nobody may go on before the network is back').to.equal(false);
+            gate.resolve();
+            yield Promise.all([first, second]);
+            yield restore.restoreSession('s1', 'later');
+            (0, chai_1.expect)(fakes.net.reset.callCount).to.equal(1);
+        }));
+        it("never throws: one service's failure doesn't stop the other's restore", () => __awaiter(this, void 0, void 0, function* () {
+            live.net.add('s1');
+            live.icp.add('s1');
+            fakes.net.reset.rejects(new Error('boom'));
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreSession('s1', 'test');
+            (0, chai_1.expect)(fakes.icp.stop.calledOnceWith('s1')).to.equal(true);
+        }));
+    });
+    it('restoreAll restores every session either service still has', () => __awaiter(this, void 0, void 0, function* () {
+        live.net.add('a');
+        live.icp.add('a');
+        live.icp.add('b');
+        yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreAll('shutdown');
+        (0, chai_1.expect)(fakes.net.reset.args.map((a) => a[0])).to.deep.equal(['a']);
+        (0, chai_1.expect)(fakes.icp.stop.args.map((a) => a[0]).sort()).to.deep.equal(['a', 'b']);
+    }));
+    describe('restoreLeftovers: what an ended run wrote down', () => {
+        beforeEach(() => {
+            adb.answers['P1 shell settings get global http_proxy'] = '127.0.0.1:11150';
+        });
+        it('turns back on what was on, and puts back the proxy it set', () => __awaiter(this, void 0, void 0, function* () {
+            yield ledger.note('old', 'P1', { offline: { wifiOn: true, dataOn: false } });
+            yield ledger.note('old', 'P1', {
+                proxy: { set: '127.0.0.1:11150', previous: null, reversePort: 11150 },
+            });
+            const done = yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreLeftovers({});
+            (0, chai_1.expect)(done).to.equal(1);
+            (0, chai_1.expect)(adb.commands('P1')).to.deep.equal([
+                'shell svc wifi enable',
+                'shell settings get global http_proxy',
+                'shell settings put global http_proxy :0',
+                'reverse --remove tcp:11150',
+            ]);
+            (0, chai_1.expect)(yield ledger.list()).to.deep.equal([]);
+        }));
+        it("puts back the phone's own proxy when it had one", () => __awaiter(this, void 0, void 0, function* () {
+            yield ledger.note('old', 'P1', {
+                proxy: { set: '127.0.0.1:11150', previous: 'proxy.corp:3128' },
+            });
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreLeftovers({});
+            (0, chai_1.expect)(adb.commands('P1')).to.include('shell settings put global http_proxy proxy.corp:3128');
+        }));
+        it('leaves a proxy alone that someone changed since, and forgets it', () => __awaiter(this, void 0, void 0, function* () {
+            adb.answers['P1 shell settings get global http_proxy'] = 'proxy.corp:3128';
+            yield ledger.note('old', 'P1', { proxy: { set: '127.0.0.1:11150', previous: null } });
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreLeftovers({});
+            (0, chai_1.expect)(adb.commands('P1')).to.deep.equal(['shell settings get global http_proxy']);
+            (0, chai_1.expect)(yield ledger.list()).to.deep.equal([]);
+        }));
+        it('skips a session this server still runs', () => __awaiter(this, void 0, void 0, function* () {
+            live.net.add('running');
+            yield ledger.note('running', 'P1', { offline: { wifiOn: true, dataOn: true } });
+            (0, chai_1.expect)(yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreLeftovers({})).to.equal(0);
+            (0, chai_1.expect)(adb.calls).to.deep.equal([]);
+            (0, chai_1.expect)((yield ledger.list()).map((e) => e.sessionId)).to.deep.equal(['running']);
+        }));
+        it('touches only the phone it is asked about', () => __awaiter(this, void 0, void 0, function* () {
+            yield ledger.note('old-1', 'P1', { offline: { wifiOn: true, dataOn: true } });
+            yield ledger.note('old-2', 'P2', { offline: { wifiOn: true, dataOn: true } });
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreLeftovers({ udid: 'P2' });
+            (0, chai_1.expect)(adb.commands('P1')).to.deep.equal([]);
+            (0, chai_1.expect)(adb.commands('P2')).to.deep.equal(['shell svc data enable', 'shell svc wifi enable']);
+            (0, chai_1.expect)((yield ledger.list()).map((e) => e.sessionId)).to.deep.equal(['old-1']);
+        }));
+        it('keeps what it could not undo for next time, and gives up after a week', () => __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            adb.answers['P1 shell svc data enable'] = new Error("device 'P1' not found");
+            adb.answers['P9 shell svc data enable'] = new Error("device 'P9' not found");
+            yield ledger.note('recent', 'P1', { offline: { wifiOn: true, dataOn: true } });
+            yield ledger.note('ancient', 'P9', { offline: { wifiOn: true, dataOn: true } });
+            const row = yield prisma_1.prisma.webConfig.findUnique({ where: { name: 'phone-network:ancient' } });
+            (0, chai_1.expect)(row, 'the record').to.not.equal(null);
+            const value = JSON.parse((_a = row === null || row === void 0 ? void 0 : row.value) !== null && _a !== void 0 ? _a : '{}');
+            yield prisma_1.prisma.webConfig.update({
+                where: { name: 'phone-network:ancient' },
+                data: { value: JSON.stringify(Object.assign(Object.assign({}, value), { at: Date.now() - 8 * DAY })) },
+            });
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().restoreLeftovers({});
+            (0, chai_1.expect)((yield ledger.list()).map((e) => e.sessionId)).to.deep.equal(['recent']);
+        }));
+    });
+    describe('cleanUpAtBoot: a proxy left pointing at a dead capture port', () => {
+        let server;
+        let livePort;
+        let deadPort;
+        beforeEach(() => __awaiter(this, void 0, void 0, function* () {
+            server = net_1.default.createServer((s) => s.destroy());
+            yield new Promise((r) => server.listen(0, '127.0.0.1', () => r()));
+            livePort = server.address().port;
+            const closed = net_1.default.createServer();
+            yield new Promise((r) => closed.listen(0, '127.0.0.1', () => r()));
+            deadPort = closed.address().port;
+            yield new Promise((r) => closed.close(() => r()));
+            range = [Math.min(livePort, deadPort), Math.max(livePort, deadPort)];
+        }));
+        afterEach(() => __awaiter(this, void 0, void 0, function* () {
+            yield new Promise((r) => server.close(() => r()));
+        }));
+        it("clears only a proxy that points at this machine's capture ports where nothing answers", () => __awaiter(this, void 0, void 0, function* () {
+            adb.answers['DEAD shell settings get global http_proxy'] = `127.0.0.1:${deadPort}`;
+            adb.answers['EMU shell settings get global http_proxy'] = `10.0.2.2:${deadPort}`;
+            adb.answers['LIVE shell settings get global http_proxy'] = `127.0.0.1:${livePort}`;
+            adb.answers['CORP shell settings get global http_proxy'] = 'proxy.corp:3128';
+            adb.answers['NONE shell settings get global http_proxy'] = ':0';
+            adb.answers['FAR shell settings get global http_proxy'] = '127.0.0.1:1';
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().cleanUpAtBoot(['DEAD', 'EMU', 'LIVE', 'CORP', 'NONE', 'FAR']);
+            (0, chai_1.expect)(adb.commands('DEAD')).to.deep.equal([
+                'shell settings get global http_proxy',
+                'shell settings put global http_proxy :0',
+                `reverse --remove tcp:${deadPort}`,
+            ]);
+            (0, chai_1.expect)(adb.commands('EMU')).to.deep.equal([
+                'shell settings get global http_proxy',
+                'shell settings put global http_proxy :0',
+            ]);
+            for (const kept of ['LIVE', 'CORP', 'NONE', 'FAR']) {
+                (0, chai_1.expect)(adb.commands(kept), kept).to.deep.equal(['shell settings get global http_proxy']);
+            }
+        }));
+        it('undoes what the ledger holds first, then checks the phones', () => __awaiter(this, void 0, void 0, function* () {
+            yield ledger.note('old', 'P1', { offline: { wifiOn: true, dataOn: true } });
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().cleanUpAtBoot([]);
+            (0, chai_1.expect)(adb.commands('P1')).to.deep.equal(['shell svc data enable', 'shell svc wifi enable']);
+        }));
+        it('never throws, even when a phone cannot be read', () => __awaiter(this, void 0, void 0, function* () {
+            adb.answers['shell settings get global http_proxy'] = new Error('unauthorized');
+            yield new PhoneNetworkRestore_1.PhoneNetworkRestore().cleanUpAtBoot(['X']);
+        }));
+    });
+    it('can be built by the container', () => {
+        restoreRegs();
+        restoreRegs = (0, container_registration_1.saveRegistrations)(PhoneNetworkRestore_1.PhoneNetworkRestore);
+        (0, chai_1.expect)(() => typedi_1.Container.get(PhoneNetworkRestore_1.PhoneNetworkRestore)).not.to.throw();
+    });
+});

package/lib/test/unit/phone-network/session-end-paths.spec.js

@@ -0,0 +1,182 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+require("reflect-metadata");
+const chai_1 = require("chai");
+const sinon_1 = __importDefault(require("sinon"));
+const typedi_1 = require("typedi");
+const plugin_1 = require("../../../src/plugin");
+const NodeDevices_1 = __importDefault(require("../../../src/device-managers/NodeDevices"));
+const DeviceService = __importStar(require("../../../src/data-service/device-service"));
+const ActiveLeases = __importStar(require("../../../src/services/lease/activeLeases"));
+const device_utils_1 = require("../../../src/device-utils");
+const SessionLifecycleService_1 = require("../../../src/services/SessionLifecycleService");
+const SessionMetricsService_1 = require("../../../src/services/metrics/SessionMetricsService");
+const OrphanSweeper_1 = require("../../../src/services/OrphanSweeper");
+const ShutdownCoordinator_1 = require("../../../src/services/ShutdownCoordinator");
+const PhoneNetworkRestore_1 = require("../../../src/services/network/PhoneNetworkRestore");
+const event_manager_1 = require("../../../src/dashboard/event-manager");
+const SessionManager_1 = require("../../../src/sessions/SessionManager");
+const prisma_1 = require("../../../src/prisma");
+const container_registration_1 = require("../../helpers/container-registration");
+const scratch_database_1 = require("../../helpers/scratch-database");
+/**
+ * A network profile or the interceptor changes the whole phone. Its network
+ * was put back only when the test deleted a session Xenon kept in memory.
+ * Every other ending (Appium's new-command timeout, Xenon's idle release, a
+ * shutdown, a stale heartbeat) left an Offline phone offline and an
+ * intercepted phone pointing at a proxy that no longer exists.
+ *
+ * Each ending now asks PhoneNetworkRestore, keyed by the session id, and
+ * before the phone is released: a phone handed to the next session while its
+ * network is still being put back could have that session's own profile
+ * undone.
+ */
+describe('a session’s phone network is put back however the session ends', function () {
+    this.timeout(30000);
+    const scratch = (0, scratch_database_1.useScratchDatabase)();
+    let restoreRegs;
+    let order;
+    let restoreSession;
+    let restoreAll;
+    beforeEach(() => {
+        restoreRegs = (0, container_registration_1.saveRegistrations)(PhoneNetworkRestore_1.PhoneNetworkRestore, SessionMetricsService_1.SessionMetricsService);
+        order = [];
+        restoreSession = sinon_1.default.stub().callsFake((id) => __awaiter(this, void 0, void 0, function* () {
+            order.push(`restore ${id}`);
+        }));
+        restoreAll = sinon_1.default.stub().callsFake(() => __awaiter(this, void 0, void 0, function* () {
+            order.push('restore all');
+        }));
+        typedi_1.Container.set(PhoneNetworkRestore_1.PhoneNetworkRestore, { restoreSession, restoreAll });
+        typedi_1.Container.set(SessionMetricsService_1.SessionMetricsService, { stop: sinon_1.default.stub().resolves() });
+        sinon_1.default.stub(DeviceService, 'releaseSessionDevices').callsFake((id) => __awaiter(this, void 0, void 0, function* () {
+            order.push(`release ${id}`);
+        }));
+        sinon_1.default.stub(event_manager_1.DASHBORD_EVENT_MANAGER, 'onSessionStopped').resolves();
+    });
+    afterEach(() => {
+        sinon_1.default.restore();
+        restoreRegs();
+    });
+    it('when the test deletes a session Xenon keeps no record of', () => __awaiter(this, void 0, void 0, function* () {
+        (0, chai_1.expect)(SessionManager_1.SESSION_MANAGER.getSession('s-untracked')).to.equal(undefined);
+        const next = sinon_1.default.stub().callsFake(() => __awaiter(this, void 0, void 0, function* () {
+            order.push('driver quit');
+        }));
+        yield typedi_1.Container.get(SessionLifecycleService_1.SessionLifecycleService).deleteSession(next, 's-untracked');
+        (0, chai_1.expect)(restoreSession.callCount).to.equal(1);
+        (0, chai_1.expect)(order.indexOf('restore s-untracked')).to.be.lessThan(order.indexOf('release s-untracked'));
+    }));
+    it("when Appium ends the session itself (new-command timeout), on this server's own phone", () => __awaiter(this, void 0, void 0, function* () {
+        const plugin = { pluginArgs: {}, xenonLog: { withSession: () => ({ info() { } }) } };
+        yield plugin_1.XenonPlugin.prototype.onUnexpectedShutdown.call(plugin, { sessionId: 's-timeout', caps: { udid: 'R5CT' } }, new Error('New command timeout'));
+        (0, chai_1.expect)(restoreSession.callCount).to.equal(1);
+        (0, chai_1.expect)(restoreSession.firstCall.args[0]).to.equal('s-timeout');
+        (0, chai_1.expect)(order).to.deep.equal(['restore s-timeout', 'release s-timeout']);
+    }));
+    it('when Appium ends the session on a node', () => __awaiter(this, void 0, void 0, function* () {
+        const unblock = sinon_1.default.stub(NodeDevices_1.default.prototype, 'unblockDevice').callsFake(() => __awaiter(this, void 0, void 0, function* () {
+            order.push('hub unblock');
+            return undefined;
+        }));
+        const plugin = { pluginArgs: { hub: 'http://hub:4723' } };
+        yield plugin_1.XenonPlugin.prototype.onUnexpectedShutdown.call(plugin, { sessionId: 's-node', caps: {} }, null);
+        (0, chai_1.expect)(unblock.calledOnce).to.equal(true);
+        (0, chai_1.expect)(order).to.deep.equal(['restore s-node', 'hub unblock']);
+    }));
+    it("when Xenon's idle check releases the phone", () => __awaiter(this, void 0, void 0, function* () {
+        const idle = {
+            udid: 'R5CT',
+            host: 'http://127.0.0.1:4723',
+            busy: true,
+            userBlocked: false,
+            session_id: 's-idle',
+            claimSessionId: 's-idle',
+            lastCmdExecutedAt: Date.now() - 10 * 60 * 1000,
+        };
+        sinon_1.default.stub(DeviceService, 'getAllDevices').resolves([idle]);
+        sinon_1.default.stub(ActiveLeases, 'activeLeasesByDevice').resolves(new Map());
+        sinon_1.default.stub(DeviceService, 'releaseSessionDevice').callsFake((_u, _h, id) => __awaiter(this, void 0, void 0, function* () {
+            order.push(`release ${id}`);
+            return true;
+        }));
+        yield (0, device_utils_1.releaseBlockedDevices)(60);
+        (0, chai_1.expect)(restoreSession.callCount).to.equal(1);
+        (0, chai_1.expect)(order).to.deep.equal(['restore s-idle', 'release s-idle']);
+    }));
+    it('when a graceful shutdown drains a session', () => __awaiter(this, void 0, void 0, function* () {
+        yield typedi_1.Container.get(SessionLifecycleService_1.SessionLifecycleService).stopSessionForShutdown('s-down', 'shutdown');
+        (0, chai_1.expect)(order.slice(0, 2)).to.deep.equal(['restore s-down', 'release s-down']);
+    }));
+    it('when a graceful shutdown ends, for sessions Xenon keeps no record of', () => __awaiter(this, void 0, void 0, function* () {
+        sinon_1.default.stub(SessionManager_1.SESSION_MANAGER, 'getAllSessions').returns([]);
+        yield new ShutdownCoordinator_1.ShutdownCoordinator().drain(1000, 500);
+        (0, chai_1.expect)(restoreAll.calledOnce).to.equal(true);
+    }));
+    it("when a session's heartbeat goes stale", () => __awaiter(this, void 0, void 0, function* () {
+        var _a;
+        yield scratch.db.session.deleteMany({});
+        yield scratch.db.session.create({
+            data: {
+                id: 's-stale',
+                status: 'running',
+                desired_capabilities: '{}',
+                session_capabilities: '{}',
+                node_id: 'n',
+                has_live_video: false,
+                device_udid: 'R5CT',
+                device_platform: 'android',
+                device_version: '14',
+                last_heartbeat_at: new Date(Date.now() - 60 * 60 * 1000),
+            },
+        });
+        yield new OrphanSweeper_1.OrphanSweeper().sweep({ heartbeatIntervalMs: 1000 });
+        (0, chai_1.expect)(order.slice(0, 2)).to.deep.equal(['restore s-stale', 'release s-stale']);
+        (0, chai_1.expect)((_a = (yield prisma_1.prisma.session.findUnique({ where: { id: 's-stale' } }))) === null || _a === void 0 ? void 0 : _a.status).to.equal('failed');
+    }));
+});

package/lib/test/unit/phone-network/session-start.spec.js

@@ -0,0 +1,110 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+require("reflect-metadata");
+const chai_1 = require("chai");
+const sinon_1 = __importDefault(require("sinon"));
+const typedi_1 = require("typedi");
+const SessionLifecycleService_1 = require("../../../src/services/SessionLifecycleService");
+const NetworkConditioningService_1 = require("../../../src/services/NetworkConditioningService");
+const InterceptorService_1 = require("../../../src/services/InterceptorService");
+const PhoneNetworkRestore_1 = require("../../../src/services/network/PhoneNetworkRestore");
+const PluginContext_1 = require("../../../src/PluginContext");
+const IPluginArgs_1 = require("../../../src/interfaces/IPluginArgs");
+const XenonCapabilityManager_1 = require("../../../src/XenonCapabilityManager");
+const SessionManager_1 = require("../../../src/sessions/SessionManager");
+const SessionType_1 = __importDefault(require("../../../src/enums/SessionType"));
+const container_registration_1 = require("../../helpers/container-registration");
+const scratch_database_1 = require("../../helpers/scratch-database");
+const android = { udid: 'R5CT', platform: 'android', deviceType: 'real', host: 'h' };
+const sessionOf = (type) => ({
+    getId: () => 's1',
+    getType: () => type,
+    getCapabilities: () => ({}),
+    apiKeyId: null,
+    userId: null,
+    startVideoRecording: sinon_1.default.stub().resolves(),
+});
+const capsWith = (extra) => (Object.assign({ [XenonCapabilityManager_1.XENON_CAPABILITIES.VIDEO_RECORDING]: false, [XenonCapabilityManager_1.XENON_CAPABILITIES.INTERCEPTOR_MOCKS]: [], [XenonCapabilityManager_1.XENON_CAPABILITIES.INTERCEPTOR_INCLUDE_HOSTS]: [], [XenonCapabilityManager_1.XENON_CAPABILITIES.INTERCEPTOR_EXCLUDE_HOSTS]: [] }, extra));
+/**
+ * Where a session's network changes are made. Only the server that drives the
+ * phone makes them: a hub used to run its own adb against a node's phone (on
+ * one Mac, the very same phone), and start a second proxy for it.
+ */
+describe('a session’s network changes at its start', function () {
+    this.timeout(30000);
+    // The row a hub writes for a session another server runs (recordRoutedSession).
+    const scratch = (0, scratch_database_1.useScratchDatabase)();
+    let restoreRegs;
+    let order;
+    let applyProfile;
+    let start;
+    let restoreLeftovers;
+    let context;
+    beforeEach(() => {
+        restoreRegs = (0, container_registration_1.saveRegistrations)(NetworkConditioningService_1.NetworkConditioningService, InterceptorService_1.InterceptorService, PhoneNetworkRestore_1.PhoneNetworkRestore, PluginContext_1.PluginContext);
+        order = [];
+        applyProfile = sinon_1.default.stub().callsFake(() => __awaiter(this, void 0, void 0, function* () { return void order.push('profile'); }));
+        start = sinon_1.default.stub().callsFake(() => __awaiter(this, void 0, void 0, function* () { return void order.push('interceptor'); }));
+        restoreLeftovers = sinon_1.default.stub().callsFake(() => __awaiter(this, void 0, void 0, function* () {
+            order.push('leftovers');
+            return 0;
+        }));
+        typedi_1.Container.set(NetworkConditioningService_1.NetworkConditioningService, { applyProfile });
+        typedi_1.Container.set(InterceptorService_1.InterceptorService, { start });
+        typedi_1.Container.set(PhoneNetworkRestore_1.PhoneNetworkRestore, { restoreLeftovers });
+        context = new PluginContext_1.PluginContext();
+        context.pluginArgs = Object.assign(Object.assign({}, IPluginArgs_1.DefaultPluginArgs), { enableDashboard: false });
+        typedi_1.Container.set(PluginContext_1.PluginContext, context);
+    });
+    afterEach(() => __awaiter(this, void 0, void 0, function* () {
+        sinon_1.default.restore();
+        SessionManager_1.SESSION_MANAGER.removeSession('s1');
+        restoreRegs();
+        yield scratch.db.session.deleteMany({});
+    }));
+    const apply = (type, caps) => typedi_1.Container.get(SessionLifecycleService_1.SessionLifecycleService).applyPostSessionLogic(sessionOf(type), caps, android);
+    it('puts back what an earlier session left on the phone, then applies its own', () => __awaiter(this, void 0, void 0, function* () {
+        yield apply(SessionType_1.default.LOCAL, capsWith({
+            [XenonCapabilityManager_1.XENON_CAPABILITIES.NETWORK_PROFILE]: 'Offline',
+            [XenonCapabilityManager_1.XENON_CAPABILITIES.INTERCEPTOR_ENABLED]: true,
+        }));
+        (0, chai_1.expect)(restoreLeftovers.calledOnceWith({ udid: 'R5CT' })).to.equal(true);
+        (0, chai_1.expect)(order).to.deep.equal(['leftovers', 'profile', 'interceptor']);
+    }));
+    it("doesn't touch a phone another server drives", () => __awaiter(this, void 0, void 0, function* () {
+        for (const type of [SessionType_1.default.REMOTE, SessionType_1.default.CLOUD]) {
+            yield apply(type, capsWith({
+                [XenonCapabilityManager_1.XENON_CAPABILITIES.NETWORK_PROFILE]: 'Offline',
+                [XenonCapabilityManager_1.XENON_CAPABILITIES.INTERCEPTOR_ENABLED]: true,
+            }));
+            SessionManager_1.SESSION_MANAGER.removeSession('s1');
+        }
+        (0, chai_1.expect)(order).to.deep.equal([]);
+    }));
+    it("starts the interceptor by the server's option when the session doesn't say", () => __awaiter(this, void 0, void 0, function* () {
+        context.pluginArgs = Object.assign(Object.assign({}, context.pluginArgs), { interceptor: { enabled: true, bufferSize: 9 } });
+        yield apply(SessionType_1.default.LOCAL, capsWith({}));
+        (0, chai_1.expect)(start.calledOnce).to.equal(true);
+        const [sessionId, device, opts] = start.firstCall.args;
+        (0, chai_1.expect)(sessionId).to.equal('s1');
+        (0, chai_1.expect)(device).to.equal(android);
+        (0, chai_1.expect)(opts).to.include({ enabled: true, bufferSize: 9, captureBodies: true });
+    }));
+    it("leaves the interceptor off when the session turns it off over the server's option", () => __awaiter(this, void 0, void 0, function* () {
+        context.pluginArgs = Object.assign(Object.assign({}, context.pluginArgs), { interceptor: { enabled: true } });
+        yield apply(SessionType_1.default.LOCAL, capsWith({ [XenonCapabilityManager_1.XENON_CAPABILITIES.INTERCEPTOR_ENABLED]: false }));
+        (0, chai_1.expect)(start.called).to.equal(false);
+    }));
+});

package/lib/test/unit/phone-network/xenon-proxy.spec.js

@@ -0,0 +1,69 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+require("reflect-metadata");
+const chai_1 = require("chai");
+const net_1 = __importDefault(require("net"));
+const xenonProxy_1 = require("../../../src/services/network/xenonProxy");
+describe('a phone proxy setting, read', () => {
+    it('parses host:port', () => {
+        (0, chai_1.expect)((0, xenonProxy_1.parseProxySetting)('127.0.0.1:11100')).to.deep.equal({ host: '127.0.0.1', port: 11100 });
+        (0, chai_1.expect)((0, xenonProxy_1.parseProxySetting)(' 10.0.2.2:11105 ')).to.deep.equal({ host: '10.0.2.2', port: 11105 });
+    });
+    it('reads no proxy from what Android stores for none', () => {
+        for (const none of [null, undefined, '', 'null', ':0', '  ']) {
+            (0, chai_1.expect)((0, xenonProxy_1.isUnsetProxy)(none), String(none)).to.equal(true);
+            (0, chai_1.expect)((0, xenonProxy_1.parseProxySetting)(none)).to.equal(null);
+        }
+        (0, chai_1.expect)((0, xenonProxy_1.isUnsetProxy)('proxy.corp:3128')).to.equal(false);
+    });
+    it('refuses a value that is not host:port', () => {
+        (0, chai_1.expect)((0, xenonProxy_1.parseProxySetting)('proxy.corp')).to.equal(null);
+        (0, chai_1.expect)((0, xenonProxy_1.parseProxySetting)('a:b')).to.equal(null);
+        (0, chai_1.expect)((0, xenonProxy_1.parseProxySetting)('a:0')).to.equal(null);
+        (0, chai_1.expect)((0, xenonProxy_1.parseProxySetting)('a:70000')).to.equal(null);
+    });
+    it('knows the addresses a phone uses to reach this machine', () => {
+        const own = ['192.168.1.20'];
+        (0, chai_1.expect)((0, xenonProxy_1.pointsAtThisMachine)('127.0.0.1', own)).to.equal(true);
+        (0, chai_1.expect)((0, xenonProxy_1.pointsAtThisMachine)('localhost', own)).to.equal(true);
+        (0, chai_1.expect)((0, xenonProxy_1.pointsAtThisMachine)('10.0.2.2', own)).to.equal(true);
+        (0, chai_1.expect)((0, xenonProxy_1.pointsAtThisMachine)('192.168.1.20', own)).to.equal(true);
+        (0, chai_1.expect)((0, xenonProxy_1.pointsAtThisMachine)('192.168.1.21', own)).to.equal(false);
+        (0, chai_1.expect)((0, xenonProxy_1.pointsAtThisMachine)('proxy.corp', own)).to.equal(false);
+    });
+});
+describe('isPortListening', function () {
+    let server;
+    let livePort;
+    let deadPort;
+    before(() => __awaiter(this, void 0, void 0, function* () {
+        server = net_1.default.createServer((s) => s.destroy());
+        yield new Promise((r) => server.listen(0, '127.0.0.1', () => r()));
+        livePort = server.address().port;
+        const closed = net_1.default.createServer();
+        yield new Promise((r) => closed.listen(0, '127.0.0.1', () => r()));
+        deadPort = closed.address().port;
+        yield new Promise((r) => closed.close(() => r()));
+    }));
+    after(() => __awaiter(this, void 0, void 0, function* () {
+        yield new Promise((r) => server.close(() => r()));
+    }));
+    it('says a port with a listener answers', () => __awaiter(this, void 0, void 0, function* () {
+        (0, chai_1.expect)(yield (0, xenonProxy_1.isPortListening)(livePort)).to.equal(true);
+    }));
+    it('says a closed port does not', () => __awaiter(this, void 0, void 0, function* () {
+        (0, chai_1.expect)(yield (0, xenonProxy_1.isPortListening)(deadPort)).to.equal(false);
+    }));
+});

package/lib/test/unit/plugin.spec.js

@@ -81,7 +81,7 @@ describe('Device filter tests', () => {
         (0, chai_1.expect)(filter).to.deep.equal({
             platform: 'ios',
             platformVersion: '14.0',
-            name: 'iPhone',
+            appleFamily: 'iphone',
             deviceType: 'real',
             udid: ['21112-1111-1111-111'],
             minSDK: undefined,
@@ -113,7 +113,7 @@ describe('Device filter tests', () => {
         (0, chai_1.expect)(filter).to.deep.equal({
             platform: 'ios',
             platformVersion: '14.0',
-            name: 'iPhone',
+            appleFamily: 'iphone',
             filterByHost: undefined,
             deviceType: 'simulator',
             udid: [],
@@ -140,7 +140,7 @@ describe('Device filter tests', () => {
             platform: 'ios',
             filterByHost: undefined,
             platformVersion: undefined,
-            name: 'iPhone',
+            appleFamily: 'iphone',
             deviceType: 'simulator',
             udid: [],
             minSDK: '10.2.0',