@itwin/core-backend 5.9.0-dev.1 → 5.9.0-dev.11

package/lib/esm/test/native/DgnDbWorker.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"DgnDbWorker.test.js","sourceRoot":"","sources":["../../../../src/test/native/DgnDbWorker.test.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAEnD,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,MAAoB,CAAC;IAEzB,SAAS,UAAU;QACjB,MAAM,WAAW,GAAG,EAAE,IAAI,EAAE,mBAAmB,EAAE,WAAW,EAAE,mBAAmB,EAAE,CAAC;QACpF,MAAM,GAAG,YAAY,CAAC,WAAW,CAAC,eAAe,CAAC,iBAAiB,CAAC,aAAa,EAAE,iBAAiB,CAAC,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;IAC1H,CAAC;IAED,MAAM,CAAC,GAAG,EAAE;QACV,UAAU,EAAE,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,EAAE;QACT,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM;QACH,OAAO,CAA4B;QACzB,OAAO,CAA4B;QAEpD;YACE,IAAI,CAAC,OAAO,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QACzE,CAAC;QAEM,KAAK,KAAK,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAChD,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACnC,QAAQ,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACvC,QAAQ,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAE9C,IAAW,UAAU,KAAc,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACtE,IAAW,WAAW,KAAc,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QACxE,IAAW,KAAK,KAAqC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAEtF,IAAW,SAAS,KAAK,OAAO,qDAA6C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC1F,IAAW,QAAQ,KAAK,OAAO,kDAA0C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtF,IAAW,SAAS,KAAK,OAAO,mDAA2C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACxF,IAAW,OAAO,KAAK,OAAO,iDAAyC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACpF,IAAW,IAAI,KAAK,OAAO,8CAAsC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9E,IAAW,SAAS,KAAK,OAAO,mDAA2C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACxF,IAAW,SAAS,KAAK,OAAO,mDAA2C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;KACzF;IAED,KAAK,UAAU,SAAS,CAAC,SAAwB;QAC/C,IAAI,SAAS,EAAE;YACb,OAAO;QAET,MAAM,IAAI,OAAO,CAAO,CAAC,OAAY,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACpE,OAAO,SAAS,CAAC,SAAS,CAAC,CAAC;IAC9B,CAAC;IAED,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QAEtC,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEvD,MAAM,CAAC,QAAQ,EAAE,CAAC;QAClB,MAAM,MAAM,CAAC,OAAO,CAAC;QAErB,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,qDAAqD;QACrD,MAAM,KAAK,GAAG,CAAC,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC,CAAC;QACvE,KAAK,MAAM,MAAM,IAAI,KAAK;YACxB,MAAM,CAAC,KAAK,EAAE,CAAC;QAEjB,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAEvD,uGAAuG;QACvG,MAAM,IAAI,GAAG,CAAC,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC,CAAC;QAC1C,KAAK,MAAM,MAAM,IAAI,IAAI;YACvB,MAAM,CAAC,KAAK,EAAE,CAAC;QAEjB,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACnD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEjD,gCAAgC;QAChC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QACpB,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACvB,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEjC,8EAA8E;QAC9E,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QACrD,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QAEnD,mBAAmB;QACnB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACnC,KAAK,MAAM,MAAM,IAAI,OAAO;YAC1B,MAAM,CAAC,QAAQ,EAAE,CAAC;QAEpB,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAsB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,gEAAgE;QACrK,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACrC,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,OAAQ,EAAE,UAAU,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,CAAC,QAAQ,EAAE,CAAC;QAClB,MAAM,MAAM,CAAC,OAAO,CAAC;QACrB,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QACtB,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,QAAQ,EAAE,CAAC;QAClB,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,OAAQ,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,IAAI,MAAM,EAAE,CAAC;QAC7B,OAAO,CAAC,QAAQ,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,QAAQ,EAAE,CAAC;QAElB,4DAA4D;QAC5D,MAAM,MAAM,GAAG,CAAC,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC,CAAC;QAEpG,4BAA4B;QAC5B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;QACjD,KAAK,MAAM,MAAM,IAAI,OAAO;YAC1B,MAAM,CAAC,KAAK,EAAE,CAAC;QAEjB,iDAAiD;QACjD,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,UAAU,EAAE,CAAC;QAEb,MAAM,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAExF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACnE,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAC1E,MAAM,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,qGAAqG;QACrG,MAAM,QAAQ,GAAG,CAAC,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC,CAAC;QAC1E,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAEnC,MAAM,SAAS,GAAG,OAAO,CAAC;QAC1B,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE;YACzC,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;YAC/B,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;YAChC,EAAE,EAAE,OAAO;YACX,WAAW,EAAE,QAAQ,CAAC,cAAc,EAAE,CAAC,MAAM,EAAE;SAChD,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAChE,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE7B,oFAAoF;QACpF,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtC,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAsB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC;QAE3G,MAAM,YAAY,CAAC;IACrB,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 { assert, expect } from \"chai\";\nimport { BeDuration } from \"@itwin/core-bentley\";\nimport { Matrix4d } from \"@itwin/core-geometry\";\nimport { IModelNative } from \"../../internal/NativePlatform\";\nimport { StandaloneDb } from \"../../IModelDb\";\nimport { IModelTestUtils } from \"../IModelTestUtils\";\nimport { IModelJsNative } from \"@bentley/imodeljs-native\";\nimport { _nativeDb } from \"../../internal/Symbols\";\n\ndescribe(\"DgnDbWorker\", () => {\n  let imodel: StandaloneDb;\n\n  function openIModel(): void {\n    const rootSubject = { name: \"DgnDbWorker tests\", description: \"DgnDbWorker tests\" };\n    imodel = StandaloneDb.createEmpty(IModelTestUtils.prepareOutputFile(\"DgnDbWorker\", \"DgnDbWorker.bim\"), { rootSubject });\n  }\n\n  before(() => {\n    openIModel();\n  });\n\n  after(() => {\n    imodel.close();\n  });\n\n  class Worker {\n    public promise: Promise<void> | undefined;\n    private readonly _worker: IModelJsNative.TestWorker;\n\n    public constructor() {\n      this._worker = new IModelNative.platform.TestWorker(imodel[_nativeDb]);\n    }\n\n    public queue() { this.promise = this._worker.queue(); }\n    public cancel() { this._worker.cancel(); }\n    public setReady() { this._worker.setReady(); }\n    public setThrow() { this._worker.setThrow(); }\n\n    public get isCanceled(): boolean { return this._worker.isCanceled(); }\n    public get wasExecuted(): boolean { return this._worker.wasExecuted(); }\n    public get state(): IModelJsNative.TestWorkerState { return this._worker.getState(); }\n\n    public get wasQueued() { return IModelJsNative.TestWorkerState.NotQueued !== this.state; }\n    public get isQueued() { return IModelJsNative.TestWorkerState.Queued === this.state; }\n    public get isRunning() { return IModelJsNative.TestWorkerState.Running === this.state; }\n    public get isError() { return IModelJsNative.TestWorkerState.Error === this.state; }\n    public get isOk() { return IModelJsNative.TestWorkerState.Ok === this.state; }\n    public get isSkipped() { return IModelJsNative.TestWorkerState.Skipped === this.state; }\n    public get isAborted() { return IModelJsNative.TestWorkerState.Aborted === this.state; }\n  }\n\n  async function waitUntil(condition: () => boolean): Promise<void> {\n    if (condition())\n      return;\n\n    await new Promise<void>((resolve: any) => setTimeout(resolve, 100));\n    return waitUntil(condition);\n  }\n\n  it(\"executes asynchronously\", async () => {\n    const worker = new Worker();\n    await BeDuration.wait(1000);\n    expect(worker.wasQueued).to.be.false;\n    expect(worker.isCanceled).to.be.false;\n\n    worker.queue();\n    expect(worker.isQueued || worker.isRunning).to.be.true;\n\n    worker.setReady();\n    await worker.promise;\n\n    expect(worker.wasExecuted).to.be.true;\n    expect(worker.isCanceled).to.be.false;\n    expect(worker.isOk).to.be.true;\n  });\n\n  it(\"executes a maximum of 4 simultaneously\", async () => {\n    // Create 4 workers. They should all start executing.\n    const first = [new Worker(), new Worker(), new Worker(), new Worker()];\n    for (const worker of first)\n      worker.queue();\n\n    await waitUntil(() => first.every((x) => x.isRunning));\n\n    // Queue up 2 more workers. They should not start executing until at least one of the first 4 resolves.\n    const next = [new Worker(), new Worker()];\n    for (const worker of next)\n      worker.queue();\n\n    await BeDuration.wait(1000);\n    expect(first.every((x) => x.isRunning)).to.be.true;\n    expect(next.every((x) => x.isQueued)).to.be.true;\n\n    // Let the first worker resolve.\n    first[0].setReady();\n    await first[0].promise;\n    expect(first[0].isOk).to.be.true;\n\n    // Exactly one of the 2 workers remaining in the queue should start executing.\n    await waitUntil(() => next.some((x) => x.isRunning));\n    await BeDuration.wait(1000);\n    expect(next.every((x) => x.isRunning)).to.be.false;\n\n    // Clear the queue.\n    const workers = first.concat(next);\n    for (const worker of workers)\n      worker.setReady();\n\n    await Promise.all(workers.map((x) => x.promise).filter((p): p is Promise<void> => p !== undefined)); // eslint-disable-line @typescript-eslint/promise-function-async\n    expect(workers.every((x) => x.isOk)).to.be.true;\n  });\n\n  it(\"skips execution if canceled before execution begins\", async () => {\n    const worker = new Worker();\n    worker.cancel();\n    expect(worker.isCanceled).to.be.true;\n    worker.queue();\n    await assert.isRejected(worker.promise!, \"canceled\");\n    expect(worker.isSkipped).to.be.true;\n    expect(worker.wasExecuted).to.be.false;\n  });\n\n  it(\"executes if canceled after execution begins\", async () => {\n    const worker = new Worker();\n    worker.queue();\n    await waitUntil(() => worker.isRunning);\n    worker.cancel();\n    worker.setReady();\n    await worker.promise;\n    expect(worker.isCanceled).to.be.true;\n    expect(worker.isSkipped).to.be.false;\n    expect(worker.wasExecuted).to.be.true;\n  });\n\n  it(\"throws\", async () => {\n    const worker = new Worker();\n    worker.setThrow();\n    worker.queue();\n    await assert.isRejected(worker.promise!, \"throw\");\n    expect(worker.isCanceled).to.be.false;\n    expect(worker.isError).to.be.true;\n  });\n\n  it(\"cancels all workers before iModel is closed\", async () => {\n    const resolve = new Worker();\n    resolve.setReady();\n    const reject = new Worker();\n    reject.setThrow();\n\n    // These 6 workers will never resolve nor reject explicitly.\n    const cancel = [new Worker(), new Worker(), new Worker(), new Worker(), new Worker(), new Worker()];\n\n    // Queue up all the workers.\n    const workers = cancel.concat([resolve, reject]);\n    for (const worker of workers)\n      worker.queue();\n\n    // Closing the iModel cancels all extant workers.\n    imodel.close();\n    openIModel();\n\n    await expect(Promise.all(workers.map(async (x) => x.promise))).rejectedWith(\"canceled\");\n\n    expect(cancel.every((x) => x.isCanceled)).to.be.true;\n    expect(cancel.every((x) => x.isAborted || x.isSkipped)).to.be.true;\n    expect(resolve.isOk || resolve.isSkipped || resolve.isAborted).to.be.true;\n    expect(reject.isError || reject.isSkipped).to.be.true;\n  });\n\n  it(\"cancels a snap request\", async () => {\n    // Block the worker thread pool with 4 workers so that our snap cannot complete before they complete.\n    const blockers = [new Worker(), new Worker(), new Worker(), new Worker()];\n    blockers.forEach((w) => w.queue());\n\n    const sessionId = \"0x222\";\n    const snap = imodel.requestSnap(sessionId, {\n      testPoint: { x: 1, y: 2, z: 3 },\n      closePoint: { x: 1, y: 2, z: 3 },\n      id: \"0x111\",\n      worldToView: Matrix4d.createIdentity().toJSON(),\n    });\n\n    const toBeRejected = expect(snap).to.be.rejectedWith(\"aborted\");\n    imodel.cancelSnap(sessionId);\n\n    // Clear the worker thread pool so the snap request (now canceled) can be processed.\n    blockers.forEach((w) => w.setReady());\n    await Promise.all(blockers.map(async (w) => w.promise).filter((p): p is Promise<void> => p !== undefined));\n\n    await toBeRejected;\n  });\n});\n"]}
+{"version":3,"file":"DgnDbWorker.test.js","sourceRoot":"","sources":["../../../../src/test/native/DgnDbWorker.test.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAE/F,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,OAAO,EAAE,SAAS,EAAE,MAAM,wBAAwB,CAAC;AAEnD,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,IAAI,MAAoB,CAAC;IAEzB,SAAS,UAAU;QACjB,MAAM,WAAW,GAAG,EAAE,IAAI,EAAE,mBAAmB,EAAE,WAAW,EAAE,mBAAmB,EAAE,CAAC;QACpF,MAAM,GAAG,YAAY,CAAC,WAAW,CAAC,eAAe,CAAC,iBAAiB,CAAC,aAAa,EAAE,iBAAiB,CAAC,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;IAC1H,CAAC;IAED,MAAM,CAAC,GAAG,EAAE;QACV,UAAU,EAAE,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,GAAG,EAAE;QACT,MAAM,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM;QACH,OAAO,CAA4B;QACzB,OAAO,CAA4B;QAEpD;YACE,IAAI,CAAC,OAAO,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;QACzE,CAAC;QAEM,KAAK,KAAK,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAChD,MAAM,KAAK,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACnC,QAAQ,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACvC,QAAQ,KAAK,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAE9C,IAAW,UAAU,KAAc,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACtE,IAAW,WAAW,KAAc,OAAO,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QACxE,IAAW,KAAK,KAAqC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAEtF,IAAW,SAAS,KAAK,OAAO,qDAA6C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC1F,IAAW,QAAQ,KAAK,OAAO,kDAA0C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtF,IAAW,SAAS,KAAK,OAAO,mDAA2C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACxF,IAAW,OAAO,KAAK,OAAO,iDAAyC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACpF,IAAW,IAAI,KAAK,OAAO,8CAAsC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9E,IAAW,SAAS,KAAK,OAAO,mDAA2C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACxF,IAAW,SAAS,KAAK,OAAO,mDAA2C,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;KACzF;IAED,KAAK,UAAU,SAAS,CAAC,SAAwB;QAC/C,IAAI,SAAS,EAAE;YACb,OAAO;QAET,MAAM,IAAI,OAAO,CAAO,CAAC,OAAY,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;QACpE,OAAO,SAAS,CAAC,SAAS,CAAC,CAAC;IAC9B,CAAC;IAED,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QAEtC,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEvD,MAAM,CAAC,QAAQ,EAAE,CAAC;QAClB,MAAM,MAAM,CAAC,OAAO,CAAC;QAErB,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,qDAAqD;QACrD,MAAM,KAAK,GAAG,CAAC,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC,CAAC;QACvE,KAAK,MAAM,MAAM,IAAI,KAAK;YACxB,MAAM,CAAC,KAAK,EAAE,CAAC;QAEjB,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAEvD,uGAAuG;QACvG,MAAM,IAAI,GAAG,CAAC,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC,CAAC;QAC1C,KAAK,MAAM,MAAM,IAAI,IAAI;YACvB,MAAM,CAAC,KAAK,EAAE,CAAC;QAEjB,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACnD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEjD,gCAAgC;QAChC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QACpB,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;QACvB,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEjC,8EAA8E;QAC9E,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QACrD,MAAM,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QAEnD,mBAAmB;QACnB,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACnC,KAAK,MAAM,MAAM,IAAI,OAAO;YAC1B,MAAM,CAAC,QAAQ,EAAE,CAAC;QAEpB,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAsB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,gEAAgE;QACrK,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACrC,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,OAAQ,EAAE,UAAU,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,EAAE,CAAC;QAChB,MAAM,CAAC,QAAQ,EAAE,CAAC;QAClB,MAAM,MAAM,CAAC,OAAO,CAAC;QACrB,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QACtB,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,QAAQ,EAAE,CAAC;QAClB,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,OAAQ,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,IAAI,MAAM,EAAE,CAAC;QAC7B,OAAO,CAAC,QAAQ,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QAC5B,MAAM,CAAC,QAAQ,EAAE,CAAC;QAElB,4DAA4D;QAC5D,MAAM,MAAM,GAAG,CAAC,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC,CAAC;QAEpG,4BAA4B;QAC5B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;QACjD,KAAK,MAAM,MAAM,IAAI,OAAO;YAC1B,MAAM,CAAC,KAAK,EAAE,CAAC;QAEjB,iDAAiD;QACjD,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,UAAU,EAAE,CAAC;QAEb,MAAM,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAExF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,IAAI,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QACnE,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAC1E,MAAM,CAAC,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,qGAAqG;QACrG,MAAM,QAAQ,GAAG,CAAC,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC,CAAC;QAC1E,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;QAEnC,MAAM,SAAS,GAAG,OAAO,CAAC;QAC1B,MAAM,IAAI,GAAG,MAAM,CAAC,WAAW,CAAC,SAAS,EAAE;YACzC,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;YAC/B,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE;YAChC,EAAE,EAAE,OAAO;YACX,WAAW,EAAE,QAAQ,CAAC,cAAc,EAAE,CAAC,MAAM,EAAE;SAChD,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAChE,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAE7B,oFAAoF;QACpF,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtC,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAsB,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC;QAE3G,MAAM,YAAY,CAAC;IACrB,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 { assert, expect } from \"chai\";\r\nimport { BeDuration } from \"@itwin/core-bentley\";\r\nimport { Matrix4d } from \"@itwin/core-geometry\";\r\nimport { IModelNative } from \"../../internal/NativePlatform\";\r\nimport { StandaloneDb } from \"../../IModelDb\";\r\nimport { IModelTestUtils } from \"../IModelTestUtils\";\r\nimport { IModelJsNative } from \"@bentley/imodeljs-native\";\r\nimport { _nativeDb } from \"../../internal/Symbols\";\r\n\r\ndescribe(\"DgnDbWorker\", () => {\r\n  let imodel: StandaloneDb;\r\n\r\n  function openIModel(): void {\r\n    const rootSubject = { name: \"DgnDbWorker tests\", description: \"DgnDbWorker tests\" };\r\n    imodel = StandaloneDb.createEmpty(IModelTestUtils.prepareOutputFile(\"DgnDbWorker\", \"DgnDbWorker.bim\"), { rootSubject });\r\n  }\r\n\r\n  before(() => {\r\n    openIModel();\r\n  });\r\n\r\n  after(() => {\r\n    imodel.close();\r\n  });\r\n\r\n  class Worker {\r\n    public promise: Promise<void> | undefined;\r\n    private readonly _worker: IModelJsNative.TestWorker;\r\n\r\n    public constructor() {\r\n      this._worker = new IModelNative.platform.TestWorker(imodel[_nativeDb]);\r\n    }\r\n\r\n    public queue() { this.promise = this._worker.queue(); }\r\n    public cancel() { this._worker.cancel(); }\r\n    public setReady() { this._worker.setReady(); }\r\n    public setThrow() { this._worker.setThrow(); }\r\n\r\n    public get isCanceled(): boolean { return this._worker.isCanceled(); }\r\n    public get wasExecuted(): boolean { return this._worker.wasExecuted(); }\r\n    public get state(): IModelJsNative.TestWorkerState { return this._worker.getState(); }\r\n\r\n    public get wasQueued() { return IModelJsNative.TestWorkerState.NotQueued !== this.state; }\r\n    public get isQueued() { return IModelJsNative.TestWorkerState.Queued === this.state; }\r\n    public get isRunning() { return IModelJsNative.TestWorkerState.Running === this.state; }\r\n    public get isError() { return IModelJsNative.TestWorkerState.Error === this.state; }\r\n    public get isOk() { return IModelJsNative.TestWorkerState.Ok === this.state; }\r\n    public get isSkipped() { return IModelJsNative.TestWorkerState.Skipped === this.state; }\r\n    public get isAborted() { return IModelJsNative.TestWorkerState.Aborted === this.state; }\r\n  }\r\n\r\n  async function waitUntil(condition: () => boolean): Promise<void> {\r\n    if (condition())\r\n      return;\r\n\r\n    await new Promise<void>((resolve: any) => setTimeout(resolve, 100));\r\n    return waitUntil(condition);\r\n  }\r\n\r\n  it(\"executes asynchronously\", async () => {\r\n    const worker = new Worker();\r\n    await BeDuration.wait(1000);\r\n    expect(worker.wasQueued).to.be.false;\r\n    expect(worker.isCanceled).to.be.false;\r\n\r\n    worker.queue();\r\n    expect(worker.isQueued || worker.isRunning).to.be.true;\r\n\r\n    worker.setReady();\r\n    await worker.promise;\r\n\r\n    expect(worker.wasExecuted).to.be.true;\r\n    expect(worker.isCanceled).to.be.false;\r\n    expect(worker.isOk).to.be.true;\r\n  });\r\n\r\n  it(\"executes a maximum of 4 simultaneously\", async () => {\r\n    // Create 4 workers. They should all start executing.\r\n    const first = [new Worker(), new Worker(), new Worker(), new Worker()];\r\n    for (const worker of first)\r\n      worker.queue();\r\n\r\n    await waitUntil(() => first.every((x) => x.isRunning));\r\n\r\n    // Queue up 2 more workers. They should not start executing until at least one of the first 4 resolves.\r\n    const next = [new Worker(), new Worker()];\r\n    for (const worker of next)\r\n      worker.queue();\r\n\r\n    await BeDuration.wait(1000);\r\n    expect(first.every((x) => x.isRunning)).to.be.true;\r\n    expect(next.every((x) => x.isQueued)).to.be.true;\r\n\r\n    // Let the first worker resolve.\r\n    first[0].setReady();\r\n    await first[0].promise;\r\n    expect(first[0].isOk).to.be.true;\r\n\r\n    // Exactly one of the 2 workers remaining in the queue should start executing.\r\n    await waitUntil(() => next.some((x) => x.isRunning));\r\n    await BeDuration.wait(1000);\r\n    expect(next.every((x) => x.isRunning)).to.be.false;\r\n\r\n    // Clear the queue.\r\n    const workers = first.concat(next);\r\n    for (const worker of workers)\r\n      worker.setReady();\r\n\r\n    await Promise.all(workers.map((x) => x.promise).filter((p): p is Promise<void> => p !== undefined)); // eslint-disable-line @typescript-eslint/promise-function-async\r\n    expect(workers.every((x) => x.isOk)).to.be.true;\r\n  });\r\n\r\n  it(\"skips execution if canceled before execution begins\", async () => {\r\n    const worker = new Worker();\r\n    worker.cancel();\r\n    expect(worker.isCanceled).to.be.true;\r\n    worker.queue();\r\n    await assert.isRejected(worker.promise!, \"canceled\");\r\n    expect(worker.isSkipped).to.be.true;\r\n    expect(worker.wasExecuted).to.be.false;\r\n  });\r\n\r\n  it(\"executes if canceled after execution begins\", async () => {\r\n    const worker = new Worker();\r\n    worker.queue();\r\n    await waitUntil(() => worker.isRunning);\r\n    worker.cancel();\r\n    worker.setReady();\r\n    await worker.promise;\r\n    expect(worker.isCanceled).to.be.true;\r\n    expect(worker.isSkipped).to.be.false;\r\n    expect(worker.wasExecuted).to.be.true;\r\n  });\r\n\r\n  it(\"throws\", async () => {\r\n    const worker = new Worker();\r\n    worker.setThrow();\r\n    worker.queue();\r\n    await assert.isRejected(worker.promise!, \"throw\");\r\n    expect(worker.isCanceled).to.be.false;\r\n    expect(worker.isError).to.be.true;\r\n  });\r\n\r\n  it(\"cancels all workers before iModel is closed\", async () => {\r\n    const resolve = new Worker();\r\n    resolve.setReady();\r\n    const reject = new Worker();\r\n    reject.setThrow();\r\n\r\n    // These 6 workers will never resolve nor reject explicitly.\r\n    const cancel = [new Worker(), new Worker(), new Worker(), new Worker(), new Worker(), new Worker()];\r\n\r\n    // Queue up all the workers.\r\n    const workers = cancel.concat([resolve, reject]);\r\n    for (const worker of workers)\r\n      worker.queue();\r\n\r\n    // Closing the iModel cancels all extant workers.\r\n    imodel.close();\r\n    openIModel();\r\n\r\n    await expect(Promise.all(workers.map(async (x) => x.promise))).rejectedWith(\"canceled\");\r\n\r\n    expect(cancel.every((x) => x.isCanceled)).to.be.true;\r\n    expect(cancel.every((x) => x.isAborted || x.isSkipped)).to.be.true;\r\n    expect(resolve.isOk || resolve.isSkipped || resolve.isAborted).to.be.true;\r\n    expect(reject.isError || reject.isSkipped).to.be.true;\r\n  });\r\n\r\n  it(\"cancels a snap request\", async () => {\r\n    // Block the worker thread pool with 4 workers so that our snap cannot complete before they complete.\r\n    const blockers = [new Worker(), new Worker(), new Worker(), new Worker()];\r\n    blockers.forEach((w) => w.queue());\r\n\r\n    const sessionId = \"0x222\";\r\n    const snap = imodel.requestSnap(sessionId, {\r\n      testPoint: { x: 1, y: 2, z: 3 },\r\n      closePoint: { x: 1, y: 2, z: 3 },\r\n      id: \"0x111\",\r\n      worldToView: Matrix4d.createIdentity().toJSON(),\r\n    });\r\n\r\n    const toBeRejected = expect(snap).to.be.rejectedWith(\"aborted\");\r\n    imodel.cancelSnap(sessionId);\r\n\r\n    // Clear the worker thread pool so the snap request (now canceled) can be processed.\r\n    blockers.forEach((w) => w.setReady());\r\n    await Promise.all(blockers.map(async (w) => w.promise).filter((p): p is Promise<void> => p !== undefined));\r\n\r\n    await toBeRejected;\r\n  });\r\n});\r\n"]}

