@itwin/core-backend 5.1.0-dev.9 → 5.1.1

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,CAAC,CAAC,gEAAgE;QAClH,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,CAAC;QAExD,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)); // 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));\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,CAAC,CAAC,gEAAgE;QAClH,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,CAAC;QAExD,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)); // 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));\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

@@ -6,14 +6,14 @@ import { assert, expect } from "chai";
 import * as sinon from "sinon";
 import * as path from "path";
 import { BisCodeSpec, Code, ConcreteEntityTypes, ECJsNames, RelatedElement, SchemaState, } from "@itwin/core-common";
-import { DefinitionElement, DefinitionModel, ElementRefersToElements, EntityReferences, IModelDb, IModelJsFs, RepositoryLink, Schema, SpatialViewDefinition, StandaloneDb, UrlLink, ViewDefinition3d, } from "../../core-backend";
+import { DefinitionElement, DefinitionModel, ElementRefersToElements, Entity, EntityReferences, IModelDb, IModelJsFs, RepositoryLink, Schema, SpatialViewDefinition, StandaloneDb, UrlLink, ViewDefinition3d, } from "../../core-backend";
 import { IModelTestUtils } from "../IModelTestUtils";
 import { KnownTestLocations } from "../KnownTestLocations";
 import { Element } from "../../Element";
 import { Schemas } from "../../Schema";
 import { ClassRegistry } from "../../ClassRegistry";
 import { OpenMode } from "@itwin/core-bentley";
-import { EntityClass } from "@itwin/ecschema-metadata";
+import { EntityClass, SchemaItemKey, SchemaKey } from "@itwin/ecschema-metadata";
 describe("Class Registry", () => {
     let imodel;
     before(() => {
@@ -95,50 +95,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
     });
@@ -250,6 +250,108 @@ describe("Class Registry - generated classes", () => {
     class Derived6 extends Derived5 {
         static get className() { return "Derived6"; }
     }
+    class NonGeneratedElement extends DefinitionElement {
+        static get className() { return "NonGeneratedElement"; }
+    }
+    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="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 {
+            static get schemaName() { return "CustomA"; }
+            static get classes() {
+                return [NonGeneratedElement];
+            }
+            static registerSchema() {
+                if (this !== Schemas.getRegisteredSchema(this.schemaName)) {
+                    Schemas.unregisterSchema(this.schemaName);
+                    Schemas.registerSchema(this);
+                    for (const ecClass of this.classes) {
+                        ClassRegistry.register(ecClass, this);
+                    }
+                }
+            }
+            static unregisterSchema() {
+                Schemas.unregisterSchema(this.schemaName);
+            }
+        }
+        CustomASchema.registerSchema();
+        class DummyElement extends DefinitionElement {
+            static get className() { return "DummyElement"; }
+        }
+        class CustomBSchema extends Schema {
+            static get schemaName() { return "CustomB"; }
+            static get classes() {
+                return [DummyElement];
+            }
+            static registerSchema() {
+                if (this !== Schemas.getRegisteredSchema(this.schemaName)) {
+                    Schemas.unregisterSchema(this.schemaName);
+                    Schemas.registerSchema(this);
+                    for (const ecClass of this.classes) {
+                        ClassRegistry.register(ecClass, this);
+                    }
+                }
+            }
+            static unregisterSchema() {
+                Schemas.unregisterSchema(this.schemaName);
+            }
+        }
+        CustomBSchema.registerSchema();
+        const nonGeneratedElement = imodel.getJsClass("CustomA:NonGeneratedElement");
+        assert.equal(nonGeneratedElement.classFullName, "CustomA:NonGeneratedElement");
+        const generatedElement = imodel.getJsClass("CustomA:GeneratedElement");
+        assert.equal(generatedElement.classFullName, "CustomA:GeneratedElement");
+        const elementC = imodel.getJsClass("CustomA:ElementC");
+        assert.equal(elementC.classFullName, "CustomA:ElementC");
+        const dummyElement = imodel.getJsClass("CustomB:DummyElement");
+        assert.equal(dummyElement.classFullName, "CustomB:DummyElement");
+        //This is the class that had the wrong schema
+        const errorElement = imodel.getJsClass("CustomB:ErrorElement");
+        assert.equal(errorElement.classFullName, "CustomB:ErrorElement");
+        assert.isTrue(errorElement.schemaItemKey.matches(new SchemaItemKey("ErrorElement", new SchemaKey("CustomB", 1, 0, 0))));
+        assert.equal(errorElement.className, "ErrorElement");
+        assert.equal(errorElement.schema.schemaName, "CustomB");
+        // Create an instance of ErrorElement
+        const errorElementInstance = Entity.instantiate(errorElement, { classFullName: errorElement.classFullName }, imodel);
+        assert.exists(errorElementInstance);
+        assert.instanceOf(errorElementInstance, errorElement);
+        assert.instanceOf(errorElementInstance, Entity);
+        assert.equal(errorElementInstance.className, "ErrorElement");
+        assert.equal(errorElementInstance.schemaName, "CustomB");
+        const metadata = await errorElementInstance.getMetaData();
+        assert.exists(metadata);
+        assert.equal(metadata.fullName, "CustomB.ErrorElement");
+    });
     // if a single inherited class is not generated, the entire hierarchy is considered not-generated
     it("should only generate automatic collectReferenceIds implementations for generated classes", async () => {
         await imodel.importSchemas([testSchemaPath]); // will throw an exception if import fails
@@ -635,36 +737,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");