@itwin/core-backend 5.4.0-dev.7 → 5.5.0-dev.1

package/lib/esm/test/hubaccess/Rebase.test.js

@@ -2,7 +2,7 @@
 * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
-import { Guid } from "@itwin/core-bentley";
+import { DbResult, Guid } from "@itwin/core-bentley";
 import { Code, IModel, QueryBinder, SubCategoryAppearance } from "@itwin/core-common";
 import * as chai from "chai";
 import * as chaiAsPromised from "chai-as-promised";
@@ -26,22 +26,22 @@ class TestIModel {
         const b1 = await HubWrappers.downloadAndOpenBriefcase({ iTwinId: HubMock.iTwinId, iModelId: this.iModelId });
         b1.channels.addAllowedChannel(ChannelControl.sharedChannelName);
         b1.saveChanges();
-        const schema1 = `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="TestDomain" alias="ts" version="01.00.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-        <ECSchemaReference name="BisCore" version="01.00.00" alias="bis"/>
-        <ECEntityClass typeName="a1">
-            <BaseClass>bis:GraphicalElement2d</BaseClass>
-            <ECProperty propertyName="prop1" typeName="string" />
-        </ECEntityClass>
-        <ECRelationshipClass typeName="A1OwnsA1" modifier="None" strength="embedding">
-            <BaseClass>bis:ElementOwnsChildElements</BaseClass>
-            <Source multiplicity="(0..1)" roleLabel="owns" polymorphic="true">
-                <Class class="a1"/>
-            </Source>
-            <Target multiplicity="(0..*)" roleLabel="is owned by" polymorphic="false">
-                <Class class="a1"/>
-            </Target>
-        </ECRelationshipClass>
+        const schema1 = `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="TestDomain" alias="ts" version="01.00.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+        <ECSchemaReference name="BisCore" version="01.00.00" alias="bis"/>
+        <ECEntityClass typeName="a1">
+            <BaseClass>bis:GraphicalElement2d</BaseClass>
+            <ECProperty propertyName="prop1" typeName="string" />
+        </ECEntityClass>
+        <ECRelationshipClass typeName="A1OwnsA1" modifier="None" strength="embedding">
+            <BaseClass>bis:ElementOwnsChildElements</BaseClass>
+            <Source multiplicity="(0..1)" roleLabel="owns" polymorphic="true">
+                <Class class="a1"/>
+            </Source>
+            <Target multiplicity="(0..*)" roleLabel="is owned by" polymorphic="false">
+                <Class class="a1"/>
+            </Target>
+        </ECRelationshipClass>
     </ECSchema>`;
         await b1.importSchemaStrings([schema1]);
         chai.expect(b1.txns.hasPendingTxns).to.be.true;
@@ -391,13 +391,13 @@ describe("rebase changes & stashing api", function () {
     });
     it("should fail to importSchemas() & importSchemaStrings() in indirect scope", async () => {
         const b1 = await testIModel.openBriefcase();
-        const schema = `<?xml version="1.0" encoding="UTF-8"?>
-    <ECSchema schemaName="MySchema" alias="ms1" version="01.00.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-        <ECSchemaReference name="BisCore" version="01.00.00" alias="bis"/>
-        <ECEntityClass typeName="my_class">
-            <BaseClass>bis:GraphicalElement2d</BaseClass>
-            <ECProperty propertyName="prop1" typeName="string" />
-        </ECEntityClass>
+        const schema = `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="MySchema" alias="ms1" version="01.00.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+        <ECSchemaReference name="BisCore" version="01.00.00" alias="bis"/>
+        <ECEntityClass typeName="my_class">
+            <BaseClass>bis:GraphicalElement2d</BaseClass>
+            <ECProperty propertyName="prop1" typeName="string" />
+        </ECEntityClass>
     </ECSchema>`;
         const schemaFile = path.join(KnownTestLocations.outputDir, "MySchema.01.00.00.ecschema.xml");
         if (existsSync(schemaFile)) {
@@ -678,23 +678,23 @@ describe("rebase changes & stashing api", function () {
     });
     it("schema change should not be stashed", async () => {
         const b1 = await testIModel.openBriefcase();
-        const schema1 = `<?xml version="1.0" encoding="UTF-8"?>
-        <ECSchema schemaName="TestDomain" alias="ts" version="01.00.01" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
-            <ECSchemaReference name="BisCore" version="01.00.00" alias="bis"/>
-            <ECEntityClass typeName="a1">
-                <BaseClass>bis:GraphicalElement2d</BaseClass>
-                <ECProperty propertyName="prop1" typeName="string" />
-                <ECProperty propertyName="prop2" typeName="string" />
-            </ECEntityClass>
-            <ECRelationshipClass typeName="A1OwnsA1" modifier="None" strength="embedding">
-                <BaseClass>bis:ElementOwnsChildElements</BaseClass>
-                <Source multiplicity="(0..1)" roleLabel="owns" polymorphic="true">
-                    <Class class="a1"/>
-                </Source>
-                <Target multiplicity="(0..*)" roleLabel="is owned by" polymorphic="false">
-                    <Class class="a1"/>
-                </Target>
-            </ECRelationshipClass>
+        const schema1 = `<?xml version="1.0" encoding="UTF-8"?>
+        <ECSchema schemaName="TestDomain" alias="ts" version="01.00.01" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+            <ECSchemaReference name="BisCore" version="01.00.00" alias="bis"/>
+            <ECEntityClass typeName="a1">
+                <BaseClass>bis:GraphicalElement2d</BaseClass>
+                <ECProperty propertyName="prop1" typeName="string" />
+                <ECProperty propertyName="prop2" typeName="string" />
+            </ECEntityClass>
+            <ECRelationshipClass typeName="A1OwnsA1" modifier="None" strength="embedding">
+                <BaseClass>bis:ElementOwnsChildElements</BaseClass>
+                <Source multiplicity="(0..1)" roleLabel="owns" polymorphic="true">
+                    <Class class="a1"/>
+                </Source>
+                <Target multiplicity="(0..*)" roleLabel="is owned by" polymorphic="false">
+                    <Class class="a1"/>
+                </Target>
+            </ECRelationshipClass>
         </ECSchema>`;
         await b1.importSchemaStrings([schema1]);
         b1.saveChanges();
@@ -987,5 +987,161 @@ describe("rebase changes & stashing api", function () {
         chai.expect(events.rebaseHandler.shouldReinstate.map((txn) => txn.id)).to.deep.equal(["0x100000000", "0x100000001", "0x100000002", "0x100000003"]);
         chai.expect(events.rebaseHandler.recompute.map((txn) => txn.id)).to.deep.equal(["0x100000000", "0x100000001", "0x100000002", "0x100000003"]);
     });
+    it("rebase multi txn", async () => {
+        const b1 = await testIModel.openBriefcase();
+        const b2 = await testIModel.openBriefcase();
+        const e1 = await testIModel.insertElement(b1);
+        const e2 = await testIModel.insertElement(b1, true);
+        b1.saveChanges();
+        await b1.pushChanges({ description: "insert element 1 direct and 1 indirect" });
+        await b2.pullChanges();
+        chai.expect(b2.txns.beginMultiTxnOperation()).to.be.equals(DbResult.BE_SQLITE_OK);
+        let elId = await testIModel.insertElement(b2);
+        b2.saveChanges(`insert element ${elId}`);
+        elId = await testIModel.insertElement(b2);
+        b2.saveChanges(`insert element ${elId}`);
+        elId = await testIModel.insertElement(b2);
+        b2.saveChanges(`insert element ${elId}`);
+        chai.expect(b2.txns.endMultiTxnOperation()).to.be.equals(DbResult.BE_SQLITE_OK);
+        b2.saveChanges();
+        elId = await testIModel.insertElement(b2);
+        b2.saveChanges(`insert element ${elId}`);
+        let txns = Array.from(b2.txns.queryTxns());
+        chai.expect(txns[0].id).to.be.equals("0x100000000"); // 1st after beginMultiTxnOperation()
+        chai.expect(txns[0].props.description).to.be.equals("insert element 0x40000000001");
+        chai.expect(txns[0].sessionId).to.be.equals(1);
+        chai.expect(txns[0].grouped).to.be.equals(false);
+        chai.expect(txns[0].reversed).to.be.equals(false);
+        chai.expect(txns[1].id).to.be.equals("0x100000001"); // 2nd after beginMultiTxnOperation()
+        chai.expect(txns[1].props.description).to.be.equals("insert element 0x40000000002");
+        chai.expect(txns[1].sessionId).to.be.equals(1);
+        chai.expect(txns[1].grouped).to.be.equals(true);
+        chai.expect(txns[1].reversed).to.be.equals(false);
+        chai.expect(txns[2].id).to.be.equals("0x100000002"); // 3rd after beginMultiTxnOperation() & before endMultiTxnOperation()
+        chai.expect(txns[2].props.description).to.be.equals("insert element 0x40000000003");
+        chai.expect(txns[2].sessionId).to.be.equals(1);
+        chai.expect(txns[2].grouped).to.be.equals(true);
+        chai.expect(txns[2].reversed).to.be.equals(false);
+        chai.expect(txns[3].id).to.be.equals("0x100000003"); // 4th after endMultiTxnOperation()
+        chai.expect(txns[3].props.description).to.be.equals("insert element 0x40000000004");
+        chai.expect(txns[3].type).to.be.equals("Data");
+        chai.expect(txns[3].grouped).to.be.equals(false);
+        chai.expect(txns[3].reversed).to.be.equals(false);
+        // reverse single txn 0x100000003
+        chai.expect(b2.txns.reverseSingleTxn()).to.be.equals(DbResult.BE_SQLITE_OK);
+        txns = Array.from(b2.txns.queryTxns());
+        chai.expect(txns[0].id).to.be.equals("0x100000000"); // 1st after beginMultiTxnOperation()
+        chai.expect(txns[0].props.description).to.be.equals("insert element 0x40000000001");
+        chai.expect(txns[0].sessionId).to.be.equals(1);
+        chai.expect(txns[0].grouped).to.be.equals(false);
+        chai.expect(txns[0].reversed).to.be.equals(false);
+        chai.expect(txns[1].id).to.be.equals("0x100000001"); // 2nd after beginMultiTxnOperation()
+        chai.expect(txns[1].props.description).to.be.equals("insert element 0x40000000002");
+        chai.expect(txns[1].sessionId).to.be.equals(1);
+        chai.expect(txns[1].grouped).to.be.equals(true);
+        chai.expect(txns[1].reversed).to.be.equals(false);
+        chai.expect(txns[2].id).to.be.equals("0x100000002"); // 3rd after beginMultiTxnOperation() & before endMultiTxnOperation()
+        chai.expect(txns[2].props.description).to.be.equals("insert element 0x40000000003");
+        chai.expect(txns[2].sessionId).to.be.equals(1);
+        chai.expect(txns[2].grouped).to.be.equals(true);
+        chai.expect(txns[2].reversed).to.be.equals(false);
+        chai.expect(txns[3].id).to.be.equals("0x100000003"); // 4th after endMultiTxnOperation()
+        chai.expect(txns[3].props.description).to.be.equals("insert element 0x40000000004");
+        chai.expect(txns[3].type).to.be.equals("Data");
+        chai.expect(txns[3].grouped).to.be.equals(false);
+        chai.expect(txns[3].reversed).to.be.equals(true);
+        // reverse multi txn. should reverse 0x100000000, 0x100000001 & 0x100000002
+        chai.expect(b2.txns.reverseSingleTxn()).to.be.equals(DbResult.BE_SQLITE_OK);
+        txns = Array.from(b2.txns.queryTxns());
+        chai.expect(txns[0].id).to.be.equals("0x100000000"); // 1st after beginMultiTxnOperation()
+        chai.expect(txns[0].props.description).to.be.equals("insert element 0x40000000001");
+        chai.expect(txns[0].sessionId).to.be.equals(1);
+        chai.expect(txns[0].grouped).to.be.equals(false);
+        chai.expect(txns[0].reversed).to.be.equals(true);
+        chai.expect(txns[1].id).to.be.equals("0x100000001"); // 2nd after beginMultiTxnOperation()
+        chai.expect(txns[1].props.description).to.be.equals("insert element 0x40000000002");
+        chai.expect(txns[1].sessionId).to.be.equals(1);
+        chai.expect(txns[1].grouped).to.be.equals(true);
+        chai.expect(txns[1].reversed).to.be.equals(true);
+        chai.expect(txns[2].id).to.be.equals("0x100000002"); // 3rd after beginMultiTxnOperation() & before endMultiTxnOperation()
+        chai.expect(txns[2].props.description).to.be.equals("insert element 0x40000000003");
+        chai.expect(txns[2].sessionId).to.be.equals(1);
+        chai.expect(txns[2].grouped).to.be.equals(true);
+        chai.expect(txns[2].reversed).to.be.equals(true);
+        chai.expect(txns[3].id).to.be.equals("0x100000003"); // 4th after endMultiTxnOperation()
+        chai.expect(txns[3].props.description).to.be.equals("insert element 0x40000000004");
+        chai.expect(txns[3].type).to.be.equals("Data");
+        chai.expect(txns[3].grouped).to.be.equals(false);
+        chai.expect(txns[3].reversed).to.be.equals(true);
+        // reinstate the transaction
+        chai.expect(b2.txns.isRedoPossible).to.be.equals(true);
+        chai.expect(b2.txns.reinstateTxn()).to.be.equals(DbResult.BE_SQLITE_OK);
+        txns = Array.from(b2.txns.queryTxns());
+        chai.expect(txns[0].id).to.be.equals("0x100000000"); // 1st after beginMultiTxnOperation()
+        chai.expect(txns[0].props.description).to.be.equals("insert element 0x40000000001");
+        chai.expect(txns[0].sessionId).to.be.equals(1);
+        chai.expect(txns[0].grouped).to.be.equals(false);
+        chai.expect(txns[0].reversed).to.be.equals(false);
+        chai.expect(txns[1].id).to.be.equals("0x100000001"); // 2nd after beginMultiTxnOperation()
+        chai.expect(txns[1].props.description).to.be.equals("insert element 0x40000000002");
+        chai.expect(txns[1].sessionId).to.be.equals(1);
+        chai.expect(txns[1].grouped).to.be.equals(true);
+        chai.expect(txns[1].reversed).to.be.equals(false);
+        chai.expect(txns[2].id).to.be.equals("0x100000002"); // 3rd after beginMultiTxnOperation() & before endMultiTxnOperation()
+        chai.expect(txns[2].props.description).to.be.equals("insert element 0x40000000003");
+        chai.expect(txns[2].sessionId).to.be.equals(1);
+        chai.expect(txns[2].grouped).to.be.equals(true);
+        chai.expect(txns[2].reversed).to.be.equals(false);
+        chai.expect(txns[3].id).to.be.equals("0x100000003"); // 4th after endMultiTxnOperation()
+        chai.expect(txns[3].props.description).to.be.equals("insert element 0x40000000004");
+        chai.expect(txns[3].type).to.be.equals("Data");
+        chai.expect(txns[3].grouped).to.be.equals(false);
+        chai.expect(txns[3].reversed).to.be.equals(true);
+        // reinstate the transaction
+        chai.expect(b2.txns.isRedoPossible).to.be.equals(true);
+        chai.expect(b2.txns.reinstateTxn()).to.be.equals(DbResult.BE_SQLITE_OK);
+        txns = Array.from(b2.txns.queryTxns());
+        chai.expect(txns[0].id).to.be.equals("0x100000000"); // 1st after beginMultiTxnOperation()
+        chai.expect(txns[0].props.description).to.be.equals("insert element 0x40000000001");
+        chai.expect(txns[0].sessionId).to.be.equals(1);
+        chai.expect(txns[0].grouped).to.be.equals(false);
+        chai.expect(txns[0].reversed).to.be.equals(false);
+        chai.expect(txns[1].id).to.be.equals("0x100000001"); // 2nd after beginMultiTxnOperation()
+        chai.expect(txns[1].props.description).to.be.equals("insert element 0x40000000002");
+        chai.expect(txns[1].sessionId).to.be.equals(1);
+        chai.expect(txns[1].grouped).to.be.equals(true);
+        chai.expect(txns[1].reversed).to.be.equals(false);
+        chai.expect(txns[2].id).to.be.equals("0x100000002"); // 3rd after beginMultiTxnOperation() & before endMultiTxnOperation()
+        chai.expect(txns[2].props.description).to.be.equals("insert element 0x40000000003");
+        chai.expect(txns[2].sessionId).to.be.equals(1);
+        chai.expect(txns[2].grouped).to.be.equals(true);
+        chai.expect(txns[2].reversed).to.be.equals(false);
+        chai.expect(txns[3].id).to.be.equals("0x100000003"); // 4th after endMultiTxnOperation()
+        chai.expect(txns[3].props.description).to.be.equals("insert element 0x40000000004");
+        chai.expect(txns[3].type).to.be.equals("Data");
+        chai.expect(txns[3].grouped).to.be.equals(false);
+        chai.expect(txns[3].reversed).to.be.equals(false);
+        chai.expect(b2.txns.isRedoPossible).to.be.equals(false);
+        await testIModel.updateElement(b1, e1);
+        await testIModel.updateElement(b1, e2, true);
+        b1.saveChanges();
+        await b1.pushChanges({ description: "update element 1 direct and 1 indirect" });
+        const recomputeTxnIds = [];
+        b2.txns.rebaser.setCustomHandler({
+            shouldReinstate: (_txn) => {
+                return true;
+            },
+            recompute: async (txn) => {
+                recomputeTxnIds.push(txn.id);
+            },
+        });
+        await b2.pullChanges();
+        chai.expect(recomputeTxnIds).to.deep.equals([
+            "0x100000000",
+            "0x100000001",
+            "0x100000002",
+            "0x100000003",
+        ]);
+    });
 });
 //# sourceMappingURL=Rebase.test.js.map