package/lib/esm/test/rpc/response.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"response.test.js","sourceRoot":"","sources":["../../../../src/test/rpc/response.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAa,IAAI,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAC5C,OAAO,EAAgE,gBAAgB,EAA2C,MAAM,oBAAoB,CAAC;AAC7J,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,oBAAoB,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAE5D,qDAAqD;AAErD,MAAM,YAAa,SAAQ,QAAQ;IAC1B,MAAM,GAAU,EAAE,CAAC;IAC1B,IAAW,MAAM,KAAa,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAElE,gEAAgE;IAChD,MAAM,CAAC,KAAU,EAAE,SAAyB,EAAE,QAAsB;QAClF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,QAAQ,EAAE,CAAC;IACb,CAAC;IAEM,IAAI,GAAyC,IAAI,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAClE,MAAM,GAA4C,IAAI,EAAE,CAAC;IACzD,GAAG,GAAsC,IAAI,EAAE,CAAC;CACxD;AAED,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,QAA2B,CAAC;IAChC,IAAI,OAA6B,CAAC;IAClC,IAAI,WAAkC,CAAC;IACvC,IAAI,GAAsB,CAAC;IAC3B,IAAI,GAAiB,CAAC;IAEtB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,QAAQ,EAAS,CAAC;QACjE,OAAO,GAAG,EAAE,SAAS,EAAE,EAAE,aAAa,EAAE,oBAAoB,EAAE,EAAS,CAAC;QACxE,WAAW,GAAG;YACZ,gBAAgB,EAAE,IAAI;YACtB,MAAM,EAAE,GAAG;YACX,MAAM,EAAE;gBACN,IAAI,EAAE,EAAE;aACT;SACK,CAAC;QACT,GAAG,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,mBAAmB,EAAS,CAAC;QACnD,GAAG,GAAG,IAAI,YAAY,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,UAAU;QACV,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC;QACpC,MAAM,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QACzD,WAAW,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,CAAC;QAE1C,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7D,SAAS;QACT,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEvE,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC;QAC3C,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAC/D,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,UAAU;QACV,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC;QACpC,MAAM,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QACzD,WAAW,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,CAAC;QAC1C,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,eAAe,CAAC;QAEnC,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7D,SAAS;QACT,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEzE,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC;QAC3C,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAC/D,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,UAAU;QACV,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC;QACpC,WAAW,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,CAAC;QAC1C,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC;QAE7B,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7D,SAAS;QACT,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QAEhE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,UAAU;QACV,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC;QACpC,MAAM,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QACzD,WAAW,CAAC,MAAM,CAAC,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAExD,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAC7D,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;QAE1D,SAAS;QACT,MAAM,cAAc,GAAG,GAAG,CAAC,MAAM,CAAC;QAClC,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC;QAC3C,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAC/D,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,UAAU;QACV,WAAW,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,EAAE,CAAC;QAC5C,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,sBAAsB,CAAC;QAE1C,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7D,SAAS;QACT,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACzE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,SAAS,YAAY,CAAC,WAAW,GAAG,IAAI;IACtC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,MAAM,GAAG,GAAG,EAAS,CAAC;IAEtB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,MAAM,GAAG,WAAW,EAAE,CAAC;QAChD,GAAG,CAAC,YAAY,WAAW,EAAE,CAAC,GAAG,SAAS,WAAW,EAAE,CAAC;QACxD,WAAW,EAAE,CAAC;IAChB,CAAC;IAED,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;AAC7B,CAAC","sourcesContent":["import { expect } from \"chai\";\nimport { SinonStub, stub } from \"sinon\";\nimport { Readable, Writable } from \"stream\";\nimport { HttpServerRequest, HttpServerResponse, RpcRequestFulfillment, RpcRequestStatus, SerializedRpcRequest, WebAppRpcProtocol } from \"@itwin/core-common\";\nimport { sendResponse } from \"../../rpc/web/response\";\nimport { brotliDecompressSync, unzipSync } from \"node:zlib\";\n\n/* eslint-disable @typescript-eslint/no-deprecated */\n\nclass StubResponse extends Writable implements HttpServerResponse {\n  public chunks: any[] = [];\n  public get buffer(): Buffer { return Buffer.concat(this.chunks); }\n\n  // eslint-disable-next-line @typescript-eslint/naming-convention\n  public override _write(chunk: any, _encoding: BufferEncoding, callback: VoidFunction): void {\n    this.chunks.push(chunk);\n    callback();\n  }\n\n  public send: SinonStub<any[], HttpServerResponse> = stub().returns(this);\n  public status: SinonStub<[number], HttpServerResponse> = stub();\n  public set: SinonStub<[string, string], void> = stub();\n}\n\ndescribe(\"sendResponse\", () => {\n  let protocol: WebAppRpcProtocol;\n  let request: SerializedRpcRequest;\n  let fulfillment: RpcRequestFulfillment;\n  let req: HttpServerRequest;\n  let res: StubResponse;\n\n  beforeEach(() => {\n    protocol = { getStatus: () => RpcRequestStatus.Resolved } as any;\n    request = { operation: { operationName: \"fake-RPC-operation\" } } as any;\n    fulfillment = {\n      allowCompression: true,\n      status: 200,\n      result: {\n        data: [],\n      },\n    } as any;\n    req = { header: () => \"gzip, deflate, br\" } as any;\n    res = new StubResponse();\n  });\n\n  it(\"should compress response using Brotli\", async () => {\n    // Arrange\n    const originalData = generateJson();\n    const originalDataSize = Buffer.byteLength(originalData);\n    fulfillment.result.objects = originalData;\n\n    // Act\n    await sendResponse(protocol, request, fulfillment, req, res);\n\n    // Assert\n    expect(res.set.calledWithExactly(\"Content-Encoding\", \"br\")).to.be.true;\n\n    const compressedData = res.send.getCall(0).args[0];\n    expect(compressedData).to.not.be.undefined;\n    expect(compressedData.length).to.be.lessThan(originalDataSize);\n    expect(brotliDecompressSync(compressedData).toString()).to.be.equal(originalData);\n  });\n\n  it(\"should compress response using Gzip if Brotli is not supported\", async () => {\n    // Arrange\n    const originalData = generateJson();\n    const originalDataSize = Buffer.byteLength(originalData);\n    fulfillment.result.objects = originalData;\n    req.header = () => \"gzip, deflate\";\n\n    // Act\n    await sendResponse(protocol, request, fulfillment, req, res);\n\n    // Assert\n    expect(res.set.calledWithExactly(\"Content-Encoding\", \"gzip\")).to.be.true;\n\n    const compressedData = res.send.getCall(0).args[0];\n    expect(compressedData).to.not.be.undefined;\n    expect(compressedData.length).to.be.lessThan(originalDataSize);\n    expect(unzipSync(compressedData).toString()).to.be.equal(originalData);\n  });\n\n  it(\"should not compress if Brotli and Gzip is not supported\", async () => {\n    // Arrange\n    const originalData = generateJson();\n    fulfillment.result.objects = originalData;\n    req.header = () => undefined;\n\n    // Act\n    await sendResponse(protocol, request, fulfillment, req, res);\n\n    // Assert\n    expect(res.set.calledWithMatch(\"Content-Encoding\")).to.be.false;\n\n    expect(res.send.getCall(0).args[0]).to.be.equal(originalData);\n  });\n\n  it(\"should compress stream\", async () => {\n    // Arrange\n    const originalData = generateJson();\n    const originalDataSize = Buffer.byteLength(originalData);\n    fulfillment.result.stream = Readable.from(originalData);\n\n    // Act\n    await sendResponse(protocol, request, fulfillment, req, res);\n    await new Promise((resolve) => res.on(\"finish\", resolve));\n\n    // Assert\n    const compressedData = res.buffer;\n    expect(compressedData).to.not.be.undefined;\n    expect(compressedData.length).to.be.lessThan(originalDataSize);\n    expect(brotliDecompressSync(compressedData).toString()).to.be.equal(originalData);\n  });\n\n  it(\"should support Accept-Encoding header value format without spaces between values\", async () => {\n    // Arrange\n    fulfillment.result.objects = generateJson();\n    req.header = () => \"gzip,deflate,br,zstd\";\n\n    // Act\n    await sendResponse(protocol, request, fulfillment, req, res);\n\n    // Assert\n    expect(res.set.calledWithExactly(\"Content-Encoding\", \"br\")).to.be.true;\n  });\n});\n\nfunction generateJson(minimumSize = 2000): string {\n  let propertyIdx = 0;\n  const obj = {} as any;\n\n  while (JSON.stringify(obj).length < minimumSize) {\n    obj[`property-${propertyIdx}`] = `value-${propertyIdx}`;\n    propertyIdx++;\n  }\n\n  return JSON.stringify(obj);\n}\n"]}
+{"version":3,"file":"response.test.js","sourceRoot":"","sources":["../../../../src/test/rpc/response.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAa,IAAI,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAC5C,OAAO,EAAgE,gBAAgB,EAA2C,MAAM,oBAAoB,CAAC;AAC7J,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,oBAAoB,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAE5D,qDAAqD;AAErD,MAAM,YAAa,SAAQ,QAAQ;IAC1B,MAAM,GAAU,EAAE,CAAC;IAC1B,IAAW,MAAM,KAAa,OAAO,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAElE,gEAAgE;IAChD,MAAM,CAAC,KAAU,EAAE,SAAyB,EAAE,QAAsB;QAClF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,QAAQ,EAAE,CAAC;IACb,CAAC;IAEM,IAAI,GAAyC,IAAI,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAClE,MAAM,GAA4C,IAAI,EAAE,CAAC;IACzD,GAAG,GAAsC,IAAI,EAAE,CAAC;CACxD;AAED,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,QAA2B,CAAC;IAChC,IAAI,OAA6B,CAAC;IAClC,IAAI,WAAkC,CAAC;IACvC,IAAI,GAAsB,CAAC;IAC3B,IAAI,GAAiB,CAAC;IAEtB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,gBAAgB,CAAC,QAAQ,EAAS,CAAC;QACjE,OAAO,GAAG,EAAE,SAAS,EAAE,EAAE,aAAa,EAAE,oBAAoB,EAAE,EAAS,CAAC;QACxE,WAAW,GAAG;YACZ,gBAAgB,EAAE,IAAI;YACtB,MAAM,EAAE,GAAG;YACX,MAAM,EAAE;gBACN,IAAI,EAAE,EAAE;aACT;SACK,CAAC;QACT,GAAG,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,mBAAmB,EAAS,CAAC;QACnD,GAAG,GAAG,IAAI,YAAY,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,UAAU;QACV,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC;QACpC,MAAM,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QACzD,WAAW,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,CAAC;QAE1C,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7D,SAAS;QACT,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEvE,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC;QAC3C,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAC/D,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,UAAU;QACV,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC;QACpC,MAAM,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QACzD,WAAW,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,CAAC;QAC1C,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,eAAe,CAAC;QAEnC,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7D,SAAS;QACT,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;QAEzE,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC;QAC3C,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAC/D,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,UAAU;QACV,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC;QACpC,WAAW,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,CAAC;QAC1C,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC;QAE7B,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7D,SAAS;QACT,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC;QAEhE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,UAAU;QACV,MAAM,YAAY,GAAG,YAAY,EAAE,CAAC;QACpC,MAAM,gBAAgB,GAAG,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QACzD,WAAW,CAAC,MAAM,CAAC,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAExD,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAC7D,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;QAE1D,SAAS;QACT,MAAM,cAAc,GAAG,GAAG,CAAC,MAAM,CAAC;QAClC,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC;QAC3C,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAC/D,MAAM,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,UAAU;QACV,WAAW,CAAC,MAAM,CAAC,OAAO,GAAG,YAAY,EAAE,CAAC;QAC5C,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC,sBAAsB,CAAC;QAE1C,MAAM;QACN,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE7D,SAAS;QACT,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,iBAAiB,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC;IACzE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,SAAS,YAAY,CAAC,WAAW,GAAG,IAAI;IACtC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,MAAM,GAAG,GAAG,EAAS,CAAC;IAEtB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,MAAM,GAAG,WAAW,EAAE,CAAC;QAChD,GAAG,CAAC,YAAY,WAAW,EAAE,CAAC,GAAG,SAAS,WAAW,EAAE,CAAC;QACxD,WAAW,EAAE,CAAC;IAChB,CAAC;IAED,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;AAC7B,CAAC","sourcesContent":["import { expect } from \"chai\";\r\nimport { SinonStub, stub } from \"sinon\";\r\nimport { Readable, Writable } from \"stream\";\r\nimport { HttpServerRequest, HttpServerResponse, RpcRequestFulfillment, RpcRequestStatus, SerializedRpcRequest, WebAppRpcProtocol } from \"@itwin/core-common\";\r\nimport { sendResponse } from \"../../rpc/web/response\";\r\nimport { brotliDecompressSync, unzipSync } from \"node:zlib\";\r\n\r\n/* eslint-disable @typescript-eslint/no-deprecated */\r\n\r\nclass StubResponse extends Writable implements HttpServerResponse {\r\n  public chunks: any[] = [];\r\n  public get buffer(): Buffer { return Buffer.concat(this.chunks); }\r\n\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  public override _write(chunk: any, _encoding: BufferEncoding, callback: VoidFunction): void {\r\n    this.chunks.push(chunk);\r\n    callback();\r\n  }\r\n\r\n  public send: SinonStub<any[], HttpServerResponse> = stub().returns(this);\r\n  public status: SinonStub<[number], HttpServerResponse> = stub();\r\n  public set: SinonStub<[string, string], void> = stub();\r\n}\r\n\r\ndescribe(\"sendResponse\", () => {\r\n  let protocol: WebAppRpcProtocol;\r\n  let request: SerializedRpcRequest;\r\n  let fulfillment: RpcRequestFulfillment;\r\n  let req: HttpServerRequest;\r\n  let res: StubResponse;\r\n\r\n  beforeEach(() => {\r\n    protocol = { getStatus: () => RpcRequestStatus.Resolved } as any;\r\n    request = { operation: { operationName: \"fake-RPC-operation\" } } as any;\r\n    fulfillment = {\r\n      allowCompression: true,\r\n      status: 200,\r\n      result: {\r\n        data: [],\r\n      },\r\n    } as any;\r\n    req = { header: () => \"gzip, deflate, br\" } as any;\r\n    res = new StubResponse();\r\n  });\r\n\r\n  it(\"should compress response using Brotli\", async () => {\r\n    // Arrange\r\n    const originalData = generateJson();\r\n    const originalDataSize = Buffer.byteLength(originalData);\r\n    fulfillment.result.objects = originalData;\r\n\r\n    // Act\r\n    await sendResponse(protocol, request, fulfillment, req, res);\r\n\r\n    // Assert\r\n    expect(res.set.calledWithExactly(\"Content-Encoding\", \"br\")).to.be.true;\r\n\r\n    const compressedData = res.send.getCall(0).args[0];\r\n    expect(compressedData).to.not.be.undefined;\r\n    expect(compressedData.length).to.be.lessThan(originalDataSize);\r\n    expect(brotliDecompressSync(compressedData).toString()).to.be.equal(originalData);\r\n  });\r\n\r\n  it(\"should compress response using Gzip if Brotli is not supported\", async () => {\r\n    // Arrange\r\n    const originalData = generateJson();\r\n    const originalDataSize = Buffer.byteLength(originalData);\r\n    fulfillment.result.objects = originalData;\r\n    req.header = () => \"gzip, deflate\";\r\n\r\n    // Act\r\n    await sendResponse(protocol, request, fulfillment, req, res);\r\n\r\n    // Assert\r\n    expect(res.set.calledWithExactly(\"Content-Encoding\", \"gzip\")).to.be.true;\r\n\r\n    const compressedData = res.send.getCall(0).args[0];\r\n    expect(compressedData).to.not.be.undefined;\r\n    expect(compressedData.length).to.be.lessThan(originalDataSize);\r\n    expect(unzipSync(compressedData).toString()).to.be.equal(originalData);\r\n  });\r\n\r\n  it(\"should not compress if Brotli and Gzip is not supported\", async () => {\r\n    // Arrange\r\n    const originalData = generateJson();\r\n    fulfillment.result.objects = originalData;\r\n    req.header = () => undefined;\r\n\r\n    // Act\r\n    await sendResponse(protocol, request, fulfillment, req, res);\r\n\r\n    // Assert\r\n    expect(res.set.calledWithMatch(\"Content-Encoding\")).to.be.false;\r\n\r\n    expect(res.send.getCall(0).args[0]).to.be.equal(originalData);\r\n  });\r\n\r\n  it(\"should compress stream\", async () => {\r\n    // Arrange\r\n    const originalData = generateJson();\r\n    const originalDataSize = Buffer.byteLength(originalData);\r\n    fulfillment.result.stream = Readable.from(originalData);\r\n\r\n    // Act\r\n    await sendResponse(protocol, request, fulfillment, req, res);\r\n    await new Promise((resolve) => res.on(\"finish\", resolve));\r\n\r\n    // Assert\r\n    const compressedData = res.buffer;\r\n    expect(compressedData).to.not.be.undefined;\r\n    expect(compressedData.length).to.be.lessThan(originalDataSize);\r\n    expect(brotliDecompressSync(compressedData).toString()).to.be.equal(originalData);\r\n  });\r\n\r\n  it(\"should support Accept-Encoding header value format without spaces between values\", async () => {\r\n    // Arrange\r\n    fulfillment.result.objects = generateJson();\r\n    req.header = () => \"gzip,deflate,br,zstd\";\r\n\r\n    // Act\r\n    await sendResponse(protocol, request, fulfillment, req, res);\r\n\r\n    // Assert\r\n    expect(res.set.calledWithExactly(\"Content-Encoding\", \"br\")).to.be.true;\r\n  });\r\n});\r\n\r\nfunction generateJson(minimumSize = 2000): string {\r\n  let propertyIdx = 0;\r\n  const obj = {} as any;\r\n\r\n  while (JSON.stringify(obj).length < minimumSize) {\r\n    obj[`property-${propertyIdx}`] = `value-${propertyIdx}`;\r\n    propertyIdx++;\r\n  }\r\n\r\n  return JSON.stringify(obj);\r\n}\r\n"]}

package/lib/esm/test/schema/ClassRegistry.test.js

@@ -5,6 +5,7 @@
 import { assert, expect } from "chai";
 import * as sinon from "sinon";
 import * as path from "path";
+import { withEditTxn } from "../../EditTxn";
 import { BisCodeSpec, Code, ConcreteEntityTypes, ECJsNames, RelatedElement, SchemaState, } from "@itwin/core-common";
 import { DefinitionElement, DefinitionModel, ElementRefersToElements, Entity, EntityReferences, IModelDb, IModelJsFs, RepositoryLink, Schema, SpatialViewDefinition, StandaloneDb, UrlLink, ViewDefinition3d, } from "../../core-backend";
 import { IModelTestUtils } from "../IModelTestUtils";
@@ -95,50 +96,50 @@ describe("Class Registry - getRootMetaData", () => {
         imodel = StandaloneDb.openFile(testFileName, OpenMode.ReadWrite);
         assert.exists(imodel);
         await imodel.importSchemaStrings([
-            `<?xml version="1.0" encoding="UTF-8"?>
-        <ECSchema schemaName="TestSchema1" alias="ts1" version="01.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
-          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
-          <ECSchemaReference name="CoreCustomAttributes" version="01.00.03" alias="CoreCA"/>
-
-          <ECEntityClass typeName="TestBase">
-            <BaseClass>bis:PhysicalElement</BaseClass>
-          </ECEntityClass>
-
-          <ECEntityClass typeName="TestDerived">
-            <BaseClass>TestBase</BaseClass>
-          </ECEntityClass>
-
-          <ECEntityClass typeName="ITestMixinForAspectsBase">
-            <ECCustomAttributes>
-              <IsMixin xmlns="CoreCustomAttributes.01.00.03">
-                <AppliesToEntityClass>bis:ElementUniqueAspect</AppliesToEntityClass>
-              </IsMixin>
-            </ECCustomAttributes>
-          </ECEntityClass>
-
-          <ECEntityClass typeName="ITestMixinForAspectsDerived">
-            <BaseClass>ITestMixinForAspectsBase</BaseClass>
-            <ECCustomAttributes>
-              <IsMixin xmlns="CoreCustomAttributes.01.00.03">
-                <AppliesToEntityClass>bis:ElementUniqueAspect</AppliesToEntityClass>
-              </IsMixin>
-            </ECCustomAttributes>
-          </ECEntityClass>
-
-          <ECEntityClass typeName="ITestMixinForElements" modifier="Abstract">
-            <ECCustomAttributes>
-              <IsMixin xmlns="CoreCustomAttributes.01.00.03">
-                <AppliesToEntityClass>bis:Element</AppliesToEntityClass>
-              </IsMixin>
-            </ECCustomAttributes>
-          </ECEntityClass>
-
-          <ECEntityClass typeName="TestMixedInAndDerived">
-            <BaseClass>TestBase</BaseClass>
-            <BaseClass>ITestMixinForElements</BaseClass>
-          </ECEntityClass>
-
-        </ECSchema>
+            `<?xml version="1.0" encoding="UTF-8"?>
+        <ECSchema schemaName="TestSchema1" alias="ts1" version="01.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
+          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
+          <ECSchemaReference name="CoreCustomAttributes" version="01.00.03" alias="CoreCA"/>
+
+          <ECEntityClass typeName="TestBase">
+            <BaseClass>bis:PhysicalElement</BaseClass>
+          </ECEntityClass>
+
+          <ECEntityClass typeName="TestDerived">
+            <BaseClass>TestBase</BaseClass>
+          </ECEntityClass>
+
+          <ECEntityClass typeName="ITestMixinForAspectsBase">
+            <ECCustomAttributes>
+              <IsMixin xmlns="CoreCustomAttributes.01.00.03">
+                <AppliesToEntityClass>bis:ElementUniqueAspect</AppliesToEntityClass>
+              </IsMixin>
+            </ECCustomAttributes>
+          </ECEntityClass>
+
+          <ECEntityClass typeName="ITestMixinForAspectsDerived">
+            <BaseClass>ITestMixinForAspectsBase</BaseClass>
+            <ECCustomAttributes>
+              <IsMixin xmlns="CoreCustomAttributes.01.00.03">
+                <AppliesToEntityClass>bis:ElementUniqueAspect</AppliesToEntityClass>
+              </IsMixin>
+            </ECCustomAttributes>
+          </ECEntityClass>
+
+          <ECEntityClass typeName="ITestMixinForElements" modifier="Abstract">
+            <ECCustomAttributes>
+              <IsMixin xmlns="CoreCustomAttributes.01.00.03">
+                <AppliesToEntityClass>bis:Element</AppliesToEntityClass>
+              </IsMixin>
+            </ECCustomAttributes>
+          </ECEntityClass>
+
+          <ECEntityClass typeName="TestMixedInAndDerived">
+            <BaseClass>TestBase</BaseClass>
+            <BaseClass>ITestMixinForElements</BaseClass>
+          </ECEntityClass>
+
+        </ECSchema>
       `,
         ]); // will throw an exception if import fails
     });
@@ -255,35 +256,35 @@ describe("Class Registry - generated classes", () => {
     }
     it("Should provide correct schemas and full-names on generated classes", async () => {
         await imodel.importSchemaStrings([
-            `<?xml version="1.0" encoding="UTF-8"?>
-        <ECSchema schemaName="CustomA" alias="custA" version="1.0.0"
-          xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
-          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
-          <ECSchemaReference name="Functional" version="1.00" alias="func"/>
-          <ECEntityClass typeName="NonGeneratedElement">
-            <BaseClass>bis:DefinitionElement</BaseClass>
-          </ECEntityClass>
-          <ECEntityClass typeName="GeneratedElement">
-            <BaseClass>NonGeneratedElement</BaseClass>
-          </ECEntityClass>
-          <ECEntityClass typeName="ElementC">
-            <BaseClass>GeneratedElement</BaseClass>
-          </ECEntityClass>
-        </ECSchema>
+            `<?xml version="1.0" encoding="UTF-8"?>
+        <ECSchema schemaName="CustomA" alias="custA" version="1.0.0"
+          xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
+          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
+          <ECSchemaReference name="Functional" version="1.00" alias="func"/>
+          <ECEntityClass typeName="NonGeneratedElement">
+            <BaseClass>bis:DefinitionElement</BaseClass>
+          </ECEntityClass>
+          <ECEntityClass typeName="GeneratedElement">
+            <BaseClass>NonGeneratedElement</BaseClass>
+          </ECEntityClass>
+          <ECEntityClass typeName="ElementC">
+            <BaseClass>GeneratedElement</BaseClass>
+          </ECEntityClass>
+        </ECSchema>
       `,
-            `<?xml version="1.0" encoding="UTF-8"?>
-        <ECSchema schemaName="CustomB" alias="custB" version="1.0.0"
-          xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
-          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
-          <ECSchemaReference name="Functional" version="1.00" alias="func"/>
-          <ECSchemaReference name="CustomA" version="01.00" alias="custA"/>
-          <ECEntityClass typeName="DummyElement">
-            <BaseClass>bis:DefinitionElement</BaseClass>
-          </ECEntityClass>
-          <ECEntityClass typeName="ErrorElement">
-            <BaseClass>custA:ElementC</BaseClass>
-          </ECEntityClass>
-        </ECSchema>
+            `<?xml version="1.0" encoding="UTF-8"?>
+        <ECSchema schemaName="CustomB" alias="custB" version="1.0.0"
+          xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
+          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
+          <ECSchemaReference name="Functional" version="1.00" alias="func"/>
+          <ECSchemaReference name="CustomA" version="01.00" alias="custA"/>
+          <ECEntityClass typeName="DummyElement">
+            <BaseClass>bis:DefinitionElement</BaseClass>
+          </ECEntityClass>
+          <ECEntityClass typeName="ErrorElement">
+            <BaseClass>custA:ElementC</BaseClass>
+          </ECEntityClass>
+        </ECSchema>
       `,
         ]);
         class CustomASchema extends Schema {
@@ -360,12 +361,12 @@ describe("Class Registry - generated classes", () => {
                 super(props, imodel);
             }
         }
-        const testEntityId = imodel.elements.insertElement({
+        const testEntityId = withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: "TestGeneratedClasses:TestEntity",
             prop: "sample-value",
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
-        });
+        }));
         const elemWithNavProp = new GeneratedTestElementWithNavProp({
             classFullName: "TestGeneratedClasses:TestElementWithNavProp",
             navProp: {
@@ -395,12 +396,12 @@ describe("Class Registry - generated classes", () => {
                 super(props, imodel);
             }
         }
-        const testEntityId = imodel.elements.insertElement({
+        const testEntityId = withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: "TestGeneratedClasses:TestEntity",
             prop: "sample-value",
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
-        });
+        }));
         const elemWithNavProp = new GeneratedTestElementWithNavProp({
             classFullName: "TestGeneratedClasses:TestElementWithNavProp",
             navProp: new RelatedElement({
@@ -426,19 +427,19 @@ describe("Class Registry - generated classes", () => {
             elemWithNavProp.parent && EntityReferences.fromEntityType(elemWithNavProp.parent?.id, ConcreteEntityTypes.Element),
             EntityReferences.fromEntityType(testEntityId, ConcreteEntityTypes.Element),
         ].filter((x) => x !== undefined));
-        const modelTestEntityIds = new Array(2).fill(undefined).map((_, index) => imodel.elements.insertElement({
+        const modelTestEntityIds = new Array(2).fill(undefined).map((_, index) => withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: "TestGeneratedClasses:TestEntity",
             prop: `model-value-${index}`,
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
-        }));
+        })));
         // eslint-disable-next-line @typescript-eslint/naming-convention
         const GeneratedTestAspectWithNavProp = imodel.getJsClass("TestGeneratedClasses:TestAspectWithNavProp");
-        const aspectWithNavPropId = imodel.elements.insertAspect({
+        const aspectWithNavPropId = withEditTxn(imodel, (txn) => txn.insertAspect({
             classFullName: GeneratedTestAspectWithNavProp.classFullName,
             navProp: { id: modelTestEntityIds[0], relClassName: "TestGeneratedClasses:NonElemRel" },
             element: { id: modelTestEntityIds[1] },
-        });
+        }));
         class GeneratedTestModelWithNavProp extends imodel.getJsClass("TestGeneratedClasses:TestModelWithNavProp") {
             constructor(props) {
                 super(props, imodel);
@@ -453,7 +454,7 @@ describe("Class Registry - generated classes", () => {
             // removed due to a bug
             // relNavProp: { id: relWithNavPropId, relClassName: "TestGeneratedClasses:ModelToRelNavRel" },
         });
-        const modelWithNavPropId = modelWithNavProp.insert();
+        const modelWithNavPropId = withEditTxn(imodel, (txn) => modelWithNavProp.insert(txn));
         expect([...modelWithNavProp.getReferenceIds()]).to.have.members([
             EntityReferences.fromEntityType(modelTestEntityIds[1], ConcreteEntityTypes.Element),
             EntityReferences.fromEntityType(IModelDb.dictionaryId, ConcreteEntityTypes.Model),
@@ -462,12 +463,12 @@ describe("Class Registry - generated classes", () => {
             // ignoring this one, because there seems to be a bug when specifying a relationship instance as a nav prop
             // EntityReferences.fromEntityType(relWithNavPropId, ConcreteEntityTypes.Relationship),
         ].filter((x) => x !== undefined));
-        const relTestEntityIds = new Array(3).fill(undefined).map((_, index) => imodel.elements.insertElement({
+        const relTestEntityIds = new Array(3).fill(undefined).map((_, index) => withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: "TestGeneratedClasses:TestEntity",
             prop: `rel-value-${index}`,
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
-        }));
+        })));
         // eslint-disable-next-line @typescript-eslint/naming-convention
         const GeneratedLinkTableRelWithNavProp = imodel.getJsClass("TestGeneratedClasses:LinkTableRelWithNavProp");
         const relWithNavProp = new GeneratedLinkTableRelWithNavProp({
@@ -487,7 +488,7 @@ describe("Class Registry - generated classes", () => {
                 relClassName: "TestGeneratedClasses:LinkTableRelToAspectNavRel",
             },
         }, imodel);
-        const _relWithNavPropId = relWithNavProp.insert();
+        const _relWithNavPropId = withEditTxn(imodel, (txn) => relWithNavProp.insert(txn));
         expect([...relWithNavProp.getReferenceIds()]).to.have.members([
             ...relTestEntityIds.map((id) => EntityReferences.fromEntityType(id, ConcreteEntityTypes.Element)),
             EntityReferences.fromEntityType(modelWithNavPropId, ConcreteEntityTypes.Model),
@@ -517,18 +518,18 @@ describe("Class Registry - generated classes", () => {
                 super(props, imodel);
             }
         }
-        const testEntity1Id = imodel.elements.insertElement({
+        const testEntity1Id = withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: "TestGeneratedClasses:TestEntity",
             prop: "sample-value-1",
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
-        });
-        const testEntity2Id = imodel.elements.insertElement({
+        }));
+        const testEntity2Id = withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: "TestGeneratedClasses:TestEntity",
             prop: "sample-value-2",
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
-        });
+        }));
         const elemWithNavProp = new ActualTestElementWithNavProp({
             classFullName: TestElementWithNavProp.classFullName,
             navProp: {
@@ -615,19 +616,19 @@ describe("Class Registry - generated classes", () => {
         assert.isTrue(ActualDerived4.prototype.hasOwnProperty("collectReferenceIds")); // manually implements so it should have the method
         assert.isFalse(ActualDerived5.prototype.hasOwnProperty("collectReferenceIds")); // ancestor is non-generated so it shouldn't get the automatic impl
         assert.isFalse(ActualDerived6.prototype.hasOwnProperty("collectReferenceIds")); // ancestor is non-generated so it shouldn't get the automatic impl
-        const testEntity1Id = imodel.elements.insertElement({
+        const testEntity1Id = withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: "TestGeneratedClasses:Derived6",
             prop: "sample-value-1",
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
-        });
-        const testEntity2Id = imodel.elements.insertElement({
+        }));
+        const testEntity2Id = withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: "TestGeneratedClasses:TestEntity",
             prop: "sample-value-2",
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
-        });
-        const derived6Id = imodel.elements.insertElement({
+        }));
+        const derived6Id = withEditTxn(imodel, (txn) => txn.insertElement({
             classFullName: Derived6.classFullName,
             model: IModelDb.dictionaryId,
             code: Code.createEmpty(),
@@ -639,7 +640,7 @@ describe("Class Registry - generated classes", () => {
                 id: testEntity2Id,
                 relClassName: "TestGeneratedClasses:DerivedElemRel",
             },
-        });
+        }));
         const derived6 = imodel.elements.getElement(derived6Id);
         /** it is not possible to make a spy of an already existing spy, so lazy try making one
          * this is necessary since due to prototypes, some "methods" we listen to are actually the same
@@ -737,36 +738,36 @@ describe("Global state of ClassRegistry", () => {
     });
     it("registering a class in different imodels should not affect each other", async () => {
         await imodel1.importSchemaStrings([
-            `<?xml version="1.0" encoding="UTF-8"?>
-        <ECSchema schemaName="TestSchema" alias="ts" version="01.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
-          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
-
-          <ECEntityClass typeName="TestClass">
-            <BaseClass>bis:PhysicalElement</BaseClass>
-            <ECProperty propertyName="foo" typeName="string"/>
-            <ECNavigationProperty propertyName="RelatedTestClass" relationshipName="MyRelationship" direction="backward"/>
-          </ECEntityClass>
-
-          <ECRelationshipClass typeName="MyRelationship" modifier="None" strength="embedding">
-            <Source multiplicity="(0..1)" roleLabel="has" polymorphic="false">
-                <Class class="TestClass"/>
-            </Source>
-            <Target multiplicity="(0..*)" roleLabel="has" polymorphic="false">
-                <Class class="TestClass"/>
-            </Target>
-          </ECRelationshipClass>
-        </ECSchema>
+            `<?xml version="1.0" encoding="UTF-8"?>
+        <ECSchema schemaName="TestSchema" alias="ts" version="01.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
+          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
+
+          <ECEntityClass typeName="TestClass">
+            <BaseClass>bis:PhysicalElement</BaseClass>
+            <ECProperty propertyName="foo" typeName="string"/>
+            <ECNavigationProperty propertyName="RelatedTestClass" relationshipName="MyRelationship" direction="backward"/>
+          </ECEntityClass>
+
+          <ECRelationshipClass typeName="MyRelationship" modifier="None" strength="embedding">
+            <Source multiplicity="(0..1)" roleLabel="has" polymorphic="false">
+                <Class class="TestClass"/>
+            </Source>
+            <Target multiplicity="(0..*)" roleLabel="has" polymorphic="false">
+                <Class class="TestClass"/>
+            </Target>
+          </ECRelationshipClass>
+        </ECSchema>
       `,
         ]);
         await imodel2.importSchemaStrings([
-            `<?xml version="1.0" encoding="UTF-8"?>
-        <ECSchema schemaName="TestSchema" alias="ts" version="01.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
-          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
-          <ECEntityClass typeName="TestClass">
-            <BaseClass>bis:PhysicalElement</BaseClass>
-            <ECProperty propertyName="bar" typeName="string"/>
-          </ECEntityClass>
-        </ECSchema>
+            `<?xml version="1.0" encoding="UTF-8"?>
+        <ECSchema schemaName="TestSchema" alias="ts" version="01.00" xmlns="http://www.bentley.com/schemas/Bentley.ECXML.3.1">
+          <ECSchemaReference name="BisCore" version="01.00" alias="bis"/>
+          <ECEntityClass typeName="TestClass">
+            <BaseClass>bis:PhysicalElement</BaseClass>
+            <ECProperty propertyName="bar" typeName="string"/>
+          </ECEntityClass>
+        </ECSchema>
       `,
         ]);
         const testClass1 = imodel1.getJsClass("TestSchema:TestClass");