@itwin/core-backend 5.8.2 → 5.9.0-dev.10

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\";\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"]}
+{"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"]}

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");