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

package/lib/esm/test/IModelTestUtils.js

@@ -8,11 +8,13 @@ import { assert, expect } from "chai";
 import * as chaiAsPromised from "chai-as-promised";
 import * as path from "path";
 import { BeEvent, DbResult, Guid, Id64, IModelStatus, omit } from "@itwin/core-bentley";
+import { withEditTxn } from "../EditTxn";
 import { Base64EncodedString, Code, CodeScopeSpec, ColorDef, Environment, FontType, GeometryParams, GeometryStreamBuilder, ImageSourceFormat, IModel, IModelError, IModelReadRpcInterface, IModelVersion, PlanProjectionSettings, RelatedElement, RpcConfiguration, RpcManager, RpcPendingResponse, SkyBoxImageType, SubCategoryAppearance, SubCategoryOverride, SyncMode, } from "@itwin/core-common";
 import { Box, Cone, LineString3d, Point2d, Point3d, Range2d, Range3d, StandardViewIndex, Vector3d, YawPitchRollAngles } from "@itwin/core-geometry";
 import { CheckpointManager, V2CheckpointManager } from "../CheckpointManager";
 import { ClassRegistry } from "../ClassRegistry";
 import { _nativeDb, AuxCoordSystem2d, BriefcaseDb, BriefcaseLocalValue, BriefcaseManager, CategorySelector, ChannelControl, DisplayStyle2d, DisplayStyle3d, DrawingCategory, DrawingViewDefinition, Element, ElementOwnsChildElements, ElementOwnsMultiAspects, ElementOwnsUniqueAspect, ExternalSource, ExternalSourceIsInRepository, FunctionalModel, FunctionalSchema, GroupModel, IModelDb, IModelHost, IModelJsFs, InformationPartitionElement, Model, ModelSelector, OrthographicViewDefinition, PhysicalModel, PhysicalObject, PhysicalPartition, RenderMaterialElement, SnapshotDb, SpatialCategory, SubCategory, SubjectOwnsPartitionElements, Texture, ViewDefinition, } from "../core-backend";
+import { _hubAccess } from "../internal/Symbols";
 import { DefinitionPartition, Drawing, DrawingGraphic, GeometryPart, LinkElement, PhysicalElement, RepositoryLink, Subject } from "../Element";
 import { DefinitionModel, DocumentListModel, DrawingModel, InformationRecordModel, SpatialLocationModel } from "../Model";
 import { DrawingGraphicRepresentsElement, ElementDrivesElement } from "../Relationship";
@@ -20,7 +22,6 @@ import { RpcBriefcaseUtility } from "../rpc-impl/RpcBriefcaseUtility";
 import { Schema, Schemas } from "../Schema";
 import { HubMock } from "../internal/HubMock";
 import { KnownTestLocations } from "./KnownTestLocations";
-import { _hubAccess } from "../internal/Symbols";
 chai.use(chaiAsPromised);
 /* eslint-disable @typescript-eslint/explicit-member-accessibility */
 /* eslint-disable @typescript-eslint/no-non-null-assertion */
@@ -34,15 +35,15 @@ export class TestElementDrivesElement extends ElementDrivesElement {
     static get className() { return "TestElementDrivesElement"; }
     static rootChanged = new BeEvent();
     static deletedDependency = new BeEvent();
-    static onRootChanged(props, imodel) { this.rootChanged.raiseEvent(props, imodel); }
-    static onDeletedDependency(props, imodel) { this.deletedDependency.raiseEvent(props, imodel); }
+    static onRootChangedArg(arg) { this.rootChanged.raiseEvent(arg); }
+    static onDeletedDependencyArg(arg) { this.deletedDependency.raiseEvent(arg); }
 }
 export class TestPhysicalObject extends PhysicalElement {
     static get className() { return "TestPhysicalObject"; }
     static beforeOutputsHandled = new BeEvent();
     static allInputsHandled = new BeEvent();
-    static onBeforeOutputsHandled(id, imodel) { this.beforeOutputsHandled.raiseEvent(id, imodel); }
-    static onAllInputsHandled(id, imodel) { this.allInputsHandled.raiseEvent(id, imodel); }
+    static onBeforeOutputsHandledArg(arg) { this.beforeOutputsHandled.raiseEvent(arg); }
+    static onAllInputsHandledArg(arg) { this.allInputsHandled.raiseEvent(arg); }
 }
 /** the types of users available for tests */
 export var TestUserType;
