@xenon-device-management/xenon 2.0.0 → 2.1.0

package/lib/test/helpers/scratch-database.js

@@ -0,0 +1,100 @@
+"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 });
+exports.useScratchDatabase = useScratchDatabase;
+const child_process_1 = require("child_process");
+const fs_1 = __importDefault(require("fs"));
+const os_1 = __importDefault(require("os"));
+const path_1 = __importDefault(require("path"));
+const sinon_1 = __importDefault(require("sinon"));
+const client_1 = require("../../src/generated/client");
+const prisma_1 = require("../../src/prisma");
+// Every model the `prisma` wrapper exposes (src/prisma.ts MODEL_DELEGATES).
+const MODELS = [
+    'build',
+    'session',
+    'sessionLog',
+    'log',
+    'profiling',
+    'app',
+    'device',
+    'pendingSession',
+    'cLIArgs',
+    'webhookConfig',
+    'webConfig',
+    'locatorEtalon',
+    'lease',
+    'portLease',
+    'apiKey',
+    'selectorState',
+    'user',
+    'userSession',
+    'passwordResetToken',
+    'team',
+    'teamMember',
+    'eventLog',
+    'project',
+    'recording',
+    'bookmark',
+    'annotation',
+];
+/**
+ * Back the whole `prisma` wrapper with a scratch SQLite file for the enclosing
+ * `describe`, so the real queries (the device store, the session lifecycle,
+ * SessionOwnerResolver) run against a real database that is not the server's.
+ * Like useScratchPortLeases, for every model.
+ *
+ * Call it inside a `describe`, never at the top of a file. The database is
+ * migrated once per suite; the stubs are made before each test and removed
+ * after it, so a spec's own `sinon.restore()` can't leave the next test on
+ * the real database. Tests empty the tables they use.
+ */
+function useScratchDatabase() {
+    const ctx = {};
+    const sandbox = sinon_1.default.createSandbox();
+    let dbPath = '';
+    before(function () {
+        this.timeout(90000);
+        dbPath = path_1.default.join(os_1.default.tmpdir(), `xenon-scratch-${process.pid}-${Date.now()}.db`);
+        const url = `file:${dbPath}`;
+        (0, child_process_1.execSync)('npx prisma migrate deploy', {
+            cwd: path_1.default.resolve(__dirname, '../..'),
+            env: Object.assign(Object.assign({}, process.env), { DATABASE_URL: url }),
+            stdio: 'pipe',
+        });
+        ctx.db = new client_1.PrismaClient({ datasources: { db: { url } } });
+    });
+    beforeEach(() => {
+        for (const model of MODELS) {
+            const wrapper = prisma_1.prisma[model];
+            const delegate = ctx.db[model];
+            for (const [name, value] of Object.entries(wrapper)) {
+                if (typeof value !== 'function' || typeof delegate[name] !== 'function')
+                    continue;
+                sandbox.stub(wrapper, name).callsFake((...args) => delegate[name](...args));
+            }
+        }
+    });
+    afterEach(() => sandbox.restore());
+    after(() => __awaiter(this, void 0, void 0, function* () {
+        sandbox.restore();
+        if (ctx.db)
+            yield ctx.db.$disconnect();
+        for (const f of [dbPath, `${dbPath}-journal`]) {
+            if (f && fs_1.default.existsSync(f))
+                fs_1.default.unlinkSync(f);
+        }
+    }));
+    return ctx;
+}

package/lib/test/helpers/scratch-port-leases.js

