@itwin/core-backend 5.15.0-dev.2 → 5.15.0-dev.20

package/lib/esm/test/element/ElementSubModelInsert.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ElementSubModelInsert.test.d.ts.map

package/lib/esm/test/element/ElementSubModelInsert.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ElementSubModelInsert.test.d.ts","sourceRoot":"","sources":["../../../../src/test/element/ElementSubModelInsert.test.ts"],"names":[],"mappings":""}

package/lib/esm/test/element/ElementSubModelInsert.test.js

@@ -0,0 +1,183 @@
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the repository root for full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+import { expect } from "chai";
+import { spy as sinonSpy } from "sinon";
+import { Id64, IModelStatus } from "@itwin/core-bentley";
+import { Code, IModel, IModelError } from "@itwin/core-common";
+import { ClassRegistry } from "../../ClassRegistry";
+import { withEditTxn } from "../../EditTxn";
+import { Document, Drawing, SectionDrawing, Sheet, TemplateRecipe2d } from "../../Element";
+import { GenericSchema } from "../../domains/GenericSchema";
+import { SnapshotDb } from "../../IModelDb";
+import { DefinitionModel, DocumentListModel, DrawingModel, SectionDrawingModel, SheetModel } from "../../Model";
+import { Schema, Schemas } from "../../Schema";
+import { IModelTestUtils } from "../IModelTestUtils";
+class TestSubModelSchema extends Schema {
+    static get schemaName() { return "TestSubModel"; }
+}
+/** A Sheet subclass that overrides the hook and defers to the inherited validation via `super`. */
+class TestSheetSpy extends Sheet {
+    static get className() { return "TestSheetSpy"; }
+    static onSubModelInsert(arg) {
+        super.onSubModelInsert(arg);
+    }
+}
+/** A Sheet subclass that widens the inherited rule to also permit a DrawingModel. */
+class TestSheetWidened extends Sheet {
+    static get className() { return "TestSheetWidened"; }
+    static get allowedSubModelClasses() { return [SheetModel, DrawingModel]; }
+}
+/** A Sheet subclass that replaces the hook without calling `super`, opting out of validation. */
+class TestSheetUnvalidated extends Sheet {
+    static get className() { return "TestSheetUnvalidated"; }
+    static onSubModelInsert(_arg) { }
+}
+/** A domain ISubModeledElement unrelated to Sheet/Drawing that opts in by overriding only the getter. */
+class TestRestrictedDocument extends Document {
+    static get className() { return "TestRestrictedDocument"; }
+    static get allowedSubModelClasses() { return [DrawingModel]; }
+}
+const testSchemaXml = `<?xml version="1.0" encoding="UTF-8"?>
+<ECSchema schemaName="TestSubModel" alias="tsm" version="01.00.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+  <ECSchemaReference name="BisCore" version="01.00.04" alias="bis"/>
+  <ECCustomAttributes>
+    <SchemaHasBehavior xmlns="BisCore.01.00.00"/>
+  </ECCustomAttributes>
+  <ECEntityClass typeName="TestSheetModel">
+    <BaseClass>bis:SheetModel</BaseClass>
+  </ECEntityClass>
+  <ECEntityClass typeName="TestSheetGenerated">
+    <BaseClass>bis:Sheet</BaseClass>
+  </ECEntityClass>
+  <ECEntityClass typeName="TestSheetSpy">
+    <BaseClass>bis:Sheet</BaseClass>
+  </ECEntityClass>
+  <ECEntityClass typeName="TestSheetWidened">
+    <BaseClass>bis:Sheet</BaseClass>
+  </ECEntityClass>
+  <ECEntityClass typeName="TestSheetUnvalidated">
+    <BaseClass>bis:Sheet</BaseClass>
+  </ECEntityClass>
+  <ECEntityClass typeName="TestHostDocument">
+    <BaseClass>bis:Document</BaseClass>
+    <BaseClass>bis:ISubModeledElement</BaseClass>
+  </ECEntityClass>
+  <ECEntityClass typeName="TestRestrictedDocument">
+    <BaseClass>bis:Document</BaseClass>
+    <BaseClass>bis:ISubModeledElement</BaseClass>
+  </ECEntityClass>
+</ECSchema>`;
+describe("Sub-model insert validation", () => {
+    let imodel;
+    let documentListModelId;
+    before(async () => {
+        GenericSchema.registerSchema();
+        Schemas.registerSchema(TestSubModelSchema);
+        ClassRegistry.register(TestSheetSpy, TestSubModelSchema);
+        ClassRegistry.register(TestSheetWidened, TestSubModelSchema);
+        ClassRegistry.register(TestSheetUnvalidated, TestSubModelSchema);
+        ClassRegistry.register(TestRestrictedDocument, TestSubModelSchema);
+        const iModelPath = IModelTestUtils.prepareOutputFile("SubModelInsert", "SubModelInsert.bim");
+        imodel = SnapshotDb.createEmpty(iModelPath, { rootSubject: { name: "SubModelInsertTest" } });
+        await imodel.importSchemaStrings([testSchemaXml]);
+        documentListModelId = withEditTxn(imodel, (txn) => DocumentListModel.insert(txn, SnapshotDb.rootSubjectId, "DocumentList"));
+    });
+    after(() => {
+        imodel.close();
+        Schemas.unregisterSchema(TestSubModelSchema.schemaName);
+    });
+    function insertHost(classFullName, modelId = documentListModelId) {
+        const id = withEditTxn(imodel, (txn) => txn.insertElement({ classFullName, model: modelId, code: Code.createEmpty() }));
+        expect(Id64.isValidId64(id)).to.be.true;
+        return id;
+    }
+    function insertSubModel(modelClassFullName, modeledElementId) {
+        return withEditTxn(imodel, (txn) => txn.insertModel({ classFullName: modelClassFullName, modeledElement: { id: modeledElementId } }));
+    }
+    function expectAccepted(hostClassFullName, modelClassFullName, hostModelId) {
+        const hostId = insertHost(hostClassFullName, hostModelId);
+        const modelId = insertSubModel(modelClassFullName, hostId);
+        expect(modelId).to.equal(hostId);
+        expect(imodel.models.getModel(modelId).classFullName).to.equal(modelClassFullName);
+    }
+    function expectRejected(hostClassFullName, modelClassFullName, hostModelId) {
+        const hostId = insertHost(hostClassFullName, hostModelId);
+        let error;
+        try {
+            insertSubModel(modelClassFullName, hostId);
+        }
+        catch (err) {
+            error = err;
+        }
+        expect(error).to.be.instanceOf(IModelError);
+        expect(error.errorNumber).to.equal(IModelStatus.WrongModel);
+        expect(error.message).to.contain("must be sub-modeled by");
+        expect(imodel.models.tryGetModel(hostId)).to.be.undefined;
+    }
+    describe("accepts", () => {
+        it("Sheet + SheetModel", () => expectAccepted(Sheet.classFullName, SheetModel.classFullName));
+        it("Drawing + DrawingModel", () => expectAccepted(Drawing.classFullName, DrawingModel.classFullName));
+        it("SectionDrawing + DrawingModel", () => expectAccepted(SectionDrawing.classFullName, DrawingModel.classFullName));
+        it("SectionDrawing + SectionDrawingModel", () => expectAccepted(SectionDrawing.classFullName, SectionDrawingModel.classFullName));
+        it("SectionDrawing + Generic:GraphicalModel3d", () => expectAccepted(SectionDrawing.classFullName, "Generic:GraphicalModel3d"));
+        it("TemplateRecipe2d + DrawingModel", () => expectAccepted(TemplateRecipe2d.classFullName, DrawingModel.classFullName, IModel.dictionaryId));
+        it("Sheet + SheetModel subclass without a JS class", () => expectAccepted(Sheet.classFullName, "TestSubModel:TestSheetModel"));
+        it("no allowedSubModelClasses → any model accepted", () => {
+            expectAccepted("TestSubModel:TestHostDocument", SheetModel.classFullName);
+            expectAccepted("TestSubModel:TestHostDocument", DrawingModel.classFullName);
+        });
+        it("Sheet subclass that widens allowedSubModelClasses + DrawingModel", () => expectAccepted(TestSheetWidened.classFullName, DrawingModel.classFullName));
+        it("Sheet subclass that overrides onSubModelInsert without super + DrawingModel", () => expectAccepted(TestSheetUnvalidated.classFullName, DrawingModel.classFullName));
+        it("Document subclass with allowedSubModelClasses + DrawingModel", () => expectAccepted(TestRestrictedDocument.classFullName, DrawingModel.classFullName));
+    });
+    describe("rejects with IModelStatus.WrongModel", () => {
+        it("Sheet + DrawingModel", () => expectRejected(Sheet.classFullName, DrawingModel.classFullName));
+        it("Drawing + SheetModel", () => expectRejected(Drawing.classFullName, SheetModel.classFullName));
+        it("SectionDrawing + SheetModel", () => expectRejected(SectionDrawing.classFullName, SheetModel.classFullName));
+        it("TemplateRecipe2d + SheetModel", () => expectRejected(TemplateRecipe2d.classFullName, SheetModel.classFullName, IModel.dictionaryId));
+        it("Sheet subclass without a JS class + DrawingModel", () => expectRejected("TestSubModel:TestSheetGenerated", DrawingModel.classFullName));
+        it("Document subclass with allowedSubModelClasses + SheetModel", () => expectRejected(TestRestrictedDocument.classFullName, SheetModel.classFullName));
+    });
+    it("invokes an overriding subclass hook that defers to the inherited validation", () => {
+        const spy = sinonSpy(TestSheetSpy, "onSubModelInsert");
+        try {
+            expectAccepted(TestSheetSpy.classFullName, SheetModel.classFullName);
+            expect(spy.calledOnce).to.be.true;
+            expect(spy.getCall(0).args[0].subModelProps.classFullName).to.equal(SheetModel.classFullName);
+            expectRejected(TestSheetSpy.classFullName, DrawingModel.classFullName);
+            expect(spy.calledTwice).to.be.true;
+            expect(spy.getCall(1).threw()).to.be.true;
+        }
+        finally {
+            spy.restore();
+        }
+    });
+    describe("missing required behavior", () => {
+        it("rejects sub-model insertion", () => {
+            const hostId = insertHost(TestRestrictedDocument.classFullName);
+            Schemas.unregisterSchema(TestSubModelSchema.schemaName);
+            try {
+                let error;
+                try {
+                    insertSubModel(DefinitionModel.classFullName, hostId);
+                }
+                catch (err) {
+                    error = err;
+                }
+                expect(error).to.be.instanceOf(IModelError);
+                expect(error.errorNumber).to.equal(IModelStatus.WrongHandler);
+                expect(imodel.models.tryGetModel(hostId)).to.be.undefined;
+            }
+            finally {
+                Schemas.registerSchema(TestSubModelSchema);
+                ClassRegistry.register(TestSheetSpy, TestSubModelSchema);
+                ClassRegistry.register(TestSheetWidened, TestSubModelSchema);
+                ClassRegistry.register(TestSheetUnvalidated, TestSubModelSchema);
+                ClassRegistry.register(TestRestrictedDocument, TestSubModelSchema);
+            }
+        });
+    });
+});
+//# sourceMappingURL=ElementSubModelInsert.test.js.map

