@itwin/core-backend 5.9.0-dev.4 → 5.9.0-dev.5

package/lib/esm/test/ecdb/QueryReaders.test.js

@@ -54,7 +54,7 @@ var __disposeResources = (this && this.__disposeResources) || (function (Suppres
 * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
-import { DbResult } from "@itwin/core-bentley";
+import { DbResult, Id64 } from "@itwin/core-bentley";
 import { Code, ColorDef, IModel, QueryBinder, QueryOptionsBuilder, QueryRowFormat } from "@itwin/core-common";
 import { DefinitionModel, ElementTreeDeleter, ElementTreeWalkerScope, PhysicalModel, PhysicalObject, SnapshotDb, Subject } from "../../core-backend";
 import { IModelTestUtils } from "../IModelTestUtils";
@@ -63,6 +63,7 @@ import { ECDbTestHelper } from "./ECDbTestHelper";
 import { Range3d } from "@itwin/core-geometry";
 import * as chai from "chai";
 import * as chaiAsPromised from "chai-as-promised";
+import { withEditTxn } from "../../EditTxn";
 chai.use(chaiAsPromised);
 const assert = chai.assert;
 const expect = chai.expect;
@@ -1377,7 +1378,7 @@ describe("createQueryReader vs withQueryReader ", () => {
         _processCodeScopes(element, scope) {
             const newScope = new ElementTreeWalkerScope(scope, element);
             // eslint-disable-next-line @typescript-eslint/no-deprecated
-            this._iModel.withPreparedStatement(`
+            this.txn.iModel.withPreparedStatement(`
         SELECT ECInstanceId
         FROM bis.Element
         WHERE CodeScope.id=?
@@ -1398,9 +1399,11 @@ describe("createQueryReader vs withQueryReader ", () => {
         const testIModelDb = SnapshotDb.createEmpty(pathForEmpty, {
             rootSubject: { name: "empty " },
         });
-        const subjectId = Subject.insert(testIModelDb, IModel.rootSubjectId, "Subject", "Subject Description");
-        const physicalModelId = PhysicalModel.insert(testIModelDb, subjectId, "Physical");
-        const definitionModelId = DefinitionModel.insert(testIModelDb, subjectId, "Definition");
+        const subjectId = withEditTxn(testIModelDb, (txn) => Subject.insert(txn, IModel.rootSubjectId, "Subject", "Subject Description"));
+        const [physicalModelId, definitionModelId] = withEditTxn(testIModelDb, (txn) => [
+            PhysicalModel.insert(txn, subjectId, "Physical"),
+            DefinitionModel.insert(txn, subjectId, "Definition"),
+        ]);
         const spatialCategoryId = IModelTestUtils.insertSpatialCategory(testIModelDb, definitionModelId, "SpatialCategory", ColorDef.green);
         const physicalObjectProps5 = {
             classFullName: PhysicalObject.classFullName,
@@ -1409,16 +1412,18 @@ describe("createQueryReader vs withQueryReader ", () => {
             code: Code.createEmpty(),
             userLabel: "ScopingElement",
         };
-        const scopingElement = testIModelDb.elements.insertElement(physicalObjectProps5);
         const childElement = {
             classFullName: PhysicalObject.classFullName,
             model: physicalModelId,
             category: spatialCategoryId,
-            code: { spec: "0x1", scope: scopingElement },
+            code: { spec: "0x1", scope: Id64.invalid },
             userLabel: "ScopedElement",
         };
-        testIModelDb.elements.insertElement(childElement);
-        testIModelDb.saveChanges();
+        withEditTxn(testIModelDb, (txn) => {
+            const scopingElement = txn.insertElement(physicalObjectProps5);
+            childElement.code = { spec: "0x1", scope: scopingElement };
+            txn.insertElement(childElement);
+        });
         return testIModelDb;
     }
     beforeEach(async () => {
@@ -1433,14 +1438,13 @@ describe("createQueryReader vs withQueryReader ", () => {
     FROM ${PhysicalObject.classFullName}
     `;
         const reader = iModelDb.createQueryReader(sql, undefined, { usePrimaryConn: true });
-        const elementTreeDeleter = new TestElementCascadingDeleter(iModelDb);
         await reader.step(); // step to initialize reader
         const firstId = reader.current[0];
-        elementTreeDeleter.deleteNormalElements(firstId);
+        withEditTxn(iModelDb, (txn) => new TestElementCascadingDeleter(txn).deleteNormalElements(firstId));
         await reader.step(); // step to initialize reader
         const secondId = reader.current[0];
         // This is because ecsqlreader built using createQueryReader caches results and so when it tries to access the second element, it is already deleted from the database and it throws "Not Found" error.
-        expect(() => elementTreeDeleter.deleteNormalElements(secondId)).to.throw();
+        expect(() => withEditTxn(iModelDb, (txn) => new TestElementCascadingDeleter(txn).deleteNormalElements(secondId))).to.throw();
     });
     it("Passing while using withQueryReader()", async () => {
         const sql = `
@@ -1448,11 +1452,10 @@ describe("createQueryReader vs withQueryReader ", () => {
     FROM ${PhysicalObject.classFullName}
     `;
         iModelDb.withQueryReader(sql, (reader) => {
-            const elementTreeDeleter = new TestElementCascadingDeleter(iModelDb);
             let cntSteps = 0;
             while (reader.step()) {
                 const id = reader.current[0];
-                elementTreeDeleter.deleteNormalElements(id);
+                withEditTxn(iModelDb, (txn) => new TestElementCascadingDeleter(txn).deleteNormalElements(id));
                 cntSteps++;
             }
             assert.equal(cntSteps, 1);