@@ -0,0 +1,119 @@
+"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 });
+exports.useScratchPortLeases = useScratchPortLeases;
+const child_process_1 = require("child_process");
+const fs_1 = __importDefault(require("fs"));
+const net_1 = __importDefault(require("net"));
+const os_1 = __importDefault(require("os"));
+const path_1 = __importDefault(require("path"));
+const sinon_1 = __importDefault(require("sinon"));
+const client_1 = require("../../src/generated/client");
+const prisma_1 = require("../../src/prisma");
+const METHODS = [
+    'findUnique',
+    'findFirst',
+    'findMany',
+    'create',
+    'update',
+    'upsert',
+    'delete',
+    'deleteMany',
+];
+/**
+ * Below every OS's ephemeral range (macOS 49152-65535, Linux 32768-60999).
+ * The blocks used to come from 40000-60000, half of it inside macOS's range,
+ * where the suite's own outgoing connections take ports: now and then one
+ * held base+1 while a two-port range needed it, and PortAllocator answered
+ * "range exhausted" in the full run only.
+ */
+const FIRST_BASE = 20000;
+const BLOCKS = 128;
+const USED_OFFSETS = [...Array(10).keys(), ...Array.from({ length: 10 }, (_, i) => 50 + i)];
+function bindable(port, host) {
+    return new Promise((resolve) => {
+        const server = net_1.default.createServer();
+        server.once('error', (err) => 
+        // A host without IPv6 can't have a listener there either.
+        resolve(err.code === 'EADDRNOTAVAIL' || err.code === 'EAFNOSUPPORT'));
+        server.listen(port, host, () => server.close(() => resolve(true)));
+    });
+}
+function blockIsFree(base) {
+    return __awaiter(this, void 0, void 0, function* () {
+        for (const offset of USED_OFFSETS) {
+            if (!(yield bindable(base + offset, '0.0.0.0')))
+                return false;
+            if (!(yield bindable(base + offset, '::')))
+                return false;
+        }
+        return true;
+    });
+}
+function freeBlock() {
+    return __awaiter(this, void 0, void 0, function* () {
+        let base = FIRST_BASE;
+        for (let attempt = 0; attempt < 20; attempt++) {
+            base = FIRST_BASE + Math.floor(Math.random() * BLOCKS) * 100;
+            if (yield blockIsFree(base))
+                return base;
+        }
+        return base;
+    });
+}
+/**
+ * Back `prisma.portLease` with a scratch SQLite file for the enclosing
+ * `describe`, so PortAllocator runs its real queries (unique constraint on
+ * `port` included) without touching the server's database.
+ *
+ * Call it inside a `describe`, never at the top of a file. The database is
+ * created once per suite; the table is emptied and the stubs are re-made
+ * before each test, so a spec's own `sinon.restore()` can't leave the next
+ * test pointed at the real database.
+ */
+function useScratchPortLeases() {
+    const ctx = {};
+    const sandbox = sinon_1.default.createSandbox();
+    let dbPath = '';
+    before(function () {
+        this.timeout(60000);
+        dbPath = path_1.default.join(os_1.default.tmpdir(), `xenon-port-leases-${process.pid}-${Date.now()}.db`);
+        const url = `file:${dbPath}`;
+        (0, child_process_1.execSync)('npx prisma migrate deploy', {
+            cwd: path_1.default.resolve(__dirname, '../..'),
+            env: Object.assign(Object.assign({}, process.env), { DATABASE_URL: url }),
+            stdio: 'pipe',
+        });
+        ctx.db = new client_1.PrismaClient({ datasources: { db: { url } } });
+    });
+    beforeEach(() => __awaiter(this, void 0, void 0, function* () {
+        yield ctx.db.portLease.deleteMany({});
+        ctx.base = yield freeBlock();
+        const wrapper = prisma_1.prisma.portLease;
+        for (const m of METHODS) {
+            sandbox.stub(wrapper, m).callsFake((...args) => ctx.db.portLease[m](...args));
+        }
+    }));
+    afterEach(() => sandbox.restore());
+    after(() => __awaiter(this, void 0, void 0, function* () {
+        sandbox.restore();
+        if (ctx.db)
+            yield ctx.db.$disconnect();
+        for (const f of [dbPath, `${dbPath}-journal`]) {
+            if (f && fs_1.default.existsSync(f))
+                fs_1.default.unlinkSync(f);
+        }
+    }));
+    return ctx;
+}