package/lib/esm/test/element/ElementSubModelInsert.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ElementSubModelInsert.test.js","sourceRoot":"","sources":["../../../../src/test/element/ElementSubModelInsert.test.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAC/F,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAE,GAAG,IAAI,QAAQ,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,IAAI,EAAc,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACrE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAsB,cAAc,EAAE,KAAK,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAE/G,OAAO,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAC5D,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,iBAAiB,EAAE,YAAY,EAAS,mBAAmB,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACvH,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,MAAM,kBAAmB,SAAQ,MAAM;IAC9B,MAAM,KAAc,UAAU,KAAa,OAAO,cAAc,CAAC,CAAC,CAAC;CAC3E;AAED,mGAAmG;AACnG,MAAM,YAAa,SAAQ,KAAK;IACvB,MAAM,KAAc,SAAS,KAAa,OAAO,cAAc,CAAC,CAAC,CAAC;IAClE,MAAM,CAAU,gBAAgB,CAAC,GAAuB;QAC7D,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;IAC9B,CAAC;CACF;AAED,qFAAqF;AACrF,MAAM,gBAAiB,SAAQ,KAAK;IAC3B,MAAM,KAAc,SAAS,KAAa,OAAO,kBAAkB,CAAC,CAAC,CAAC;IACnE,MAAM,KAAc,sBAAsB,KAAoC,OAAO,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC;CAC7H;AAED,iGAAiG;AACjG,MAAM,oBAAqB,SAAQ,KAAK;IAC/B,MAAM,KAAc,SAAS,KAAa,OAAO,sBAAsB,CAAC,CAAC,CAAC;IAC1E,MAAM,CAAU,gBAAgB,CAAC,IAAwB,IAAU,CAAC;CAC5E;AAED,yGAAyG;AACzG,MAAM,sBAAuB,SAAQ,QAAQ;IACpC,MAAM,KAAc,SAAS,KAAa,OAAO,wBAAwB,CAAC,CAAC,CAAC;IACzE,MAAM,KAAc,sBAAsB,KAAoC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;CACjH;AAED,MAAM,aAAa,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;YA6BV,CAAC;AAEb,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,MAAkB,CAAC;IACvB,IAAI,mBAA+B,CAAC;IAEpC,MAAM,CAAC,KAAK,IAAI,EAAE;QAChB,aAAa,CAAC,cAAc,EAAE,CAAC;QAC/B,OAAO,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;QAC3C,aAAa,CAAC,QAAQ,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;QACzD,aAAa,CAAC,QAAQ,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;QAC7D,aAAa,CAAC,QAAQ,CAAC,oBAAoB,EAAE,kBAAkB,CAAC,CAAC;QACjE,aAAa,CAAC,QAAQ,CAAC,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;QAEnE,MAAM,UAAU,GAAG,eAAe,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,oBAAoB,CAAC,CAAC;QAC7F,MAAM,GAAG,UAAU,CAAC,WAAW,CAAC,UAAU,EAAE,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,oBAAoB,EAAE,EAAE,CAAC,CAAC;QAC7F,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QAClD,mBAAmB,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,iBAAiB,CAAC,MAAM,CAAC,GAAG,EAAE,UAAU,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC,CAAC;IAC9H,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,EAAE;QACT,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,SAAS,UAAU,CAAC,aAAqB,EAAE,UAAsB,mBAAmB;QAClF,MAAM,EAAE,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;QACxH,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACxC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,SAAS,cAAc,CAAC,kBAA0B,EAAE,gBAA4B;QAC9E,OAAO,WAAW,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,aAAa,EAAE,kBAAkB,EAAE,cAAc,EAAE,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC,CAAC;IACxI,CAAC;IAED,SAAS,cAAc,CAAC,iBAAyB,EAAE,kBAA0B,EAAE,WAAwB;QACrG,MAAM,MAAM,GAAG,UAAU,CAAC,iBAAiB,EAAE,WAAW,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,cAAc,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;QAC3D,MAAM,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACrF,CAAC;IAED,SAAS,cAAc,CAAC,iBAAyB,EAAE,kBAA0B,EAAE,WAAwB;QACrG,MAAM,MAAM,GAAG,UAAU,CAAC,iBAAiB,EAAE,WAAW,CAAC,CAAC;QAC1D,IAAI,KAAc,CAAC;QACnB,IAAI,CAAC;YACH,cAAc,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,KAAK,GAAG,GAAG,CAAC;QACd,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QAC5C,MAAM,CAAE,KAAqB,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAC7E,MAAM,CAAE,KAAqB,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;QAC5E,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;IAC5D,CAAC;IAED,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,aAAa,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC;QAC9F,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,OAAO,CAAC,aAAa,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;QACtG,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,cAAc,CAAC,aAAa,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;QACpH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,cAAc,CAAC,aAAa,EAAE,mBAAmB,CAAC,aAAa,CAAC,CAAC,CAAC;QAClI,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,cAAc,CAAC,aAAa,EAAE,0BAA0B,CAAC,CAAC,CAAC;QAChI,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,gBAAgB,CAAC,aAAa,EAAE,YAAY,CAAC,aAAa,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;QAC7I,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,aAAa,EAAE,6BAA6B,CAAC,CAAC,CAAC;QAC/H,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,cAAc,CAAC,+BAA+B,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC;YAC1E,cAAc,CAAC,+BAA+B,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,gBAAgB,CAAC,aAAa,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;QACzJ,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,oBAAoB,CAAC,aAAa,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;QACxK,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,sBAAsB,CAAC,aAAa,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;IAC7J,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;QACpD,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,aAAa,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;QAClG,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,OAAO,CAAC,aAAa,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC;QAClG,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,cAAc,CAAC,aAAa,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC;QAChH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,gBAAgB,CAAC,aAAa,EAAE,UAAU,CAAC,aAAa,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;QACzI,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,iCAAiC,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC;QAC5I,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,sBAAsB,CAAC,aAAa,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC;IACzJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;QACrF,MAAM,GAAG,GAAG,QAAQ,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;QACvD,IAAI,CAAC;YACH,cAAc,CAAC,YAAY,CAAC,aAAa,EAAE,UAAU,CAAC,aAAa,CAAC,CAAC;YACrE,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;YAClC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;YAE9F,cAAc,CAAC,YAAY,CAAC,aAAa,EAAE,YAAY,CAAC,aAAa,CAAC,CAAC;YACvE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;YACnC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAC5C,CAAC;gBAAS,CAAC;YACT,GAAG,CAAC,OAAO,EAAE,CAAC;QAChB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,MAAM,GAAG,UAAU,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;YAChE,OAAO,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAExD,IAAI,CAAC;gBACH,IAAI,KAAc,CAAC;gBACnB,IAAI,CAAC;oBACH,cAAc,CAAC,eAAe,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;gBACxD,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,KAAK,GAAG,GAAG,CAAC;gBACd,CAAC;gBAED,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;gBAC5C,MAAM,CAAE,KAAqB,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;gBAC/E,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;YAC5D,CAAC;oBAAS,CAAC;gBACT,OAAO,CAAC,cAAc,CAAC,kBAAkB,CAAC,CAAC;gBAC3C,aAAa,CAAC,QAAQ,CAAC,YAAY,EAAE,kBAAkB,CAAC,CAAC;gBACzD,aAAa,CAAC,QAAQ,CAAC,gBAAgB,EAAE,kBAAkB,CAAC,CAAC;gBAC7D,aAAa,CAAC,QAAQ,CAAC,oBAAoB,EAAE,kBAAkB,CAAC,CAAC;gBACjE,aAAa,CAAC,QAAQ,CAAC,sBAAsB,EAAE,kBAAkB,CAAC,CAAC;YACrE,CAAC;QACH,CAAC,CAAC,CAAC;IACL,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 repository root for full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\nimport { expect } from \"chai\";\r\nimport { spy as sinonSpy } from \"sinon\";\r\nimport { Id64, Id64String, IModelStatus } from \"@itwin/core-bentley\";\r\nimport { Code, IModel, IModelError } from \"@itwin/core-common\";\r\nimport { ClassRegistry } from \"../../ClassRegistry\";\r\nimport { withEditTxn } from \"../../EditTxn\";\r\nimport { Document, Drawing, OnSubModelPropsArg, SectionDrawing, Sheet, TemplateRecipe2d } from \"../../Element\";\r\nimport { EntityClassType } from \"../../Entity\";\r\nimport { GenericSchema } from \"../../domains/GenericSchema\";\r\nimport { SnapshotDb } from \"../../IModelDb\";\r\nimport { DefinitionModel, DocumentListModel, DrawingModel, Model, SectionDrawingModel, SheetModel } from \"../../Model\";\r\nimport { Schema, Schemas } from \"../../Schema\";\r\nimport { IModelTestUtils } from \"../IModelTestUtils\";\r\n\r\nclass TestSubModelSchema extends Schema {\r\n  public static override get schemaName(): string { return \"TestSubModel\"; }\r\n}\r\n\r\n/** A Sheet subclass that overrides the hook and defers to the inherited validation via `super`. */\r\nclass TestSheetSpy extends Sheet {\r\n  public static override get className(): string { return \"TestSheetSpy\"; }\r\n  public static override onSubModelInsert(arg: OnSubModelPropsArg): void {\r\n    super.onSubModelInsert(arg);\r\n  }\r\n}\r\n\r\n/** A Sheet subclass that widens the inherited rule to also permit a DrawingModel. */\r\nclass TestSheetWidened extends Sheet {\r\n  public static override get className(): string { return \"TestSheetWidened\"; }\r\n  protected static override get allowedSubModelClasses(): Array<EntityClassType<Model>> { return [SheetModel, DrawingModel]; }\r\n}\r\n\r\n/** A Sheet subclass that replaces the hook without calling `super`, opting out of validation. */\r\nclass TestSheetUnvalidated extends Sheet {\r\n  public static override get className(): string { return \"TestSheetUnvalidated\"; }\r\n  public static override onSubModelInsert(_arg: OnSubModelPropsArg): void { }\r\n}\r\n\r\n/** A domain ISubModeledElement unrelated to Sheet/Drawing that opts in by overriding only the getter. */\r\nclass TestRestrictedDocument extends Document {\r\n  public static override get className(): string { return \"TestRestrictedDocument\"; }\r\n  protected static override get allowedSubModelClasses(): Array<EntityClassType<Model>> { return [DrawingModel]; }\r\n}\r\n\r\nconst testSchemaXml = `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n<ECSchema schemaName=\"TestSubModel\" alias=\"tsm\" version=\"01.00.00\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n  <ECSchemaReference name=\"BisCore\" version=\"01.00.04\" alias=\"bis\"/>\r\n  <ECCustomAttributes>\r\n    <SchemaHasBehavior xmlns=\"BisCore.01.00.00\"/>\r\n  </ECCustomAttributes>\r\n  <ECEntityClass typeName=\"TestSheetModel\">\r\n    <BaseClass>bis:SheetModel</BaseClass>\r\n  </ECEntityClass>\r\n  <ECEntityClass typeName=\"TestSheetGenerated\">\r\n    <BaseClass>bis:Sheet</BaseClass>\r\n  </ECEntityClass>\r\n  <ECEntityClass typeName=\"TestSheetSpy\">\r\n    <BaseClass>bis:Sheet</BaseClass>\r\n  </ECEntityClass>\r\n  <ECEntityClass typeName=\"TestSheetWidened\">\r\n    <BaseClass>bis:Sheet</BaseClass>\r\n  </ECEntityClass>\r\n  <ECEntityClass typeName=\"TestSheetUnvalidated\">\r\n    <BaseClass>bis:Sheet</BaseClass>\r\n  </ECEntityClass>\r\n  <ECEntityClass typeName=\"TestHostDocument\">\r\n    <BaseClass>bis:Document</BaseClass>\r\n    <BaseClass>bis:ISubModeledElement</BaseClass>\r\n  </ECEntityClass>\r\n  <ECEntityClass typeName=\"TestRestrictedDocument\">\r\n    <BaseClass>bis:Document</BaseClass>\r\n    <BaseClass>bis:ISubModeledElement</BaseClass>\r\n  </ECEntityClass>\r\n</ECSchema>`;\r\n\r\ndescribe(\"Sub-model insert validation\", () => {\r\n  let imodel: SnapshotDb;\r\n  let documentListModelId: Id64String;\r\n\r\n  before(async () => {\r\n    GenericSchema.registerSchema();\r\n    Schemas.registerSchema(TestSubModelSchema);\r\n    ClassRegistry.register(TestSheetSpy, TestSubModelSchema);\r\n    ClassRegistry.register(TestSheetWidened, TestSubModelSchema);\r\n    ClassRegistry.register(TestSheetUnvalidated, TestSubModelSchema);\r\n    ClassRegistry.register(TestRestrictedDocument, TestSubModelSchema);\r\n\r\n    const iModelPath = IModelTestUtils.prepareOutputFile(\"SubModelInsert\", \"SubModelInsert.bim\");\r\n    imodel = SnapshotDb.createEmpty(iModelPath, { rootSubject: { name: \"SubModelInsertTest\" } });\r\n    await imodel.importSchemaStrings([testSchemaXml]);\r\n    documentListModelId = withEditTxn(imodel, (txn) => DocumentListModel.insert(txn, SnapshotDb.rootSubjectId, \"DocumentList\"));\r\n  });\r\n\r\n  after(() => {\r\n    imodel.close();\r\n    Schemas.unregisterSchema(TestSubModelSchema.schemaName);\r\n  });\r\n\r\n  function insertHost(classFullName: string, modelId: Id64String = documentListModelId): Id64String {\r\n    const id = withEditTxn(imodel, (txn) => txn.insertElement({ classFullName, model: modelId, code: Code.createEmpty() }));\r\n    expect(Id64.isValidId64(id)).to.be.true;\r\n    return id;\r\n  }\r\n\r\n  function insertSubModel(modelClassFullName: string, modeledElementId: Id64String): Id64String {\r\n    return withEditTxn(imodel, (txn) => txn.insertModel({ classFullName: modelClassFullName, modeledElement: { id: modeledElementId } }));\r\n  }\r\n\r\n  function expectAccepted(hostClassFullName: string, modelClassFullName: string, hostModelId?: Id64String): void {\r\n    const hostId = insertHost(hostClassFullName, hostModelId);\r\n    const modelId = insertSubModel(modelClassFullName, hostId);\r\n    expect(modelId).to.equal(hostId);\r\n    expect(imodel.models.getModel(modelId).classFullName).to.equal(modelClassFullName);\r\n  }\r\n\r\n  function expectRejected(hostClassFullName: string, modelClassFullName: string, hostModelId?: Id64String): void {\r\n    const hostId = insertHost(hostClassFullName, hostModelId);\r\n    let error: unknown;\r\n    try {\r\n      insertSubModel(modelClassFullName, hostId);\r\n    } catch (err) {\r\n      error = err;\r\n    }\r\n\r\n    expect(error).to.be.instanceOf(IModelError);\r\n    expect((error as IModelError).errorNumber).to.equal(IModelStatus.WrongModel);\r\n    expect((error as IModelError).message).to.contain(\"must be sub-modeled by\");\r\n    expect(imodel.models.tryGetModel(hostId)).to.be.undefined;\r\n  }\r\n\r\n  describe(\"accepts\", () => {\r\n    it(\"Sheet + SheetModel\", () => expectAccepted(Sheet.classFullName, SheetModel.classFullName));\r\n    it(\"Drawing + DrawingModel\", () => expectAccepted(Drawing.classFullName, DrawingModel.classFullName));\r\n    it(\"SectionDrawing + DrawingModel\", () => expectAccepted(SectionDrawing.classFullName, DrawingModel.classFullName));\r\n    it(\"SectionDrawing + SectionDrawingModel\", () => expectAccepted(SectionDrawing.classFullName, SectionDrawingModel.classFullName));\r\n    it(\"SectionDrawing + Generic:GraphicalModel3d\", () => expectAccepted(SectionDrawing.classFullName, \"Generic:GraphicalModel3d\"));\r\n    it(\"TemplateRecipe2d + DrawingModel\", () => expectAccepted(TemplateRecipe2d.classFullName, DrawingModel.classFullName, IModel.dictionaryId));\r\n    it(\"Sheet + SheetModel subclass without a JS class\", () => expectAccepted(Sheet.classFullName, \"TestSubModel:TestSheetModel\"));\r\n    it(\"no allowedSubModelClasses → any model accepted\", () => {\r\n      expectAccepted(\"TestSubModel:TestHostDocument\", SheetModel.classFullName);\r\n      expectAccepted(\"TestSubModel:TestHostDocument\", DrawingModel.classFullName);\r\n    });\r\n    it(\"Sheet subclass that widens allowedSubModelClasses + DrawingModel\", () => expectAccepted(TestSheetWidened.classFullName, DrawingModel.classFullName));\r\n    it(\"Sheet subclass that overrides onSubModelInsert without super + DrawingModel\", () => expectAccepted(TestSheetUnvalidated.classFullName, DrawingModel.classFullName));\r\n    it(\"Document subclass with allowedSubModelClasses + DrawingModel\", () => expectAccepted(TestRestrictedDocument.classFullName, DrawingModel.classFullName));\r\n  });\r\n\r\n  describe(\"rejects with IModelStatus.WrongModel\", () => {\r\n    it(\"Sheet + DrawingModel\", () => expectRejected(Sheet.classFullName, DrawingModel.classFullName));\r\n    it(\"Drawing + SheetModel\", () => expectRejected(Drawing.classFullName, SheetModel.classFullName));\r\n    it(\"SectionDrawing + SheetModel\", () => expectRejected(SectionDrawing.classFullName, SheetModel.classFullName));\r\n    it(\"TemplateRecipe2d + SheetModel\", () => expectRejected(TemplateRecipe2d.classFullName, SheetModel.classFullName, IModel.dictionaryId));\r\n    it(\"Sheet subclass without a JS class + DrawingModel\", () => expectRejected(\"TestSubModel:TestSheetGenerated\", DrawingModel.classFullName));\r\n    it(\"Document subclass with allowedSubModelClasses + SheetModel\", () => expectRejected(TestRestrictedDocument.classFullName, SheetModel.classFullName));\r\n  });\r\n\r\n  it(\"invokes an overriding subclass hook that defers to the inherited validation\", () => {\r\n    const spy = sinonSpy(TestSheetSpy, \"onSubModelInsert\");\r\n    try {\r\n      expectAccepted(TestSheetSpy.classFullName, SheetModel.classFullName);\r\n      expect(spy.calledOnce).to.be.true;\r\n      expect(spy.getCall(0).args[0].subModelProps.classFullName).to.equal(SheetModel.classFullName);\r\n\r\n      expectRejected(TestSheetSpy.classFullName, DrawingModel.classFullName);\r\n      expect(spy.calledTwice).to.be.true;\r\n      expect(spy.getCall(1).threw()).to.be.true;\r\n    } finally {\r\n      spy.restore();\r\n    }\r\n  });\r\n\r\n  describe(\"missing required behavior\", () => {\r\n    it(\"rejects sub-model insertion\", () => {\r\n      const hostId = insertHost(TestRestrictedDocument.classFullName);\r\n      Schemas.unregisterSchema(TestSubModelSchema.schemaName);\r\n\r\n      try {\r\n        let error: unknown;\r\n        try {\r\n          insertSubModel(DefinitionModel.classFullName, hostId);\r\n        } catch (err) {\r\n          error = err;\r\n        }\r\n\r\n        expect(error).to.be.instanceOf(IModelError);\r\n        expect((error as IModelError).errorNumber).to.equal(IModelStatus.WrongHandler);\r\n        expect(imodel.models.tryGetModel(hostId)).to.be.undefined;\r\n      } finally {\r\n        Schemas.registerSchema(TestSubModelSchema);\r\n        ClassRegistry.register(TestSheetSpy, TestSubModelSchema);\r\n        ClassRegistry.register(TestSheetWidened, TestSubModelSchema);\r\n        ClassRegistry.register(TestSheetUnvalidated, TestSubModelSchema);\r\n        ClassRegistry.register(TestRestrictedDocument, TestSubModelSchema);\r\n      }\r\n    });\r\n  });\r\n});\r\n"]}

