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

package/lib/cjs/internal/HubMock.js.map

@@ -1 +1 @@
-{"version":3,"file":"HubMock.js","sourceRoot":"","sources":["../../../src/internal/HubMock.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;AAE/F,+BAA4B;AAC5B,sDAAuD;AASvD,4DAA+I;AAC/I,8CAA2C;AAC3C,8CAA2C;AAC3C,0CAAuC;AAEvC,uCAA0E;AAC1E,0DAAuD;AACvD,6BAA6B;AAE7B,SAAS,UAAU,CAAC,GAAuB;IACzC,IAAI,SAAS,KAAK,GAAG;QACnB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;AAClD,CAAC;AAED,SAAS,UAAU,CAAC,IAA6E;IAC/F,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AACD,MAAM,cAAc,GAAmB;IACrC,UAAU,EAAE,CAAC,UAA2B,EAAE,EAAE;QAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,mCAAgB,CAAC,oBAAoB,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK,MAAM,CAAC,CAAC;QAC9H,UAAU,CAAC,EAAE,GAAG,UAAU,EAAE,UAAU,EAAE,CAAC,CAAA;QACzC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,YAAY,EAAE,CAAC,OAAwB,EAAE,EAAE;QACzC,UAAU,CAAC,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,UAAU,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IACvE,CAAC;CACF,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AACH,MAAa,OAAO;IACV,MAAM,CAAC,QAAQ,CAA2B;IAC1C,MAAM,CAAC,IAAI,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC1C,MAAM,CAAC,cAAc,CAA+B;IACpD,MAAM,CAAC,QAAQ,CAAyB;IAEhD,oEAAoE;IAC7D,MAAM,KAAK,OAAO,KAAK,OAAO,SAAS,KAAK,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC5D,MAAM,KAAK,OAAO;QACvB,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;;;OAIG;IACI,MAAM,CAAC,OAAO,CAAC,QAAsB,EAAE,SAAiB;QAC7D,IAAI,IAAI,CAAC,OAAO;YACd,MAAM,IAAI,KAAK,CAAC,0JAA0J,CAAC,CAAC;QAE9K,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,IAAI,CAAC,QAAQ,GAAG,IAAA,WAAI,EAAC,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACrD,uBAAU,CAAC,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,uBAAU,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,uBAAU,CAAC,uBAAa,CAAC,EAAE,CAAC;QAElD,uBAAU,CAAC,uBAAa,CAAC,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,CAAC,QAAQ,GAAG,mBAAI,CAAC,WAAW,EAAE,CAAC,CAAC,mDAAmD;QAE1F,uCAAmB,CAAC,yBAAe,CAAC,GAAG,cAAc,CAAC;IACxD,CAAC;IAED;;OAEG;IACI,MAAM,CAAC,QAAQ;QACpB,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS;YAC7B,OAAO;QAET,uCAAmB,CAAC,yBAAe,CAAC,GAAG,SAAS,CAAC;QAEjD,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC;QAC7B,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI;YACzB,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAEnB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,uBAAU,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,uBAAU,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,uBAAU,CAAC,uBAAa,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/C,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;IAC5B,CAAC;IAEM,MAAM,CAAC,YAAY,CAAC,QAAoB;QAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,CAAC,GAAG;YACN,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,cAAc,CAAC,CAAC;QAClE,OAAO,GAAG,CAAC;IACb,CAAC;IAED,4CAA4C;IACrC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAyB;QAC3D,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1B,MAAM,KAAK,GAAG,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,mBAAI,CAAC,WAAW,EAAE,EAAE,CAAC;QACvD,MAAM,IAAI,GAAG,IAAI,mBAAQ,CAAC,IAAA,WAAI,EAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;QACtE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACpC,OAAO,KAAK,CAAC,QAAQ,CAAC;IACxB,CAAC;IAED,4CAA4C;IACrC,MAAM,CAAC,OAAO,CAAC,QAAoB;QACxC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC;QACtC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAED,wEAAwE;IAEjE,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,GAA0C;QACzF,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC3E,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAAC,GAAiB;QACpD,OAAO,CAAC,SAAS,KAAK,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACzI,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,GAA6C;QACvF,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QAC5B,IAAI,OAAO,CAAC,OAAO;YACjB,OAAO,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QAEpC,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;QACxC,IAAI,IAAI;YACN,OAAO,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAEpC,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;QACtC,IAAI,WAAW;YACb,OAAO,GAAG,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;QAE3C,OAAO,GAAG,CAAC,kBAAkB,EAAE,CAAC;IAClC,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,GAAgB;QACrD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,kBAAkB,EAAE,CAAC;IAC9D,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,GAAa;QAC/C,OAAO,GAAG,CAAC,WAAW,IAAI,MAAM,uBAAU,CAAC,cAAc,EAAE,CAAC;IAC9D,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAgB;QACpD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;IACtE,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,GAA6B;QACrE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,WAAW,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC;IAChG,CAAC;IAED,gHAAgH;IACzG,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,GAAmB;QACtD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC7E,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAyB;QAC7D,MAAM,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;QAE/I,IAAI,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,uBAAU,CAAC,SAAS,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC;YACtE,IAAI,SAAS;gBACX,MAAM,OAAO,CAAC,qBAAqB,CAAC,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,cAAc,CAAC;IACxB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,GAA8B;QACnE,MAAM,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;QAE1H,IAAI,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,uBAAU,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACpH,MAAM,OAAO,CAAC,qBAAqB,CAAC,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;QACvE,CAAC;QAED,OAAO,cAAc,CAAC;IACxB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,GAAiB;QAClD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC;IAC9F,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAA6C;QAC/E,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAyD;QACzF,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAC1E,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAoB;QACxD,OAAO;YACL,WAAW,EAAE,MAAM;YACnB,QAAQ,EAAE,MAAM;YAChB,WAAW,EAAE,mBAAI,CAAC,WAAW,EAAE;YAC/B,MAAM,EAAE,GAAG,GAAG,CAAC,SAAS,CAAC,KAAK,IAAI,CAAC,MAAM;YACzC,WAAW,EAAE,MAAM;YACnB,MAAM,EAAE,IAAI;YACZ,UAAU,EAAE,GAAG;SACW,CAAC;IAC/B,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAmB;QACrD,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,GAAG,CAAC,eAAe,CAAC,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,cAAc,EAAE,GAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAClH,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,IAAoB;QACpD,OAAO,EAAE,CAAC;IACZ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAmB,EAAE,KAAc;QAClE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC3D,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAkB;QACtD,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;YACxB,IAAI,QAAQ,CAAC,OAAO,KAAK,GAAG,CAAC,OAAO,IAAI,QAAQ,CAAC,UAAU,KAAK,GAAG,CAAC,UAAU;gBAC5E,OAAO,QAAQ,CAAC,QAAQ,CAAC;QAC7B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAA0C;QACzE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,gBAAkC,EAAE,SAAiB;QAC9F,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACpC,IAAI,QAAQ,GAAG,KAAK,CAAC;YAErB,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,EAAE;gBACrC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC5D,IAAI,CAAC,QAAQ,IAAI,gBAAgB,CAAC,eAAe,EAAE,SAAS,CAAC,KAAK,kCAAc,CAAC,KAAK,EAAE,CAAC;oBACvF,QAAQ,GAAG,IAAI,CAAC;oBAChB,MAAM,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC,CAAC;YAEF,YAAY,CAAC,CAAC,CAAC,CAAC;YAChB,UAAU,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtC,UAAU,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACvC,UAAU,CAAC,GAAG,EAAE;gBACd,YAAY,CAAC,CAAC,CAAC,CAAC;gBAChB,OAAO,CAAC,SAAS,CAAC,CAAC;YACrB,CAAC,EAAE,GAAG,CAAC,CAAC;QACV,CAAC,CAAC,CAAC;IACL,CAAC;;AA3NH,0BA4NC","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 { join } from \"path\";\nimport { Guid, GuidString } from \"@itwin/core-bentley\";\nimport {\n  ChangesetFileProps, ChangesetIndex, ChangesetIndexOrId, ChangesetProps, ChangesetRange, IModelVersion, LocalDirName,\n} from \"@itwin/core-common\";\nimport {\n  AcquireNewBriefcaseIdArg,\n  BackendHubAccess, BriefcaseDbArg, BriefcaseIdArg, ChangesetArg, CreateNewIModelProps, DownloadChangesetArg, DownloadChangesetRangeArg, IModelIdArg, IModelNameArg,\n  LockMap, LockProps, V2CheckpointAccessProps,\n} from \"../BackendHubAccess\";\nimport { CheckpointProps, DownloadRequest, MockCheckpoint, ProgressFunction, ProgressStatus, V2CheckpointManager } from \"../CheckpointManager\";\nimport { IModelHost } from \"../IModelHost\";\nimport { IModelJsFs } from \"../IModelJsFs\";\nimport { LocalHub } from \"../LocalHub\";\nimport { TokenArg } from \"../IModelDb\";\nimport { _getHubAccess, _mockCheckpoint, _setHubAccess } from \"./Symbols\";\nimport { BriefcaseManager } from \"../BriefcaseManager\";\nimport * as path from \"path\";\n\nfunction wasStarted(val: string | undefined): asserts val is string {\n  if (undefined === val)\n    throw new Error(\"Call HubMock.startup first\");\n}\n\nfunction doDownload(args: { iModelId: string, changeset: ChangesetIndexOrId, targetFile: string }) {\n  HubMock.findLocalHub(args.iModelId).downloadCheckpoint(args);\n}\nconst mockCheckpoint: MockCheckpoint = {\n  mockAttach: (checkpoint: CheckpointProps) => {\n    const targetFile = path.join(BriefcaseManager.getBriefcaseBasePath(checkpoint.iModelId), `${checkpoint.changeset.index}.bim`);\n    doDownload({ ...checkpoint, targetFile })\n    return targetFile;\n  },\n\n  mockDownload: (request: DownloadRequest) => {\n    doDownload({ ...request.checkpoint, targetFile: request.localFile });\n  }\n};\n\n/**\n * Mocks iModelHub for testing creating Briefcases, downloading checkpoints, and simulating multiple users pushing and pulling changesets, etc.\n *\n * Generally, tests for apis that *create or modify* iModels can and should be mocked. Otherwise they:\n * - create tremendous load on the test servers when they run on programmer's desktops and in CI jobs\n * - waste network and data center resources (i.e. $$$s),\n * - interfere with other tests running on the same or other systems, and\n * - (far worse) are the source of test flakiness outside of the api being tested.\n *\n * This class can be used to create tests that do not require authentication, are synchronous,\n * are guaranteed to be self-contained (i.e. do not interfere with other tests running at the same time or later), and do not fail for reasons outside\n * of the control of the test itself. As a bonus, in addition to making tests more reliable, mocking IModelHub generally makes tests run *much* faster.\n *\n * On the other hand, tests that expect to find an existing iModels, checkpoints, changesets, etc. in IModelHub cannot be mocked. In that case, those tests\n * should be careful to NOT modify the data, since doing so causes interference with other tests running simultaneously. These tests should be limited to\n * low level testing of the core apis only.\n *\n * To initialize HubMock, call [[startup]] at the beginning of your test, usually in `describe.before`. Thereafter, all access to iModelHub for an iModel will be\n * directed to a [[LocalHub]] - your test code does not change. After the test(s) complete, call [[shutdown]] (usually in `describe.after`) to stop mocking IModelHub and clean\n * up any resources used by the test(s). If you want to mock a single test, call [[startup]] as the first line and [[shutdown]] as the last. If you wish to run the\n * test against a \"real\" IModelHub, you can simply comment off the call [[startup]], though in that case you should make sure the name of your\n * iModel is unique so your test won't collide with other tests (iModel name uniqueness is not necessary for mocked tests.)\n *\n * Mocked tests must always start by creating a new iModel via [[IModelHost[_hubAccess].createNewIModel]] with a `version0` iModel.\n * They use mock (aka \"bogus\") credentials for `AccessTokens`, which is fine since [[HubMock]] never accesses resources outside the current\n * computer.\n *\n * @note Only one HubMock at a time, *running in a single process*, may be active. The comments above about multiple simultaneous tests refer to tests\n * running on different computers, or on a single computer in multiple processes. All of those scenarios are problematic without mocking.\n *\n * @internal\n */\nexport class HubMock {\n  private static mockRoot: LocalDirName | undefined;\n  private static hubs = new Map<string, LocalHub>();\n  private static _saveHubAccess: BackendHubAccess | undefined;\n  private static _iTwinId: GuidString | undefined;\n\n  /** Determine whether a test us currently being run under HubMock */\n  public static get isValid() { return undefined !== this.mockRoot; }\n  public static get iTwinId() {\n    wasStarted(this._iTwinId);\n    return this._iTwinId;\n  }\n\n  /**\n   * Begin mocking IModelHub access. After this call, all access to IModelHub will be directed to a [[LocalHub]].\n   * @param mockName a unique name (e.g. \"MyTest\") for this HubMock to disambiguate tests when more than one is simultaneously active.\n   * It is used to create a private directory used by the HubMock for a test. That directory is removed when [[shutdown]] is called.\n   */\n  public static startup(mockName: LocalDirName, outputDir: string) {\n    if (this.isValid)\n      throw new Error(\"Either a previous test did not call HubMock.shutdown() properly, or more than one test is simultaneously attempting to use HubMock, which is not allowed\");\n\n    this.hubs.clear();\n    this.mockRoot = join(outputDir, \"HubMock\", mockName);\n    IModelJsFs.recursiveMkDirSync(this.mockRoot);\n    IModelJsFs.purgeDirSync(this.mockRoot);\n    this._saveHubAccess = IModelHost[_getHubAccess]();\n\n    IModelHost[_setHubAccess](this);\n    HubMock._iTwinId = Guid.createValue(); // all iModels for this test get the same \"iTwinId\"\n\n    V2CheckpointManager[_mockCheckpoint] = mockCheckpoint;\n  }\n\n  /** Stop a HubMock that was previously started with [[startup]]\n   * @note this function throws an exception if any of the iModels used during the tests are left open.\n   */\n  public static shutdown() {\n    if (this.mockRoot === undefined)\n      return;\n\n    V2CheckpointManager[_mockCheckpoint] = undefined;\n\n    HubMock._iTwinId = undefined;\n    for (const hub of this.hubs)\n      hub[1].cleanup();\n\n    this.hubs.clear();\n    IModelJsFs.purgeDirSync(this.mockRoot);\n    IModelJsFs.removeSync(this.mockRoot);\n    IModelHost[_setHubAccess](this._saveHubAccess);\n    this.mockRoot = undefined;\n  }\n\n  public static findLocalHub(iModelId: GuidString): LocalHub {\n    const hub = this.hubs.get(iModelId);\n    if (!hub)\n      throw new Error(`local hub for iModel ${iModelId} not created`);\n    return hub;\n  }\n\n  /** create a [[LocalHub]] for an iModel.  */\n  public static async createNewIModel(arg: CreateNewIModelProps): Promise<GuidString> {\n    wasStarted(this.mockRoot);\n    const props = { ...arg, iModelId: Guid.createValue() };\n    const mock = new LocalHub(join(this.mockRoot, props.iModelId), props);\n    this.hubs.set(props.iModelId, mock);\n    return props.iModelId;\n  }\n\n  /** remove the [[LocalHub]] for an iModel */\n  public static destroy(iModelId: GuidString) {\n    this.findLocalHub(iModelId).cleanup();\n    this.hubs.delete(iModelId);\n  }\n\n  /** All methods below are mocks of the [[BackendHubAccess]] interface */\n\n  public static async getChangesetFromNamedVersion(arg: IModelIdArg & { versionName: string }): Promise<ChangesetProps> {\n    return this.findLocalHub(arg.iModelId).findNamedVersion(arg.versionName);\n  }\n\n  private static changesetIndexFromArg(arg: ChangesetArg) {\n    return (undefined !== arg.changeset.index) ? arg.changeset.index : this.findLocalHub(arg.iModelId).getChangesetIndex(arg.changeset.id);\n  }\n\n  public static async getChangesetFromVersion(arg: IModelIdArg & { version: IModelVersion }): Promise<ChangesetProps> {\n    const hub = this.findLocalHub(arg.iModelId);\n    const version = arg.version;\n    if (version.isFirst)\n      return hub.getChangesetByIndex(0);\n\n    const asOf = version.getAsOfChangeSet();\n    if (asOf)\n      return hub.getChangesetById(asOf);\n\n    const versionName = version.getName();\n    if (versionName)\n      return hub.findNamedVersion(versionName);\n\n    return hub.getLatestChangeset();\n  }\n\n  public static async getLatestChangeset(arg: IModelIdArg): Promise<ChangesetProps> {\n    return this.findLocalHub(arg.iModelId).getLatestChangeset();\n  }\n\n  private static async getAccessToken(arg: TokenArg) {\n    return arg.accessToken ?? await IModelHost.getAccessToken();\n  }\n\n  public static async getMyBriefcaseIds(arg: IModelIdArg): Promise<number[]> {\n    const accessToken = await this.getAccessToken(arg);\n    return this.findLocalHub(arg.iModelId).getBriefcaseIds(accessToken);\n  }\n\n  public static async acquireNewBriefcaseId(arg: AcquireNewBriefcaseIdArg): Promise<number> {\n    const accessToken = await this.getAccessToken(arg);\n    return this.findLocalHub(arg.iModelId).acquireNewBriefcaseId(accessToken, arg.briefcaseAlias);\n  }\n\n  /** Release a briefcaseId. After this call it is illegal to generate changesets for the released briefcaseId. */\n  public static async releaseBriefcase(arg: BriefcaseIdArg): Promise<void> {\n    return this.findLocalHub(arg.iModelId).releaseBriefcaseId(arg.briefcaseId);\n  }\n\n  public static async downloadChangeset(arg: DownloadChangesetArg): Promise<ChangesetFileProps> {\n    const changesetProps = this.findLocalHub(arg.iModelId).downloadChangeset({ index: this.changesetIndexFromArg(arg), targetDir: arg.targetDir });\n\n    if (arg.progressCallback) {\n      const totalSize = IModelJsFs.lstatSync(changesetProps.pathname)?.size;\n      if (totalSize)\n        await HubMock.mockProgressReporting(arg.progressCallback, totalSize);\n    }\n\n    return changesetProps;\n  }\n\n  public static async downloadChangesets(arg: DownloadChangesetRangeArg): Promise<ChangesetFileProps[]> {\n    const changesetProps = this.findLocalHub(arg.iModelId).downloadChangesets({ range: arg.range, targetDir: arg.targetDir });\n\n    if (arg.progressCallback) {\n      const totalSize = changesetProps.reduce((sum, props) => sum + (IModelJsFs.lstatSync(props.pathname)?.size ?? 0), 0);\n      await HubMock.mockProgressReporting(arg.progressCallback, totalSize);\n    }\n\n    return changesetProps;\n  }\n\n  public static async queryChangeset(arg: ChangesetArg): Promise<ChangesetProps> {\n    return this.findLocalHub(arg.iModelId).getChangesetByIndex(this.changesetIndexFromArg(arg));\n  }\n\n  public static async queryChangesets(arg: IModelIdArg & { range?: ChangesetRange }): Promise<ChangesetProps[]> {\n    return this.findLocalHub(arg.iModelId).queryChangesets(arg.range);\n  }\n\n  public static async pushChangeset(arg: IModelIdArg & { changesetProps: ChangesetFileProps }): Promise<ChangesetIndex> {\n    return this.findLocalHub(arg.iModelId).addChangeset(arg.changesetProps);\n  }\n\n  public static async queryV2Checkpoint(arg: CheckpointProps): Promise<V2CheckpointAccessProps | undefined> {\n    return {\n      accountName: \"none\",\n      sasToken: \"none\",\n      containerId: Guid.createValue(),\n      dbName: `${arg.changeset.index ?? 0}.bim`,\n      storageType: \"mock\",\n      isMock: true,\n      checkpoint: arg,\n    } as V2CheckpointAccessProps;\n  }\n\n  public static async releaseAllLocks(arg: BriefcaseDbArg) {\n    const hub = this.findLocalHub(arg.iModelId);\n    hub.releaseAllLocks({ briefcaseId: arg.briefcaseId, changesetIndex: hub.getIndexFromChangeset(arg.changeset) });\n  }\n\n  public static async queryAllLocks(_arg: BriefcaseDbArg): Promise<LockProps[]> {\n    return [];\n  }\n\n  public static async acquireLocks(arg: BriefcaseDbArg, locks: LockMap): Promise<void> {\n    this.findLocalHub(arg.iModelId).acquireLocks(locks, arg);\n  }\n\n  public static async queryIModelByName(arg: IModelNameArg): Promise<GuidString | undefined> {\n    for (const hub of this.hubs) {\n      const localHub = hub[1];\n      if (localHub.iTwinId === arg.iTwinId && localHub.iModelName === arg.iModelName)\n        return localHub.iModelId;\n    }\n    return undefined;\n  }\n\n  public static async deleteIModel(arg: IModelIdArg & { iTwinId: GuidString }): Promise<void> {\n    return this.destroy(arg.iModelId);\n  }\n\n  private static async mockProgressReporting(progressCallback: ProgressFunction, totalSize: number): Promise<void> {\n    await new Promise((resolve, reject) => {\n      let rejected = false;\n\n      const mockProgress = (index: number) => {\n        const bytesDownloaded = Math.floor(totalSize * (index / 4));\n        if (!rejected && progressCallback(bytesDownloaded, totalSize) === ProgressStatus.Abort) {\n          rejected = true;\n          reject(new Error(\"AbortError\"));\n        }\n      };\n\n      mockProgress(1);\n      setTimeout(() => mockProgress(2), 50);\n      setTimeout(() => mockProgress(3), 100);\n      setTimeout(() => {\n        mockProgress(4);\n        resolve(undefined);\n      }, 150);\n    });\n  }\n}\n"]}
+{"version":3,"file":"HubMock.js","sourceRoot":"","sources":["../../../src/internal/HubMock.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;AAE/F,+BAA4B;AAC5B,sDAAuD;AASvD,4DAA+I;AAC/I,8CAA2C;AAC3C,8CAA2C;AAC3C,0CAAuC;AAEvC,uCAA0E;AAC1E,0DAAuD;AACvD,6BAA6B;AAE7B,SAAS,UAAU,CAAC,GAAuB;IACzC,IAAI,SAAS,KAAK,GAAG;QACnB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;AAClD,CAAC;AAED,SAAS,UAAU,CAAC,IAA6E;IAC/F,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;AAC/D,CAAC;AACD,MAAM,cAAc,GAAmB;IACrC,UAAU,EAAE,CAAC,UAA2B,EAAE,EAAE;QAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,mCAAgB,CAAC,oBAAoB,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK,MAAM,CAAC,CAAC;QAC9H,UAAU,CAAC,EAAE,GAAG,UAAU,EAAE,UAAU,EAAE,CAAC,CAAA;QACzC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,YAAY,EAAE,CAAC,OAAwB,EAAE,EAAE;QACzC,UAAU,CAAC,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,UAAU,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IACvE,CAAC;CACF,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AACH,MAAa,OAAO;IACV,MAAM,CAAC,QAAQ,CAA2B;IAC1C,MAAM,CAAC,IAAI,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC1C,MAAM,CAAC,cAAc,CAA+B;IACpD,MAAM,CAAC,QAAQ,CAAyB;IAEhD,oEAAoE;IAC7D,MAAM,KAAK,OAAO,KAAK,OAAO,SAAS,KAAK,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC5D,MAAM,KAAK,OAAO;QACvB,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;;;OAIG;IACI,MAAM,CAAC,OAAO,CAAC,QAAsB,EAAE,SAAiB;QAC7D,IAAI,IAAI,CAAC,OAAO;YACd,MAAM,IAAI,KAAK,CAAC,0JAA0J,CAAC,CAAC;QAE9K,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,IAAI,CAAC,QAAQ,GAAG,IAAA,WAAI,EAAC,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACrD,uBAAU,CAAC,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,uBAAU,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,CAAC,cAAc,GAAG,uBAAU,CAAC,uBAAa,CAAC,EAAE,CAAC;QAElD,uBAAU,CAAC,uBAAa,CAAC,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,CAAC,QAAQ,GAAG,mBAAI,CAAC,WAAW,EAAE,CAAC,CAAC,mDAAmD;QAE1F,uCAAmB,CAAC,yBAAe,CAAC,GAAG,cAAc,CAAC;IACxD,CAAC;IAED;;OAEG;IACI,MAAM,CAAC,QAAQ;QACpB,IAAI,IAAI,CAAC,QAAQ,KAAK,SAAS;YAC7B,OAAO;QAET,uCAAmB,CAAC,yBAAe,CAAC,GAAG,SAAS,CAAC;QAEjD,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC;QAC7B,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI;YACzB,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC;QAEnB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,uBAAU,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,uBAAU,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrC,uBAAU,CAAC,uBAAa,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC/C,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;IAC5B,CAAC;IAEM,MAAM,CAAC,YAAY,CAAC,QAAoB;QAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,CAAC,GAAG;YACN,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,cAAc,CAAC,CAAC;QAClE,OAAO,GAAG,CAAC;IACb,CAAC;IAED,4CAA4C;IACrC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAyB;QAC3D,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1B,MAAM,KAAK,GAAG,EAAE,GAAG,GAAG,EAAE,QAAQ,EAAE,mBAAI,CAAC,WAAW,EAAE,EAAE,CAAC;QACvD,MAAM,IAAI,GAAG,IAAI,mBAAQ,CAAC,IAAA,WAAI,EAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,CAAC,CAAC;QACtE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QACpC,OAAO,KAAK,CAAC,QAAQ,CAAC;IACxB,CAAC;IAED,4CAA4C;IACrC,MAAM,CAAC,OAAO,CAAC,QAAoB;QACxC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC;QACtC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAED,wEAAwE;IAEjE,MAAM,CAAC,KAAK,CAAC,4BAA4B,CAAC,GAA0C;QACzF,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC3E,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAAC,GAAiB;QACpD,OAAO,CAAC,SAAS,KAAK,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACzI,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,GAA6C;QACvF,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QAC5B,IAAI,OAAO,CAAC,OAAO;YACjB,OAAO,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QAEpC,MAAM,IAAI,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;QACxC,IAAI,IAAI;YACN,OAAO,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAEpC,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;QACtC,IAAI,WAAW;YACb,OAAO,GAAG,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;QAE3C,OAAO,GAAG,CAAC,kBAAkB,EAAE,CAAC;IAClC,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,GAAgB;QACrD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,kBAAkB,EAAE,CAAC;IAC9D,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,GAAa;QAC/C,OAAO,GAAG,CAAC,WAAW,IAAI,MAAM,uBAAU,CAAC,cAAc,EAAE,CAAC;IAC9D,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAgB;QACpD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;IACtE,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,GAA6B;QACrE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC;QACnD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,WAAW,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC;IAChG,CAAC;IAED,gHAAgH;IACzG,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,GAAmB;QACtD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC7E,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAyB;QAC7D,MAAM,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;QAE/I,IAAI,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,uBAAU,CAAC,SAAS,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC;YACtE,IAAI,SAAS;gBACX,MAAM,OAAO,CAAC,qBAAqB,CAAC,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;QACzE,CAAC;QAED,OAAO,cAAc,CAAC;IACxB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,GAA8B;QACnE,MAAM,cAAc,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;QAE1H,IAAI,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACzB,MAAM,SAAS,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,uBAAU,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACpH,MAAM,OAAO,CAAC,qBAAqB,CAAC,GAAG,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;QACvE,CAAC;QAED,OAAO,cAAc,CAAC;IACxB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC,GAAiB;QAClD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,mBAAmB,CAAC,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC;IAC9F,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAA6C;QAC/E,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACpE,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,GAAyD;QACzF,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IAC1E,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAoB;QACxD,OAAO;YACL,WAAW,EAAE,MAAM;YACnB,QAAQ,EAAE,MAAM;YAChB,WAAW,EAAE,mBAAI,CAAC,WAAW,EAAE;YAC/B,MAAM,EAAE,GAAG,GAAG,CAAC,SAAS,CAAC,KAAK,IAAI,CAAC,MAAM;YACzC,WAAW,EAAE,MAAM;YACnB,MAAM,EAAE,IAAI;YACZ,UAAU,EAAE,GAAG;SACW,CAAC;IAC/B,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAmB;QACrD,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,GAAG,CAAC,eAAe,CAAC,EAAE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,cAAc,EAAE,GAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAClH,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAmB;QACrD,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;IAC3B,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,IAAoB;QACpD,OAAO,EAAE,CAAC;IACZ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAmB,EAAE,KAAc;QAClE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC3D,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAmB,EAAE,KAAc;QAClE,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAC3D,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAkB;QACtD,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC;YACxB,IAAI,QAAQ,CAAC,OAAO,KAAK,GAAG,CAAC,OAAO,IAAI,QAAQ,CAAC,UAAU,KAAK,GAAG,CAAC,UAAU;gBAC5E,OAAO,QAAQ,CAAC,QAAQ,CAAC;QAC7B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAA0C;QACzE,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,gBAAkC,EAAE,SAAiB;QAC9F,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACpC,IAAI,QAAQ,GAAG,KAAK,CAAC;YAErB,MAAM,YAAY,GAAG,CAAC,KAAa,EAAE,EAAE;gBACrC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC5D,IAAI,CAAC,QAAQ,IAAI,gBAAgB,CAAC,eAAe,EAAE,SAAS,CAAC,KAAK,kCAAc,CAAC,KAAK,EAAE,CAAC;oBACvF,QAAQ,GAAG,IAAI,CAAC;oBAChB,MAAM,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;gBAClC,CAAC;YACH,CAAC,CAAC;YAEF,YAAY,CAAC,CAAC,CAAC,CAAC;YAChB,UAAU,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACtC,UAAU,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACvC,UAAU,CAAC,GAAG,EAAE;gBACd,YAAY,CAAC,CAAC,CAAC,CAAC;gBAChB,OAAO,CAAC,SAAS,CAAC,CAAC;YACrB,CAAC,EAAE,GAAG,CAAC,CAAC;QACV,CAAC,CAAC,CAAC;IACL,CAAC;;AApOH,0BAqOC","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 { join } from \"path\";\nimport { Guid, GuidString } from \"@itwin/core-bentley\";\nimport {\n  ChangesetFileProps, ChangesetIndex, ChangesetIndexOrId, ChangesetProps, ChangesetRange, IModelVersion, LocalDirName,\n} from \"@itwin/core-common\";\nimport {\n  AcquireNewBriefcaseIdArg,\n  BackendHubAccess, BriefcaseDbArg, BriefcaseIdArg, ChangesetArg, CreateNewIModelProps, DownloadChangesetArg, DownloadChangesetRangeArg, IModelIdArg, IModelNameArg,\n  LockMap, LockProps, V2CheckpointAccessProps,\n} from \"../BackendHubAccess\";\nimport { CheckpointProps, DownloadRequest, MockCheckpoint, ProgressFunction, ProgressStatus, V2CheckpointManager } from \"../CheckpointManager\";\nimport { IModelHost } from \"../IModelHost\";\nimport { IModelJsFs } from \"../IModelJsFs\";\nimport { LocalHub } from \"../LocalHub\";\nimport { TokenArg } from \"../IModelDb\";\nimport { _getHubAccess, _mockCheckpoint, _setHubAccess } from \"./Symbols\";\nimport { BriefcaseManager } from \"../BriefcaseManager\";\nimport * as path from \"path\";\n\nfunction wasStarted(val: string | undefined): asserts val is string {\n  if (undefined === val)\n    throw new Error(\"Call HubMock.startup first\");\n}\n\nfunction doDownload(args: { iModelId: string, changeset: ChangesetIndexOrId, targetFile: string }) {\n  HubMock.findLocalHub(args.iModelId).downloadCheckpoint(args);\n}\nconst mockCheckpoint: MockCheckpoint = {\n  mockAttach: (checkpoint: CheckpointProps) => {\n    const targetFile = path.join(BriefcaseManager.getBriefcaseBasePath(checkpoint.iModelId), `${checkpoint.changeset.index}.bim`);\n    doDownload({ ...checkpoint, targetFile })\n    return targetFile;\n  },\n\n  mockDownload: (request: DownloadRequest) => {\n    doDownload({ ...request.checkpoint, targetFile: request.localFile });\n  }\n};\n\n/**\n * Mocks iModelHub for testing creating Briefcases, downloading checkpoints, and simulating multiple users pushing and pulling changesets, etc.\n *\n * Generally, tests for apis that *create or modify* iModels can and should be mocked. Otherwise they:\n * - create tremendous load on the test servers when they run on programmer's desktops and in CI jobs\n * - waste network and data center resources (i.e. $$$s),\n * - interfere with other tests running on the same or other systems, and\n * - (far worse) are the source of test flakiness outside of the api being tested.\n *\n * This class can be used to create tests that do not require authentication, are synchronous,\n * are guaranteed to be self-contained (i.e. do not interfere with other tests running at the same time or later), and do not fail for reasons outside\n * of the control of the test itself. As a bonus, in addition to making tests more reliable, mocking IModelHub generally makes tests run *much* faster.\n *\n * On the other hand, tests that expect to find an existing iModels, checkpoints, changesets, etc. in IModelHub cannot be mocked. In that case, those tests\n * should be careful to NOT modify the data, since doing so causes interference with other tests running simultaneously. These tests should be limited to\n * low level testing of the core apis only.\n *\n * To initialize HubMock, call [[startup]] at the beginning of your test, usually in `describe.before`. Thereafter, all access to iModelHub for an iModel will be\n * directed to a [[LocalHub]] - your test code does not change. After the test(s) complete, call [[shutdown]] (usually in `describe.after`) to stop mocking IModelHub and clean\n * up any resources used by the test(s). If you want to mock a single test, call [[startup]] as the first line and [[shutdown]] as the last. If you wish to run the\n * test against a \"real\" IModelHub, you can simply comment off the call [[startup]], though in that case you should make sure the name of your\n * iModel is unique so your test won't collide with other tests (iModel name uniqueness is not necessary for mocked tests.)\n *\n * Mocked tests must always start by creating a new iModel via [[IModelHost[_hubAccess].createNewIModel]] with a `version0` iModel.\n * They use mock (aka \"bogus\") credentials for `AccessTokens`, which is fine since [[HubMock]] never accesses resources outside the current\n * computer.\n *\n * @note Only one HubMock at a time, *running in a single process*, may be active. The comments above about multiple simultaneous tests refer to tests\n * running on different computers, or on a single computer in multiple processes. All of those scenarios are problematic without mocking.\n *\n * @internal\n */\nexport class HubMock {\n  private static mockRoot: LocalDirName | undefined;\n  private static hubs = new Map<string, LocalHub>();\n  private static _saveHubAccess: BackendHubAccess | undefined;\n  private static _iTwinId: GuidString | undefined;\n\n  /** Determine whether a test us currently being run under HubMock */\n  public static get isValid() { return undefined !== this.mockRoot; }\n  public static get iTwinId() {\n    wasStarted(this._iTwinId);\n    return this._iTwinId;\n  }\n\n  /**\n   * Begin mocking IModelHub access. After this call, all access to IModelHub will be directed to a [[LocalHub]].\n   * @param mockName a unique name (e.g. \"MyTest\") for this HubMock to disambiguate tests when more than one is simultaneously active.\n   * It is used to create a private directory used by the HubMock for a test. That directory is removed when [[shutdown]] is called.\n   */\n  public static startup(mockName: LocalDirName, outputDir: string) {\n    if (this.isValid)\n      throw new Error(\"Either a previous test did not call HubMock.shutdown() properly, or more than one test is simultaneously attempting to use HubMock, which is not allowed\");\n\n    this.hubs.clear();\n    this.mockRoot = join(outputDir, \"HubMock\", mockName);\n    IModelJsFs.recursiveMkDirSync(this.mockRoot);\n    IModelJsFs.purgeDirSync(this.mockRoot);\n    this._saveHubAccess = IModelHost[_getHubAccess]();\n\n    IModelHost[_setHubAccess](this);\n    HubMock._iTwinId = Guid.createValue(); // all iModels for this test get the same \"iTwinId\"\n\n    V2CheckpointManager[_mockCheckpoint] = mockCheckpoint;\n  }\n\n  /** Stop a HubMock that was previously started with [[startup]]\n   * @note this function throws an exception if any of the iModels used during the tests are left open.\n   */\n  public static shutdown() {\n    if (this.mockRoot === undefined)\n      return;\n\n    V2CheckpointManager[_mockCheckpoint] = undefined;\n\n    HubMock._iTwinId = undefined;\n    for (const hub of this.hubs)\n      hub[1].cleanup();\n\n    this.hubs.clear();\n    IModelJsFs.purgeDirSync(this.mockRoot);\n    IModelJsFs.removeSync(this.mockRoot);\n    IModelHost[_setHubAccess](this._saveHubAccess);\n    this.mockRoot = undefined;\n  }\n\n  public static findLocalHub(iModelId: GuidString): LocalHub {\n    const hub = this.hubs.get(iModelId);\n    if (!hub)\n      throw new Error(`local hub for iModel ${iModelId} not created`);\n    return hub;\n  }\n\n  /** create a [[LocalHub]] for an iModel.  */\n  public static async createNewIModel(arg: CreateNewIModelProps): Promise<GuidString> {\n    wasStarted(this.mockRoot);\n    const props = { ...arg, iModelId: Guid.createValue() };\n    const mock = new LocalHub(join(this.mockRoot, props.iModelId), props);\n    this.hubs.set(props.iModelId, mock);\n    return props.iModelId;\n  }\n\n  /** remove the [[LocalHub]] for an iModel */\n  public static destroy(iModelId: GuidString) {\n    this.findLocalHub(iModelId).cleanup();\n    this.hubs.delete(iModelId);\n  }\n\n  /** All methods below are mocks of the [[BackendHubAccess]] interface */\n\n  public static async getChangesetFromNamedVersion(arg: IModelIdArg & { versionName: string }): Promise<ChangesetProps> {\n    return this.findLocalHub(arg.iModelId).findNamedVersion(arg.versionName);\n  }\n\n  private static changesetIndexFromArg(arg: ChangesetArg) {\n    return (undefined !== arg.changeset.index) ? arg.changeset.index : this.findLocalHub(arg.iModelId).getChangesetIndex(arg.changeset.id);\n  }\n\n  public static async getChangesetFromVersion(arg: IModelIdArg & { version: IModelVersion }): Promise<ChangesetProps> {\n    const hub = this.findLocalHub(arg.iModelId);\n    const version = arg.version;\n    if (version.isFirst)\n      return hub.getChangesetByIndex(0);\n\n    const asOf = version.getAsOfChangeSet();\n    if (asOf)\n      return hub.getChangesetById(asOf);\n\n    const versionName = version.getName();\n    if (versionName)\n      return hub.findNamedVersion(versionName);\n\n    return hub.getLatestChangeset();\n  }\n\n  public static async getLatestChangeset(arg: IModelIdArg): Promise<ChangesetProps> {\n    return this.findLocalHub(arg.iModelId).getLatestChangeset();\n  }\n\n  private static async getAccessToken(arg: TokenArg) {\n    return arg.accessToken ?? await IModelHost.getAccessToken();\n  }\n\n  public static async getMyBriefcaseIds(arg: IModelIdArg): Promise<number[]> {\n    const accessToken = await this.getAccessToken(arg);\n    return this.findLocalHub(arg.iModelId).getBriefcaseIds(accessToken);\n  }\n\n  public static async acquireNewBriefcaseId(arg: AcquireNewBriefcaseIdArg): Promise<number> {\n    const accessToken = await this.getAccessToken(arg);\n    return this.findLocalHub(arg.iModelId).acquireNewBriefcaseId(accessToken, arg.briefcaseAlias);\n  }\n\n  /** Release a briefcaseId. After this call it is illegal to generate changesets for the released briefcaseId. */\n  public static async releaseBriefcase(arg: BriefcaseIdArg): Promise<void> {\n    return this.findLocalHub(arg.iModelId).releaseBriefcaseId(arg.briefcaseId);\n  }\n\n  public static async downloadChangeset(arg: DownloadChangesetArg): Promise<ChangesetFileProps> {\n    const changesetProps = this.findLocalHub(arg.iModelId).downloadChangeset({ index: this.changesetIndexFromArg(arg), targetDir: arg.targetDir });\n\n    if (arg.progressCallback) {\n      const totalSize = IModelJsFs.lstatSync(changesetProps.pathname)?.size;\n      if (totalSize)\n        await HubMock.mockProgressReporting(arg.progressCallback, totalSize);\n    }\n\n    return changesetProps;\n  }\n\n  public static async downloadChangesets(arg: DownloadChangesetRangeArg): Promise<ChangesetFileProps[]> {\n    const changesetProps = this.findLocalHub(arg.iModelId).downloadChangesets({ range: arg.range, targetDir: arg.targetDir });\n\n    if (arg.progressCallback) {\n      const totalSize = changesetProps.reduce((sum, props) => sum + (IModelJsFs.lstatSync(props.pathname)?.size ?? 0), 0);\n      await HubMock.mockProgressReporting(arg.progressCallback, totalSize);\n    }\n\n    return changesetProps;\n  }\n\n  public static async queryChangeset(arg: ChangesetArg): Promise<ChangesetProps> {\n    return this.findLocalHub(arg.iModelId).getChangesetByIndex(this.changesetIndexFromArg(arg));\n  }\n\n  public static async queryChangesets(arg: IModelIdArg & { range?: ChangesetRange }): Promise<ChangesetProps[]> {\n    return this.findLocalHub(arg.iModelId).queryChangesets(arg.range);\n  }\n\n  public static async pushChangeset(arg: IModelIdArg & { changesetProps: ChangesetFileProps }): Promise<ChangesetIndex> {\n    return this.findLocalHub(arg.iModelId).addChangeset(arg.changesetProps);\n  }\n\n  public static async queryV2Checkpoint(arg: CheckpointProps): Promise<V2CheckpointAccessProps | undefined> {\n    return {\n      accountName: \"none\",\n      sasToken: \"none\",\n      containerId: Guid.createValue(),\n      dbName: `${arg.changeset.index ?? 0}.bim`,\n      storageType: \"mock\",\n      isMock: true,\n      checkpoint: arg,\n    } as V2CheckpointAccessProps;\n  }\n\n  public static async releaseAllLocks(arg: BriefcaseDbArg) {\n    const hub = this.findLocalHub(arg.iModelId);\n    hub.releaseAllLocks({ briefcaseId: arg.briefcaseId, changesetIndex: hub.getIndexFromChangeset(arg.changeset) });\n  }\n\n  public static async abandonAllLocks(arg: BriefcaseIdArg): Promise<void> {\n    const hub = this.findLocalHub(arg.iModelId);\n    hub.abandonAllLocks(arg);\n  }\n\n  public static async queryAllLocks(_arg: BriefcaseDbArg): Promise<LockProps[]> {\n    return [];\n  }\n\n  public static async acquireLocks(arg: BriefcaseDbArg, locks: LockMap): Promise<void> {\n    this.findLocalHub(arg.iModelId).acquireLocks(locks, arg);\n  }\n\n  public static async abandonLocks(arg: BriefcaseIdArg, locks: LockMap): Promise<void> {\n    this.findLocalHub(arg.iModelId).abandonLocks(locks, arg);\n  }\n\n  public static async queryIModelByName(arg: IModelNameArg): Promise<GuidString | undefined> {\n    for (const hub of this.hubs) {\n      const localHub = hub[1];\n      if (localHub.iTwinId === arg.iTwinId && localHub.iModelName === arg.iModelName)\n        return localHub.iModelId;\n    }\n    return undefined;\n  }\n\n  public static async deleteIModel(arg: IModelIdArg & { iTwinId: GuidString }): Promise<void> {\n    return this.destroy(arg.iModelId);\n  }\n\n  private static async mockProgressReporting(progressCallback: ProgressFunction, totalSize: number): Promise<void> {\n    await new Promise((resolve, reject) => {\n      let rejected = false;\n\n      const mockProgress = (index: number) => {\n        const bytesDownloaded = Math.floor(totalSize * (index / 4));\n        if (!rejected && progressCallback(bytesDownloaded, totalSize) === ProgressStatus.Abort) {\n          rejected = true;\n          reject(new Error(\"AbortError\"));\n        }\n      };\n\n      mockProgress(1);\n      setTimeout(() => mockProgress(2), 50);\n      setTimeout(() => mockProgress(3), 100);\n      setTimeout(() => {\n        mockProgress(4);\n        resolve(undefined);\n      }, 150);\n    });\n  }\n}\n"]}