package/lib/test/helpers/test-container.js

@@ -23,20 +23,38 @@ const PluginContext_1 = require("../../src/PluginContext");
 const IPluginArgs_1 = require("../../src/interfaces/IPluginArgs");
 const uuid_1 = require("uuid");
 const db_1 = require("../../src/data-service/db");
-const device_store_1 = require("../../src/data-service/device-store");
 const AndroidDeviceManager_1 = __importDefault(require("../../src/device-managers/AndroidDeviceManager"));
 const IOSDeviceManager_1 = __importDefault(require("../../src/device-managers/IOSDeviceManager"));
 const device_managers_1 = require("../../src/device-managers");
-const sinon_1 = __importDefault(require("sinon"));
 const IOSDiscoveryService_1 = require("../../src/device-managers/ios/IOSDiscoveryService");
-const sandbox = sinon_1.default.createSandbox();
+const container_registration_1 = require("./container-registration");
 /**
- * Test utility to initialize the TypeDI Container with a PluginContext
- * configured for testing purposes.
+ * The classes setupTestContainer() builds. Each call makes new ones, with its
+ * own PluginContext; the ones already registered are put back by
+ * resetTestContainer().
+ */
+const BUILT = [
+    PluginContext_1.PluginContext,
+    IOSDiscoveryService_1.IOSDiscoveryService,
+    AndroidDeviceManager_1.default,
+    IOSDeviceManager_1.default,
+    device_managers_1.XenonManager,
+];
+/** What each setupTestContainer() call not yet undone replaced, newest last. */
+const undo = [];
+/**
+ * Registers a PluginContext configured for testing, and the device managers
+ * built with it. Undo it with resetTestContainer() in the same `describe`.
+ *
+ * Only the ids it builds are replaced. It used to Container.reset(), which
+ * discarded every singleton in the process, including the ones other specs
+ * and the server's own modules had built and still held.
  */
 function setupTestContainer(overrides) {
-    // Reset container to ensure clean state between tests
-    typedi_1.Container.reset();
+    undo.push((0, container_registration_1.saveRegistrations)('LocalStorage', ...BUILT));
+    // Unbuilt again, so the Container.get() calls below make new instances.
+    for (const id of BUILT)
+        typedi_1.Container.set({ id, type: id });
     const nodeId = (0, uuid_1.v4)();
     const port = 4723;
     const pluginArgs = Object.assign({}, IPluginArgs_1.DefaultPluginArgs, overrides || {});
@@ -85,28 +103,18 @@ function createTestXenonManager(pluginArgs) {
     xenonManager.init();
     return xenonManager;
 }
+/**
+ * Puts back every registration the setupTestContainer() calls since the last
+ * reset replaced, newest first, and empties the in-memory device database.
+ *
+ * It changes nothing else. It used to stub device discovery on the
+ * AndroidDeviceManager and IOSDiscoveryService prototypes and never restore
+ * the stubs, so every spec that ran later got no devices from discovery.
+ */
 function resetTestContainer() {
     return __awaiter(this, void 0, void 0, function* () {
-        sandbox.restore();
-        typedi_1.Container.reset();
-        // Stub discovery methods to prevent background pollution
-        sandbox.stub(AndroidDeviceManager_1.default.prototype, 'getDevices').resolves([]);
-        sandbox.stub(IOSDiscoveryService_1.IOSDiscoveryService.prototype, 'getDevices').resolves([]);
-        // @ts-ignore - Stubbing internal methods to be extra safe
-        sandbox.stub(AndroidDeviceManager_1.default.prototype, 'fetchAndroidDevices').resolves([]);
-        // @ts-ignore
-        sandbox.stub(IOSDiscoveryService_1.IOSDiscoveryService.prototype, 'fetchLocalIOSDevices').resolves([]);
-        // @ts-ignore
-        sandbox.stub(IOSDiscoveryService_1.IOSDiscoveryService.prototype, 'fetchLocalSimulators').resolves([]);
-        // Clear DeviceStoreFactory static caches to prevent cross-test pollution
-        // @ts-ignore - Accessing private static members for test cleanup
-        device_store_1.DeviceStoreFactory._deviceStore = undefined;
-        // @ts-ignore
-        device_store_1.DeviceStoreFactory._pendingSessionStore = undefined;
-        // @ts-ignore
-        device_store_1.DeviceStoreFactory._cliArgsStore = undefined;
-        // @ts-ignore
-        device_store_1.DeviceStoreFactory._healEtalonStore = undefined;
+        for (const restore of undo.splice(0).reverse())
+            restore();
         yield db_1.XenonDatabase.reset();
     });
 }

package/lib/test/integration/androidDevices.spec.js

@@ -49,6 +49,7 @@ require("reflect-metadata");
 const chai_1 = __importStar(require("chai"));
 const db_1 = require("../../src/data-service/db");
 const device_utils_1 = require("../../src/device-utils");
+const localDeviceHosts_1 = require("../../src/device-managers/localDeviceHosts");
 const IPluginArgs_1 = require("../../src/interfaces/IPluginArgs");
 const ip_1 = __importDefault(require("ip"));
 const async_wait_until_1 = __importDefault(require("async-wait-until"));
@@ -73,7 +74,7 @@ describe('Android Test', () => {
         yield (0, device_utils_1.initializeStorage)();
         yield deviceManager.getDevices();
         const hub = pluginArgs.hub;
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp, hub);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723), hub);
         yield (0, device_utils_1.cleanPendingSessions)(0);
         yield (0, device_service_1.unblockDeviceMatchingFilter)({});
         const capabilities = {
@@ -95,7 +96,7 @@ describe('Android Test', () => {
     it.skip('Allocate second free device and verify both the device state is busy in db', () => __awaiter(void 0, void 0, void 0, function* () {
         yield (0, device_utils_1.initializeStorage)();
         deviceManager = (0, test_container_1.createTestXenonManager)(Object.assign({}, pluginArgs, { platform: 'android' }));
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723));
         const capabilities = {
             alwaysMatch: {
                 platformName: 'android',
@@ -108,7 +109,7 @@ describe('Android Test', () => {
         // wait until there are two devices and both are not offline
         let devices;
         (0, async_wait_until_1.default)(() => __awaiter(void 0, void 0, void 0, function* () {
-            yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp);
+            yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723));
             devices = yield deviceManager.getDevices();
             return devices.length === 2 && devices.every((device) => !device.offline);
         }));
@@ -120,7 +121,7 @@ describe('Android Test', () => {
         yield (0, device_utils_1.initializeStorage)();
         deviceManager = (0, test_container_1.createTestXenonManager)(pluginArgs);
         const hub = pluginArgs.hub;
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp, hub);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723), hub);
         const capabilities = {
             alwaysMatch: {
                 platformName: 'android',

package/lib/test/integration/ios/01iOSSimulator.spec.js

@@ -22,6 +22,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 require("reflect-metadata");
 const chai_1 = require("chai");
 const device_utils_1 = require("../../../src/device-utils");
+const localDeviceHosts_1 = require("../../../src/device-managers/localDeviceHosts");
 const db_1 = require("../../../src/data-service/db");
 const node_simctl_1 = __importDefault(require("node-simctl"));
 const pluginArgs_1 = require("../../../src/data-service/pluginArgs");
@@ -56,7 +57,7 @@ function initDeviceFarm(iosDeviceType) {
         }));
         (0, chai_1.expect)(deviceManager.getMaxSessionCount()).to.be.eql(1);
         const hub = pluginArgs.hub;
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp, hub);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723), hub);
         yield markSimulatorsAsBooted();
         yield (0, device_service_1.unblockDeviceMatchingFilter)({});
         yield (0, device_utils_1.cleanPendingSessions)(0);
@@ -155,7 +156,7 @@ describe('IOS Simulator Test', () => {
         yield (0, device_utils_1.initializeStorage)();
         const deviceManager = (0, test_container_1.createTestXenonManager)(pluginArgs);
         const hub = pluginArgs.hub;
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp, hub);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723), hub);
         yield markSimulatorsAsBooted();
         yield (0, device_service_1.unblockDeviceMatchingFilter)({});
         yield (0, device_utils_1.cleanPendingSessions)(0);