package/lib/esm/test/element/RelationshipBatchLookup.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=RelationshipBatchLookup.test.d.ts.map

package/lib/esm/test/element/RelationshipBatchLookup.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"RelationshipBatchLookup.test.d.ts","sourceRoot":"","sources":["../../../../src/test/element/RelationshipBatchLookup.test.ts"],"names":[],"mappings":""}

package/lib/esm/test/element/RelationshipBatchLookup.test.js

@@ -0,0 +1,202 @@
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+import { expect } from "chai";
+import * as sinon from "sinon";
+import { Guid, Id64 } from "@itwin/core-bentley";
+import { Code, IModel } from "@itwin/core-common";
+import { withEditTxn } from "../../EditTxn";
+import { RepositoryLink } from "../../Element";
+import { SnapshotDb } from "../../IModelDb";
+import { ElementDrivesElement, ElementGroupsMembers, ElementRefersToElements } from "../../Relationship";
+import { IModelTestUtils } from "../IModelTestUtils";
+describe("Relationships batched lookup", () => {
+    const testSchema = `<?xml version="1.0" encoding="UTF-8"?>
+    <ECSchema schemaName="BatchLookupTest" alias="blt" version="01.00.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.2">
+      <ECSchemaReference name="BisCore" version="01.00.00" alias="bis"/>
+      <ECStructClass typeName="TestStruct">
+        <ECProperty propertyName="label" typeName="string"/>
+        <ECProperty propertyName="origin" typeName="point2d"/>
+      </ECStructClass>
+      <ECRelationshipClass typeName="TestRefersToElements" strength="referencing" modifier="Sealed">
+        <BaseClass>bis:ElementRefersToElements</BaseClass>
+        <Source multiplicity="(0..*)" roleLabel="refers to" polymorphic="true">
+          <Class class="bis:Element"/>
+        </Source>
+        <Target multiplicity="(0..*)" roleLabel="is referenced by" polymorphic="true">
+          <Class class="bis:Element"/>
+        </Target>
+        <ECProperty propertyName="Location" typeName="point3d"/>
+        <ECStructProperty propertyName="Info" typeName="TestStruct"/>
+        <ECArrayProperty propertyName="Points" typeName="point3d" minOccurs="0" maxOccurs="unbounded"/>
+      </ECRelationshipClass>
+    </ECSchema>`;
+    const testRelClass = "BatchLookupTest:TestRefersToElements";
+    let imodel;
+    let elementIds;
+    let relIds;
+    before(async () => {
+        imodel = SnapshotDb.createEmpty(IModelTestUtils.prepareOutputFile("Relationship", "BatchLookup.bim"), { rootSubject: { name: "BatchLookup" } });
+        await imodel.importSchemaStrings([testSchema]);
+        withEditTxn(imodel, (txn) => {
+            elementIds = [];
+            for (let i = 0; i < 6; ++i) {
+                const props = {
+                    classFullName: RepositoryLink.classFullName,
+                    model: IModel.repositoryModelId,
+                    code: Code.createEmpty(),
+                    repositoryGuid: Guid.createValue(),
+                };
+                elementIds.push(txn.insertElement(props));
+            }
+            // Chain e0->e1->e2->e3->e4 with memberPriority equal to the source index.
+            relIds = [];
+            for (let i = 0; i < 4; ++i)
+                relIds.push(txn.insertRelationship(ElementGroupsMembers.create(imodel, elementIds[i], elementIds[i + 1], i).toJSON()));
+        });
+    });
+    after(() => imodel.close());
+    it("matches tryGetInstanceProps for ids and source/target pairs, preserving order", async () => {
+        const [e0, e1, e2, e3, e4, e5] = elementIds;
+        const criteria = [
+            relIds[2],
+            { sourceId: e0, targetId: e1 },
+            Id64.fromLocalAndBriefcaseIds(0x7fff, 0), // non-existent id
+            { sourceId: e3, targetId: e4 },
+            { sourceId: e0, targetId: e2 }, // both ends requested elsewhere, but not related as a pair
+            { sourceId: e4, targetId: e5 }, // unrelated
+            Id64.invalid,
+            relIds[2], // duplicate
+            { sourceId: e1, targetId: e2 },
+            relIds[0],
+        ];
+        const batch = await imodel.relationships.queryInstancesProps(ElementGroupsMembers.classFullName, criteria);
+        expect(batch.length).to.equal(criteria.length);
+        criteria.forEach((criterion, i) => {
+            expect(batch[i]).to.deep.equal(imodel.relationships.tryGetInstanceProps(ElementGroupsMembers.classFullName, criterion), `criterion ${i}`);
+        });
+        expect(batch[0]?.id).to.equal(relIds[2]);
+        expect(batch[0]?.classFullName).to.equal(ElementGroupsMembers.classFullName);
+        expect(batch[1].memberPriority).to.equal(0);
+        expect(batch[1]?.id).to.equal(relIds[0]);
+        expect(batch[2]).to.be.undefined;
+        expect(batch[3]?.id).to.equal(relIds[3]);
+        expect(batch[4]).to.be.undefined;
+        expect(batch[5]).to.be.undefined;
+        expect(batch[6]).to.be.undefined;
+        expect(batch[7]?.id).to.equal(relIds[2]);
+        expect(batch[8]?.id).to.equal(relIds[1]);
+        // The concurrent query connection sees the same saved data.
+        expect(await imodel.relationships.queryInstancesProps(ElementGroupsMembers.classFullName, criteria, { usePrimaryConn: false })).to.deep.equal(batch);
+    });
+    it("returns an empty array for empty criteria", async () => {
+        expect(await imodel.relationships.queryInstancesProps(ElementGroupsMembers.classFullName, [])).to.deep.equal([]);
+        expect(await imodel.relationships.queryInstances(ElementGroupsMembers.classFullName, [])).to.deep.equal([]);
+    });
+    it("queries polymorphically and respects the relationship class", async () => {
+        const criteria = [relIds[0], { sourceId: elementIds[1], targetId: elementIds[2] }];
+        const viaBase = await imodel.relationships.queryInstancesProps(ElementRefersToElements.classFullName, criteria);
+        expect(viaBase.map((p) => p?.classFullName)).to.deep.equal([ElementGroupsMembers.classFullName, ElementGroupsMembers.classFullName]);
+        criteria.forEach((criterion, i) => expect(viaBase[i]).to.deep.equal(imodel.relationships.tryGetInstanceProps(ElementRefersToElements.classFullName, criterion), `criterion ${i}`));
+        const viaOther = await imodel.relationships.queryInstancesProps(ElementDrivesElement.classFullName, criteria);
+        expect(viaOther).to.deep.equal([undefined, undefined]);
+    });
+    it("queryInstances constructs Relationship entities", async () => {
+        const criteria = [relIds[1], { sourceId: elementIds[3], targetId: elementIds[4] }, Id64.invalid];
+        const rels = await imodel.relationships.queryInstances(ElementGroupsMembers.classFullName, criteria);
+        expect(rels.length).to.equal(3);
+        expect(rels[0]).to.be.instanceOf(ElementGroupsMembers);
+        expect(rels[0]?.id).to.equal(relIds[1]);
+        expect(rels[0]?.memberPriority).to.equal(1);
+        expect(rels[1]).to.be.instanceOf(ElementGroupsMembers);
+        expect(rels[1]?.sourceId).to.equal(elementIds[3]);
+        expect(rels[1]?.targetId).to.equal(elementIds[4]);
+        expect(rels[2]).to.be.undefined;
+        criteria.forEach((criterion, i) => expect(rels[i]?.toJSON()).to.deep.equal(imodel.relationships.tryGetInstance(ElementGroupsMembers.classFullName, criterion)?.toJSON(), `criterion ${i}`));
+    });
+    it("sees unsaved changes by default", async () => {
+        const cls = ElementGroupsMembers.classFullName;
+        const pair = { sourceId: elementIds[4], targetId: elementIds[5] };
+        await withEditTxn(imodel, async (txn) => {
+            const id = txn.insertRelationship(ElementGroupsMembers.create(imodel, pair.sourceId, pair.targetId, 4).toJSON());
+            const found = await imodel.relationships.queryInstances(cls, [id, pair]);
+            expect(found.map((r) => r?.id)).to.deep.equal([id, id]);
+            txn.deleteRelationship(imodel.relationships.getInstanceProps(cls, id));
+        });
+    });
+    it("converts point, struct and array properties like tryGetInstanceProps", async () => {
+        const pair = { sourceId: elementIds[0], targetId: elementIds[5] };
+        await withEditTxn(imodel, async (txn) => {
+            const id = txn.insertRelationship({
+                classFullName: testRelClass,
+                ...pair,
+                location: { x: 1, y: 2, z: 3 },
+                info: { label: "a", origin: { x: 4, y: 5 } },
+                points: [{ x: 6, y: 7, z: 8 }, { x: 9, y: 10, z: 11 }],
+            });
+            const criteria = [id, pair];
+            const batch = await imodel.relationships.queryInstancesProps(testRelClass, criteria);
+            criteria.forEach((criterion, i) => expect(batch[i]).to.deep.equal(imodel.relationships.tryGetInstanceProps(testRelClass, criterion), `criterion ${i}`));
+            expect(batch[0]).to.include({ id, classFullName: testRelClass });
+            expect(batch[0]).to.deep.include({
+                location: { x: 1, y: 2, z: 3 },
+                info: { label: "a", origin: { x: 4, y: 5 } },
+                points: [{ x: 6, y: 7, z: 8 }, { x: 9, y: 10, z: 11 }],
+            });
+            txn.deleteRelationship(batch[0]);
+        });
+    });
+    it("reads only the requested pairs in a dense many-to-many graph", async () => {
+        const cls = ElementGroupsMembers.classFullName;
+        const n = 5;
+        await withEditTxn(imodel, async (txn) => {
+            const sources = [];
+            const targets = [];
+            for (let i = 0; i < 2 * n; ++i) {
+                const id = txn.insertElement({ classFullName: RepositoryLink.classFullName, model: IModel.repositoryModelId, code: Code.createEmpty(), repositoryGuid: Guid.createValue() });
+                (i < n ? sources : targets).push(id);
+            }
+            // Connect every source to every target, then request only the "diagonal" pairs.
+            for (const sourceId of sources)
+                for (const targetId of targets)
+                    txn.insertRelationship(ElementGroupsMembers.create(imodel, sourceId, targetId, 0).toJSON());
+            const pairs = sources.map((sourceId, i) => ({ sourceId, targetId: targets[i] }));
+            let rowsRead = 0;
+            const createQueryReader = imodel.createQueryReader.bind(imodel);
+            const stub = sinon.stub(imodel, "createQueryReader").callsFake((ecsql, params, config) => {
+                const reader = createQueryReader(ecsql, params, config);
+                const step = reader.step.bind(reader);
+                reader.step = async () => {
+                    const hasRow = await step();
+                    if (hasRow)
+                        ++rowsRead;
+                    return hasRow;
+                };
+                return reader;
+            });
+            try {
+                const found = await imodel.relationships.queryInstancesProps(cls, pairs);
+                expect(found.map((p) => p && { sourceId: p.sourceId, targetId: p.targetId })).to.deep.equal(pairs);
+            }
+            finally {
+                stub.restore();
+            }
+            expect(rowsRead).to.equal(n);
+        });
+    });
+    it("supports relationships stored as navigation properties and ids from any briefcase", async () => {
+        const cls = "BisCore:ModelContainsElements";
+        const highBriefcaseId = Id64.fromLocalAndBriefcaseIds(1, 0xfffffd);
+        const criteria = [
+            elementIds[0],
+            { sourceId: IModel.repositoryModelId, targetId: elementIds[1] },
+            highBriefcaseId,
+            { sourceId: highBriefcaseId, targetId: elementIds[1] },
+        ];
+        const batch = await imodel.relationships.queryInstancesProps(cls, criteria);
+        criteria.forEach((criterion, i) => expect(batch[i]).to.deep.equal(imodel.relationships.tryGetInstanceProps(cls, criterion), `criterion ${i}`));
+        expect(batch.map((p) => p?.targetId)).to.deep.equal([elementIds[0], elementIds[1], undefined, undefined]);
+    });
+});
+//# sourceMappingURL=RelationshipBatchLookup.test.js.map