@@ -58,6 +59,10 @@ export var TestUserType;
  */
 export class HubWrappers {
     static get hubMock() { return HubMock; }
+    static isTransientBriefcaseOpenError(error) {
+        const message = error instanceof Error ? error.message : String(error);
+        return message.includes("EBUSY") || message.includes("EPERM") || message.includes("database is locked");
+    }
     static async getAccessToken(user) {
         return TestUserType[user];
     }
@@ -102,14 +107,24 @@ export class HubWrappers {
     }
     /** Helper to open a briefcase db directly with the BriefcaseManager API */
     static async downloadAndOpenBriefcase(args) {
-        const props = await BriefcaseManager.downloadBriefcase(args);
-        if (args.noLock) {
-            const briefcase = await BriefcaseDb.open({ fileName: props.fileName });
-            briefcase[_nativeDb].saveLocalValue(BriefcaseLocalValue.NoLocking, "true");
-            briefcase.saveChanges();
-            briefcase.close();
+        const maxAttempts = 10;
+        for (let attempt = 1;; attempt++) {
+            try {
+                const props = await BriefcaseManager.downloadBriefcase(args);
+                if (args.noLock) {
+                    const briefcase = await BriefcaseDb.open({ fileName: props.fileName });
+                    briefcase[_nativeDb].saveLocalValue(BriefcaseLocalValue.NoLocking, "true");
+                    briefcase[_nativeDb].saveChanges();
+                    briefcase.close();
+                }
+                return await BriefcaseDb.open({ fileName: props.fileName });
+            }
+            catch (error) {
+                if (attempt >= maxAttempts || !this.isTransientBriefcaseOpenError(error))
+                    throw error;
+                await new Promise((resolve) => setTimeout(resolve, attempt * 250));
+            }
         }
-        return BriefcaseDb.open({ fileName: props.fileName });
     }
     /** Opens the specific iModel as a Briefcase through the same workflow the IModelReadRpc.getConnectionProps method will use. Replicates the way a frontend would open the iModel. */
     static async openBriefcaseUsingRpc(args) {
@@ -291,8 +306,8 @@ export class IModelTestUtils {
         return { id: result };
     }
     /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */
-    static createAndInsertPhysicalPartition(testDb, newModelCode, parentId) {
-        const model = parentId ? testDb.elements.getElement(parentId).model : IModel.repositoryModelId;
+    static createAndInsertPhysicalPartition(txn, newModelCode, parentId) {
+        const model = parentId ? txn.iModel.elements.getElement(parentId).model : IModel.repositoryModelId;
         const parent = new SubjectOwnsPartitionElements(parentId || IModel.rootSubjectId);
         const modeledElementProps = {
             classFullName: PhysicalPartition.classFullName,
@@ -300,12 +315,12 @@ export class IModelTestUtils {
             model,
             code: newModelCode,
         };
-        const modeledElement = testDb.elements.createElement(modeledElementProps);
-        return testDb.elements.insertElement(modeledElement.toJSON());
+        const modeledElement = txn.iModel.elements.createElement(modeledElementProps);
+        return txn.insertElement(modeledElement.toJSON());
     }
     /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */
-    static async createAndInsertPhysicalPartitionAsync(testDb, newModelCode, parentId) {
-        const model = parentId ? testDb.elements.getElement(parentId).model : IModel.repositoryModelId;
+    static async createAndInsertPhysicalPartitionAsync(txn, newModelCode, parentId) {
+        const model = parentId ? txn.iModel.elements.getElement(parentId).model : IModel.repositoryModelId;
         const parent = new SubjectOwnsPartitionElements(parentId || IModel.rootSubjectId);
         const modeledElementProps = {
             classFullName: PhysicalPartition.classFullName,
@@ -313,23 +328,25 @@ export class IModelTestUtils {
             model,
             code: newModelCode,
         };
-        const modeledElement = testDb.elements.createElement(modeledElementProps);
-        await testDb.locks.acquireLocks({ shared: model });
-        return testDb.elements.insertElement(modeledElement.toJSON());
+        const modeledElement = txn.iModel.elements.createElement(modeledElementProps);
+        await txn.iModel.locks.acquireLocks({ shared: model });
+        return txn.insertElement(modeledElement.toJSON());
     }
     /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */
-    static createAndInsertPhysicalModel(testDb, modeledElementRef, privateModel = false) {
-        const newModel = testDb.models.createModel({ modeledElement: modeledElementRef, classFullName: PhysicalModel.classFullName, isPrivate: privateModel });
-        const newModelId = newModel.id = testDb.models.insertModel(newModel.toJSON());
+    static createAndInsertPhysicalModel(txn, modeledElementRef, privateModel = false) {
+        const newModel = txn.iModel.models.createModel({ modeledElement: modeledElementRef, classFullName: PhysicalModel.classFullName, isPrivate: privateModel });
+        const newModelId = txn.insertModel(newModel.toJSON());
+        newModel.id = newModelId;
         assert.isTrue(Id64.isValidId64(newModelId));
         assert.isTrue(Id64.isValidId64(newModel.id));
         assert.deepEqual(newModelId, newModel.id);
         return newModelId;
     }
     /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */
-    static async createAndInsertPhysicalModelAsync(testDb, modeledElementRef, privateModel = false) {
-        const newModel = testDb.models.createModel({ modeledElement: modeledElementRef, classFullName: PhysicalModel.classFullName, isPrivate: privateModel });
-        const newModelId = newModel.insert();
+    static async createAndInsertPhysicalModelAsync(txn, modeledElementRef, privateModel = false) {
+        const newModel = txn.iModel.models.createModel({ modeledElement: modeledElementRef, classFullName: PhysicalModel.classFullName, isPrivate: privateModel });
+        const newModelId = txn.insertModel(newModel.toJSON());
+        newModel.id = newModelId;
         assert.isTrue(Id64.isValidId64(newModelId));
         assert.isTrue(Id64.isValidId64(newModel.id));
         assert.deepEqual(newModelId, newModel.id);
@@ -339,25 +356,25 @@ export class IModelTestUtils {
      * Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel.
      * @return [modeledElementId, modelId]
      */
-    static createAndInsertPhysicalPartitionAndModel(testImodel, newModelCode, privateModel = false, parent) {
-        const eid = IModelTestUtils.createAndInsertPhysicalPartition(testImodel, newModelCode, parent);
+    static createAndInsertPhysicalPartitionAndModel(txn, newModelCode, privateModel = false, parent) {
+        const eid = IModelTestUtils.createAndInsertPhysicalPartition(txn, newModelCode, parent);
         const modeledElementRef = new RelatedElement({ id: eid });
-        const mid = IModelTestUtils.createAndInsertPhysicalModel(testImodel, modeledElementRef, privateModel);
+        const mid = IModelTestUtils.createAndInsertPhysicalModel(txn, modeledElementRef, privateModel);
         return [eid, mid];
     }
     /**
      * Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel.
      * @return [modeledElementId, modelId]
      */
-    static async createAndInsertPhysicalPartitionAndModelAsync(testImodel, newModelCode, privateModel = false, parentId) {
-        const eid = await IModelTestUtils.createAndInsertPhysicalPartitionAsync(testImodel, newModelCode, parentId);
+    static async createAndInsertPhysicalPartitionAndModelAsync(txn, newModelCode, privateModel = false, parentId) {
+        const eid = await IModelTestUtils.createAndInsertPhysicalPartitionAsync(txn, newModelCode, parentId);
         const modeledElementRef = new RelatedElement({ id: eid });
-        const mid = await IModelTestUtils.createAndInsertPhysicalModelAsync(testImodel, modeledElementRef, privateModel);
+        const mid = await IModelTestUtils.createAndInsertPhysicalModelAsync(txn, modeledElementRef, privateModel);
         return [eid, mid];
     }
     /** Create and insert a Drawing Partition element (in the repositoryModel). */
-    static createAndInsertDrawingPartition(testDb, newModelCode, parentId) {
-        const model = parentId ? testDb.elements.getElement(parentId).model : IModel.repositoryModelId;
+    static createAndInsertDrawingPartition(txn, newModelCode, parentId) {
+        const model = parentId ? txn.iModel.elements.getElement(parentId).model : IModel.repositoryModelId;
         const parent = new SubjectOwnsPartitionElements(parentId || IModel.rootSubjectId);
         const modeledElementProps = {
             classFullName: Drawing.classFullName,
@@ -365,13 +382,14 @@ export class IModelTestUtils {
             model,
             code: newModelCode,
         };
-        const modeledElement = testDb.elements.createElement(modeledElementProps);
-        return testDb.elements.insertElement(modeledElement.toJSON());
+        const modeledElement = txn.iModel.elements.createElement(modeledElementProps);
+        return txn.insertElement(modeledElement.toJSON());
     }
     /** Create and insert a DrawingModel associated with Drawing Partition. */
-    static createAndInsertDrawingModel(testDb, modeledElementRef, privateModel = false) {
-        const newModel = testDb.models.createModel({ modeledElement: modeledElementRef, classFullName: DrawingModel.classFullName, isPrivate: privateModel });
-        const newModelId = newModel.insert();
+    static createAndInsertDrawingModel(txn, modeledElementRef, privateModel = false) {
+        const newModel = txn.iModel.models.createModel({ modeledElement: modeledElementRef, classFullName: DrawingModel.classFullName, isPrivate: privateModel });
+        const newModelId = txn.insertModel(newModel.toJSON());
+        newModel.id = newModelId;
         assert.isTrue(Id64.isValidId64(newModelId));
         assert.isTrue(Id64.isValidId64(newModel.id));
         assert.deepEqual(newModelId, newModel.id);
@@ -381,10 +399,10 @@ export class IModelTestUtils {
      * Create and insert a Drawing Partition element (in the repositoryModel) and an associated DrawingModel.
      * @return [modeledElementId, modelId]
      */
-    static createAndInsertDrawingPartitionAndModel(testImodel, newModelCode, privateModel = false, parent) {
-        const eid = IModelTestUtils.createAndInsertDrawingPartition(testImodel, newModelCode, parent);
+    static createAndInsertDrawingPartitionAndModel(txn, newModelCode, privateModel = false, parent) {
+        const eid = IModelTestUtils.createAndInsertDrawingPartition(txn, newModelCode, parent);
         const modeledElementRef = new RelatedElement({ id: eid });
-        const mid = IModelTestUtils.createAndInsertDrawingModel(testImodel, modeledElementRef, privateModel);
+        const mid = IModelTestUtils.createAndInsertDrawingModel(txn, modeledElementRef, privateModel);
         return [eid, mid];
     }
     static getUniqueSpatialCategoryCode(scopeModel, newCodeBaseValue) {
@@ -475,7 +493,7 @@ export class IModelTestUtils {
             transp: 0,
             invisible: false,
         };
-        return SpatialCategory.insert(iModelDb, modelId, categoryName, appearance);
+        return withEditTxn(iModelDb, (txn) => SpatialCategory.insert(txn, modelId, categoryName, appearance));
     }
     static createBoxes(subCategoryIds) {
         const length = 1.0;
@@ -524,12 +542,6 @@ export class IModelTestUtils {
         ]));
         return geometryStreamBuilder.geometryStream;
     }
-    static insertTextureElement(iModelDb, modelId, textureName) {
-        // This is an encoded png containing a 3x3 square with white in top left pixel, blue in middle pixel, and green in bottom right pixel. The rest of the square is red.
-        const pngData = [137, 80, 78, 71, 13, 10, 26, 10, 0, 0, 0, 13, 73, 72, 68, 82, 0, 0, 0, 3, 0, 0, 0, 3, 8, 2, 0, 0, 0, 217, 74, 34, 232, 0, 0, 0, 1, 115, 82, 71, 66, 0, 174, 206, 28, 233, 0, 0, 0, 4, 103, 65, 77, 65, 0, 0, 177, 143, 11, 252, 97, 5, 0, 0, 0, 9, 112, 72, 89, 115, 0, 0, 14, 195, 0, 0, 14, 195, 1, 199, 111, 168, 100, 0, 0, 0, 24, 73, 68, 65, 84, 24, 87, 99, 248, 15, 4, 12, 12, 64, 4, 198, 64, 46, 132, 5, 162, 254, 51, 0, 0, 195, 90, 10, 246, 127, 175, 154, 145, 0, 0, 0, 0, 73, 69, 78, 68, 174, 66, 96, 130];
-        const textureData = Buffer.from(pngData).toString("base64");
-        return Texture.insertTexture(iModelDb, modelId, textureName, ImageSourceFormat.Png, textureData, `Description for ${textureName}`);
-    }
     static queryByUserLabel(iModelDb, userLabel) {
         // eslint-disable-next-line @typescript-eslint/no-deprecated
         return iModelDb.withPreparedStatement(`SELECT ECInstanceId FROM ${Element.classFullName} WHERE UserLabel=:userLabel`, (statement) => {
@@ -544,17 +556,17 @@ export class IModelTestUtils {
             return DbResult.BE_SQLITE_ROW === statement.step() ? statement.getValue(0).getId() : Id64.invalid;
         });
     }
-    static insertRepositoryLink(iModelDb, codeValue, url, format) {
+    static insertRepositoryLink(txn, codeValue, url, format) {
         const repositoryLinkProps = {
             classFullName: RepositoryLink.classFullName,
             model: IModel.repositoryModelId,
-            code: LinkElement.createCode(iModelDb, IModel.repositoryModelId, codeValue),
+            code: LinkElement.createCode(txn.iModel, IModel.repositoryModelId, codeValue),
             url,
             format,
         };
-        return iModelDb.elements.insertElement(repositoryLinkProps);
+        return txn.insertElement(repositoryLinkProps);
     }
-    static insertExternalSource(iModelDb, repositoryId, userLabel) {
+    static insertExternalSource(txn, repositoryId, userLabel) {
         const externalSourceProps = {
             classFullName: ExternalSource.classFullName,
             model: IModel.repositoryModelId,
@@ -564,7 +576,7 @@ export class IModelTestUtils {
             connectorName: "Connector",
             connectorVersion: "0.0.1",
         };
-        return iModelDb.elements.insertElement(externalSourceProps);
+        return txn.insertElement(externalSourceProps);
     }
     static dumpIModelInfo(iModelDb) {
         const outputFileName = `${iModelDb.pathName}.info.txt`;
@@ -647,403 +659,408 @@ export class ExtensiveTestScenario {
         expect(arialFontId).greaterThan(0);
         // Initialize project extents
         const projectExtents = new Range3d(-1000, -1000, -1000, 1000, 1000, 1000);
-        sourceDb.updateProjectExtents(projectExtents);
-        // Insert CodeSpecs
-        const codeSpecId1 = sourceDb.codeSpecs.insert("SourceCodeSpec", CodeScopeSpec.Type.Model);
-        const codeSpecId2 = sourceDb.codeSpecs.insert("ExtraCodeSpec", CodeScopeSpec.Type.ParentElement);
-        const codeSpecId3 = sourceDb.codeSpecs.insert("InformationRecords", CodeScopeSpec.Type.Model);
-        assert.isTrue(Id64.isValidId64(codeSpecId1));
-        assert.isTrue(Id64.isValidId64(codeSpecId2));
-        assert.isTrue(Id64.isValidId64(codeSpecId3));
-        // Insert RepositoryModel structure
-        const subjectId = Subject.insert(sourceDb, IModel.rootSubjectId, "Subject", "Subject Description");
-        assert.isTrue(Id64.isValidId64(subjectId));
-        const sourceOnlySubjectId = Subject.insert(sourceDb, IModel.rootSubjectId, "Only in Source");
-        assert.isTrue(Id64.isValidId64(sourceOnlySubjectId));
-        const definitionModelId = DefinitionModel.insert(sourceDb, subjectId, "Definition");
-        assert.isTrue(Id64.isValidId64(definitionModelId));
-        const informationModelId = InformationRecordModel.insert(sourceDb, subjectId, "Information");
-        assert.isTrue(Id64.isValidId64(informationModelId));
-        const groupModelId = GroupModel.insert(sourceDb, subjectId, "Group");
-        assert.isTrue(Id64.isValidId64(groupModelId));
-        const physicalModelId = PhysicalModel.insert(sourceDb, subjectId, "Physical");
-        assert.isTrue(Id64.isValidId64(physicalModelId));
-        const spatialLocationModelId = SpatialLocationModel.insert(sourceDb, subjectId, "SpatialLocation", true);
-        assert.isTrue(Id64.isValidId64(spatialLocationModelId));
-        const functionalModelId = FunctionalModel.insert(sourceDb, subjectId, "Functional");
-        assert.isTrue(Id64.isValidId64(functionalModelId));
-        const documentListModelId = DocumentListModel.insert(sourceDb, subjectId, "Document");
-        assert.isTrue(Id64.isValidId64(documentListModelId));
-        const drawingId = Drawing.insert(sourceDb, documentListModelId, "Drawing");
-        assert.isTrue(Id64.isValidId64(drawingId));
-        // Insert DefinitionElements
-        const modelSelectorId = ModelSelector.insert(sourceDb, definitionModelId, "SpatialModels", [physicalModelId, spatialLocationModelId]);
-        assert.isTrue(Id64.isValidId64(modelSelectorId));
-        const spatialCategoryId = IModelTestUtils.insertSpatialCategory(sourceDb, definitionModelId, "SpatialCategory", ColorDef.green);
-        assert.isTrue(Id64.isValidId64(spatialCategoryId));
-        const sourcePhysicalCategoryId = IModelTestUtils.insertSpatialCategory(sourceDb, definitionModelId, "SourcePhysicalCategory", ColorDef.blue);
-        assert.isTrue(Id64.isValidId64(sourcePhysicalCategoryId));
-        const subCategoryId = SubCategory.insert(sourceDb, spatialCategoryId, "SubCategory", { color: ColorDef.blue.toJSON() });
-        assert.isTrue(Id64.isValidId64(subCategoryId));
-        const filteredSubCategoryId = SubCategory.insert(sourceDb, spatialCategoryId, "FilteredSubCategory", { color: ColorDef.green.toJSON() });
-        assert.isTrue(Id64.isValidId64(filteredSubCategoryId));
-        const drawingCategoryId = DrawingCategory.insert(sourceDb, definitionModelId, "DrawingCategory", new SubCategoryAppearance());
-        assert.isTrue(Id64.isValidId64(drawingCategoryId));
-        const spatialCategorySelectorId = CategorySelector.insert(sourceDb, definitionModelId, "SpatialCategories", [spatialCategoryId, sourcePhysicalCategoryId]);
-        assert.isTrue(Id64.isValidId64(spatialCategorySelectorId));
-        const drawingCategorySelectorId = CategorySelector.insert(sourceDb, definitionModelId, "DrawingCategories", [drawingCategoryId]);
-        assert.isTrue(Id64.isValidId64(drawingCategorySelectorId));
-        const auxCoordSystemProps = {
-            classFullName: AuxCoordSystem2d.classFullName,
-            model: definitionModelId,
-            code: AuxCoordSystem2d.createCode(sourceDb, definitionModelId, "AuxCoordSystem2d"),
-        };
-        const auxCoordSystemId = sourceDb.elements.insertElement(auxCoordSystemProps);
-        assert.isTrue(Id64.isValidId64(auxCoordSystemId));
-        const textureId = IModelTestUtils.insertTextureElement(sourceDb, definitionModelId, "Texture");
-        assert.isTrue(Id64.isValidId64(textureId));
-        const renderMaterialId = RenderMaterialElement.insert(sourceDb, definitionModelId, "RenderMaterial", { paletteName: "PaletteName" });
-        assert.isTrue(Id64.isValidId64(renderMaterialId));
-        const geometryPartProps = {
-            classFullName: GeometryPart.classFullName,
-            model: definitionModelId,
-            code: GeometryPart.createCode(sourceDb, definitionModelId, "GeometryPart"),
-            geom: IModelTestUtils.createBox(Point3d.create(3, 3, 3)),
-        };
-        const geometryPartId = sourceDb.elements.insertElement(geometryPartProps);
-        assert.isTrue(Id64.isValidId64(geometryPartId));
-        // Insert InformationRecords
-        const informationRecordProps1 = {
-            classFullName: "ExtensiveTestScenario:SourceInformationRecord",
-            model: informationModelId,
-            code: { spec: codeSpecId3, scope: informationModelId, value: "InformationRecord1" },
-            commonString: "Common1",
-            sourceString: "One",
-        };
-        const informationRecordId1 = sourceDb.elements.insertElement(informationRecordProps1);
-        assert.isTrue(Id64.isValidId64(informationRecordId1));
-        const informationRecordProps2 = {
-            classFullName: "ExtensiveTestScenario:SourceInformationRecord",
-            model: informationModelId,
-            code: { spec: codeSpecId3, scope: informationModelId, value: "InformationRecord2" },
-            commonString: "Common2",
-            sourceString: "Two",
-        };
-        const informationRecordId2 = sourceDb.elements.insertElement(informationRecordProps2);
-        assert.isTrue(Id64.isValidId64(informationRecordId2));
-        const informationRecordProps3 = {
-            classFullName: "ExtensiveTestScenario:SourceInformationRecord",
-            model: informationModelId,
-            code: { spec: codeSpecId3, scope: informationModelId, value: "InformationRecord3" },
-            commonString: "Common3",
-            sourceString: "Three",
-        };
-        const informationRecordId3 = sourceDb.elements.insertElement(informationRecordProps3);
-        assert.isTrue(Id64.isValidId64(informationRecordId3));
-        // Insert PhysicalObject1
-        const physicalObjectProps1 = {
-            classFullName: PhysicalObject.classFullName,
-            model: physicalModelId,
-            category: spatialCategoryId,
-            code: Code.createEmpty(),
-            userLabel: "PhysicalObject1",
-            geom: IModelTestUtils.createBox(Point3d.create(1, 1, 1), spatialCategoryId, subCategoryId, renderMaterialId, geometryPartId),
-            placement: {
-                origin: Point3d.create(1, 1, 1),
-                angles: YawPitchRollAngles.createDegrees(0, 0, 0),
-            },
-        };
-        const physicalObjectId1 = sourceDb.elements.insertElement(physicalObjectProps1);
-        assert.isTrue(Id64.isValidId64(physicalObjectId1));
-        // Insert PhysicalObject1 children
-        const childObjectProps1A = physicalObjectProps1;
-        childObjectProps1A.userLabel = "ChildObject1A";
-        childObjectProps1A.parent = new ElementOwnsChildElements(physicalObjectId1);
-        childObjectProps1A.placement.origin = Point3d.create(0, 1, 1);
-        const childObjectId1A = sourceDb.elements.insertElement(childObjectProps1A);
-        assert.isTrue(Id64.isValidId64(childObjectId1A));
-        const childObjectProps1B = childObjectProps1A;
-        childObjectProps1B.userLabel = "ChildObject1B";
-        childObjectProps1B.placement.origin = Point3d.create(1, 0, 1);
-        const childObjectId1B = sourceDb.elements.insertElement(childObjectProps1B);
-        assert.isTrue(Id64.isValidId64(childObjectId1B));
-        // Insert PhysicalObject2
-        const physicalObjectProps2 = {
-            classFullName: PhysicalObject.classFullName,
-            model: physicalModelId,
-            category: sourcePhysicalCategoryId,
-            code: Code.createEmpty(),
-            userLabel: "PhysicalObject2",
-            geom: IModelTestUtils.createBox(Point3d.create(2, 2, 2)),
-            placement: {
-                origin: Point3d.create(2, 2, 2),
-                angles: YawPitchRollAngles.createDegrees(0, 0, 0),
-            },
-        };
-        const physicalObjectId2 = sourceDb.elements.insertElement(physicalObjectProps2);
-        assert.isTrue(Id64.isValidId64(physicalObjectId2));
-        // Insert PhysicalObject3
-        const physicalObjectProps3 = {
-            classFullName: PhysicalObject.classFullName,
-            model: physicalModelId,
-            category: sourcePhysicalCategoryId,
-            code: Code.createEmpty(),
-            federationGuid: ExtensiveTestScenario.federationGuid3,
-            userLabel: "PhysicalObject3",
-        };
-        const physicalObjectId3 = sourceDb.elements.insertElement(physicalObjectProps3);
-        assert.isTrue(Id64.isValidId64(physicalObjectId3));
-        // Insert PhysicalObject4
-        const physicalObjectProps4 = {
-            classFullName: PhysicalObject.classFullName,
-            model: physicalModelId,
-            category: spatialCategoryId,
-            code: Code.createEmpty(),
-            userLabel: "PhysicalObject4",
-            geom: IModelTestUtils.createBoxes([subCategoryId, filteredSubCategoryId]),
-            placement: {
-                origin: Point3d.create(4, 4, 4),
-                angles: YawPitchRollAngles.createDegrees(0, 0, 0),
-            },
-        };
-        const physicalObjectId4 = sourceDb.elements.insertElement(physicalObjectProps4);
-        assert.isTrue(Id64.isValidId64(physicalObjectId4));
-        // Insert PhysicalElement1
-        const sourcePhysicalElementProps = {
-            classFullName: "ExtensiveTestScenario:SourcePhysicalElement",
-            model: physicalModelId,
-            category: sourcePhysicalCategoryId,
-            code: Code.createEmpty(),
-            userLabel: "PhysicalElement1",
-            geom: IModelTestUtils.createBox(Point3d.create(2, 2, 2)),
-            placement: {
-                origin: Point3d.create(4, 4, 4),
-                angles: YawPitchRollAngles.createDegrees(0, 0, 0),
-            },
-            sourceString: "S1",
-            sourceDouble: 1.1,
-            sourceNavigation: { id: sourcePhysicalCategoryId, relClassName: "ExtensiveTestScenario:SourcePhysicalElementUsesSourceDefinition" },
-            commonNavigation: { id: sourcePhysicalCategoryId },
-            commonString: "Common",
-            commonDouble: 7.3,
-            sourceBinary: new Uint8Array([1, 3, 5, 7]),
-            commonBinary: Base64EncodedString.fromUint8Array(new Uint8Array([2, 4, 6, 8])),
-            extraString: "Extra",
-        };
-        const sourcePhysicalElementId = sourceDb.elements.insertElement(sourcePhysicalElementProps);
-        assert.isTrue(Id64.isValidId64(sourcePhysicalElementId));
-        assert.doesNotThrow(() => sourceDb.elements.getElement(sourcePhysicalElementId));
-        // Insert ElementAspects
-        const aspectProps = {
-            classFullName: "ExtensiveTestScenario:SourceUniqueAspect",
-            element: new ElementOwnsUniqueAspect(physicalObjectId1),
-            commonDouble: 1.1,
-            commonString: "Unique",
-            commonLong: physicalObjectId1,
-            commonBinary: Base64EncodedString.fromUint8Array(new Uint8Array([2, 4, 6, 8])),
-            sourceDouble: 11.1,
-            sourceString: "UniqueAspect",
-            sourceLong: physicalObjectId1,
-            sourceGuid: ExtensiveTestScenario.uniqueAspectGuid,
-            extraString: "Extra",
-        };
-        sourceDb.elements.insertAspect(aspectProps);
-        const sourceUniqueAspect = sourceDb.elements.getAspects(physicalObjectId1, "ExtensiveTestScenario:SourceUniqueAspect")[0];
-        expect(sourceUniqueAspect).to.deep.subsetEqual(omit(aspectProps, ["commonBinary"]), { normalizeClassNameProps: true });
-        sourceDb.elements.insertAspect({
-            classFullName: "ExtensiveTestScenario:SourceMultiAspect",
-            element: new ElementOwnsMultiAspects(physicalObjectId1),
-            commonDouble: 2.2,
-            commonString: "Multi",
-            commonLong: physicalObjectId1,
-            sourceDouble: 22.2,
-            sourceString: "MultiAspect",
-            sourceLong: physicalObjectId1,
-            sourceGuid: Guid.createValue(),
-            extraString: "Extra",
-        });
-        sourceDb.elements.insertAspect({
-            classFullName: "ExtensiveTestScenario:SourceMultiAspect",
-            element: new ElementOwnsMultiAspects(physicalObjectId1),
-            commonDouble: 3.3,
-            commonString: "Multi",
-            commonLong: physicalObjectId1,
-            sourceDouble: 33.3,
-            sourceString: "MultiAspect",
-            sourceLong: physicalObjectId1,
-            sourceGuid: Guid.createValue(),
-            extraString: "Extra",
-        });
-        sourceDb.elements.insertAspect({
-            classFullName: "ExtensiveTestScenario:SourceUniqueAspectToExclude",
-            element: new ElementOwnsUniqueAspect(physicalObjectId1),
-            description: "SourceUniqueAspect1",
-        });
-        sourceDb.elements.insertAspect({
-            classFullName: "ExtensiveTestScenario:SourceMultiAspectToExclude",
-            element: new ElementOwnsMultiAspects(physicalObjectId1),
-            description: "SourceMultiAspect1",
-        });
-        // Insert DrawingGraphics
-        const drawingGraphicProps1 = {
-            classFullName: DrawingGraphic.classFullName,
-            model: drawingId,
-            category: drawingCategoryId,
-            code: Code.createEmpty(),
-            userLabel: "DrawingGraphic1",
-            geom: IModelTestUtils.createRectangle(Point2d.create(1, 1)),
-            placement: { origin: Point2d.create(2, 2), angle: 0 },
-        };
-        const drawingGraphicId1 = sourceDb.elements.insertElement(drawingGraphicProps1);
-        assert.isTrue(Id64.isValidId64(drawingGraphicId1));
-        const drawingGraphicRepresentsId1 = DrawingGraphicRepresentsElement.insert(sourceDb, drawingGraphicId1, physicalObjectId1);
-        assert.isTrue(Id64.isValidId64(drawingGraphicRepresentsId1));
-        const drawingGraphicProps2 = {
-            classFullName: DrawingGraphic.classFullName,
-            model: drawingId,
-            category: drawingCategoryId,
-            code: Code.createEmpty(),
-            userLabel: "DrawingGraphic2",
-            geom: IModelTestUtils.createRectangle(Point2d.create(1, 1)),
-            placement: { origin: Point2d.create(3, 3), angle: 0 },
-        };
-        const drawingGraphicId2 = sourceDb.elements.insertElement(drawingGraphicProps2);
-        assert.isTrue(Id64.isValidId64(drawingGraphicId2));
-        const drawingGraphicRepresentsId2 = DrawingGraphicRepresentsElement.insert(sourceDb, drawingGraphicId2, physicalObjectId1);
-        assert.isTrue(Id64.isValidId64(drawingGraphicRepresentsId2));
-        // Insert DisplayStyles
-        const displayStyle2dId = DisplayStyle2d.insert(sourceDb, definitionModelId, "DisplayStyle2d");
-        assert.isTrue(Id64.isValidId64(displayStyle2dId));
-        const displayStyle3d = DisplayStyle3d.create(sourceDb, definitionModelId, "DisplayStyle3d");
-        const subCategoryOverride = SubCategoryOverride.fromJSON({ color: ColorDef.from(1, 2, 3).toJSON() });
-        displayStyle3d.settings.overrideSubCategory(subCategoryId, subCategoryOverride);
-        displayStyle3d.settings.addExcludedElements(physicalObjectId1);
-        displayStyle3d.settings.setPlanProjectionSettings(spatialLocationModelId, PlanProjectionSettings.fromJSON({ elevation: 10.0 }));
-        displayStyle3d.settings.environment = Environment.fromJSON({
-            sky: {
-                image: {
-                    type: SkyBoxImageType.Spherical,
-                    texture: textureId,
+        await withEditTxn(sourceDb, async (txn) => {
+            txn.updateProjectExtents(projectExtents);
+            // Insert CodeSpecs
+            const codeSpecId1 = sourceDb.codeSpecs.insert(txn, "SourceCodeSpec", CodeScopeSpec.Type.Model);
+            const codeSpecId2 = sourceDb.codeSpecs.insert(txn, "ExtraCodeSpec", CodeScopeSpec.Type.ParentElement);
+            const codeSpecId3 = sourceDb.codeSpecs.insert(txn, "InformationRecords", CodeScopeSpec.Type.Model);
+            assert.isTrue(Id64.isValidId64(codeSpecId1));
+            assert.isTrue(Id64.isValidId64(codeSpecId2));
+            assert.isTrue(Id64.isValidId64(codeSpecId3));
+            // Insert RepositoryModel structure
+            const subjectId = Subject.insert(txn, IModel.rootSubjectId, "Subject", "Subject Description");
+            assert.isTrue(Id64.isValidId64(subjectId));
+            const sourceOnlySubjectId = Subject.insert(txn, IModel.rootSubjectId, "Only in Source");
+            assert.isTrue(Id64.isValidId64(sourceOnlySubjectId));
+            const definitionModelId = DefinitionModel.insert(txn, subjectId, "Definition");
+            assert.isTrue(Id64.isValidId64(definitionModelId));
+            const informationModelId = InformationRecordModel.insert(txn, subjectId, "Information");
+            assert.isTrue(Id64.isValidId64(informationModelId));
+            const groupModelId = GroupModel.insert(txn, subjectId, "Group");
+            assert.isTrue(Id64.isValidId64(groupModelId));
+            const physicalModelId = PhysicalModel.insert(txn, subjectId, "Physical");
+            assert.isTrue(Id64.isValidId64(physicalModelId));
+            const spatialLocationModelId = SpatialLocationModel.insert(txn, subjectId, "SpatialLocation", true);
+            assert.isTrue(Id64.isValidId64(spatialLocationModelId));
+            const functionalModelId = FunctionalModel.insert(txn, subjectId, "Functional");
+            assert.isTrue(Id64.isValidId64(functionalModelId));
+            const documentListModelId = DocumentListModel.insert(txn, subjectId, "Document");
+            assert.isTrue(Id64.isValidId64(documentListModelId));
+            const drawingId = Drawing.insert(txn, documentListModelId, "Drawing");
+            assert.isTrue(Id64.isValidId64(drawingId));
+            // Insert DefinitionElements
+            const modelSelectorId = ModelSelector.insert(txn, definitionModelId, "SpatialModels", [physicalModelId, spatialLocationModelId]);
+            assert.isTrue(Id64.isValidId64(modelSelectorId));
+            const spatialCategoryId = SpatialCategory.insert(txn, definitionModelId, "SpatialCategory", { color: ColorDef.green.toJSON(), transp: 0, invisible: false });
+            assert.isTrue(Id64.isValidId64(spatialCategoryId));
+            const sourcePhysicalCategoryId = SpatialCategory.insert(txn, definitionModelId, "SourcePhysicalCategory", { color: ColorDef.blue.toJSON(), transp: 0, invisible: false });
+            assert.isTrue(Id64.isValidId64(sourcePhysicalCategoryId));
+            const subCategoryId = SubCategory.insert(txn, spatialCategoryId, "SubCategory", { color: ColorDef.blue.toJSON() });
+            assert.isTrue(Id64.isValidId64(subCategoryId));
+            const filteredSubCategoryId = SubCategory.insert(txn, spatialCategoryId, "FilteredSubCategory", { color: ColorDef.green.toJSON() });
+            assert.isTrue(Id64.isValidId64(filteredSubCategoryId));
+            const drawingCategoryId = DrawingCategory.insert(txn, definitionModelId, "DrawingCategory", new SubCategoryAppearance());
+            assert.isTrue(Id64.isValidId64(drawingCategoryId));
+            const spatialCategorySelectorId = CategorySelector.insert(txn, definitionModelId, "SpatialCategories", [spatialCategoryId, sourcePhysicalCategoryId]);
+            assert.isTrue(Id64.isValidId64(spatialCategorySelectorId));
+            const drawingCategorySelectorId = CategorySelector.insert(txn, definitionModelId, "DrawingCategories", [drawingCategoryId]);
+            assert.isTrue(Id64.isValidId64(drawingCategorySelectorId));
+            const auxCoordSystemProps = {
+                classFullName: AuxCoordSystem2d.classFullName,
+                model: definitionModelId,
+                code: AuxCoordSystem2d.createCode(sourceDb, definitionModelId, "AuxCoordSystem2d"),
+            };
+            const auxCoordSystemId = txn.insertElement(auxCoordSystemProps);
+            assert.isTrue(Id64.isValidId64(auxCoordSystemId));
+            const pngData = [137, 80, 78, 71, 13, 10, 26, 10, 0, 0, 0, 13, 73, 72, 68, 82, 0, 0, 0, 3, 0, 0, 0, 3, 8, 2, 0, 0, 0, 217, 74, 34, 232, 0, 0, 0, 1, 115, 82, 71, 66, 0, 174, 206, 28, 233, 0, 0, 0, 4, 103, 65, 77, 65, 0, 0, 177, 143, 11, 252, 97, 5, 0, 0, 0, 9, 112, 72, 89, 115, 0, 0, 14, 195, 0, 0, 14, 195, 1, 199, 111, 168, 100, 0, 0, 0, 24, 73, 68, 65, 84, 24, 87, 99, 248, 15, 4, 12, 12, 64, 4, 198, 64, 46, 132, 5, 162, 254, 51, 0, 0, 195, 90, 10, 246, 127, 175, 154, 145, 0, 0, 0, 0, 73, 69, 78, 68, 174, 66, 96, 130];
+            const textureId = Texture.insertTexture(txn, definitionModelId, "Texture", ImageSourceFormat.Png, Buffer.from(pngData).toString("base64"), "Description for Texture");
+            assert.isTrue(Id64.isValidId64(textureId));
+            const renderMaterialId = RenderMaterialElement.insert(txn, definitionModelId, "RenderMaterial", { paletteName: "PaletteName" });
+            assert.isTrue(Id64.isValidId64(renderMaterialId));
+            const geometryPartProps = {
+                classFullName: GeometryPart.classFullName,
+                model: definitionModelId,
+                code: GeometryPart.createCode(sourceDb, definitionModelId, "GeometryPart"),
+                geom: IModelTestUtils.createBox(Point3d.create(3, 3, 3)),
+            };
+            const geometryPartId = txn.insertElement(geometryPartProps);
+            assert.isTrue(Id64.isValidId64(geometryPartId));
+            // Insert InformationRecords
+            const informationRecordProps1 = {
+                classFullName: "ExtensiveTestScenario:SourceInformationRecord",
+                model: informationModelId,
+                code: { spec: codeSpecId3, scope: informationModelId, value: "InformationRecord1" },
+                commonString: "Common1",
+                sourceString: "One",
+            };
+            const informationRecordId1 = txn.insertElement(informationRecordProps1);
+            assert.isTrue(Id64.isValidId64(informationRecordId1));
+            const informationRecordProps2 = {
+                classFullName: "ExtensiveTestScenario:SourceInformationRecord",
+                model: informationModelId,
+                code: { spec: codeSpecId3, scope: informationModelId, value: "InformationRecord2" },
+                commonString: "Common2",
+                sourceString: "Two",
+            };
+            const informationRecordId2 = txn.insertElement(informationRecordProps2);
+            assert.isTrue(Id64.isValidId64(informationRecordId2));
+            const informationRecordProps3 = {
+                classFullName: "ExtensiveTestScenario:SourceInformationRecord",
+                model: informationModelId,
+                code: { spec: codeSpecId3, scope: informationModelId, value: "InformationRecord3" },
+                commonString: "Common3",
+                sourceString: "Three",
+            };
+            const informationRecordId3 = txn.insertElement(informationRecordProps3);
+            assert.isTrue(Id64.isValidId64(informationRecordId3));
+            // Insert PhysicalObject1
+            const physicalObjectProps1 = {
+                classFullName: PhysicalObject.classFullName,
+                model: physicalModelId,
+                category: spatialCategoryId,
+                code: Code.createEmpty(),
+                userLabel: "PhysicalObject1",
+                geom: IModelTestUtils.createBox(Point3d.create(1, 1, 1), spatialCategoryId, subCategoryId, renderMaterialId, geometryPartId),
+                placement: {
+                    origin: Point3d.create(1, 1, 1),
+                    angles: YawPitchRollAngles.createDegrees(0, 0, 0),
                 },
-            },
-        });
-        const displayStyle3dId = displayStyle3d.insert();
-        assert.isTrue(Id64.isValidId64(displayStyle3dId));
-        // Insert ViewDefinitions
-        const viewId = OrthographicViewDefinition.insert(sourceDb, definitionModelId, "Orthographic View", modelSelectorId, spatialCategorySelectorId, displayStyle3dId, projectExtents, StandardViewIndex.Iso);
-        assert.isTrue(Id64.isValidId64(viewId));
-        const drawingViewRange = new Range2d(0, 0, 100, 100);
-        const drawingViewId = DrawingViewDefinition.insert(sourceDb, definitionModelId, "Drawing View", drawingId, drawingCategorySelectorId, displayStyle2dId, drawingViewRange);
-        assert.isTrue(Id64.isValidId64(drawingViewId));
-        // Insert instance of SourceRelToExclude to test relationship exclusion by class
-        const relationship1 = sourceDb.relationships.createInstance({
-            classFullName: "ExtensiveTestScenario:SourceRelToExclude",
-            sourceId: spatialCategorySelectorId,
-            targetId: drawingCategorySelectorId,
-        });
-        const relationshipId1 = sourceDb.relationships.insertInstance(relationship1.toJSON());
-        assert.isTrue(Id64.isValidId64(relationshipId1));
-        // Insert instance of RelWithProps to test relationship property remapping
-        const relationship2 = sourceDb.relationships.createInstance({
-            classFullName: "ExtensiveTestScenario:SourceRelWithProps",
-            sourceId: spatialCategorySelectorId,
-            targetId: drawingCategorySelectorId,
-            sourceString: "One",
-            sourceDouble: 1.1,
-            sourceLong: spatialCategoryId,
-            sourceGuid: Guid.createValue(),
-        });
-        const relationshipId2 = sourceDb.relationships.insertInstance(relationship2.toJSON());
-        assert.isTrue(Id64.isValidId64(relationshipId2));
+            };
+            const physicalObjectId1 = txn.insertElement(physicalObjectProps1);
+            assert.isTrue(Id64.isValidId64(physicalObjectId1));
+            // Insert PhysicalObject1 children
+            const childObjectProps1A = physicalObjectProps1;
+            childObjectProps1A.userLabel = "ChildObject1A";
+            childObjectProps1A.parent = new ElementOwnsChildElements(physicalObjectId1);
+            childObjectProps1A.placement.origin = Point3d.create(0, 1, 1);
+            const childObjectId1A = txn.insertElement(childObjectProps1A);
+            assert.isTrue(Id64.isValidId64(childObjectId1A));
+            const childObjectProps1B = childObjectProps1A;
+            childObjectProps1B.userLabel = "ChildObject1B";
+            childObjectProps1B.placement.origin = Point3d.create(1, 0, 1);
+            const childObjectId1B = txn.insertElement(childObjectProps1B);
+            assert.isTrue(Id64.isValidId64(childObjectId1B));
+            // Insert PhysicalObject2
+            const physicalObjectProps2 = {
+                classFullName: PhysicalObject.classFullName,
+                model: physicalModelId,
+                category: sourcePhysicalCategoryId,
+                code: Code.createEmpty(),
+                userLabel: "PhysicalObject2",
+                geom: IModelTestUtils.createBox(Point3d.create(2, 2, 2)),
+                placement: {
+                    origin: Point3d.create(2, 2, 2),
+                    angles: YawPitchRollAngles.createDegrees(0, 0, 0),
+                },
+            };
+            const physicalObjectId2 = txn.insertElement(physicalObjectProps2);
+            assert.isTrue(Id64.isValidId64(physicalObjectId2));
+            // Insert PhysicalObject3
+            const physicalObjectProps3 = {
+                classFullName: PhysicalObject.classFullName,
+                model: physicalModelId,
+                category: sourcePhysicalCategoryId,
+                code: Code.createEmpty(),
+                federationGuid: ExtensiveTestScenario.federationGuid3,
+                userLabel: "PhysicalObject3",
+            };
+            const physicalObjectId3 = txn.insertElement(physicalObjectProps3);
+            assert.isTrue(Id64.isValidId64(physicalObjectId3));
+            // Insert PhysicalObject4
+            const physicalObjectProps4 = {
+                classFullName: PhysicalObject.classFullName,
+                model: physicalModelId,
+                category: spatialCategoryId,
+                code: Code.createEmpty(),
+                userLabel: "PhysicalObject4",
+                geom: IModelTestUtils.createBoxes([subCategoryId, filteredSubCategoryId]),
+                placement: {
+                    origin: Point3d.create(4, 4, 4),
+                    angles: YawPitchRollAngles.createDegrees(0, 0, 0),
+                },
+            };
+            const physicalObjectId4 = txn.insertElement(physicalObjectProps4);
+            assert.isTrue(Id64.isValidId64(physicalObjectId4));
+            // Insert PhysicalElement1
+            const sourcePhysicalElementProps = {
+                classFullName: "ExtensiveTestScenario:SourcePhysicalElement",
+                model: physicalModelId,
+                category: sourcePhysicalCategoryId,
+                code: Code.createEmpty(),
+                userLabel: "PhysicalElement1",
+                geom: IModelTestUtils.createBox(Point3d.create(2, 2, 2)),
+                placement: {
+                    origin: Point3d.create(4, 4, 4),
+                    angles: YawPitchRollAngles.createDegrees(0, 0, 0),
+                },
+                sourceString: "S1",
+                sourceDouble: 1.1,
+                sourceNavigation: { id: sourcePhysicalCategoryId, relClassName: "ExtensiveTestScenario:SourcePhysicalElementUsesSourceDefinition" },
+                commonNavigation: { id: sourcePhysicalCategoryId },
+                commonString: "Common",
+                commonDouble: 7.3,
+                sourceBinary: new Uint8Array([1, 3, 5, 7]),
+                commonBinary: Base64EncodedString.fromUint8Array(new Uint8Array([2, 4, 6, 8])),
+                extraString: "Extra",
+            };
+            const sourcePhysicalElementId = txn.insertElement(sourcePhysicalElementProps);
+            assert.isTrue(Id64.isValidId64(sourcePhysicalElementId));
+            assert.doesNotThrow(() => sourceDb.elements.getElement(sourcePhysicalElementId));
+            // Insert ElementAspects
+            const aspectProps = {
+                classFullName: "ExtensiveTestScenario:SourceUniqueAspect",
+                element: new ElementOwnsUniqueAspect(physicalObjectId1),
+                commonDouble: 1.1,
+                commonString: "Unique",
+                commonLong: physicalObjectId1,
+                commonBinary: Base64EncodedString.fromUint8Array(new Uint8Array([2, 4, 6, 8])),
+                sourceDouble: 11.1,
+                sourceString: "UniqueAspect",
+                sourceLong: physicalObjectId1,
+                sourceGuid: ExtensiveTestScenario.uniqueAspectGuid,
+                extraString: "Extra",
+            };
+            txn.insertAspect(aspectProps);
+            const sourceUniqueAspect = sourceDb.elements.getAspects(physicalObjectId1, "ExtensiveTestScenario:SourceUniqueAspect")[0];
+            expect(sourceUniqueAspect).to.deep.subsetEqual(omit(aspectProps, ["commonBinary"]), { normalizeClassNameProps: true });
+            txn.insertAspect({
+                classFullName: "ExtensiveTestScenario:SourceMultiAspect",
+                element: new ElementOwnsMultiAspects(physicalObjectId1),
+                commonDouble: 2.2,
+                commonString: "Multi",
+                commonLong: physicalObjectId1,
+                sourceDouble: 22.2,
+                sourceString: "MultiAspect",
+                sourceLong: physicalObjectId1,
+                sourceGuid: Guid.createValue(),
+                extraString: "Extra",
+            });
+            txn.insertAspect({
+                classFullName: "ExtensiveTestScenario:SourceMultiAspect",
+                element: new ElementOwnsMultiAspects(physicalObjectId1),
+                commonDouble: 3.3,
+                commonString: "Multi",
+                commonLong: physicalObjectId1,
+                sourceDouble: 33.3,
+                sourceString: "MultiAspect",
+                sourceLong: physicalObjectId1,
+                sourceGuid: Guid.createValue(),
+                extraString: "Extra",
+            });
+            txn.insertAspect({
+                classFullName: "ExtensiveTestScenario:SourceUniqueAspectToExclude",
+                element: new ElementOwnsUniqueAspect(physicalObjectId1),
+                description: "SourceUniqueAspect1",
+            });
+            txn.insertAspect({
+                classFullName: "ExtensiveTestScenario:SourceMultiAspectToExclude",
+                element: new ElementOwnsMultiAspects(physicalObjectId1),
+                description: "SourceMultiAspect1",
+            });
+            // Insert DrawingGraphics
+            const drawingGraphicProps1 = {
+                classFullName: DrawingGraphic.classFullName,
+                model: drawingId,
+                category: drawingCategoryId,
+                code: Code.createEmpty(),
+                userLabel: "DrawingGraphic1",
+                geom: IModelTestUtils.createRectangle(Point2d.create(1, 1)),
+                placement: { origin: Point2d.create(2, 2), angle: 0 },
+            };
+            const drawingGraphicId1 = txn.insertElement(drawingGraphicProps1);
+            assert.isTrue(Id64.isValidId64(drawingGraphicId1));
+            const drawingGraphicRepresentsId1 = DrawingGraphicRepresentsElement.insert(txn, drawingGraphicId1, physicalObjectId1);
+            assert.isTrue(Id64.isValidId64(drawingGraphicRepresentsId1));
+            const drawingGraphicProps2 = {
+                classFullName: DrawingGraphic.classFullName,
+                model: drawingId,
+                category: drawingCategoryId,
+                code: Code.createEmpty(),
+                userLabel: "DrawingGraphic2",
+                geom: IModelTestUtils.createRectangle(Point2d.create(1, 1)),
+                placement: { origin: Point2d.create(3, 3), angle: 0 },
+            };
+            const drawingGraphicId2 = txn.insertElement(drawingGraphicProps2);
+            assert.isTrue(Id64.isValidId64(drawingGraphicId2));
+            const drawingGraphicRepresentsId2 = DrawingGraphicRepresentsElement.insert(txn, drawingGraphicId2, physicalObjectId1);
+            assert.isTrue(Id64.isValidId64(drawingGraphicRepresentsId2));
+            // Insert DisplayStyles
+            const displayStyle2dId = DisplayStyle2d.insert(txn, definitionModelId, "DisplayStyle2d");
+            assert.isTrue(Id64.isValidId64(displayStyle2dId));
+            const displayStyle3d = DisplayStyle3d.create(sourceDb, definitionModelId, "DisplayStyle3d");
+            const subCategoryOverride = SubCategoryOverride.fromJSON({ color: ColorDef.from(1, 2, 3).toJSON() });
+            displayStyle3d.settings.overrideSubCategory(subCategoryId, subCategoryOverride);
+            displayStyle3d.settings.addExcludedElements(physicalObjectId1);
+            displayStyle3d.settings.setPlanProjectionSettings(spatialLocationModelId, PlanProjectionSettings.fromJSON({ elevation: 10.0 }));
+            displayStyle3d.settings.environment = Environment.fromJSON({
+                sky: {
+                    image: {
+                        type: SkyBoxImageType.Spherical,
+                        texture: textureId,
+                    },
+                },
+            });
+            const displayStyle3dId = displayStyle3d.insert(txn);
+            assert.isTrue(Id64.isValidId64(displayStyle3dId));
+            // Insert ViewDefinitions
+            const viewId = OrthographicViewDefinition.insert(txn, definitionModelId, "Orthographic View", modelSelectorId, spatialCategorySelectorId, displayStyle3dId, projectExtents, StandardViewIndex.Iso);
+            assert.isTrue(Id64.isValidId64(viewId));
+            const drawingViewRange = new Range2d(0, 0, 100, 100);
+            const drawingViewId = DrawingViewDefinition.insert(txn, definitionModelId, "Drawing View", drawingId, drawingCategorySelectorId, displayStyle2dId, drawingViewRange);
+            assert.isTrue(Id64.isValidId64(drawingViewId));
+            // Insert instance of SourceRelToExclude to test relationship exclusion by class
+            const relationship1 = sourceDb.relationships.createInstance({
+                classFullName: "ExtensiveTestScenario:SourceRelToExclude",
+                sourceId: spatialCategorySelectorId,
+                targetId: drawingCategorySelectorId,
+            });
+            const relationshipId1 = txn.insertRelationship(relationship1.toJSON());
+            assert.isTrue(Id64.isValidId64(relationshipId1));
+            // Insert instance of RelWithProps to test relationship property remapping
+            const relationship2 = sourceDb.relationships.createInstance({
+                classFullName: "ExtensiveTestScenario:SourceRelWithProps",
+                sourceId: spatialCategorySelectorId,
+                targetId: drawingCategorySelectorId,
+                sourceString: "One",
+                sourceDouble: 1.1,
+                sourceLong: spatialCategoryId,
+                sourceGuid: Guid.createValue(),
+            });
+            const relationshipId2 = txn.insertRelationship(relationship2.toJSON());
+            assert.isTrue(Id64.isValidId64(relationshipId2));
+        }); // end withEditTxn
     }
     static updateDb(sourceDb) {
-        // Update Subject element
-        const subjectId = sourceDb.elements.queryElementIdByCode(Subject.createCode(sourceDb, IModel.rootSubjectId, "Subject"));
-        assert.isTrue(Id64.isValidId64(subjectId));
-        const subject = sourceDb.elements.getElement(subjectId);
-        subject.description = "Subject description (Updated)";
-        sourceDb.elements.updateElement(subject.toJSON());
-        // Update spatialCategory element
-        const definitionModelId = sourceDb.elements.queryElementIdByCode(InformationPartitionElement.createCode(sourceDb, subjectId, "Definition"));
-        assert.isTrue(Id64.isValidId64(definitionModelId));
-        const spatialCategoryId = sourceDb.elements.queryElementIdByCode(SpatialCategory.createCode(sourceDb, definitionModelId, "SpatialCategory"));
-        assert.isTrue(Id64.isValidId64(spatialCategoryId));
-        const spatialCategory = sourceDb.elements.getElement(spatialCategoryId);
-        spatialCategory.federationGuid = Guid.createValue();
-        sourceDb.elements.updateElement(spatialCategory.toJSON());
-        // Update relationship properties
-        const spatialCategorySelectorId = sourceDb.elements.queryElementIdByCode(CategorySelector.createCode(sourceDb, definitionModelId, "SpatialCategories"));
-        assert.isTrue(Id64.isValidId64(spatialCategorySelectorId));
-        const drawingCategorySelectorId = sourceDb.elements.queryElementIdByCode(CategorySelector.createCode(sourceDb, definitionModelId, "DrawingCategories"));
-        assert.isTrue(Id64.isValidId64(drawingCategorySelectorId));
-        const relWithProps = sourceDb.relationships.getInstanceProps("ExtensiveTestScenario:SourceRelWithProps", { sourceId: spatialCategorySelectorId, targetId: drawingCategorySelectorId });
-        assert.equal(relWithProps.sourceString, "One");
-        assert.equal(relWithProps.sourceDouble, 1.1);
-        relWithProps.sourceString += "-Updated";
-        relWithProps.sourceDouble = 1.2;
-        sourceDb.relationships.updateInstance(relWithProps);
-        // Update ElementAspect properties
-        const physicalObjectId1 = IModelTestUtils.queryByUserLabel(sourceDb, "PhysicalObject1");
-        const sourceUniqueAspects = sourceDb.elements.getAspects(physicalObjectId1, "ExtensiveTestScenario:SourceUniqueAspect");
-        assert.equal(sourceUniqueAspects.length, 1);
-        sourceUniqueAspects[0].asAny.commonString += "-Updated";
-        sourceUniqueAspects[0].asAny.sourceString += "-Updated";
-        sourceDb.elements.updateAspect(sourceUniqueAspects[0].toJSON());
-        const sourceMultiAspects = sourceDb.elements.getAspects(physicalObjectId1, "ExtensiveTestScenario:SourceMultiAspect");
-        assert.equal(sourceMultiAspects.length, 2);
-        sourceMultiAspects[1].asAny.commonString += "-Updated";
-        sourceMultiAspects[1].asAny.sourceString += "-Updated";
-        sourceDb.elements.updateAspect(sourceMultiAspects[1].toJSON());
-        // clear NavigationProperty of PhysicalElement1
-        const physicalElementId1 = IModelTestUtils.queryByUserLabel(sourceDb, "PhysicalElement1");
-        let physicalElement1 = sourceDb.elements.getElement(physicalElementId1);
-        physicalElement1.asAny.commonNavigation = RelatedElement.none;
-        physicalElement1.update();
-        physicalElement1 = sourceDb.elements.getElement(physicalElementId1);
-        assert.isUndefined(physicalElement1.asAny.commonNavigation);
-        // delete PhysicalObject3
-        const physicalObjectId3 = IModelTestUtils.queryByUserLabel(sourceDb, "PhysicalObject3");
-        assert.isTrue(Id64.isValidId64(physicalObjectId3));
-        sourceDb.elements.deleteElement(physicalObjectId3);
-        assert.equal(Id64.invalid, IModelTestUtils.queryByUserLabel(sourceDb, "PhysicalObject3"));
-        // Insert PhysicalObject5
-        const physicalObjectProps5 = {
-            classFullName: PhysicalObject.classFullName,
-            model: physicalElement1.model,
-            category: spatialCategoryId,
-            code: Code.createEmpty(),
-            userLabel: "PhysicalObject5",
-            geom: IModelTestUtils.createBox(Point3d.create(1, 1, 1)),
-            placement: {
-                origin: Point3d.create(5, 5, 5),
-                angles: YawPitchRollAngles.createDegrees(0, 0, 0),
-            },
-        };
-        const physicalObjectId5 = sourceDb.elements.insertElement(physicalObjectProps5);
-        assert.isTrue(Id64.isValidId64(physicalObjectId5));
-        // delete relationship
-        const drawingGraphicId1 = IModelTestUtils.queryByUserLabel(sourceDb, "DrawingGraphic1");
-        const drawingGraphicId2 = IModelTestUtils.queryByUserLabel(sourceDb, "DrawingGraphic2");
-        const relationship = sourceDb.relationships.getInstance(DrawingGraphicRepresentsElement.classFullName, { sourceId: drawingGraphicId2, targetId: physicalObjectId1 });
-        relationship.delete();
-        // insert relationships
-        DrawingGraphicRepresentsElement.insert(sourceDb, drawingGraphicId1, physicalObjectId5);
-        DrawingGraphicRepresentsElement.insert(sourceDb, drawingGraphicId2, physicalObjectId5);
-        // update InformationRecord2
-        const informationRecordCodeSpec = sourceDb.codeSpecs.getByName("InformationRecords");
-        const informationModelId = sourceDb.elements.queryElementIdByCode(InformationPartitionElement.createCode(sourceDb, subjectId, "Information"));
-        const informationRecodeCode2 = new Code({ spec: informationRecordCodeSpec.id, scope: informationModelId, value: "InformationRecord2" });
-        const informationRecordId2 = sourceDb.elements.queryElementIdByCode(informationRecodeCode2);
-        assert.isTrue(Id64.isValidId64(informationRecordId2));
-        const informationRecord2 = sourceDb.elements.getElement(informationRecordId2);
-        informationRecord2.commonString = `${informationRecord2.commonString}-Updated`;
-        informationRecord2.sourceString = `${informationRecord2.sourceString}-Updated`;
-        informationRecord2.update();
-        // delete InformationRecord3
-        const informationRecodeCode3 = new Code({ spec: informationRecordCodeSpec.id, scope: informationModelId, value: "InformationRecord3" });
-        const informationRecordId3 = sourceDb.elements.queryElementIdByCode(informationRecodeCode3);
-        assert.isTrue(Id64.isValidId64(informationRecordId3));
-        sourceDb.elements.deleteElement(informationRecordId3);
+        withEditTxn(sourceDb, (txn) => {
+            // Update Subject element
+            const subjectId = sourceDb.elements.queryElementIdByCode(Subject.createCode(sourceDb, IModel.rootSubjectId, "Subject"));
+            assert.isTrue(Id64.isValidId64(subjectId));
+            const subject = sourceDb.elements.getElement(subjectId);
+            subject.description = "Subject description (Updated)";
+            txn.updateElement(subject.toJSON());
+            // Update spatialCategory element
+            const definitionModelId = sourceDb.elements.queryElementIdByCode(InformationPartitionElement.createCode(sourceDb, subjectId, "Definition"));
+            assert.isTrue(Id64.isValidId64(definitionModelId));
+            const spatialCategoryId = sourceDb.elements.queryElementIdByCode(SpatialCategory.createCode(sourceDb, definitionModelId, "SpatialCategory"));
+            assert.isTrue(Id64.isValidId64(spatialCategoryId));
+            const spatialCategory = sourceDb.elements.getElement(spatialCategoryId);
+            spatialCategory.federationGuid = Guid.createValue();
+            txn.updateElement(spatialCategory.toJSON());
+            // Update relationship properties
+            const spatialCategorySelectorId = sourceDb.elements.queryElementIdByCode(CategorySelector.createCode(sourceDb, definitionModelId, "SpatialCategories"));
+            assert.isTrue(Id64.isValidId64(spatialCategorySelectorId));
+            const drawingCategorySelectorId = sourceDb.elements.queryElementIdByCode(CategorySelector.createCode(sourceDb, definitionModelId, "DrawingCategories"));
+            assert.isTrue(Id64.isValidId64(drawingCategorySelectorId));
+            const relWithProps = sourceDb.relationships.getInstanceProps("ExtensiveTestScenario:SourceRelWithProps", { sourceId: spatialCategorySelectorId, targetId: drawingCategorySelectorId });
+            assert.equal(relWithProps.sourceString, "One");
+            assert.equal(relWithProps.sourceDouble, 1.1);
+            relWithProps.sourceString += "-Updated";
+            relWithProps.sourceDouble = 1.2;
+            txn.updateRelationship(relWithProps);
+            // Update ElementAspect properties
+            const physicalObjectId1 = IModelTestUtils.queryByUserLabel(sourceDb, "PhysicalObject1");
+            const sourceUniqueAspects = sourceDb.elements.getAspects(physicalObjectId1, "ExtensiveTestScenario:SourceUniqueAspect");
+            assert.equal(sourceUniqueAspects.length, 1);
+            sourceUniqueAspects[0].asAny.commonString += "-Updated";
+            sourceUniqueAspects[0].asAny.sourceString += "-Updated";
+            txn.updateAspect(sourceUniqueAspects[0].toJSON());
+            const sourceMultiAspects = sourceDb.elements.getAspects(physicalObjectId1, "ExtensiveTestScenario:SourceMultiAspect");
+            assert.equal(sourceMultiAspects.length, 2);
+            sourceMultiAspects[1].asAny.commonString += "-Updated";
+            sourceMultiAspects[1].asAny.sourceString += "-Updated";
+            txn.updateAspect(sourceMultiAspects[1].toJSON());
+            // clear NavigationProperty of PhysicalElement1
+            const physicalElementId1 = IModelTestUtils.queryByUserLabel(sourceDb, "PhysicalElement1");
+            let physicalElement1 = sourceDb.elements.getElement(physicalElementId1);
+            physicalElement1.asAny.commonNavigation = RelatedElement.none;
+            txn.updateElement(physicalElement1.toJSON());
+            physicalElement1 = sourceDb.elements.getElement(physicalElementId1);
+            assert.isUndefined(physicalElement1.asAny.commonNavigation);
+            // delete PhysicalObject3
+            const physicalObjectId3 = IModelTestUtils.queryByUserLabel(sourceDb, "PhysicalObject3");
+            assert.isTrue(Id64.isValidId64(physicalObjectId3));
+            txn.deleteElement(physicalObjectId3);
+            assert.equal(Id64.invalid, IModelTestUtils.queryByUserLabel(sourceDb, "PhysicalObject3"));
+            // Insert PhysicalObject5
+            const physicalObjectProps5 = {
+                classFullName: PhysicalObject.classFullName,
+                model: physicalElement1.model,
+                category: spatialCategoryId,
+                code: Code.createEmpty(),
+                userLabel: "PhysicalObject5",
+                geom: IModelTestUtils.createBox(Point3d.create(1, 1, 1)),
+                placement: {
+                    origin: Point3d.create(5, 5, 5),
+                    angles: YawPitchRollAngles.createDegrees(0, 0, 0),
+                },
+            };
+            const physicalObjectId5 = txn.insertElement(physicalObjectProps5);
+            assert.isTrue(Id64.isValidId64(physicalObjectId5));
+            // delete relationship
+            const drawingGraphicId1 = IModelTestUtils.queryByUserLabel(sourceDb, "DrawingGraphic1");
+            const drawingGraphicId2 = IModelTestUtils.queryByUserLabel(sourceDb, "DrawingGraphic2");
+            const relationship = sourceDb.relationships.getInstance(DrawingGraphicRepresentsElement.classFullName, { sourceId: drawingGraphicId2, targetId: physicalObjectId1 });
+            txn.deleteRelationship(relationship.toJSON());
+            // insert relationships
+            DrawingGraphicRepresentsElement.insert(txn, drawingGraphicId1, physicalObjectId5);
+            DrawingGraphicRepresentsElement.insert(txn, drawingGraphicId2, physicalObjectId5);
+            // update InformationRecord2
+            const informationRecordCodeSpec = sourceDb.codeSpecs.getByName("InformationRecords");
+            const informationModelId = sourceDb.elements.queryElementIdByCode(InformationPartitionElement.createCode(sourceDb, subjectId, "Information"));
+            const informationRecodeCode2 = new Code({ spec: informationRecordCodeSpec.id, scope: informationModelId, value: "InformationRecord2" });
+            const informationRecordId2 = sourceDb.elements.queryElementIdByCode(informationRecodeCode2);
+            assert.isTrue(Id64.isValidId64(informationRecordId2));
+            const informationRecord2 = sourceDb.elements.getElement(informationRecordId2);
+            informationRecord2.commonString = `${informationRecord2.commonString}-Updated`;
+            informationRecord2.sourceString = `${informationRecord2.sourceString}-Updated`;
+            txn.updateElement(informationRecord2);
+            // delete InformationRecord3
+            const informationRecodeCode3 = new Code({ spec: informationRecordCodeSpec.id, scope: informationModelId, value: "InformationRecord3" });
+            const informationRecordId3 = sourceDb.elements.queryElementIdByCode(informationRecodeCode3);
+            assert.isTrue(Id64.isValidId64(informationRecordId3));
+            txn.deleteElement(informationRecordId3);
+        }); // end withEditTxn
     }
     static assertUpdatesInDb(iModelDb, assertDeletes = true) {
         // determine which schema was imported