@@ -184,7 +185,7 @@ describe('IOS Simulator Test', () => {
         yield (0, device_utils_1.initializeStorage)();
         const deviceManager = (0, test_container_1.createTestXenonManager)(pluginArgs);
         const hub = pluginArgs.hub;
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp, hub);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723), hub);
         yield markSimulatorsAsBooted();
         yield (0, device_service_1.unblockDeviceMatchingFilter)({});
         yield (0, device_utils_1.cleanPendingSessions)(0);
@@ -215,7 +216,7 @@ describe('IOS Simulator Test', () => {
                 yield (0, device_utils_1.initializeStorage)();
                 const deviceManager = (0, test_container_1.createTestXenonManager)(pluginArgs);
                 const hub = pluginArgs.hub;
-                yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp, hub);
+                yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723), hub);
                 yield markSimulatorsAsBooted();
                 yield (0, device_service_1.unblockDeviceMatchingFilter)({});
                 yield (0, device_utils_1.cleanPendingSessions)(0);
@@ -265,7 +266,7 @@ describe('Boot simulator test', () => __awaiter(void 0, void 0, void 0, function
         yield (0, device_utils_1.initializeStorage)();
         const deviceManager = (0, test_container_1.createTestXenonManager)(pluginArgs);
         const hub = pluginArgs.hub;
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp, hub);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723), hub);
         const capabilities = {
             alwaysMatch: {
                 platformName: 'iOS',

package/lib/test/integration/ios/02iOSDevices.spec.js

@@ -16,6 +16,7 @@ require("reflect-metadata");
 const chai_1 = require("chai");
 const ip_1 = __importDefault(require("ip"));
 const device_utils_1 = require("../../../src/device-utils");
+const localDeviceHosts_1 = require("../../../src/device-managers/localDeviceHosts");
 const db_1 = require("../../../src/data-service/db");
 const IPluginArgs_1 = require("../../../src/interfaces/IPluginArgs");
 const device_service_1 = require("../../../src/data-service/device-service");
@@ -39,7 +40,7 @@ describe('IOS Test', () => {
             d.plugin['xenon'].iosDeviceType = 'real';
         });
         const deviceManager = (0, test_container_1.createTestXenonManager)(pluginArgs);
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723));
         const capabilities = {
             alwaysMatch: {
                 platformName: 'iOS',
@@ -57,7 +58,7 @@ describe('IOS Test', () => {
     it('Should throw error if the IPA does not match with device type real', () => __awaiter(void 0, void 0, void 0, function* () {
         yield (0, device_utils_1.initializeStorage)();
         const deviceManager = (0, test_container_1.createTestXenonManager)(pluginArgs);
-        yield (0, device_utils_1.updateDeviceList)(pluginArgs.bindHostOrIp);
+        yield (0, device_utils_1.updateDeviceList)((0, localDeviceHosts_1.localDeviceHosts)(pluginArgs, 4723));
         const capabilities = {
             alwaysMatch: {
                 platformName: 'iOS',

package/lib/test/integration/prisma-device-store-add-devices.spec.js

@@ -65,10 +65,20 @@ describe('PrismaDeviceStore.addDevices (integration)', function () {
         (0, chai_1.expect)(yield row(NEW)).to.include({ name: 'Pixel 8', busy: false });
         (0, chai_1.expect)(yield row(KNOWN)).to.include({ busy: true });
     }));
-    it('with mirror, writes every column the node sent', () => __awaiter(this, void 0, void 0, function* () {
-        yield store.addDevices([phone(KNOWN, { busy: false, session_id: 'node-sess' })], {
-            mirror: true,
+    it("as a node's report, records the node's busy apart and never writes the session", () => __awaiter(this, void 0, void 0, function* () {
+        yield prisma_1.prisma.device.updateMany({
+            where: { udid: KNOWN, host: HOST },
+            data: { claimSessionId: 'sess-1', claimedAt: 41 },
+        });
+        yield store.addDevices([phone(KNOWN, { busy: false, session_id: 'node-sess', sdk: '11' })], {
+            nodeReport: true,
+        });
+        (0, chai_1.expect)(yield row(KNOWN)).to.include({
+            sdk: '11',
+            nodeBusy: false,
+            busy: true,
+            session_id: 'sess-1',
+            claimSessionId: 'sess-1',
         });
-        (0, chai_1.expect)(yield row(KNOWN)).to.include({ busy: false, session_id: 'node-sess' });
     }));
 });

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

@@ -18,9 +18,13 @@ const adbkit_1 = __importDefault(require("@devicefarmer/adbkit"));
 const appium_adb_1 = require("appium-adb");
 const chai_1 = require("chai");
 const test_container_1 = require("../helpers/test-container");
+const scratch_database_1 = require("../helpers/scratch-database");
 const GetAdbOriginal_1 = require("./GetAdbOriginal");
 const sandbox = sinon_1.default.createSandbox();
 describe('Android Device Manager', () => {
+    // deviceInfo() leases the phone's ports through Prisma. Without a database
+    // of its own that was the developer's ~/.cache/xenon/xenon.db.
+    (0, scratch_database_1.useScratchDatabase)();
     let adb;
     beforeEach(() => __awaiter(void 0, void 0, void 0, function* () {
         sandbox.restore();
@@ -37,7 +41,10 @@ describe('Android Device Manager', () => {
         });
     }));
     afterEach(function () {
-        sandbox.restore();
+        return __awaiter(this, void 0, void 0, function* () {
+            sandbox.restore();
+            yield (0, test_container_1.resetTestContainer)();
+        });
     });
     it('Android Device List to have added state', () => __awaiter(void 0, void 0, void 0, function* () {
         const androidDevices = (0, test_container_1.createTestAndroidManager)({ platform: 'android' });

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

@@ -19,18 +19,20 @@ const device_service_1 = require("../../src/data-service/device-service");
 const devices_1 = require("./fixtures/devices");
 const chai_2 = require("chai");
 const device_service_2 = require("../../src/data-service/device-service");
-const sinon_1 = __importDefault(require("sinon"));
-chai_1.default.should();
-const sandbox = sinon_1.default.createSandbox();
 const test_container_1 = require("../helpers/test-container");
-describe('Model Test', () => __awaiter(void 0, void 0, void 0, function* () {
+const loki_stores_1 = require("../helpers/loki-stores");
+chai_1.default.should();
+describe('Model Test', () => {
+    // The in-memory store this spec is about. Run on its own, without the
+    // NODE_ENV=test that test:all sets, it used the Prisma store instead: the
+    // developer's own database, whose tables it wrote to.
+    (0, loki_stores_1.useLokiStores)();
     before('Add device collection', () => __awaiter(void 0, void 0, void 0, function* () {
         yield (0, test_container_1.resetTestContainer)();
         (0, chai_2.expect)((yield db_1.XenonDatabase.DeviceModel).chain().find().data().length).to.be.equal(0);
         (0, chai_2.expect)(devices_1.deviceMock.length).to.be.greaterThanOrEqual(1);
         // console.log(`deviceMock length: ${deviceMock.length}`);
         (yield db_1.XenonDatabase.DeviceModel).insert(devices_1.deviceMock);
-        (yield db_1.XenonDatabase.db).saveDatabase();
         // console.log(`result: ${result}`);
         // console.log(`device model length: ${deviceModel.find().length}`);
         (yield db_1.XenonDatabase.DeviceModel)
@@ -39,9 +41,7 @@ describe('Model Test', () => __awaiter(void 0, void 0, void 0, function* () {
             .data()
             .length.should.be.equal(devices_1.deviceMock.length);
     }));
-    after('clean', () => {
-        sandbox.restore();
-    });
+    after('clean', () => (0, test_container_1.resetTestContainer)());
     it('Should remove device from old pool when new poll call does not have the device', () => __awaiter(void 0, void 0, void 0, function* () {
         const findDevice = (yield db_1.XenonDatabase.DeviceModel)
             .chain()
@@ -157,4 +157,4 @@ describe('Model Test', () => __awaiter(void 0, void 0, void 0, function* () {
         const updatedDeviceUdids = updatedDeviceList.map((device) => device.udid);
         addedDeviceUdids.forEach((udid) => updatedDeviceUdids.includes(udid).should.be.true);
     }));
-}));
+});

package/lib/test/unit/HealingCascade.test.js

@@ -9,12 +9,14 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+require("reflect-metadata");
 const chai_1 = require("chai");
 const HealingOrchestrator_1 = require("../../src/services/healing/HealingOrchestrator");
 const types_1 = require("../../src/services/healing/types");
 const typedi_1 = require("typedi");
 const HealEtalonService_1 = require("../../src/services/healing/HealEtalonService");
 const HealedLocatorGenerator_1 = require("../../src/services/healing/HealedLocatorGenerator");
+const container_registration_1 = require("../helpers/container-registration");
 // Mock Tesseract
 const TesseractMock = {
     recognize: () => __awaiter(void 0, void 0, void 0, function* () {
@@ -38,8 +40,9 @@ describe('Healing Orchestrator - Cascade Strategy', () => {
     let orchestrator;
     let mockEtalonService;
     let mockGenerator;
+    let restoreContainer;
     beforeEach(() => {
-        typedi_1.Container.reset();
+        restoreContainer = (0, container_registration_1.saveRegistrations)(HealEtalonService_1.HealEtalonService, HealedLocatorGenerator_1.HealedLocatorGenerator);
         mockEtalonService = {
             getSignature: () => __awaiter(void 0, void 0, void 0, function* () { return null; }),
             saveSignature: () => __awaiter(void 0, void 0, void 0, function* () { }),
@@ -49,6 +52,7 @@ describe('Healing Orchestrator - Cascade Strategy', () => {
         typedi_1.Container.set(HealedLocatorGenerator_1.HealedLocatorGenerator, mockGenerator);
         orchestrator = new HealingOrchestrator_1.HealingOrchestrator(mockEtalonService);
     });
+    afterEach(() => restoreContainer());
     it('should cascade to OCR (Tier 3) when Fuzzy XML (Tier 2) fails to find a high-confidence match', () => __awaiter(void 0, void 0, void 0, function* () {
         // 1. Setup a page source that has NO matching text or buttons for "Login"
         // This will cause Fuzzy XML Provider to return null (or score below 0.5 threshold)

package/lib/test/unit/OrphanSweeper.test.js

@@ -48,22 +48,29 @@ Object.defineProperty(exports, "__esModule", { value: true });
 require("reflect-metadata");
 const chai_1 = require("chai");
 const sinon_1 = __importDefault(require("sinon"));
-const prisma_1 = require("../../src/prisma");
 const test_container_1 = require("../helpers/test-container");
+const scratch_database_1 = require("../helpers/scratch-database");
+const device_store_1 = require("../../src/data-service/device-store");
+const prisma_store_1 = require("../../src/data-service/prisma-store");
 const event_manager_1 = require("../../src/dashboard/event-manager");
 const SessionStatus_1 = require("../../src/types/SessionStatus");
 describe('OrphanSweeper', () => {
-    let prismaClient;
+    // Its own migrated database, never the developer's ~/.cache/xenon/xenon.db:
+    // that file's schema is whatever the last local migrate left, so a new
+    // column failed this spec there and nowhere else.
+    const scratch = (0, scratch_database_1.useScratchDatabase)();
+    let savedStore;
     const STALE_SESSION_ID = 'orphan-stale-sess-001';
     const STALE_UDID = 'udid-orphan-1';
     const NODE_ID = 'node-a';
     const FRESH_SESSION_ID = 'orphan-fresh-sess-002';
     before(() => __awaiter(void 0, void 0, void 0, function* () {
         (0, test_container_1.setupTestContainer)();
-        prismaClient = (0, prisma_1.getPrismaClient)();
+        savedStore = device_store_1.DeviceStoreFactory._deviceStore;
+        device_store_1.DeviceStoreFactory._deviceStore = new prisma_store_1.PrismaDeviceStore();
         // Create a stale session (last_heartbeat_at is old / null with old updatedAt)
         const oldDate = new Date(Date.now() - 5 * 60 * 1000); // 5 minutes ago
-        yield prismaClient.session.upsert({
+        yield scratch.db.session.upsert({
             where: { id: STALE_SESSION_ID },
             create: {
                 id: STALE_SESSION_ID,
@@ -87,7 +94,7 @@ describe('OrphanSweeper', () => {
             },
         });
         // Create a device for the stale session
-        yield prismaClient.device.upsert({
+        yield scratch.db.device.upsert({
             where: { udid_host: { udid: STALE_UDID, host: 'http://localhost:4723' } },
             create: {
                 udid: STALE_UDID,
@@ -105,10 +112,7 @@ describe('OrphanSweeper', () => {
         });
     }));
     after(() => __awaiter(void 0, void 0, void 0, function* () {
-        yield prismaClient.session.deleteMany({
-            where: { id: { in: [STALE_SESSION_ID, FRESH_SESSION_ID] } },
-        });
-        yield prismaClient.device.deleteMany({ where: { udid: STALE_UDID } });
+        device_store_1.DeviceStoreFactory._deviceStore = savedStore;
         yield (0, test_container_1.resetTestContainer)();
     }));
     afterEach(() => {
@@ -122,13 +126,13 @@ describe('OrphanSweeper', () => {
         const { OrphanSweeper } = yield Promise.resolve().then(() => __importStar(require('../../src/services/OrphanSweeper')));
         const sweeper = new OrphanSweeper();
         yield sweeper.sweep({ heartbeatIntervalMs: 30000 });
-        const session = yield prismaClient.session.findUnique({
+        const session = yield scratch.db.session.findUnique({
             where: { id: STALE_SESSION_ID },
             select: { status: true, failure_reason: true },
         });
         (0, chai_1.expect)(session.status).to.equal('failed');
         (0, chai_1.expect)(session.failure_reason).to.match(/heartbeat timeout/i);
-        const device = yield prismaClient.device.findFirst({
+        const device = yield scratch.db.device.findFirst({
             where: { udid: STALE_UDID },
             select: { busy: true, owningSessionId: true },
         });
@@ -138,7 +142,7 @@ describe('OrphanSweeper', () => {
     }));
     it('leaves fresh sessions alone', () => __awaiter(void 0, void 0, void 0, function* () {
         // Create a fresh session (heartbeat just now)
-        yield prismaClient.session.upsert({
+        yield scratch.db.session.upsert({
             where: { id: FRESH_SESSION_ID },
             create: {
                 id: FRESH_SESSION_ID,
@@ -164,7 +168,7 @@ describe('OrphanSweeper', () => {
         const { OrphanSweeper } = yield Promise.resolve().then(() => __importStar(require('../../src/services/OrphanSweeper')));
         const sweeper = new OrphanSweeper();
         yield sweeper.sweep({ heartbeatIntervalMs: 30000 });
-        const session = yield prismaClient.session.findUnique({
+        const session = yield scratch.db.session.findUnique({
             where: { id: FRESH_SESSION_ID },
             select: { status: true },
         });