package/lib/esm/test/element/RelationshipBatchLookup.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"RelationshipBatchLookup.test.js","sourceRoot":"","sources":["../../../../src/test/element/RelationshipBatchLookup.test.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAC/F,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,KAAK,KAAK,MAAM,OAAO,CAAC;AAC/B,OAAO,EAAE,IAAI,EAAE,IAAI,EAAc,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAAE,IAAI,EAAE,MAAM,EAA2D,MAAM,oBAAoB,CAAC;AAC3G,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AACzG,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,MAAM,UAAU,GAAG;;;;;;;;;;;;;;;;;;;gBAmBL,CAAC;IACf,MAAM,YAAY,GAAG,sCAAsC,CAAC;IAE5D,IAAI,MAAkB,CAAC;IACvB,IAAI,UAAwB,CAAC;IAC7B,IAAI,MAAoB,CAAC;IAEzB,MAAM,CAAC,KAAK,IAAI,EAAE;QAChB,MAAM,GAAG,UAAU,CAAC,WAAW,CAAC,eAAe,CAAC,iBAAiB,CAAC,cAAc,EAAE,iBAAiB,CAAC,EAAE,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;QAChJ,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAC/C,WAAW,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE;YAC1B,UAAU,GAAG,EAAE,CAAC;YAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC3B,MAAM,KAAK,GAAwB;oBACjC,aAAa,EAAE,cAAc,CAAC,aAAa;oBAC3C,KAAK,EAAE,MAAM,CAAC,iBAAiB;oBAC/B,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE;oBACxB,cAAc,EAAE,IAAI,CAAC,WAAW,EAAE;iBACnC,CAAC;gBACF,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;YAC5C,CAAC;YAED,0EAA0E;YAC1E,MAAM,GAAG,EAAE,CAAC;YACZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACxB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC3H,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;IAE5B,EAAE,CAAC,+EAA+E,EAAE,KAAK,IAAI,EAAE;QAC7F,MAAM,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,GAAG,UAAU,CAAC;QAC5C,MAAM,QAAQ,GAAwC;YACpD,MAAM,CAAC,CAAC,CAAC;YACT,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;YAC9B,IAAI,CAAC,wBAAwB,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,kBAAkB;YAC5D,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;YAC9B,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,2DAA2D;YAC3F,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,YAAY;YAC5C,IAAI,CAAC,OAAO;YACZ,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY;YACvB,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE;YAC9B,MAAM,CAAC,CAAC,CAAC;SACV,CAAC;QAEF,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,oBAAoB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QAC3G,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC/C,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE;YAChC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,oBAAoB,CAAC,aAAa,EAAE,SAAS,CAAC,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC;QAC5I,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAC7E,MAAM,CAAE,KAAK,CAAC,CAAC,CAAS,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAEzC,4DAA4D;QAC5D,MAAM,CAAC,MAAM,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,oBAAoB,CAAC,aAAa,EAAE,QAAQ,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACvJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,CAAC,MAAM,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,oBAAoB,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjH,MAAM,CAAC,MAAM,MAAM,CAAC,aAAa,CAAC,cAAc,CAAC,oBAAoB,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC9G,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,QAAQ,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAEnF,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QAChH,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,aAAa,EAAE,oBAAoB,CAAC,aAAa,CAAC,CAAC,CAAC;QACrI,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,aAAa,EAAE,SAAS,CAAC,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;QAEnL,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,oBAAoB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QAC9G,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,QAAQ,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACjG,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,cAAc,CAAuB,oBAAoB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QAC3H,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC;QACvD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC;QACvD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,CAAC;QAChC,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,cAAc,CAAC,oBAAoB,CAAC,aAAa,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9L,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,GAAG,GAAG,oBAAoB,CAAC,aAAa,CAAC;QAC/C,MAAM,IAAI,GAAG,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QAClE,MAAM,WAAW,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACtC,MAAM,EAAE,GAAG,GAAG,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACjH,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,cAAc,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;YACzE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,GAAG,CAAC,kBAAkB,CAAC,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,IAAI,GAAG,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QAClE,MAAM,WAAW,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACtC,MAAM,EAAE,GAAG,GAAG,CAAC,kBAAkB,CAAC;gBAChC,aAAa,EAAE,YAAY;gBAC3B,GAAG,IAAI;gBACP,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;gBAC9B,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;gBAC5C,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;aAClC,CAAC,CAAC;YAExB,MAAM,QAAQ,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YAC5B,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;YACrF,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,YAAY,EAAE,SAAS,CAAC,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;YACxJ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;gBAC/B,QAAQ,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;gBAC9B,IAAI,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE;gBAC5C,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;aACvD,CAAC,CAAC;YACH,GAAG,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,GAAG,GAAG,oBAAoB,CAAC,aAAa,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,CAAC;QACZ,MAAM,WAAW,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACtC,MAAM,OAAO,GAAiB,EAAE,CAAC;YACjC,MAAM,OAAO,GAAiB,EAAE,CAAC;YACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;gBAC/B,MAAM,EAAE,GAAG,GAAG,CAAC,aAAa,CAAC,EAAE,aAAa,EAAE,cAAc,CAAC,aAAa,EAAE,KAAK,EAAE,MAAM,CAAC,iBAAiB,EAAE,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,WAAW,EAAE,EAAyB,CAAC,CAAC;gBACpM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvC,CAAC;YAED,gFAAgF;YAChF,KAAK,MAAM,QAAQ,IAAI,OAAO;gBAC5B,KAAK,MAAM,QAAQ,IAAI,OAAO;oBAC5B,GAAG,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,MAAM,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAEhG,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACjF,IAAI,QAAQ,GAAG,CAAC,CAAC;YACjB,MAAM,iBAAiB,GAAG,MAAM,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAChE,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE;gBACvF,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;gBACxD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtC,MAAM,CAAC,IAAI,GAAG,KAAK,IAAI,EAAE;oBACvB,MAAM,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;oBAC5B,IAAI,MAAM;wBACR,EAAE,QAAQ,CAAC;oBACb,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC;gBACF,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,CAAC;YACH,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBACzE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACrG,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC;YACD,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,GAAG,GAAG,+BAA+B,CAAC;QAC5C,MAAM,eAAe,GAAG,IAAI,CAAC,wBAAwB,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QACnE,MAAM,QAAQ,GAAwC;YACpD,UAAU,CAAC,CAAC,CAAC;YACb,EAAE,QAAQ,EAAE,MAAM,CAAC,iBAAiB,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE;YAC/D,eAAe;YACf,EAAE,QAAQ,EAAE,eAAe,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE;SACvD,CAAC;QACF,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QAC5E,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,GAAG,EAAE,SAAS,CAAC,EAAE,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/I,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;IAC5G,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\nimport { expect } from \"chai\";\r\nimport * as sinon from \"sinon\";\r\nimport { Guid, Id64, Id64String } from \"@itwin/core-bentley\";\r\nimport { Code, IModel, RelationshipProps, RepositoryLinkProps, SourceAndTarget } from \"@itwin/core-common\";\r\nimport { withEditTxn } from \"../../EditTxn\";\r\nimport { RepositoryLink } from \"../../Element\";\r\nimport { SnapshotDb } from \"../../IModelDb\";\r\nimport { ElementDrivesElement, ElementGroupsMembers, ElementRefersToElements } from \"../../Relationship\";\r\nimport { IModelTestUtils } from \"../IModelTestUtils\";\r\n\r\ndescribe(\"Relationships batched lookup\", () => {\r\n  const testSchema = `<?xml version=\"1.0\" encoding=\"UTF-8\"?>\r\n    <ECSchema schemaName=\"BatchLookupTest\" alias=\"blt\" version=\"01.00.00\" xmlns=\"http://www.bentley.com/schemas/Bentley.ECXML.3.2\">\r\n      <ECSchemaReference name=\"BisCore\" version=\"01.00.00\" alias=\"bis\"/>\r\n      <ECStructClass typeName=\"TestStruct\">\r\n        <ECProperty propertyName=\"label\" typeName=\"string\"/>\r\n        <ECProperty propertyName=\"origin\" typeName=\"point2d\"/>\r\n      </ECStructClass>\r\n      <ECRelationshipClass typeName=\"TestRefersToElements\" strength=\"referencing\" modifier=\"Sealed\">\r\n        <BaseClass>bis:ElementRefersToElements</BaseClass>\r\n        <Source multiplicity=\"(0..*)\" roleLabel=\"refers to\" polymorphic=\"true\">\r\n          <Class class=\"bis:Element\"/>\r\n        </Source>\r\n        <Target multiplicity=\"(0..*)\" roleLabel=\"is referenced by\" polymorphic=\"true\">\r\n          <Class class=\"bis:Element\"/>\r\n        </Target>\r\n        <ECProperty propertyName=\"Location\" typeName=\"point3d\"/>\r\n        <ECStructProperty propertyName=\"Info\" typeName=\"TestStruct\"/>\r\n        <ECArrayProperty propertyName=\"Points\" typeName=\"point3d\" minOccurs=\"0\" maxOccurs=\"unbounded\"/>\r\n      </ECRelationshipClass>\r\n    </ECSchema>`;\r\n  const testRelClass = \"BatchLookupTest:TestRefersToElements\";\r\n\r\n  let imodel: SnapshotDb;\r\n  let elementIds: Id64String[];\r\n  let relIds: Id64String[];\r\n\r\n  before(async () => {\r\n    imodel = SnapshotDb.createEmpty(IModelTestUtils.prepareOutputFile(\"Relationship\", \"BatchLookup.bim\"), { rootSubject: { name: \"BatchLookup\" } });\r\n    await imodel.importSchemaStrings([testSchema]);\r\n    withEditTxn(imodel, (txn) => {\r\n      elementIds = [];\r\n      for (let i = 0; i < 6; ++i) {\r\n        const props: RepositoryLinkProps = {\r\n          classFullName: RepositoryLink.classFullName,\r\n          model: IModel.repositoryModelId,\r\n          code: Code.createEmpty(),\r\n          repositoryGuid: Guid.createValue(),\r\n        };\r\n        elementIds.push(txn.insertElement(props));\r\n      }\r\n\r\n      // Chain e0->e1->e2->e3->e4 with memberPriority equal to the source index.\r\n      relIds = [];\r\n      for (let i = 0; i < 4; ++i)\r\n        relIds.push(txn.insertRelationship(ElementGroupsMembers.create(imodel, elementIds[i], elementIds[i + 1], i).toJSON()));\r\n    });\r\n  });\r\n\r\n  after(() => imodel.close());\r\n\r\n  it(\"matches tryGetInstanceProps for ids and source/target pairs, preserving order\", async () => {\r\n    const [e0, e1, e2, e3, e4, e5] = elementIds;\r\n    const criteria: Array<Id64String | SourceAndTarget> = [\r\n      relIds[2],\r\n      { sourceId: e0, targetId: e1 },\r\n      Id64.fromLocalAndBriefcaseIds(0x7fff, 0), // non-existent id\r\n      { sourceId: e3, targetId: e4 },\r\n      { sourceId: e0, targetId: e2 }, // both ends requested elsewhere, but not related as a pair\r\n      { sourceId: e4, targetId: e5 }, // unrelated\r\n      Id64.invalid,\r\n      relIds[2], // duplicate\r\n      { sourceId: e1, targetId: e2 },\r\n      relIds[0],\r\n    ];\r\n\r\n    const batch = await imodel.relationships.queryInstancesProps(ElementGroupsMembers.classFullName, criteria);\r\n    expect(batch.length).to.equal(criteria.length);\r\n    criteria.forEach((criterion, i) => {\r\n      expect(batch[i]).to.deep.equal(imodel.relationships.tryGetInstanceProps(ElementGroupsMembers.classFullName, criterion), `criterion ${i}`);\r\n    });\r\n\r\n    expect(batch[0]?.id).to.equal(relIds[2]);\r\n    expect(batch[0]?.classFullName).to.equal(ElementGroupsMembers.classFullName);\r\n    expect((batch[1] as any).memberPriority).to.equal(0);\r\n    expect(batch[1]?.id).to.equal(relIds[0]);\r\n    expect(batch[2]).to.be.undefined;\r\n    expect(batch[3]?.id).to.equal(relIds[3]);\r\n    expect(batch[4]).to.be.undefined;\r\n    expect(batch[5]).to.be.undefined;\r\n    expect(batch[6]).to.be.undefined;\r\n    expect(batch[7]?.id).to.equal(relIds[2]);\r\n    expect(batch[8]?.id).to.equal(relIds[1]);\r\n\r\n    // The concurrent query connection sees the same saved data.\r\n    expect(await imodel.relationships.queryInstancesProps(ElementGroupsMembers.classFullName, criteria, { usePrimaryConn: false })).to.deep.equal(batch);\r\n  });\r\n\r\n  it(\"returns an empty array for empty criteria\", async () => {\r\n    expect(await imodel.relationships.queryInstancesProps(ElementGroupsMembers.classFullName, [])).to.deep.equal([]);\r\n    expect(await imodel.relationships.queryInstances(ElementGroupsMembers.classFullName, [])).to.deep.equal([]);\r\n  });\r\n\r\n  it(\"queries polymorphically and respects the relationship class\", async () => {\r\n    const criteria = [relIds[0], { sourceId: elementIds[1], targetId: elementIds[2] }];\r\n\r\n    const viaBase = await imodel.relationships.queryInstancesProps(ElementRefersToElements.classFullName, criteria);\r\n    expect(viaBase.map((p) => p?.classFullName)).to.deep.equal([ElementGroupsMembers.classFullName, ElementGroupsMembers.classFullName]);\r\n    criteria.forEach((criterion, i) => expect(viaBase[i]).to.deep.equal(imodel.relationships.tryGetInstanceProps(ElementRefersToElements.classFullName, criterion), `criterion ${i}`));\r\n\r\n    const viaOther = await imodel.relationships.queryInstancesProps(ElementDrivesElement.classFullName, criteria);\r\n    expect(viaOther).to.deep.equal([undefined, undefined]);\r\n  });\r\n\r\n  it(\"queryInstances constructs Relationship entities\", async () => {\r\n    const criteria = [relIds[1], { sourceId: elementIds[3], targetId: elementIds[4] }, Id64.invalid];\r\n    const rels = await imodel.relationships.queryInstances<ElementGroupsMembers>(ElementGroupsMembers.classFullName, criteria);\r\n    expect(rels.length).to.equal(3);\r\n    expect(rels[0]).to.be.instanceOf(ElementGroupsMembers);\r\n    expect(rels[0]?.id).to.equal(relIds[1]);\r\n    expect(rels[0]?.memberPriority).to.equal(1);\r\n    expect(rels[1]).to.be.instanceOf(ElementGroupsMembers);\r\n    expect(rels[1]?.sourceId).to.equal(elementIds[3]);\r\n    expect(rels[1]?.targetId).to.equal(elementIds[4]);\r\n    expect(rels[2]).to.be.undefined;\r\n    criteria.forEach((criterion, i) => expect(rels[i]?.toJSON()).to.deep.equal(imodel.relationships.tryGetInstance(ElementGroupsMembers.classFullName, criterion)?.toJSON(), `criterion ${i}`));\r\n  });\r\n\r\n  it(\"sees unsaved changes by default\", async () => {\r\n    const cls = ElementGroupsMembers.classFullName;\r\n    const pair = { sourceId: elementIds[4], targetId: elementIds[5] };\r\n    await withEditTxn(imodel, async (txn) => {\r\n      const id = txn.insertRelationship(ElementGroupsMembers.create(imodel, pair.sourceId, pair.targetId, 4).toJSON());\r\n      const found = await imodel.relationships.queryInstances(cls, [id, pair]);\r\n      expect(found.map((r) => r?.id)).to.deep.equal([id, id]);\r\n      txn.deleteRelationship(imodel.relationships.getInstanceProps(cls, id));\r\n    });\r\n  });\r\n\r\n  it(\"converts point, struct and array properties like tryGetInstanceProps\", async () => {\r\n    const pair = { sourceId: elementIds[0], targetId: elementIds[5] };\r\n    await withEditTxn(imodel, async (txn) => {\r\n      const id = txn.insertRelationship({\r\n        classFullName: testRelClass,\r\n        ...pair,\r\n        location: { x: 1, y: 2, z: 3 },\r\n        info: { label: \"a\", origin: { x: 4, y: 5 } },\r\n        points: [{ x: 6, y: 7, z: 8 }, { x: 9, y: 10, z: 11 }],\r\n      } as RelationshipProps);\r\n\r\n      const criteria = [id, pair];\r\n      const batch = await imodel.relationships.queryInstancesProps(testRelClass, criteria);\r\n      criteria.forEach((criterion, i) => expect(batch[i]).to.deep.equal(imodel.relationships.tryGetInstanceProps(testRelClass, criterion), `criterion ${i}`));\r\n      expect(batch[0]).to.include({ id, classFullName: testRelClass });\r\n      expect(batch[0]).to.deep.include({\r\n        location: { x: 1, y: 2, z: 3 },\r\n        info: { label: \"a\", origin: { x: 4, y: 5 } },\r\n        points: [{ x: 6, y: 7, z: 8 }, { x: 9, y: 10, z: 11 }],\r\n      });\r\n      txn.deleteRelationship(batch[0]!);\r\n    });\r\n  });\r\n\r\n  it(\"reads only the requested pairs in a dense many-to-many graph\", async () => {\r\n    const cls = ElementGroupsMembers.classFullName;\r\n    const n = 5;\r\n    await withEditTxn(imodel, async (txn) => {\r\n      const sources: Id64String[] = [];\r\n      const targets: Id64String[] = [];\r\n      for (let i = 0; i < 2 * n; ++i) {\r\n        const id = txn.insertElement({ classFullName: RepositoryLink.classFullName, model: IModel.repositoryModelId, code: Code.createEmpty(), repositoryGuid: Guid.createValue() } as RepositoryLinkProps);\r\n        (i < n ? sources : targets).push(id);\r\n      }\r\n\r\n      // Connect every source to every target, then request only the \"diagonal\" pairs.\r\n      for (const sourceId of sources)\r\n        for (const targetId of targets)\r\n          txn.insertRelationship(ElementGroupsMembers.create(imodel, sourceId, targetId, 0).toJSON());\r\n\r\n      const pairs = sources.map((sourceId, i) => ({ sourceId, targetId: targets[i] }));\r\n      let rowsRead = 0;\r\n      const createQueryReader = imodel.createQueryReader.bind(imodel);\r\n      const stub = sinon.stub(imodel, \"createQueryReader\").callsFake((ecsql, params, config) => {\r\n        const reader = createQueryReader(ecsql, params, config);\r\n        const step = reader.step.bind(reader);\r\n        reader.step = async () => {\r\n          const hasRow = await step();\r\n          if (hasRow)\r\n            ++rowsRead;\r\n          return hasRow;\r\n        };\r\n        return reader;\r\n      });\r\n      try {\r\n        const found = await imodel.relationships.queryInstancesProps(cls, pairs);\r\n        expect(found.map((p) => p && { sourceId: p.sourceId, targetId: p.targetId })).to.deep.equal(pairs);\r\n      } finally {\r\n        stub.restore();\r\n      }\r\n      expect(rowsRead).to.equal(n);\r\n    });\r\n  });\r\n\r\n  it(\"supports relationships stored as navigation properties and ids from any briefcase\", async () => {\r\n    const cls = \"BisCore:ModelContainsElements\";\r\n    const highBriefcaseId = Id64.fromLocalAndBriefcaseIds(1, 0xfffffd);\r\n    const criteria: Array<Id64String | SourceAndTarget> = [\r\n      elementIds[0],\r\n      { sourceId: IModel.repositoryModelId, targetId: elementIds[1] },\r\n      highBriefcaseId,\r\n      { sourceId: highBriefcaseId, targetId: elementIds[1] },\r\n    ];\r\n    const batch = await imodel.relationships.queryInstancesProps(cls, criteria);\r\n    criteria.forEach((criterion, i) => expect(batch[i]).to.deep.equal(imodel.relationships.tryGetInstanceProps(cls, criterion), `criterion ${i}`));\r\n    expect(batch.map((p) => p?.targetId)).to.deep.equal([elementIds[0], elementIds[1], undefined, undefined]);\r\n  });\r\n});\r\n"]}