package/lib/cjs/internal/NoLocks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"NoLocks.d.ts","sourceRoot":"","sources":["../../../src/internal/NoLocks.ts"],"names":[],"mappings":"AAKA;;GAEG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAmB7C,wBAAgB,qBAAqB,IAAI,WAAW,CAEnD"}
+{"version":3,"file":"NoLocks.d.ts","sourceRoot":"","sources":["../../../src/internal/NoLocks.ts"],"names":[],"mappings":"AAKA;;GAEG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAyB7C,wBAAgB,qBAAqB,IAAI,WAAW,CAEnD"}

package/lib/cjs/internal/NoLocks.js

@@ -20,6 +20,12 @@ class NoLocks {
     async acquireLocks() { }
     async [Symbols_1._releaseAllLocks]() { }
     async releaseAllLocks() { }
+    async abandonAllLocks() { }
+    async abandonLocksForReversedTxn() { return false; }
+    async abandonLocksForCurrentUnsavedTxn() { return false; }
+    async acquireLocksForReinstatingTxn() { return false; }
+    holdsNecessaryLocksForReinstatingTxn() { return true; }
+    clearTxnLockRecords() { }
 }
 function createNoOpLockControl() {
     return new NoLocks();

package/lib/cjs/internal/NoLocks.js.map

@@ -1 +1 @@
-{"version":3,"file":"NoLocks.js","sourceRoot":"","sources":["../../../src/internal/NoLocks.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;AAyB/F,sDAEC;AApBD,uCAAoG;AAEpG,yJAAyJ;AACzJ,MAAM,OAAO;IACK,CAAC,mCAAyB,CAAC,GAAG,SAAS,CAAC;IACxD,IAAW,aAAa,KAAK,OAAO,KAAK,CAAC,CAAC,CAAC;IACrC,CAAC,gBAAM,CAAC,KAAW,CAAC;IACpB,aAAa,KAAW,CAAC;IACzB,kBAAkB,KAAc,OAAO,KAAK,CAAC,CAAC,CAAC;IAC/C,eAAe,KAAc,OAAO,KAAK,CAAC,CAAC,CAAC;IAC5C,kBAAkB,KAAW,CAAC;IAC9B,eAAe,KAAW,CAAC;IAC3B,CAAC,4BAAkB,CAAC,KAAW,CAAC;IAChC,KAAK,CAAC,YAAY,KAAK,CAAC;IACxB,KAAK,CAAC,CAAC,0BAAgB,CAAC,KAAoB,CAAC;IAC7C,KAAK,CAAC,eAAe,KAAoB,CAAC;CAClD;AAED,SAAgB,qBAAqB;IACnC,OAAO,IAAI,OAAO,EAAE,CAAC;AACvB,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\n/** @packageDocumentation\n * @module iModels\n */\n\nimport { LockControl } from \"../LockControl\";\nimport { _close, _elementWasCreated, _implementationProhibited, _releaseAllLocks } from \"./Symbols\";\n\n/** A null-implementation of LockControl that does not attempt to limit access between briefcases. This relies on change-merging to resolve conflicts. */\nclass NoLocks implements LockControl {\n  public readonly [_implementationProhibited] = undefined;\n  public get isServerBased() { return false; }\n  public [_close](): void { }\n  public clearAllLocks(): void { }\n  public holdsExclusiveLock(): boolean { return false; }\n  public holdsSharedLock(): boolean { return false; }\n  public checkExclusiveLock(): void { }\n  public checkSharedLock(): void { }\n  public [_elementWasCreated](): void { }\n  public async acquireLocks() { }\n  public async [_releaseAllLocks](): Promise<void> { }\n  public async releaseAllLocks(): Promise<void> { }\n}\n\nexport function createNoOpLockControl(): LockControl {\n  return new NoLocks();\n}\n"]}
+{"version":3,"file":"NoLocks.js","sourceRoot":"","sources":["../../../src/internal/NoLocks.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;AA+B/F,sDAEC;AA1BD,uCAAoG;AAEpG,yJAAyJ;AACzJ,MAAM,OAAO;IACK,CAAC,mCAAyB,CAAC,GAAG,SAAS,CAAC;IACxD,IAAW,aAAa,KAAK,OAAO,KAAK,CAAC,CAAC,CAAC;IACrC,CAAC,gBAAM,CAAC,KAAW,CAAC;IACpB,aAAa,KAAW,CAAC;IACzB,kBAAkB,KAAc,OAAO,KAAK,CAAC,CAAC,CAAC;IAC/C,eAAe,KAAc,OAAO,KAAK,CAAC,CAAC,CAAC;IAC5C,kBAAkB,KAAW,CAAC;IAC9B,eAAe,KAAW,CAAC;IAC3B,CAAC,4BAAkB,CAAC,KAAW,CAAC;IAChC,KAAK,CAAC,YAAY,KAAK,CAAC;IACxB,KAAK,CAAC,CAAC,0BAAgB,CAAC,KAAoB,CAAC;IAC7C,KAAK,CAAC,eAAe,KAAoB,CAAC;IAC1C,KAAK,CAAC,eAAe,KAAoB,CAAC;IAC1C,KAAK,CAAC,0BAA0B,KAAuB,OAAO,KAAK,CAAC,CAAC,CAAC;IACtE,KAAK,CAAC,gCAAgC,KAAuB,OAAO,KAAK,CAAC,CAAC,CAAC;IAC5E,KAAK,CAAC,6BAA6B,KAAuB,OAAO,KAAK,CAAC,CAAC,CAAC;IACzE,oCAAoC,KAAc,OAAO,IAAI,CAAC,CAAC,CAAC;IAChE,mBAAmB,KAAW,CAAC;CACvC;AAED,SAAgB,qBAAqB;IACnC,OAAO,IAAI,OAAO,EAAE,CAAC;AACvB,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\n/** @packageDocumentation\n * @module iModels\n */\n\nimport { LockControl } from \"../LockControl\";\nimport { _close, _elementWasCreated, _implementationProhibited, _releaseAllLocks } from \"./Symbols\";\n\n/** A null-implementation of LockControl that does not attempt to limit access between briefcases. This relies on change-merging to resolve conflicts. */\nclass NoLocks implements LockControl {\n  public readonly [_implementationProhibited] = undefined;\n  public get isServerBased() { return false; }\n  public [_close](): void { }\n  public clearAllLocks(): void { }\n  public holdsExclusiveLock(): boolean { return false; }\n  public holdsSharedLock(): boolean { return false; }\n  public checkExclusiveLock(): void { }\n  public checkSharedLock(): void { }\n  public [_elementWasCreated](): void { }\n  public async acquireLocks() { }\n  public async [_releaseAllLocks](): Promise<void> { }\n  public async releaseAllLocks(): Promise<void> { }\n  public async abandonAllLocks(): Promise<void> { }\n  public async abandonLocksForReversedTxn(): Promise<boolean> { return false; }\n  public async abandonLocksForCurrentUnsavedTxn(): Promise<boolean> { return false; }\n  public async acquireLocksForReinstatingTxn(): Promise<boolean> { return false; }\n  public holdsNecessaryLocksForReinstatingTxn(): boolean { return true; }\n  public clearTxnLockRecords(): void { }\n}\n\nexport function createNoOpLockControl(): LockControl {\n  return new NoLocks();\n}\n"]}

package/lib/cjs/internal/ServerBasedLocks.d.ts

@@ -12,6 +12,8 @@ export declare class ServerBasedLocks implements LockControl {
     get isServerBased(): boolean;
     protected readonly lockDb: SQLiteDb;
     protected readonly briefcase: BriefcaseDb;
+    private _removeOnCommitListener;
+    private readonly _unsavedChangesTxnId;
     constructor(iModel: BriefcaseDb);
     [_close](): void;
     private getOwners;
@@ -24,7 +26,9 @@ export declare class ServerBasedLocks implements LockControl {
     getLockCount(state: LockState): number;
     [_releaseAllLocks](): Promise<void>;
     releaseAllLocks(): Promise<void>;
+    abandonAllLocks(): Promise<void>;
     private insertLock;
+    private insertTxnLockRecord;
     private ownerHoldsExclusiveLock;
     /** Determine whether an the exclusive lock is already held by an element (or one of its owners) */
     holdsExclusiveLock(id: Id64String): boolean;
@@ -39,8 +43,16 @@ export declare class ServerBasedLocks implements LockControl {
         shared?: Id64Arg;
         exclusive?: Id64Arg;
     }): Promise<void>;
+    private abandonLocks;
+    abandonLocksForCurrentUnsavedTxn(): Promise<boolean>;
+    abandonLocksForReversedTxn(txnId: Id64String): Promise<boolean>;
+    private getAbandonedLocksForTxn;
+    acquireLocksForReinstatingTxn(txnId: Id64String): Promise<boolean>;
+    holdsNecessaryLocksForReinstatingTxn(txnId: Id64String): boolean;
+    clearTxnLockRecords(txnId: Id64String): void;
     /** When an element is newly created in a session, we hold the lock on it implicitly. Save that fact. */
     [_elementWasCreated](id: Id64String): void;
+    private clearDiscoveredLocks;
     /** locks are not necessary during change propagation. */
     private get _locksAreRequired();
     /** throw if locks are currently required and the exclusive lock is not held on the supplied element */

package/lib/cjs/internal/ServerBasedLocks.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ServerBasedLocks.d.ts","sourceRoot":"","sources":["../../../src/internal/ServerBasedLocks.ts"],"names":[],"mappings":"AAKA;;GAEG;AAEH,OAAO,EAAkB,OAAO,EAAE,UAAU,EAA0B,MAAM,qBAAqB,CAAC;AAClG,OAAO,EAAuB,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEpE,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAc,yBAAyB,EAAa,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAmB3H,qBAAa,gBAAiB,YAAW,WAAW;IAClD,SAAgB,CAAC,yBAAyB,CAAC,YAAa;IAExD,IAAW,aAAa,YAAmB;IAC3C,SAAS,CAAC,QAAQ,CAAC,MAAM,WAAkB;IAC3C,SAAS,CAAC,QAAQ,CAAC,SAAS,EAAE,WAAW,CAAC;gBAEvB,MAAM,EAAE,WAAW;IAY/B,CAAC,MAAM,CAAC;IAKf,OAAO,CAAC,SAAS;IAWjB,OAAO,CAAC,YAAY;IAOpB;;OAEG;IACH,OAAO,CAAC,aAAa;IAKrB,qBAAqB;IACd,YAAY,CAAC,KAAK,EAAE,SAAS,GAAG,MAAM;IAQhC,CAAC,gBAAgB,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC;IAKnC,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ7C,OAAO,CAAC,UAAU;IAYlB,OAAO,CAAC,uBAAuB;IAgB/B,mGAAmG;IAC5F,kBAAkB,CAAC,EAAE,EAAE,UAAU,GAAG,OAAO;IAK3C,eAAe,CAAC,EAAE,EAAE,UAAU,GAAG,OAAO;IAM/C,sIAAsI;IACtI,OAAO,CAAC,aAAa;IAUrB,sGAAsG;IACtG,OAAO,CAAC,mBAAmB;IAM3B,mEAAmE;YACrD,eAAe;IAmBhB,YAAY,CAAC,GAAG,EAAE;QAAE,MAAM,CAAC,EAAE,OAAO,CAAC;QAAC,SAAS,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBxF,wGAAwG;IACjG,CAAC,kBAAkB,CAAC,CAAC,EAAE,EAAE,UAAU;IAK1C,yDAAyD;IACzD,OAAO,KAAK,iBAAiB,GAAqD;IAElF,uGAAuG;IAChG,kBAAkB,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;IAKzE,kGAAkG;IAC3F,eAAe,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;CAKvE;AAED,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,WAAW,GAAG,WAAW,CAEvE"}
+{"version":3,"file":"ServerBasedLocks.d.ts","sourceRoot":"","sources":["../../../src/internal/ServerBasedLocks.ts"],"names":[],"mappings":"AAKA;;GAEG;AAEH,OAAO,EAAkB,OAAO,EAAE,UAAU,EAA0B,MAAM,qBAAqB,CAAC;AAClG,OAAO,EAAuB,SAAS,EAAyB,MAAM,oBAAoB,CAAC;AAE3F,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAc,yBAAyB,EAAa,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAmB3H,qBAAa,gBAAiB,YAAW,WAAW;IAClD,SAAgB,CAAC,yBAAyB,CAAC,YAAa;IAExD,IAAW,aAAa,YAAmB;IAC3C,SAAS,CAAC,QAAQ,CAAC,MAAM,WAAkB;IAC3C,SAAS,CAAC,QAAQ,CAAC,SAAS,EAAE,WAAW,CAAC;IAC1C,OAAO,CAAC,uBAAuB,CAAa;IAC5C,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAwB;gBAE1C,MAAM,EAAE,WAAW;IA2C/B,CAAC,MAAM,CAAC;IAOf,OAAO,CAAC,SAAS;IAWjB,OAAO,CAAC,YAAY;IAOpB;;OAEG;IACH,OAAO,CAAC,aAAa;IAMrB,qBAAqB;IACd,YAAY,CAAC,KAAK,EAAE,SAAS,GAAG,MAAM;IAQhC,CAAC,gBAAgB,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC;IAKnC,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAQhC,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAoB7C,OAAO,CAAC,UAAU;IAalB,OAAO,CAAC,mBAAmB;IAuB3B,OAAO,CAAC,uBAAuB;IAgB/B,mGAAmG;IAC5F,kBAAkB,CAAC,EAAE,EAAE,UAAU,GAAG,OAAO;IAK3C,eAAe,CAAC,EAAE,EAAE,UAAU,GAAG,OAAO;IAM/C,sIAAsI;IACtI,OAAO,CAAC,aAAa;IAUrB,sGAAsG;IACtG,OAAO,CAAC,mBAAmB;IAM3B,mEAAmE;YACrD,eAAe;IAqBhB,YAAY,CAAC,GAAG,EAAE;QAAE,MAAM,CAAC,EAAE,OAAO,CAAC;QAAC,SAAS,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;YAiB1E,YAAY;IAcb,gCAAgC,IAAI,OAAO,CAAC,OAAO,CAAC;IAIpD,0BAA0B,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC;IA8H5E,OAAO,CAAC,uBAAuB;IAyBlB,6BAA6B,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC;IAiDxE,oCAAoC,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO;IAKhE,mBAAmB,CAAC,KAAK,EAAE,UAAU;IAY5C,wGAAwG;IACjG,CAAC,kBAAkB,CAAC,CAAC,EAAE,EAAE,UAAU;IAM1C,OAAO,CAAC,oBAAoB;IAS5B,yDAAyD;IACzD,OAAO,KAAK,iBAAiB,GAAqD;IAElF,uGAAuG;IAChG,kBAAkB,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;IAKzE,kGAAkG;IAC3F,eAAe,CAAC,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM;CAKvE;AAED,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,WAAW,GAAG,WAAW,CAEvE"}

package/lib/cjs/internal/ServerBasedLocks.js

@@ -19,6 +19,8 @@ class ServerBasedLocks {
     get isServerBased() { return true; }
     lockDb = new SQLiteDb_1.SQLiteDb();
     briefcase;
+    _removeOnCommitListener;
+    _unsavedChangesTxnId = "0x7FFFFFFFFFFFFFFF"; // a placeholder txn id for locks acquired in the current unsaved Txn
     constructor(iModel) {
         this.briefcase = iModel;
         const dbName = `${iModel[Symbols_1._nativeDb].getTempFileBaseName()}-locks`;
@@ -27,11 +29,37 @@ class ServerBasedLocks {
         }
         catch {
             this.lockDb.createDb(dbName);
-            this.lockDb.executeSQL("CREATE TABLE locks(id INTEGER PRIMARY KEY NOT NULL,state INTEGER NOT NULL,origin INTEGER)");
-            this.lockDb.saveChanges();
         }
+        // Tracks the locks that are actively held.
+        this.lockDb.executeSQL("CREATE TABLE IF NOT EXISTS locks(id INTEGER PRIMARY KEY NOT NULL,state INTEGER NOT NULL,origin INTEGER)");
+        // Tracks the locks that are required by each Txn. They may or may not currently be held.
+        this.lockDb.executeSQL(`
+      CREATE TABLE IF NOT EXISTS txn_locks(
+        txnId INTEGER NOT NULL,
+        elementId INTEGER NOT NULL,
+        state INTEGER NOT NULL,
+        origin INTEGER NOT NULL,
+        abandoned BOOLEAN NOT NULL,
+        PRIMARY KEY (txnId, elementId))`);
+        this.lockDb.saveChanges();
+        this._removeOnCommitListener = this.briefcase.txns.onCommit.addListener(() => {
+            const committedTxnId = this.briefcase.txns.queryPreviousTxnId(this.briefcase.txns.getCurrentTxnId());
+            // With this commit, any reversed txns with the committed txn's ID or greater are no longer reinstatable,
+            // so clear out the record of their locks. If the locks are still held, sorry, it's too late!
+            this.clearTxnLockRecords(committedTxnId);
+            // All of the "current" changes are now part of a real txn, so update the txn id accordingly.
+            this.lockDb.withPreparedSqliteStatement("UPDATE txn_locks SET txnId=? WHERE txnId=?", (stmt) => {
+                stmt.bindId(1, committedTxnId);
+                stmt.bindId(2, this._unsavedChangesTxnId);
+                const rc = stmt.step();
+                if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                    core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't update locks database with txn ID of unsaved changes txn upon saving (error code ${rc})`);
+            });
+            this.lockDb.saveChanges();
+        });
     }
     [Symbols_1._close]() {
+        this._removeOnCommitListener();
         if (this.lockDb.isOpen)
             this.lockDb.closeDb();
     }
@@ -55,6 +83,7 @@ class ServerBasedLocks {
      */
     clearAllLocks() {
         this.lockDb.executeSQL("DELETE FROM locks");
+        this.lockDb.executeSQL("DELETE FROM txn_locks");
         this.lockDb.saveChanges();
     }
     /** only for tests */
@@ -71,10 +100,28 @@ class ServerBasedLocks {
     }
     async releaseAllLocks() {
         if (this.briefcase.txns.hasLocalChanges) {
-            throw new Error("Locks cannot be released while the briefcase contains local changes");
+            core_common_1.ServerBasedLocksError.throwError("has-unsaved-changes", "Locks cannot be released while the briefcase contains local changes");
         }
         return this[Symbols_1._releaseAllLocks]();
     }
+    async abandonAllLocks() {
+        if (this.briefcase.txns.hasLocalChanges) {
+            core_common_1.ServerBasedLocksError.throwError("has-unsaved-changes", "Locks cannot be abandoned while the briefcase contains local changes");
+        }
+        if (IModelHost_1.IModelHost[Symbols_1._hubAccess].abandonAllLocks === undefined) {
+            // If the IModelHub doesn't support an explicit abandon, call release with a blank changeset to indicate
+            // that locks should be released without updating the changeset associated with the locks.
+            await IModelHost_1.IModelHost[Symbols_1._hubAccess].releaseAllLocks({
+                iModelId: this.briefcase.iModelId,
+                briefcaseId: this.briefcase.briefcaseId,
+                changeset: { id: "", index: 0 }
+            });
+        }
+        else {
+            await IModelHost_1.IModelHost[Symbols_1._hubAccess].abandonAllLocks(this.briefcase);
+        }
+        this.clearAllLocks();
+    }
     insertLock(id, state, origin) {
         this.lockDb.withPreparedSqliteStatement("INSERT INTO locks(id,state,origin) VALUES (?,?,?) ON CONFLICT(id) DO UPDATE SET state=excluded.state,origin=excluded.origin", (stmt) => {
             stmt.bindId(1, id);
@@ -86,6 +133,28 @@ class ServerBasedLocks {
         });
         return true;
     }
+    insertTxnLockRecord(id, state, origin) {
+        // Locks are always acquired in the current txn, which isn't a real txn until it's committed.
+        // So use a placeholder txn id for now, and we'll update to the real txn id on commit.
+        // This is important to distinguish new locks acquired in the current txn from locks acquired in previous reversed txns, which will only
+        // be cleared (no longer reinstateable) on commit.
+        this.lockDb.withPreparedSqliteStatement(`
+      INSERT INTO txn_locks(txnId,elementId,state,origin,abandoned)
+      VALUES (?,?,?,?,FALSE)
+      ON CONFLICT(txnId,elementId)
+        DO UPDATE SET
+          state=excluded.state,
+          origin=excluded.origin,
+          abandoned=excluded.abandoned`, (stmt) => {
+            stmt.bindId(1, this._unsavedChangesTxnId);
+            stmt.bindId(2, id);
+            stmt.bindInteger(3, state);
+            stmt.bindInteger(4, origin);
+            const rc = stmt.step();
+            if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't insert txn lock record into database (error code ${rc})`);
+        });
+    }
     ownerHoldsExclusiveLock(id) {
         if (id === undefined || id === core_common_1.IModel.rootSubjectId)
             return false; // has no owners
@@ -135,8 +204,10 @@ class ServerBasedLocks {
                 locks.set(shared, core_common_1.LockState.Shared);
         }
         await IModelHost_1.IModelHost[Symbols_1._hubAccess].acquireLocks(this.briefcase, locks); // throws if unsuccessful
-        for (const lock of locks)
+        for (const lock of locks) {
             this.insertLock(lock[0], lock[1], 0 /* LockOrigin.Acquired */);
+            this.insertTxnLockRecord(lock[0], lock[1], 0 /* LockOrigin.Acquired */);
+        }
         this.lockDb.saveChanges();
     }
     async acquireLocks(arg) {
@@ -155,11 +226,221 @@ class ServerBasedLocks {
         }
         return this.acquireAllLocks(locks);
     }
+    async abandonLocks(locks) {
+        if (IModelHost_1.IModelHost[Symbols_1._hubAccess].abandonLocks === undefined) {
+            // If the IModelHub doesn't support an explicit abandon, call acquireLocks with a blank changeset to indicate
+            // that locks should be released without updating the changeset associated with the locks.
+            await IModelHost_1.IModelHost[Symbols_1._hubAccess].acquireLocks({
+                iModelId: this.briefcase.iModelId,
+                briefcaseId: this.briefcase.briefcaseId,
+                changeset: { id: "", index: 0 }
+            }, locks);
+        }
+        else {
+            await IModelHost_1.IModelHost[Symbols_1._hubAccess].abandonLocks(this.briefcase, locks);
+        }
+    }
+    async abandonLocksForCurrentUnsavedTxn() {
+        return this.abandonLocksForReversedTxn(this._unsavedChangesTxnId);
+    }
+    async abandonLocksForReversedTxn(txnId) {
+        if (this.briefcase.txns.hasUnsavedChanges)
+            core_common_1.ServerBasedLocksError.throwError("has-unsaved-changes", `cannot abandon locks for txn ${txnId} because the current txn has unsaved changes`);
+        const txnProps = this.briefcase.txns.getTxnProps(txnId);
+        if (txnProps === undefined) {
+            // The current txn commonly won't exist on the TxnManager yet. It's often just a placeholder for not-yet-saved changes.
+            // (Sometimes it will exist and refer to a reversed Txn).
+            // The unsavedChangesTxnId won't exist on the TxnManager either, of course.
+            // But all other txn ids must be known to the TxnManager or it is an error.
+            if (txnId !== this.briefcase.txns.getCurrentTxnId() && txnId !== this._unsavedChangesTxnId)
+                core_common_1.ServerBasedLocksError.throwError("txn-id-not-found", `cannot abandon locks for txn ${txnId} because it does not exist`);
+        }
+        else {
+            // If the txn id is known to the TxnManager, then we require that it has already been reversed.
+            if (!txnProps.reversed)
+                core_common_1.ServerBasedLocksError.throwError("txn-not-reversed", `cannot abandon locks for txn ${txnId} because it has not been reversed`);
+        }
+        let locksReleased = false;
+        // Abandon locks for unsaved (and now abandoned) changes.
+        if (txnId !== this._unsavedChangesTxnId) {
+            locksReleased = await this.abandonLocksForCurrentUnsavedTxn();
+        }
+        // At this point, we know:
+        // 1. There are no unsaved changes, and the associated locks have been abandoned.
+        // 2. The given txn ID has been reversed, which means any later txns are sure to have been reversed, too.
+        // So we simply have to find all non-abandoned locks associated with the given txn or later, abandon them (or restore
+        // them to their previous state), and mark them as abandoned in txn_locks.
+        // Find all locks associated with the given txnId or later.
+        // For each elementId, find the previous state of the lock before this Txn (if any), or None otherwise.
+        // This is the state that we will restore the element's lock to. The reason we do this is to account for
+        // lock upgrades. If an earlier Txn acquired a Shared lock on this element, and this Txn acquired an
+        // Exclusive lock, we should restore the Shared lock.
+        const allTxnLocks = new Map();
+        const locksToRelease = new Map();
+        this.lockDb.withPreparedSqliteStatement(`
+        SELECT
+          current.elementId,
+          current.origin,
+          IFNULL(
+            (SELECT previous.state
+              FROM txn_locks previous
+              WHERE previous.elementId = current.elementId
+                AND previous.txnId < ?2
+                AND previous.abandoned=FALSE
+              ORDER BY previous.txnId DESC
+              LIMIT 1
+            ),
+            ?1
+          ) AS previousState
+        FROM txn_locks current
+        WHERE current.txnId>=?2
+          AND current.abandoned=FALSE
+        ORDER BY current.txnId DESC
+      `, (stmt) => {
+            stmt.bindInteger(1, core_common_1.LockState.None);
+            stmt.bindId(2, txnId);
+            while (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step()) {
+                const elementId = stmt.getValueId(0);
+                const origin = stmt.getValueInteger(1);
+                const previousState = stmt.getValueInteger(2);
+                allTxnLocks.set(elementId, previousState);
+                if (origin !== 1 /* LockOrigin.NewElement */)
+                    locksToRelease.set(elementId, previousState);
+            }
+        });
+        // Release the locks on the server.
+        if (locksToRelease.size > 0)
+            await this.abandonLocks(locksToRelease);
+        // Mark the txn locks as abandoned.
+        if (txnId === this._unsavedChangesTxnId) {
+            // After abandoning locks held for the "unsaved" txn, we clear them completely because they are not reinstateable.
+            this.lockDb.withPreparedSqliteStatement("DELETE FROM txn_locks WHERE txnId=?", (stmt) => {
+                stmt.bindId(1, this._unsavedChangesTxnId);
+                const rc = stmt.step();
+                if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                    core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't delete txn locks for unsaved changes in database (error code ${rc})`);
+            });
+        }
+        else {
+            this.lockDb.withPreparedSqliteStatement("UPDATE txn_locks SET abandoned=TRUE WHERE txnId>=?", (stmt) => {
+                stmt.bindId(1, txnId);
+                const rc = stmt.step();
+                if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                    core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't mark txn locks as abandoned in database (error code ${rc})`);
+            });
+        }
+        // Restore each lock to its previous state (if any) in the local cache. Usually this means deleting it.
+        for (const [elementId, previousState] of allTxnLocks) {
+            if (previousState === core_common_1.LockState.None) {
+                this.lockDb.withPreparedSqliteStatement("DELETE FROM locks WHERE id=?", (stmt) => {
+                    stmt.bindId(1, elementId);
+                    const rc = stmt.step();
+                    if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                        core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't delete lock from database (error code ${rc})`);
+                });
+            }
+            else {
+                this.lockDb.withPreparedSqliteStatement("UPDATE locks SET state=? WHERE id=?", (stmt) => {
+                    stmt.bindInteger(1, previousState);
+                    stmt.bindId(2, elementId);
+                    const rc = stmt.step();
+                    if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                        core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't update lock in database (error code ${rc})`);
+                });
+            }
+        }
+        // Ideally we'd only invalidate "Discovered" locks that are related to this Txn's Shared and
+        // Exclusive locks. But that is a lot of added complexity for little benefit.
+        // Clearing them all will have no impact on correctness and a minimal impact on performance.
+        this.clearDiscoveredLocks();
+        this.lockDb.saveChanges();
+        return locksReleased || allTxnLocks.size > 0;
+    }
+    getAbandonedLocksForTxn(txnId) {
+        const newElementLocks = new Map();
+        const locksToAcquire = new Map();
+        this.lockDb.withPreparedSqliteStatement("SELECT elementId, state, origin FROM txn_locks WHERE txnId<=? AND abandoned=TRUE", (stmt) => {
+            stmt.bindId(1, txnId);
+            while (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step()) {
+                const elementId = stmt.getValueId(0);
+                const state = stmt.getValueInteger(1);
+                const origin = stmt.getValueInteger(2);
+                if (origin === 1 /* LockOrigin.NewElement */)
+                    newElementLocks.set(elementId, state);
+                else
+                    locksToAcquire.set(elementId, state);
+            }
+        });
+        return { newElementLocks, locksToAcquire };
+    }
+    async acquireLocksForReinstatingTxn(txnId) {
+        if (this.briefcase.txns.hasUnsavedChanges)
+            core_common_1.ServerBasedLocksError.throwError("has-unsaved-changes", `cannot acquire locks for reinstating txn ${txnId} because the current txn has unsaved changes`);
+        // If the Txn is known to the TxnManager, we can proceed. We don't need to check if it is currently
+        // reversed, because if it isn't, then abandonLocksForReversedTxn couldn't have been called, and so the
+        // locks are still held. Proceeding with this method will be a no-op, but it will be harmless.
+        // However, if the Txn Id is unknown, it may have been canceled or refer to the current Txn
+        // whose unsaved changes were just abandoned. Or it's just plain-old invalid. In any case, we can't
+        // re-acquire the associated locks.
+        const txnProps = this.briefcase.txns.getTxnProps(txnId);
+        if (txnProps === undefined) {
+            core_common_1.ServerBasedLocksError.throwError("txn-id-not-found", `cannot acquire locks for txn ${txnId} because it does not exist or has not been saved`);
+        }
+        // Find all locks associated with the given txnId.
+        const { newElementLocks, locksToAcquire } = this.getAbandonedLocksForTxn(txnId);
+        // Attempt to acquire the locks on the server. This may fail if the locks are no longer available!
+        if (locksToAcquire.size > 0)
+            await IModelHost_1.IModelHost[Symbols_1._hubAccess].acquireLocks(this.briefcase, locksToAcquire); // throws if unsuccessful
+        // Mark the txn locks as no longer abandoned.
+        this.lockDb.withPreparedSqliteStatement("UPDATE txn_locks SET abandoned=FALSE WHERE txnId<=?", (stmt) => {
+            stmt.bindId(1, txnId);
+            const rc = stmt.step();
+            if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't mark txn locks as no longer abandoned in database (error code ${rc})`);
+        });
+        // Insert the newly-acquired locks in the local cache. Note that we don't need to insert entries in the txn_locks table,
+        // because these locks are already associated with the Txn.
+        for (const [elementId, state] of locksToAcquire) {
+            this.insertLock(elementId, state, 0 /* LockOrigin.Acquired */);
+        }
+        for (const [elementId, state] of newElementLocks) {
+            this.insertLock(elementId, state, 1 /* LockOrigin.NewElement */);
+        }
+        // Ideally we'd only invalidate "Discovered" locks that are related to this Txn's Shared and
+        // Exclusive locks. But that is a lot of added complexity for little benefit.
+        // Clearing them all will have no impact on correctness and a minimal impact on performance.
+        this.clearDiscoveredLocks();
+        this.lockDb.saveChanges();
+        return locksToAcquire.size > 0 || newElementLocks.size > 0;
+    }
+    holdsNecessaryLocksForReinstatingTxn(txnId) {
+        const locks = this.getAbandonedLocksForTxn(txnId);
+        return locks.locksToAcquire.size === 0 && locks.newElementLocks.size === 0;
+    }
+    clearTxnLockRecords(txnId) {
+        this.lockDb.withPreparedSqliteStatement("DELETE FROM txn_locks WHERE txnId!=? AND txnId>=?", (stmt) => {
+            stmt.bindId(1, this._unsavedChangesTxnId);
+            stmt.bindId(2, txnId);
+            const rc = stmt.step();
+            if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't delete txn lock records from database (error code ${rc})`);
+        });
+        this.lockDb.saveChanges();
+    }
     /** When an element is newly created in a session, we hold the lock on it implicitly. Save that fact. */
     [Symbols_1._elementWasCreated](id) {
         this.insertLock(id, core_common_1.LockState.Exclusive, 1 /* LockOrigin.NewElement */);
+        this.insertTxnLockRecord(id, core_common_1.LockState.Exclusive, 1 /* LockOrigin.NewElement */);
         this.lockDb.saveChanges();
     }
+    clearDiscoveredLocks() {
+        this.lockDb.withPreparedSqliteStatement("DELETE FROM locks WHERE origin=?", (stmt) => {
+            stmt.bindInteger(1, 2 /* LockOrigin.Discovered */);
+            const rc = stmt.step();
+            if (core_bentley_1.DbResult.BE_SQLITE_DONE !== rc)
+                core_common_1.ServerBasedLocksError.throwError("lock-database-problem", `can't delete discovered locks from database (error code ${rc})`);
+        });
+    }
     /** locks are not necessary during change propagation. */
     get _locksAreRequired() { return !this.briefcase.txns.isIndirectChanges; }
     /** throw if locks are currently required and the exclusive lock is not held on the supplied element */