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

package/lib/esm/test/SquashSchemaAndDataChanges.test.js

@@ -11,151 +11,152 @@ import { HubMock } from "../internal/HubMock";
 import { KnownTestLocations } from "./KnownTestLocations";
 import * as chai from "chai";
 import { TestUtils } from "./TestUtils";
+import { EditTxn, withEditTxn } from "../EditTxn";
 const schemas = {
     /** Base schema v01.00.00 with classes A, C, D */
-    v01x00x00: `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="td" version="01.00.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
-      <ECEntityClass typeName="A">
-        <BaseClass>bis:GraphicalElement2d</BaseClass>
-        <ECProperty propertyName="PropA" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="C">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropC" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="D">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropD" typeName="string"/>
-      </ECEntityClass>
+    v01x00x00: `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="td" version="01.00.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
+      <ECEntityClass typeName="A">
+        <BaseClass>bis:GraphicalElement2d</BaseClass>
+        <ECProperty propertyName="PropA" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="C">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropC" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="D">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropD" typeName="string"/>
+      </ECEntityClass>
     </ECSchema>`,
     /** v01.00.01 - Adds PropC2 to class C (trivial additive change) */
-    v01x00x01AddPropC2: `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="td" version="01.00.01" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
-      <ECEntityClass typeName="A">
-        <BaseClass>bis:GraphicalElement2d</BaseClass>
-        <ECProperty propertyName="PropA" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="C">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropC" typeName="string"/>
-        <ECProperty propertyName="PropC2" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="D">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropD" typeName="string"/>
-      </ECEntityClass>
+    v01x00x01AddPropC2: `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="td" version="01.00.01" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
+      <ECEntityClass typeName="A">
+        <BaseClass>bis:GraphicalElement2d</BaseClass>
+        <ECProperty propertyName="PropA" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="C">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropC" typeName="string"/>
+        <ECProperty propertyName="PropC2" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="D">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropD" typeName="string"/>
+      </ECEntityClass>
     </ECSchema>`,
     /** v01.00.02 - Adds PropD2 to class D (trivial additive change) */
-    v01x00x02AddPropD2: `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="td" version="01.00.02" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
-      <ECEntityClass typeName="A">
-        <BaseClass>bis:GraphicalElement2d</BaseClass>
-        <ECProperty propertyName="PropA" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="C">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropC" typeName="string"/>
-        <ECProperty propertyName="PropC2" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="D">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropD" typeName="string"/>
-        <ECProperty propertyName="PropD2" typeName="string"/>
-      </ECEntityClass>
+    v01x00x02AddPropD2: `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="td" version="01.00.02" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
+      <ECEntityClass typeName="A">
+        <BaseClass>bis:GraphicalElement2d</BaseClass>
+        <ECProperty propertyName="PropA" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="C">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropC" typeName="string"/>
+        <ECProperty propertyName="PropC2" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="D">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropD" typeName="string"/>
+        <ECProperty propertyName="PropD2" typeName="string"/>
+      </ECEntityClass>
     </ECSchema>`,
     /** v01.00.02 - Moves PropC from C to A (requires data transformation) on top of v01.00.01 */
-    v01x00x02MovePropCToA: `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="td" version="01.00.02" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
-      <ECEntityClass typeName="A">
-        <BaseClass>bis:GraphicalElement2d</BaseClass>
-        <ECProperty propertyName="PropA" typeName="string"/>
-        <ECProperty propertyName="PropC" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="C">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropC2" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="D">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropD" typeName="string"/>
-      </ECEntityClass>
+    v01x00x02MovePropCToA: `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="td" version="01.00.02" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
+      <ECEntityClass typeName="A">
+        <BaseClass>bis:GraphicalElement2d</BaseClass>
+        <ECProperty propertyName="PropA" typeName="string"/>
+        <ECProperty propertyName="PropC" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="C">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropC2" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="D">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropD" typeName="string"/>
+      </ECEntityClass>
     </ECSchema>`,
     /** v01.00.03 - Builds on top of v01.00.02 and in addition moves PropD to base, so we can have incoming and local transforming changes */
-    v01x00x03MovePropCAndD: `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="td" version="01.00.03" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
-      <ECEntityClass typeName="A">
-        <BaseClass>bis:GraphicalElement2d</BaseClass>
-        <ECProperty propertyName="PropA" typeName="string"/>
-        <ECProperty propertyName="PropC" typeName="string"/>
-        <ECProperty propertyName="PropD" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="C">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropC2" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="D">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropD2" typeName="string"/>
-      </ECEntityClass>
+    v01x00x03MovePropCAndD: `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="td" version="01.00.03" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
+      <ECEntityClass typeName="A">
+        <BaseClass>bis:GraphicalElement2d</BaseClass>
+        <ECProperty propertyName="PropA" typeName="string"/>
+        <ECProperty propertyName="PropC" typeName="string"/>
+        <ECProperty propertyName="PropD" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="C">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropC2" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="D">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropD2" typeName="string"/>
+      </ECEntityClass>
     </ECSchema>`,
     /** v01.00.01 (incompatible variant) - Adds PropC3 instead of PropC2 to class C (same version) */
-    v01x00x01AddPropC3Incompatible: `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="td" version="01.00.01" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
-      <ECEntityClass typeName="A">
-        <BaseClass>bis:GraphicalElement2d</BaseClass>
-        <ECProperty propertyName="PropA" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="C">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropC" typeName="string"/>
-        <ECProperty propertyName="PropC3" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="D">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropD" typeName="string"/>
-      </ECEntityClass>
+    v01x00x01AddPropC3Incompatible: `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="td" version="01.00.01" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
+      <ECEntityClass typeName="A">
+        <BaseClass>bis:GraphicalElement2d</BaseClass>
+        <ECProperty propertyName="PropA" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="C">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropC" typeName="string"/>
+        <ECProperty propertyName="PropC3" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="D">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropD" typeName="string"/>
+      </ECEntityClass>
     </ECSchema>`,
     /** v01.00.02 (incompatible variant) - Adds PropC3 instead of PropC2 to class C (higher version) */
-    v01x00x02AddPropC3Incompatible: `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="td" version="01.00.02" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
-      <ECEntityClass typeName="A">
-        <BaseClass>bis:GraphicalElement2d</BaseClass>
-        <ECProperty propertyName="PropA" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="C">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropC" typeName="string"/>
-        <ECProperty propertyName="PropC3" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="D">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropD" typeName="string"/>
-      </ECEntityClass>
+    v01x00x02AddPropC3Incompatible: `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="td" version="01.00.02" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
+      <ECEntityClass typeName="A">
+        <BaseClass>bis:GraphicalElement2d</BaseClass>
+        <ECProperty propertyName="PropA" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="C">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropC" typeName="string"/>
+        <ECProperty propertyName="PropC3" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="D">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropD" typeName="string"/>
+      </ECEntityClass>
     </ECSchema>`,
     /** v01.00.02 (incompatible variant) - Adds PropC2 (higher version, different type) */
-    v01x00x02AddPropC2Incompatible: `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="td" version="01.00.02" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
-      <ECEntityClass typeName="A">
-        <BaseClass>bis:GraphicalElement2d</BaseClass>
-        <ECProperty propertyName="PropA" typeName="string"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="C">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropC" typeName="string"/>
-        <ECProperty propertyName="PropC2" typeName="int"/>
-      </ECEntityClass>
-      <ECEntityClass typeName="D">
-        <BaseClass>A</BaseClass>
-        <ECProperty propertyName="PropD" typeName="string"/>
-      </ECEntityClass>
+    v01x00x02AddPropC2Incompatible: `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="td" version="01.00.02" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.23" alias="bis"/>
+      <ECEntityClass typeName="A">
+        <BaseClass>bis:GraphicalElement2d</BaseClass>
+        <ECProperty propertyName="PropA" typeName="string"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="C">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropC" typeName="string"/>
+        <ECProperty propertyName="PropC2" typeName="int"/>
+      </ECEntityClass>
+      <ECEntityClass typeName="D">
+        <BaseClass>A</BaseClass>
+        <ECProperty propertyName="PropD" typeName="string"/>
+      </ECEntityClass>
     </ECSchema>`,
 };
 describe("SquashSchemaAndDataChanges", () => {
@@ -166,12 +167,13 @@ describe("SquashSchemaAndDataChanges", () => {
     const createModelAndCategory = async (db) => {
         const modelCode = IModelTestUtils.getUniqueModelCode(db, "DrawingModel");
         await db.locks.acquireLocks({ shared: IModel.dictionaryId });
-        const [, newDrawingModelId] = IModelTestUtils.createAndInsertDrawingPartitionAndModel(db, modelCode);
-        const newDrawingCategoryId = DrawingCategory.insert(db, IModel.dictionaryId, "DrawingCategory", new SubCategoryAppearance());
-        db.saveChanges();
-        return [newDrawingModelId, newDrawingCategoryId];
+        return withEditTxn(db, (txn) => {
+            const [, newDrawingModelId] = IModelTestUtils.createAndInsertDrawingPartitionAndModel(txn, modelCode);
+            const newDrawingCategoryId = DrawingCategory.insert(txn, IModel.dictionaryId, "DrawingCategory", new SubCategoryAppearance());
+            return [newDrawingModelId, newDrawingCategoryId];
+        });
     };
-    const insertElement = (briefcase, className, properties) => {
+    const insertElement = (txn, className, properties) => {
         const elementProps = {
             classFullName: className,
             model: drawingModelId,
@@ -179,8 +181,8 @@ describe("SquashSchemaAndDataChanges", () => {
             code: Code.createEmpty(),
             ...properties,
         };
-        const element = briefcase.elements.createElement(elementProps);
-        return briefcase.elements.insertElement(element.toJSON());
+        const element = txn.iModel.elements.createElement(elementProps);
+        return txn.insertElement(element.toJSON());
     };
     before(async () => {
         HubMock.startup("MergeSchemaAndDataChanges", KnownTestLocations.outputDir);
@@ -191,7 +193,6 @@ describe("SquashSchemaAndDataChanges", () => {
         iModelId = await HubWrappers.createIModel("user1", HubMock.iTwinId, `Test-${Guid.createValue()}`);
         imodel = await HubWrappers.downloadAndOpenBriefcase({ accessToken: "user1", iTwinId: HubMock.iTwinId, iModelId });
         imodel.channels.addAllowedChannel(ChannelControl.sharedChannelName);
-        imodel.saveChanges();
         [drawingModelId, drawingCategoryId] = await createModelAndCategory(imodel);
         await imodel.importSchemaStrings([schemas.v01x00x00, schemas.v01x00x01AddPropC2]);
         await imodel.pushChanges({ description: "create model and category and imported schemas" });
@@ -205,22 +206,29 @@ describe("SquashSchemaAndDataChanges", () => {
         await TestUtils.shutdownBackend();
         await TestUtils.startBackend(); // restart normal backend so subsequent test suites aren't left without IModelHost
     });
-    it("should throw error if tried to import schema while unsaved changes are present", async () => {
+    it("should allow schema import while an EditTxn is active", async () => {
         await imodel.locks.acquireLocks({ shared: drawingModelId });
-        insertElement(imodel, "TestDomain:C", {
-            propA: "local_value_a",
-            propC: "local_value_c",
-        });
-        await chai.expect(imodel.importSchemaStrings([schemas.v01x00x02MovePropCToA])).to.be.rejectedWith("Cannot import schemas with unsaved changes when useSemanticRebase flag is on");
-        await imodel.discardChanges();
+        const txn = new EditTxn(imodel, "schema and data changes unsaved state");
+        txn.start();
+        try {
+            txn.saveChanges("prepare for schema import");
+            await imodel.importSchemaStrings([schemas.v01x00x02MovePropCToA]);
+            chai.assert(imodel.containsClass("TestDomain:C"));
+        }
+        finally {
+            if (txn.isActive)
+                txn.end("abandon");
+        }
     });
     it("should squash schema and data changes if useSemanticRebase flag is on", async () => {
         await imodel.locks.acquireLocks({ shared: drawingModelId });
-        insertElement(imodel, "TestDomain:C", {
+        const txn = new EditTxn(imodel, "squash schema and data changes");
+        txn.start();
+        insertElement(txn, "TestDomain:C", {
             propA: "local_value_a",
             propC: "local_value_c",
         });
-        imodel.saveChanges("local data change");
+        txn.saveChanges("local data change");
         await imodel.importSchemaStrings([schemas.v01x00x02MovePropCToA]); // transforming data change
         const lastTxnProps = imodel.txns.getLastSavedTxnProps();
         chai.assert(lastTxnProps !== undefined);
@@ -235,7 +243,8 @@ describe("SquashSchemaAndDataChanges", () => {
         chai.assert(thirdLastTxnProps !== undefined);
         chai.assert(thirdLastTxnProps?.type === "Data");
         chai.assert(thirdLastTxnProps?.prevId === undefined);
-        await imodel.discardChanges();
+        if (txn.isActive)
+            txn.end("abandon");
     });
 });
 //# sourceMappingURL=SquashSchemaAndDataChanges.test.js.map

package/lib/esm/test/SquashSchemaAndDataChanges.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"SquashSchemaAndDataChanges.test.js","sourceRoot":"","sources":["../../../src/test/SquashSchemaAndDataChanges.test.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,IAAI,EAAc,MAAM,qBAAqB,CAAC;AAEvD,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,IAAI,EAAyB,MAAM,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChG,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAExC,MAAM,OAAO,GAAG;IACd,iDAAiD;IACjD,SAAS,EAAE;;;;;;;;;;;;;;;gBAeG;IAEd,mEAAmE;IACnE,kBAAkB,EAAE;;;;;;;;;;;;;;;;gBAgBN;IAEd,mEAAmE;IACnE,kBAAkB,EAAE;;;;;;;;;;;;;;;;;gBAiBN;IAEd,6FAA6F;IAC7F,qBAAqB,EAAE;;;;;;;;;;;;;;;;gBAgBT;IAEd,yIAAyI;IACzI,sBAAsB,EAAE;;;;;;;;;;;;;;;;;gBAiBV;IAEd,iGAAiG;IACjG,8BAA8B,EAAE;;;;;;;;;;;;;;;;gBAgBlB;IAEd,mGAAmG;IACnG,8BAA8B,EAAE;;;;;;;;;;;;;;;;gBAgBlB;IAEd,sFAAsF;IACtF,8BAA8B,EAAE;;;;;;;;;;;;;;;;gBAgBlB;CACf,CAAC;AAEF,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,IAAI,MAAmB,CAAC;IACxB,IAAI,QAAgB,CAAC;IACrB,IAAI,cAAsB,CAAC;IAC3B,IAAI,iBAAyB,CAAC;IAE9B,MAAM,sBAAsB,GAAG,KAAK,EAAE,EAAe,EAAE,EAAE;QACvD,MAAM,SAAS,GAAG,eAAe,CAAC,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;QACzE,MAAM,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,EAAE,iBAAiB,CAAC,GAAG,eAAe,CAAC,uCAAuC,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QACrG,MAAM,oBAAoB,GAAG,eAAe,CAAC,MAAM,CACjD,EAAE,EACF,MAAM,CAAC,YAAY,EACnB,iBAAiB,EACjB,IAAI,qBAAqB,EAAE,CAC5B,CAAC;QACF,EAAE,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,CAAC;IACnD,CAAC,CAAC;IAEF,MAAM,aAAa,GAAG,CACpB,SAAsB,EACtB,SAAiB,EACjB,UAA+B,EACnB,EAAE;QACd,MAAM,YAAY,GAA0B;YAC1C,aAAa,EAAE,SAAS;YACxB,KAAK,EAAE,cAAc;YACrB,QAAQ,EAAE,iBAAiB;YAC3B,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE;YACxB,GAAG,UAAU;SACd,CAAC;QACF,MAAM,OAAO,GAAG,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QAC/D,OAAO,SAAS,CAAC,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAA;IAED,MAAM,CAAC,KAAK,IAAI,EAAE;QAChB,OAAO,CAAC,OAAO,CAAC,2BAA2B,EAAE,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC3E,MAAM,SAAS,CAAC,eAAe,EAAE,CAAC;QAClC,MAAM,SAAS,CAAC,YAAY,CAAC,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,QAAQ,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAElG,MAAM,GAAG,MAAM,WAAW,CAAC,wBAAwB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QAElH,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;QAEpE,MAAM,CAAC,WAAW,EAAE,CAAC;QACrB,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,MAAM,sBAAsB,CAAC,MAAM,CAAC,CAAC;QAC3E,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAClF,MAAM,MAAM,CAAC,WAAW,CAAC,EAAE,WAAW,EAAE,gDAAgD,EAAE,CAAC,CAAC;IAC9F,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,KAAK,IAAI,EAAE;QACf,OAAO,CAAC,QAAQ,EAAE,CAAC;QACnB,MAAM,SAAS,CAAC,eAAe,EAAE,CAAC;QAClC,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,kFAAkF;IACpH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,MAAM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;QAC5D,aAAa,CAAC,MAAM,EAAE,cAAc,EAAE;YACpC,KAAK,EAAE,eAAe;YACtB,KAAK,EAAE,eAAe;SACvB,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,8EAA8E,CAAC,CAAC;QAClL,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;QAC5D,aAAa,CAAC,MAAM,EAAE,cAAc,EAAE;YACpC,KAAK,EAAE,eAAe;YACtB,KAAK,EAAE,eAAe;SACvB,CAAC,CAAC;QACH,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;QACxC,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,2BAA2B;QAE9F,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACxD,IAAI,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;QAChD,qEAAqE;QAErE,MAAM,kBAAkB,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACxE,IAAI,CAAC,MAAM,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,IAAI,KAAK,KAAK,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;QAEtD,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC7E,IAAI,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;QAErD,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n* See LICENSE.md in the project root for license terms and full copyright notice.\n*--------------------------------------------------------------------------------------------*/\n\nimport { Guid, Id64String } from \"@itwin/core-bentley\";\nimport { BriefcaseDb } from \"../IModelDb\";\nimport { HubWrappers, IModelTestUtils } from \"./IModelTestUtils\";\nimport { ChannelControl } from \"../ChannelControl\";\nimport { Code, GeometricElementProps, IModel, SubCategoryAppearance } from \"@itwin/core-common\";\nimport { DrawingCategory } from \"../Category\";\nimport { HubMock } from \"../internal/HubMock\";\nimport { KnownTestLocations } from \"./KnownTestLocations\";\nimport * as chai from \"chai\";\nimport { TestUtils } from \"./TestUtils\";\n\nconst schemas = {\n  /** Base schema v01.00.00 with classes A, C, D */\n  v01x00x00: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.00\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\n      <ECEntityClass typeName=\"A\">\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"C\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"D\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\n      </ECEntityClass>\n    </ECSchema>`,\n\n  /** v01.00.01 - Adds PropC2 to class C (trivial additive change) */\n  v01x00x01AddPropC2: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.01\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\n      <ECEntityClass typeName=\"A\">\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"C\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropC2\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"D\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\n      </ECEntityClass>\n    </ECSchema>`,\n\n  /** v01.00.02 - Adds PropD2 to class D (trivial additive change) */\n  v01x00x02AddPropD2: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.02\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\n      <ECEntityClass typeName=\"A\">\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"C\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropC2\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"D\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropD2\" typeName=\"string\"/>\n      </ECEntityClass>\n    </ECSchema>`,\n\n  /** v01.00.02 - Moves PropC from C to A (requires data transformation) on top of v01.00.01 */\n  v01x00x02MovePropCToA: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.02\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\n      <ECEntityClass typeName=\"A\">\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"C\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropC2\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"D\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\n      </ECEntityClass>\n    </ECSchema>`,\n\n  /** v01.00.03 - Builds on top of v01.00.02 and in addition moves PropD to base, so we can have incoming and local transforming changes */\n  v01x00x03MovePropCAndD: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.03\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\n      <ECEntityClass typeName=\"A\">\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"C\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropC2\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"D\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropD2\" typeName=\"string\"/>\n      </ECEntityClass>\n    </ECSchema>`,\n\n  /** v01.00.01 (incompatible variant) - Adds PropC3 instead of PropC2 to class C (same version) */\n  v01x00x01AddPropC3Incompatible: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.01\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\n      <ECEntityClass typeName=\"A\">\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"C\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropC3\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"D\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\n      </ECEntityClass>\n    </ECSchema>`,\n\n  /** v01.00.02 (incompatible variant) - Adds PropC3 instead of PropC2 to class C (higher version) */\n  v01x00x02AddPropC3Incompatible: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.02\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\n      <ECEntityClass typeName=\"A\">\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"C\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropC3\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"D\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\n      </ECEntityClass>\n    </ECSchema>`,\n\n  /** v01.00.02 (incompatible variant) - Adds PropC2 (higher version, different type) */\n  v01x00x02AddPropC2Incompatible: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.02\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\n      <ECEntityClass typeName=\"A\">\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"C\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\n        <ECProperty propertyName=\"PropC2\" typeName=\"int\"/>\n      </ECEntityClass>\n      <ECEntityClass typeName=\"D\">\n        <BaseClass>A</BaseClass>\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\n      </ECEntityClass>\n    </ECSchema>`,\n};\n\ndescribe(\"SquashSchemaAndDataChanges\", () => {\n  let imodel: BriefcaseDb;\n  let iModelId: string;\n  let drawingModelId: string;\n  let drawingCategoryId: string;\n\n  const createModelAndCategory = async (db: BriefcaseDb) => {\n    const modelCode = IModelTestUtils.getUniqueModelCode(db, \"DrawingModel\");\n    await db.locks.acquireLocks({ shared: IModel.dictionaryId });\n    const [, newDrawingModelId] = IModelTestUtils.createAndInsertDrawingPartitionAndModel(db, modelCode);\n    const newDrawingCategoryId = DrawingCategory.insert(\n      db,\n      IModel.dictionaryId,\n      \"DrawingCategory\",\n      new SubCategoryAppearance()\n    );\n    db.saveChanges();\n    return [newDrawingModelId, newDrawingCategoryId];\n  };\n\n  const insertElement = (\n    briefcase: BriefcaseDb,\n    className: string,\n    properties: Record<string, any>\n  ): Id64String => {\n    const elementProps: GeometricElementProps = {\n      classFullName: className,\n      model: drawingModelId,\n      category: drawingCategoryId,\n      code: Code.createEmpty(),\n      ...properties,\n    };\n    const element = briefcase.elements.createElement(elementProps);\n    return briefcase.elements.insertElement(element.toJSON());\n  }\n\n  before(async () => {\n    HubMock.startup(\"MergeSchemaAndDataChanges\", KnownTestLocations.outputDir);\n    await TestUtils.shutdownBackend();\n    await TestUtils.startBackend({ useSemanticRebase: true });\n  });\n\n  beforeEach(async () => {\n    iModelId = await HubWrappers.createIModel(\"user1\", HubMock.iTwinId, `Test-${Guid.createValue()}`);\n\n    imodel = await HubWrappers.downloadAndOpenBriefcase({ accessToken: \"user1\", iTwinId: HubMock.iTwinId, iModelId });\n\n    imodel.channels.addAllowedChannel(ChannelControl.sharedChannelName);\n\n    imodel.saveChanges();\n    [drawingModelId, drawingCategoryId] = await createModelAndCategory(imodel);\n    await imodel.importSchemaStrings([schemas.v01x00x00, schemas.v01x00x01AddPropC2]);\n    await imodel.pushChanges({ description: \"create model and category and imported schemas\" });\n  });\n\n  afterEach(async () => {\n    imodel.close();\n    await HubMock.deleteIModel({ accessToken: \"user1\", iTwinId: HubMock.iTwinId, iModelId });\n  });\n\n  after(async () => {\n    HubMock.shutdown();\n    await TestUtils.shutdownBackend();\n    await TestUtils.startBackend(); // restart normal backend so subsequent test suites aren't left without IModelHost\n  });\n\n  it(\"should throw error if tried to import schema while unsaved changes are present\", async () => {\n    await imodel.locks.acquireLocks({ shared: drawingModelId });\n    insertElement(imodel, \"TestDomain:C\", {\n      propA: \"local_value_a\",\n      propC: \"local_value_c\",\n    });\n\n    await chai.expect(imodel.importSchemaStrings([schemas.v01x00x02MovePropCToA])).to.be.rejectedWith(\"Cannot import schemas with unsaved changes when useSemanticRebase flag is on\");\n    await imodel.discardChanges();\n  });\n\n  it(\"should squash schema and data changes if useSemanticRebase flag is on\", async () => {\n    await imodel.locks.acquireLocks({ shared: drawingModelId });\n    insertElement(imodel, \"TestDomain:C\", {\n      propA: \"local_value_a\",\n      propC: \"local_value_c\",\n    });\n    imodel.saveChanges(\"local data change\");\n    await imodel.importSchemaStrings([schemas.v01x00x02MovePropCToA]); // transforming data change\n\n    const lastTxnProps = imodel.txns.getLastSavedTxnProps();\n    chai.assert(lastTxnProps !== undefined);\n    chai.assert(lastTxnProps?.type === \"Schema\");\n    chai.assert(lastTxnProps?.prevId !== undefined);\n    // both schema and data(migration) changes are merged into single txn\n\n    const secondLastTxnProps = imodel.txns.getTxnProps(lastTxnProps.prevId);\n    chai.assert(secondLastTxnProps !== undefined);\n    chai.assert(secondLastTxnProps?.type === \"Ddl\");\n    chai.assert(secondLastTxnProps?.prevId !== undefined);\n\n    const thirdLastTxnProps = imodel.txns.getTxnProps(secondLastTxnProps.prevId);\n    chai.assert(thirdLastTxnProps !== undefined);\n    chai.assert(thirdLastTxnProps?.type === \"Data\");\n    chai.assert(thirdLastTxnProps?.prevId === undefined);\n\n    await imodel.discardChanges();\n  });\n});"]}
+{"version":3,"file":"SquashSchemaAndDataChanges.test.js","sourceRoot":"","sources":["../../../src/test/SquashSchemaAndDataChanges.test.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,IAAI,EAAc,MAAM,qBAAqB,CAAC;AAEvD,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,IAAI,EAAyB,MAAM,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAChG,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAElD,MAAM,OAAO,GAAG;IACd,iDAAiD;IACjD,SAAS,EAAE;;;;;;;;;;;;;;;gBAeG;IAEd,mEAAmE;IACnE,kBAAkB,EAAE;;;;;;;;;;;;;;;;gBAgBN;IAEd,mEAAmE;IACnE,kBAAkB,EAAE;;;;;;;;;;;;;;;;;gBAiBN;IAEd,6FAA6F;IAC7F,qBAAqB,EAAE;;;;;;;;;;;;;;;;gBAgBT;IAEd,yIAAyI;IACzI,sBAAsB,EAAE;;;;;;;;;;;;;;;;;gBAiBV;IAEd,iGAAiG;IACjG,8BAA8B,EAAE;;;;;;;;;;;;;;;;gBAgBlB;IAEd,mGAAmG;IACnG,8BAA8B,EAAE;;;;;;;;;;;;;;;;gBAgBlB;IAEd,sFAAsF;IACtF,8BAA8B,EAAE;;;;;;;;;;;;;;;;gBAgBlB;CACf,CAAC;AAEF,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,IAAI,MAAmB,CAAC;IACxB,IAAI,QAAgB,CAAC;IACrB,IAAI,cAAsB,CAAC;IAC3B,IAAI,iBAAyB,CAAC;IAE9B,MAAM,sBAAsB,GAAG,KAAK,EAAE,EAAe,EAAE,EAAE;QACvD,MAAM,SAAS,GAAG,eAAe,CAAC,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;QACzE,MAAM,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QAC7D,OAAO,WAAW,CAAC,EAAE,EAAE,CAAC,GAAG,EAAE,EAAE;YAC7B,MAAM,CAAC,EAAE,iBAAiB,CAAC,GAAG,eAAe,CAAC,uCAAuC,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;YACtG,MAAM,oBAAoB,GAAG,eAAe,CAAC,MAAM,CACjD,GAAG,EACH,MAAM,CAAC,YAAY,EACnB,iBAAiB,EACjB,IAAI,qBAAqB,EAAE,CAC5B,CAAC;YACF,OAAO,CAAC,iBAAiB,EAAE,oBAAoB,CAAU,CAAC;QAC5D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,aAAa,GAAG,CACpB,GAAY,EACZ,SAAiB,EACjB,UAA+B,EACnB,EAAE;QACd,MAAM,YAAY,GAA0B;YAC1C,aAAa,EAAE,SAAS;YACxB,KAAK,EAAE,cAAc;YACrB,QAAQ,EAAE,iBAAiB;YAC3B,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE;YACxB,GAAG,UAAU;SACd,CAAC;QACF,MAAM,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;QAChE,OAAO,GAAG,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC,CAAA;IAED,MAAM,CAAC,KAAK,IAAI,EAAE;QAChB,OAAO,CAAC,OAAO,CAAC,2BAA2B,EAAE,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAC3E,MAAM,SAAS,CAAC,eAAe,EAAE,CAAC;QAClC,MAAM,SAAS,CAAC,YAAY,CAAC,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,QAAQ,GAAG,MAAM,WAAW,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAElG,MAAM,GAAG,MAAM,WAAW,CAAC,wBAAwB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QAElH,MAAM,CAAC,QAAQ,CAAC,iBAAiB,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;QAEpE,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,MAAM,sBAAsB,CAAC,MAAM,CAAC,CAAC;QAC3E,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAClF,MAAM,MAAM,CAAC,WAAW,CAAC,EAAE,WAAW,EAAE,gDAAgD,EAAE,CAAC,CAAC;IAC9F,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,KAAK,IAAI,EAAE;QACf,OAAO,CAAC,QAAQ,EAAE,CAAC;QACnB,MAAM,SAAS,CAAC,eAAe,EAAE,CAAC;QAClC,MAAM,SAAS,CAAC,YAAY,EAAE,CAAC,CAAC,kFAAkF;IACpH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;QAC5D,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC,MAAM,EAAE,uCAAuC,CAAC,CAAC;QACzE,GAAG,CAAC,KAAK,EAAE,CAAC;QACZ,IAAI,CAAC;YACH,GAAG,CAAC,WAAW,CAAC,2BAA2B,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC;YAClE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC;QACpD,CAAC;gBAAS,CAAC;YACT,IAAI,GAAG,CAAC,QAAQ;gBACd,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;QAC5D,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC,MAAM,EAAE,gCAAgC,CAAC,CAAC;QAClE,GAAG,CAAC,KAAK,EAAE,CAAC;QACZ,aAAa,CAAC,GAAG,EAAE,cAAc,EAAE;YACjC,KAAK,EAAE,eAAe;YACtB,KAAK,EAAE,eAAe;SACvB,CAAC,CAAC;QACH,GAAG,CAAC,WAAW,CAAC,mBAAmB,CAAC,CAAC;QACrC,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,2BAA2B;QAE9F,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;QACxD,IAAI,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;QAChD,qEAAqE;QAErE,MAAM,kBAAkB,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QACxE,IAAI,CAAC,MAAM,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,IAAI,KAAK,KAAK,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;QAEtD,MAAM,iBAAiB,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC7E,IAAI,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,IAAI,KAAK,MAAM,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;QAErD,IAAI,GAAG,CAAC,QAAQ;YACd,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { Guid, Id64String } from \"@itwin/core-bentley\";\r\nimport { BriefcaseDb } from \"../IModelDb\";\r\nimport { HubWrappers, IModelTestUtils } from \"./IModelTestUtils\";\r\nimport { ChannelControl } from \"../ChannelControl\";\r\nimport { Code, GeometricElementProps, IModel, SubCategoryAppearance } from \"@itwin/core-common\";\r\nimport { DrawingCategory } from \"../Category\";\r\nimport { HubMock } from \"../internal/HubMock\";\r\nimport { KnownTestLocations } from \"./KnownTestLocations\";\r\nimport * as chai from \"chai\";\r\nimport { TestUtils } from \"./TestUtils\";\r\nimport { EditTxn, withEditTxn } from \"../EditTxn\";\r\n\r\nconst schemas = {\r\n  /** Base schema v01.00.00 with classes A, C, D */\r\n  v01x00x00: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.00\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\r\n      <ECEntityClass typeName=\"A\">\r\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\r\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"C\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"D\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n    </ECSchema>`,\r\n\r\n  /** v01.00.01 - Adds PropC2 to class C (trivial additive change) */\r\n  v01x00x01AddPropC2: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.01\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\r\n      <ECEntityClass typeName=\"A\">\r\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\r\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"C\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropC2\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"D\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n    </ECSchema>`,\r\n\r\n  /** v01.00.02 - Adds PropD2 to class D (trivial additive change) */\r\n  v01x00x02AddPropD2: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.02\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\r\n      <ECEntityClass typeName=\"A\">\r\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\r\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"C\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropC2\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"D\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropD2\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n    </ECSchema>`,\r\n\r\n  /** v01.00.02 - Moves PropC from C to A (requires data transformation) on top of v01.00.01 */\r\n  v01x00x02MovePropCToA: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.02\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\r\n      <ECEntityClass typeName=\"A\">\r\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\r\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"C\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropC2\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"D\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n    </ECSchema>`,\r\n\r\n  /** v01.00.03 - Builds on top of v01.00.02 and in addition moves PropD to base, so we can have incoming and local transforming changes */\r\n  v01x00x03MovePropCAndD: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.03\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\r\n      <ECEntityClass typeName=\"A\">\r\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\r\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"C\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropC2\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"D\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropD2\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n    </ECSchema>`,\r\n\r\n  /** v01.00.01 (incompatible variant) - Adds PropC3 instead of PropC2 to class C (same version) */\r\n  v01x00x01AddPropC3Incompatible: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.01\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\r\n      <ECEntityClass typeName=\"A\">\r\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\r\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"C\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropC3\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"D\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n    </ECSchema>`,\r\n\r\n  /** v01.00.02 (incompatible variant) - Adds PropC3 instead of PropC2 to class C (higher version) */\r\n  v01x00x02AddPropC3Incompatible: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.02\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\r\n      <ECEntityClass typeName=\"A\">\r\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\r\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"C\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropC3\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"D\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n    </ECSchema>`,\r\n\r\n  /** v01.00.02 (incompatible variant) - Adds PropC2 (higher version, different type) */\r\n  v01x00x02AddPropC2Incompatible: `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"TestDomain\" alias=\"td\" version=\"01.00.02\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.23\" alias=\"bis\"/>\r\n      <ECEntityClass typeName=\"A\">\r\n        <BaseClass>bis:GraphicalElement2d</BaseClass>\r\n        <ECProperty propertyName=\"PropA\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"C\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropC\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"PropC2\" typeName=\"int\"/>\r\n      </ECEntityClass>\r\n      <ECEntityClass typeName=\"D\">\r\n        <BaseClass>A</BaseClass>\r\n        <ECProperty propertyName=\"PropD\" typeName=\"string\"/>\r\n      </ECEntityClass>\r\n    </ECSchema>`,\r\n};\r\n\r\ndescribe(\"SquashSchemaAndDataChanges\", () => {\r\n  let imodel: BriefcaseDb;\r\n  let iModelId: string;\r\n  let drawingModelId: string;\r\n  let drawingCategoryId: string;\r\n\r\n  const createModelAndCategory = async (db: BriefcaseDb) => {\r\n    const modelCode = IModelTestUtils.getUniqueModelCode(db, \"DrawingModel\");\r\n    await db.locks.acquireLocks({ shared: IModel.dictionaryId });\r\n    return withEditTxn(db, (txn) => {\r\n      const [, newDrawingModelId] = IModelTestUtils.createAndInsertDrawingPartitionAndModel(txn, modelCode);\r\n      const newDrawingCategoryId = DrawingCategory.insert(\r\n        txn,\r\n        IModel.dictionaryId,\r\n        \"DrawingCategory\",\r\n        new SubCategoryAppearance()\r\n      );\r\n      return [newDrawingModelId, newDrawingCategoryId] as const;\r\n    });\r\n  };\r\n\r\n  const insertElement = (\r\n    txn: EditTxn,\r\n    className: string,\r\n    properties: Record<string, any>\r\n  ): Id64String => {\r\n    const elementProps: GeometricElementProps = {\r\n      classFullName: className,\r\n      model: drawingModelId,\r\n      category: drawingCategoryId,\r\n      code: Code.createEmpty(),\r\n      ...properties,\r\n    };\r\n    const element = txn.iModel.elements.createElement(elementProps);\r\n    return txn.insertElement(element.toJSON());\r\n  }\r\n\r\n  before(async () => {\r\n    HubMock.startup(\"MergeSchemaAndDataChanges\", KnownTestLocations.outputDir);\r\n    await TestUtils.shutdownBackend();\r\n    await TestUtils.startBackend({ useSemanticRebase: true });\r\n  });\r\n\r\n  beforeEach(async () => {\r\n    iModelId = await HubWrappers.createIModel(\"user1\", HubMock.iTwinId, `Test-${Guid.createValue()}`);\r\n\r\n    imodel = await HubWrappers.downloadAndOpenBriefcase({ accessToken: \"user1\", iTwinId: HubMock.iTwinId, iModelId });\r\n\r\n    imodel.channels.addAllowedChannel(ChannelControl.sharedChannelName);\r\n\r\n    [drawingModelId, drawingCategoryId] = await createModelAndCategory(imodel);\r\n    await imodel.importSchemaStrings([schemas.v01x00x00, schemas.v01x00x01AddPropC2]);\r\n    await imodel.pushChanges({ description: \"create model and category and imported schemas\" });\r\n  });\r\n\r\n  afterEach(async () => {\r\n    imodel.close();\r\n    await HubMock.deleteIModel({ accessToken: \"user1\", iTwinId: HubMock.iTwinId, iModelId });\r\n  });\r\n\r\n  after(async () => {\r\n    HubMock.shutdown();\r\n    await TestUtils.shutdownBackend();\r\n    await TestUtils.startBackend(); // restart normal backend so subsequent test suites aren't left without IModelHost\r\n  });\r\n\r\n  it(\"should allow schema import while an EditTxn is active\", async () => {\r\n    await imodel.locks.acquireLocks({ shared: drawingModelId });\r\n    const txn = new EditTxn(imodel, \"schema and data changes unsaved state\");\r\n    txn.start();\r\n    try {\r\n      txn.saveChanges(\"prepare for schema import\");\r\n      await imodel.importSchemaStrings([schemas.v01x00x02MovePropCToA]);\r\n      chai.assert(imodel.containsClass(\"TestDomain:C\"));\r\n    } finally {\r\n      if (txn.isActive)\r\n        txn.end(\"abandon\");\r\n    }\r\n  });\r\n\r\n  it(\"should squash schema and data changes if useSemanticRebase flag is on\", async () => {\r\n    await imodel.locks.acquireLocks({ shared: drawingModelId });\r\n    const txn = new EditTxn(imodel, \"squash schema and data changes\");\r\n    txn.start();\r\n    insertElement(txn, \"TestDomain:C\", {\r\n      propA: \"local_value_a\",\r\n      propC: \"local_value_c\",\r\n    });\r\n    txn.saveChanges(\"local data change\");\r\n    await imodel.importSchemaStrings([schemas.v01x00x02MovePropCToA]); // transforming data change\r\n\r\n    const lastTxnProps = imodel.txns.getLastSavedTxnProps();\r\n    chai.assert(lastTxnProps !== undefined);\r\n    chai.assert(lastTxnProps?.type === \"Schema\");\r\n    chai.assert(lastTxnProps?.prevId !== undefined);\r\n    // both schema and data(migration) changes are merged into single txn\r\n\r\n    const secondLastTxnProps = imodel.txns.getTxnProps(lastTxnProps.prevId);\r\n    chai.assert(secondLastTxnProps !== undefined);\r\n    chai.assert(secondLastTxnProps?.type === \"Ddl\");\r\n    chai.assert(secondLastTxnProps?.prevId !== undefined);\r\n\r\n    const thirdLastTxnProps = imodel.txns.getTxnProps(secondLastTxnProps.prevId);\r\n    chai.assert(thirdLastTxnProps !== undefined);\r\n    chai.assert(thirdLastTxnProps?.type === \"Data\");\r\n    chai.assert(thirdLastTxnProps?.prevId === undefined);\r\n\r\n    if (txn.isActive)\r\n      txn.end(\"abandon\");\r\n  });\r\n});\r\n"]}

package/lib/esm/test/TestChangeSetUtility.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TestChangeSetUtility.d.ts","sourceRoot":"","sources":["../../../src/test/TestChangeSetUtility.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAE9D,OAAO,EAAE,WAAW,EAAmC,MAAM,iBAAiB,CAAC;AAI/E,oDAAoD;AACpD,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IAE9B,OAAO,EAAG,UAAU,CAAC;IACrB,QAAQ,EAAG,UAAU,CAAC;IAC7B,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,YAAY,CAAc;IAElC,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,WAAW,CAAU;gBAEjB,WAAW,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM;YAK1C,YAAY;YAKZ,eAAe;YAKf,eAAe;IAM7B;;OAEG;IACU,gBAAgB,IAAI,OAAO,CAAC,WAAW,CAAC;IAmBxC,iBAAiB;IAOjB,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAM/C"}
+{"version":3,"file":"TestChangeSetUtility.d.ts","sourceRoot":"","sources":["../../../src/test/TestChangeSetUtility.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAE9D,OAAO,EAAE,WAAW,EAAmC,MAAM,iBAAiB,CAAC;AAK/E,oDAAoD;AACpD,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IAE9B,OAAO,EAAG,UAAU,CAAC;IACrB,QAAQ,EAAG,UAAU,CAAC;IAC7B,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,YAAY,CAAc;IAElC,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,WAAW,CAAU;gBAEjB,WAAW,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM;YAK1C,YAAY;YAMZ,eAAe;YAMf,eAAe;IAO7B;;OAEG;IACU,gBAAgB,IAAI,OAAO,CAAC,WAAW,CAAC;IAmBxC,iBAAiB;IAOjB,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAM/C"}

package/lib/esm/test/TestChangeSetUtility.js

@@ -6,6 +6,7 @@ import { ColorDef, IModel, SubCategoryAppearance } from "@itwin/core-common";
 import { ChannelControl, SpatialCategory } from "../core-backend";
 import { HubMock } from "../internal/HubMock";
 import { HubWrappers, IModelTestUtils } from "./IModelTestUtils";
+import { withEditTxn } from "../EditTxn";
 /** Test utility to push an iModel and ChangeSets */
 export class TestChangeSetUtility {
     _iModelName;
@@ -20,17 +21,20 @@ export class TestChangeSetUtility {
         this._iModelName = IModelTestUtils.generateUniqueName(iModelName); // Generate a unique name for the iModel (so that this test can be run simultaneously by multiple users+hosts simultaneously)
     }
     async addTestModel() {
-        [, this._modelId] = IModelTestUtils.createAndInsertPhysicalPartitionAndModel(this._iModel, IModelTestUtils.getUniqueModelCode(this._iModel, "TestPhysicalModel"), true);
-        this._iModel.saveChanges("Added test model");
+        withEditTxn(this._iModel, "Added test model", (txn) => {
+            [, this._modelId] = IModelTestUtils.createAndInsertPhysicalPartitionAndModel(txn, IModelTestUtils.getUniqueModelCode(txn.iModel, "TestPhysicalModel"), true);
+        });
     }
     async addTestCategory() {
-        this._categoryId = SpatialCategory.insert(this._iModel, IModel.dictionaryId, "TestSpatialCategory", new SubCategoryAppearance({ color: ColorDef.fromString("rgb(255,0,0)").toJSON() }));
-        this._iModel.saveChanges("Added test category");
+        withEditTxn(this._iModel, "Added test category", (txn) => {
+            this._categoryId = SpatialCategory.insert(txn, IModel.dictionaryId, "TestSpatialCategory", new SubCategoryAppearance({ color: ColorDef.fromString("rgb(255,0,0)").toJSON() }));
+        });
     }
     async addTestElements() {
-        this._iModel.elements.insertElement(IModelTestUtils.createPhysicalObject(this._iModel, this._modelId, this._categoryId).toJSON());
-        this._iModel.elements.insertElement(IModelTestUtils.createPhysicalObject(this._iModel, this._modelId, this._categoryId).toJSON());
-        this._iModel.saveChanges("Added test elements");
+        withEditTxn(this._iModel, "Added test elements", (txn) => {
+            txn.insertElement(IModelTestUtils.createPhysicalObject(txn.iModel, this._modelId, this._categoryId).toJSON());
+            txn.insertElement(IModelTestUtils.createPhysicalObject(txn.iModel, this._modelId, this._categoryId).toJSON());
+        });
     }
     /** Create a new iModel, populate it, push the changes and returns the opened db.
      * Uses the iTwinId from HubMock.

package/lib/esm/test/TestChangeSetUtility.js.map

@@ -1 +1 @@
-{"version":3,"file":"TestChangeSetUtility.js","sourceRoot":"","sources":["../../../src/test/TestChangeSetUtility.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAG/F,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC7E,OAAO,EAAe,cAAc,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAC/E,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEjE,oDAAoD;AACpD,MAAM,OAAO,oBAAoB;IACd,WAAW,CAAS;IAE9B,OAAO,CAAc;IACrB,QAAQ,CAAc;IACrB,OAAO,CAAe;IACtB,YAAY,CAAc;IAE1B,QAAQ,CAAU;IAClB,WAAW,CAAU;IAE7B,YAAY,WAAwB,EAAE,UAAkB;QACtD,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;QAChC,IAAI,CAAC,WAAW,GAAG,eAAe,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,6HAA6H;IAClM,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,eAAe,CAAC,wCAAwC,CAAC,IAAI,CAAC,OAAO,EAAE,eAAe,CAAC,kBAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,mBAAmB,CAAC,EAAE,IAAI,CAAC,CAAC;QACxK,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;IAC/C,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,WAAW,GAAG,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,YAAY,EAAE,qBAAqB,EAAE,IAAI,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;QACxL,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,qBAAqB,CAAC,CAAC;IAClD,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,eAAe,CAAC,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAClI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,eAAe,CAAC,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAClI,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,qBAAqB,CAAC,CAAC;IAClD,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAE/B,gCAAgC;QAChC,IAAI,CAAC,QAAQ,GAAG,MAAM,WAAW,CAAC,cAAc,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAEzJ,IAAI,CAAC,OAAO,GAAG,MAAM,WAAW,CAAC,wBAAwB,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9I,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;QAE1E,uBAAuB;QACvB,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1B,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAE7B,0BAA0B;QAC1B,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC,CAAC;QACpG,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEM,KAAK,CAAC,iBAAiB;QAC5B,IAAI,CAAC,IAAI,CAAC,OAAO;YACf,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,WAAW,EAAE,qBAAqB,EAAE,CAAC,CAAC;IACzG,CAAC;IAEM,KAAK,CAAC,gBAAgB;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO;YACf,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,MAAM,WAAW,CAAC,yBAAyB,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7E,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IACjH,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n* See LICENSE.md in the project root for license terms and full copyright notice.\n*--------------------------------------------------------------------------------------------*/\n\nimport { AccessToken, GuidString } from \"@itwin/core-bentley\";\nimport { ColorDef, IModel, SubCategoryAppearance } from \"@itwin/core-common\";\nimport { BriefcaseDb, ChannelControl, SpatialCategory } from \"../core-backend\";\nimport { HubMock } from \"../internal/HubMock\";\nimport { HubWrappers, IModelTestUtils } from \"./IModelTestUtils\";\n\n/** Test utility to push an iModel and ChangeSets */\nexport class TestChangeSetUtility {\n  private readonly _iModelName: string;\n\n  public iTwinId!: GuidString;\n  public iModelId!: GuidString;\n  private _iModel!: BriefcaseDb;\n  private _accessToken: AccessToken;\n\n  private _modelId!: string;\n  private _categoryId!: string;\n\n  constructor(accessToken: AccessToken, iModelName: string) {\n    this._accessToken = accessToken;\n    this._iModelName = IModelTestUtils.generateUniqueName(iModelName); // Generate a unique name for the iModel (so that this test can be run simultaneously by multiple users+hosts simultaneously)\n  }\n\n  private async addTestModel(): Promise<void> {\n    [, this._modelId] = IModelTestUtils.createAndInsertPhysicalPartitionAndModel(this._iModel, IModelTestUtils.getUniqueModelCode(this._iModel, \"TestPhysicalModel\"), true);\n    this._iModel.saveChanges(\"Added test model\");\n  }\n\n  private async addTestCategory(): Promise<void> {\n    this._categoryId = SpatialCategory.insert(this._iModel, IModel.dictionaryId, \"TestSpatialCategory\", new SubCategoryAppearance({ color: ColorDef.fromString(\"rgb(255,0,0)\").toJSON() }));\n    this._iModel.saveChanges(\"Added test category\");\n  }\n\n  private async addTestElements(): Promise<void> {\n    this._iModel.elements.insertElement(IModelTestUtils.createPhysicalObject(this._iModel, this._modelId, this._categoryId).toJSON());\n    this._iModel.elements.insertElement(IModelTestUtils.createPhysicalObject(this._iModel, this._modelId, this._categoryId).toJSON());\n    this._iModel.saveChanges(\"Added test elements\");\n  }\n\n  /** Create a new iModel, populate it, push the changes and returns the opened db.\n   * Uses the iTwinId from HubMock.\n   */\n  public async createTestIModel(): Promise<BriefcaseDb> {\n    this.iTwinId = HubMock.iTwinId;\n\n    // Re-create iModel on iModelHub\n    this.iModelId = await HubWrappers.recreateIModel({ accessToken: this._accessToken, iTwinId: this.iTwinId, iModelName: this._iModelName, noLocks: true });\n\n    this._iModel = await HubWrappers.downloadAndOpenBriefcase({ accessToken: this._accessToken, iTwinId: this.iTwinId, iModelId: this.iModelId });\n    this._iModel.channels.addAllowedChannel(ChannelControl.sharedChannelName);\n\n    // Populate sample data\n    await this.addTestModel();\n    await this.addTestCategory();\n    await this.addTestElements();\n\n    // Push changes to the hub\n    await this._iModel.pushChanges({ accessToken: this._accessToken, description: \"Setup test model\" });\n    return this._iModel;\n  }\n\n  public async pushTestChangeSet() {\n    if (!this._iModel)\n      throw new Error(\"Must first call createTestIModel\");\n    await this.addTestElements();\n    await this._iModel.pushChanges({ accessToken: this._accessToken, description: \"Added test elements\" });\n  }\n\n  public async deleteTestIModel(): Promise<void> {\n    if (!this._iModel)\n      throw new Error(\"Must first call createTestIModel\");\n    await HubWrappers.closeAndDeleteBriefcaseDb(this._accessToken, this._iModel);\n    await HubMock.deleteIModel({ accessToken: this._accessToken, iTwinId: this.iTwinId, iModelId: this.iModelId });\n  }\n}\n"]}
+{"version":3,"file":"TestChangeSetUtility.js","sourceRoot":"","sources":["../../../src/test/TestChangeSetUtility.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAG/F,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC7E,OAAO,EAAe,cAAc,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAC/E,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAEzC,oDAAoD;AACpD,MAAM,OAAO,oBAAoB;IACd,WAAW,CAAS;IAE9B,OAAO,CAAc;IACrB,QAAQ,CAAc;IACrB,OAAO,CAAe;IACtB,YAAY,CAAc;IAE1B,QAAQ,CAAU;IAClB,WAAW,CAAU;IAE7B,YAAY,WAAwB,EAAE,UAAkB;QACtD,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC;QAChC,IAAI,CAAC,WAAW,GAAG,eAAe,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,6HAA6H;IAClM,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,kBAAkB,EAAE,CAAC,GAAG,EAAE,EAAE;YACpD,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,eAAe,CAAC,wCAAwC,CAAC,GAAG,EAAE,eAAe,CAAC,kBAAkB,CAAC,GAAG,CAAC,MAAM,EAAE,mBAAmB,CAAC,EAAE,IAAI,CAAC,CAAC;QAC/J,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,qBAAqB,EAAE,CAAC,GAAG,EAAE,EAAE;YACvD,IAAI,CAAC,WAAW,GAAG,eAAe,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,YAAY,EAAE,qBAAqB,EAAE,IAAI,qBAAqB,CAAC,EAAE,KAAK,EAAE,QAAQ,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;QACjL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,qBAAqB,EAAE,CAAC,GAAG,EAAE,EAAE;YACvD,GAAG,CAAC,aAAa,CAAC,eAAe,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC9G,GAAG,CAAC,aAAa,CAAC,eAAe,CAAC,oBAAoB,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAChH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,gBAAgB;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAE/B,gCAAgC;QAChC,IAAI,CAAC,QAAQ,GAAG,MAAM,WAAW,CAAC,cAAc,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAEzJ,IAAI,CAAC,OAAO,GAAG,MAAM,WAAW,CAAC,wBAAwB,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9I,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;QAE1E,uBAAuB;QACvB,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAC1B,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAE7B,0BAA0B;QAC1B,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC,CAAC;QACpG,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEM,KAAK,CAAC,iBAAiB;QAC5B,IAAI,CAAC,IAAI,CAAC,OAAO;YACf,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAC7B,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,WAAW,EAAE,qBAAqB,EAAE,CAAC,CAAC;IACzG,CAAC;IAEM,KAAK,CAAC,gBAAgB;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO;YACf,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;QACtD,MAAM,WAAW,CAAC,yBAAyB,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7E,MAAM,OAAO,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IACjH,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n\r\nimport { AccessToken, GuidString } from \"@itwin/core-bentley\";\r\nimport { ColorDef, IModel, SubCategoryAppearance } from \"@itwin/core-common\";\r\nimport { BriefcaseDb, ChannelControl, SpatialCategory } from \"../core-backend\";\r\nimport { HubMock } from \"../internal/HubMock\";\r\nimport { HubWrappers, IModelTestUtils } from \"./IModelTestUtils\";\r\nimport { withEditTxn } from \"../EditTxn\";\r\n\r\n/** Test utility to push an iModel and ChangeSets */\r\nexport class TestChangeSetUtility {\r\n  private readonly _iModelName: string;\r\n\r\n  public iTwinId!: GuidString;\r\n  public iModelId!: GuidString;\r\n  private _iModel!: BriefcaseDb;\r\n  private _accessToken: AccessToken;\r\n\r\n  private _modelId!: string;\r\n  private _categoryId!: string;\r\n\r\n  constructor(accessToken: AccessToken, iModelName: string) {\r\n    this._accessToken = accessToken;\r\n    this._iModelName = IModelTestUtils.generateUniqueName(iModelName); // Generate a unique name for the iModel (so that this test can be run simultaneously by multiple users+hosts simultaneously)\r\n  }\r\n\r\n  private async addTestModel(): Promise<void> {\r\n    withEditTxn(this._iModel, \"Added test model\", (txn) => {\r\n      [, this._modelId] = IModelTestUtils.createAndInsertPhysicalPartitionAndModel(txn, IModelTestUtils.getUniqueModelCode(txn.iModel, \"TestPhysicalModel\"), true);\r\n    });\r\n  }\r\n\r\n  private async addTestCategory(): Promise<void> {\r\n    withEditTxn(this._iModel, \"Added test category\", (txn) => {\r\n      this._categoryId = SpatialCategory.insert(txn, IModel.dictionaryId, \"TestSpatialCategory\", new SubCategoryAppearance({ color: ColorDef.fromString(\"rgb(255,0,0)\").toJSON() }));\r\n    });\r\n  }\r\n\r\n  private async addTestElements(): Promise<void> {\r\n    withEditTxn(this._iModel, \"Added test elements\", (txn) => {\r\n      txn.insertElement(IModelTestUtils.createPhysicalObject(txn.iModel, this._modelId, this._categoryId).toJSON());\r\n      txn.insertElement(IModelTestUtils.createPhysicalObject(txn.iModel, this._modelId, this._categoryId).toJSON());\r\n    });\r\n  }\r\n\r\n  /** Create a new iModel, populate it, push the changes and returns the opened db.\r\n   * Uses the iTwinId from HubMock.\r\n   */\r\n  public async createTestIModel(): Promise<BriefcaseDb> {\r\n    this.iTwinId = HubMock.iTwinId;\r\n\r\n    // Re-create iModel on iModelHub\r\n    this.iModelId = await HubWrappers.recreateIModel({ accessToken: this._accessToken, iTwinId: this.iTwinId, iModelName: this._iModelName, noLocks: true });\r\n\r\n    this._iModel = await HubWrappers.downloadAndOpenBriefcase({ accessToken: this._accessToken, iTwinId: this.iTwinId, iModelId: this.iModelId });\r\n    this._iModel.channels.addAllowedChannel(ChannelControl.sharedChannelName);\r\n\r\n    // Populate sample data\r\n    await this.addTestModel();\r\n    await this.addTestCategory();\r\n    await this.addTestElements();\r\n\r\n    // Push changes to the hub\r\n    await this._iModel.pushChanges({ accessToken: this._accessToken, description: \"Setup test model\" });\r\n    return this._iModel;\r\n  }\r\n\r\n  public async pushTestChangeSet() {\r\n    if (!this._iModel)\r\n      throw new Error(\"Must first call createTestIModel\");\r\n    await this.addTestElements();\r\n    await this._iModel.pushChanges({ accessToken: this._accessToken, description: \"Added test elements\" });\r\n  }\r\n\r\n  public async deleteTestIModel(): Promise<void> {\r\n    if (!this._iModel)\r\n      throw new Error(\"Must first call createTestIModel\");\r\n    await HubWrappers.closeAndDeleteBriefcaseDb(this._accessToken, this._iModel);\r\n    await HubMock.deleteIModel({ accessToken: this._accessToken, iTwinId: this.iTwinId, iModelId: this.iModelId });\r\n  }\r\n}\r\n"]}

package/lib/esm/test/TestEditTxn.d.ts

@@ -0,0 +1,8 @@
+import { SaveChangesArgs } from "@itwin/core-common";
+import { EditTxn } from "../EditTxn";
+import type { IModelDb } from "../IModelDb";
+export declare function withEditTxn<T>(iModel: IModelDb, fn: (txn: EditTxn) => T): T;
+export declare function withEditTxn<T>(iModel: IModelDb, saveArgs: string | SaveChangesArgs, fn: (txn: EditTxn) => T): T;
+export declare function withEditTxn<T>(iModel: IModelDb, fn: (txn: EditTxn) => Promise<T>): Promise<T>;
+export declare function withEditTxn<T>(iModel: IModelDb, saveArgs: string | SaveChangesArgs, fn: (txn: EditTxn) => Promise<T>): Promise<T>;
+//# sourceMappingURL=TestEditTxn.d.ts.map

package/lib/esm/test/TestEditTxn.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestEditTxn.d.ts","sourceRoot":"","sources":["../../../src/test/TestEditTxn.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACrC,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAE5C,wBAAgB,WAAW,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC;AAC7E,wBAAgB,WAAW,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,GAAG,eAAe,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC;AACjH,wBAAgB,WAAW,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;AAC/F,wBAAgB,WAAW,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,GAAG,eAAe,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC"}

package/lib/esm/test/TestEditTxn.js

@@ -0,0 +1,34 @@
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+import { EditTxn } from "../EditTxn";
+export function withEditTxn(iModel, saveArgsOrFn, maybeFn) {
+    const saveArgs = "function" === typeof saveArgsOrFn ? undefined : saveArgsOrFn;
+    const fn = "function" === typeof saveArgsOrFn ? saveArgsOrFn : maybeFn;
+    if (undefined === fn)
+        throw new Error("withEditTxn requires a callback");
+    const txn = new EditTxn(iModel, "test");
+    txn.start();
+    try {
+        const result = fn(txn);
+        if (result instanceof Promise) {
+            return result.then((value) => {
+                txn.end("save", saveArgs);
+                return value;
+            }, (err) => {
+                if (txn.isActive)
+                    txn.end("abandon");
+                throw err;
+            });
+        }
+        txn.end("save", saveArgs);
+        return result;
+    }
+    catch (err) {
+        if (txn.isActive)
+            txn.end("abandon");
+        throw err;
+    }
+}
+//# sourceMappingURL=TestEditTxn.js.map