@itwin/presentation-testing 3.4.0-dev.9 → 3.4.1

package/lib/cjs/presentation-testing/HierarchyBuilder.js

@@ -1,69 +1,69 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.HierarchyBuilder = exports.defaultNodeMappingFunc = void 0;
-/*---------------------------------------------------------------------------------------------
-* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
-* See LICENSE.md in the project root for license terms and full copyright notice.
-*--------------------------------------------------------------------------------------------*/
-/** @packageDocumentation
- * @module Hierarchies
- */
-const core_bentley_1 = require("@itwin/core-bentley");
-const presentation_components_1 = require("@itwin/presentation-components");
-const presentation_frontend_1 = require("@itwin/presentation-frontend");
-/**
- * Default [[NodeMappingFunc]] implementation that outputs the whole `TreeNodeItem` object.
- * @public
- */
-const defaultNodeMappingFunc = (node) => {
-    // Skip properties 'id', 'parentId' as they contain  internal stuff
-    // that callers are most likely not interested in. Otherwise they can supply
-    // a custom `NodeMappingFunc` that does return those properties as well.
-    const { id, parentId, ...resultNode } = node; // eslint-disable-line @typescript-eslint/no-unused-vars
-    return resultNode;
-};
-exports.defaultNodeMappingFunc = defaultNodeMappingFunc;
-/**
- * A class that constructs simple node hierarchy from specified
- * imodel and ruleset.
- *
- * @public
- */
-class HierarchyBuilder {
-    /** Constructor */
-    constructor(props) {
-        var _a;
-        this._iModel = props.imodel;
-        this._nodeMappingFunc = (_a = props.nodeMappingFunc) !== null && _a !== void 0 ? _a : exports.defaultNodeMappingFunc;
-    }
-    async createSubHierarchy(nodes, dataProvider) {
-        const hierarchy = [];
-        for (const node of nodes) {
-            // istanbul ignore next: for some reason coverage tool thinks the below statement is conditional and one of branches is not covered...
-            const { [presentation_components_1.PRESENTATION_TREE_NODE_KEY]: key, ...nodeNoKey } = node; // eslint-disable-line @typescript-eslint/no-unused-vars
-            const nodeIndex = hierarchy.push(this._nodeMappingFunc(nodeNoKey)) - 1;
-            const childNodes = await dataProvider.getNodes(node);
-            if (childNodes.length > 0)
-                hierarchy[nodeIndex].children = await this.createSubHierarchy(childNodes, dataProvider);
-        }
-        return hierarchy;
-    }
-    async doCreateHierarchy(rulesetId) {
-        const dataProvider = new presentation_components_1.PresentationTreeDataProvider({ imodel: this._iModel, ruleset: rulesetId });
-        const rootNodes = await dataProvider.getNodes();
-        return this.createSubHierarchy(rootNodes, dataProvider);
-    }
-    /**
-     * Create a hierarchy using the supplied presentation ruleset.
-     * @param rulesetOrId Either a [[Ruleset]] object or a ruleset id.
-     */
-    async createHierarchy(rulesetOrId) {
-        if (typeof rulesetOrId === "string")
-            return this.doCreateHierarchy(rulesetOrId);
-        return (0, core_bentley_1.using)(await presentation_frontend_1.Presentation.presentation.rulesets().add(rulesetOrId), async (ruleset) => {
-            return this.doCreateHierarchy(ruleset.id);
-        });
-    }
-}
-exports.HierarchyBuilder = HierarchyBuilder;
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HierarchyBuilder = exports.defaultNodeMappingFunc = void 0;
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+/** @packageDocumentation
+ * @module Hierarchies
+ */
+const core_bentley_1 = require("@itwin/core-bentley");
+const presentation_components_1 = require("@itwin/presentation-components");
+const presentation_frontend_1 = require("@itwin/presentation-frontend");
+/**
+ * Default [[NodeMappingFunc]] implementation that outputs the whole `TreeNodeItem` object.
+ * @public
+ */
+const defaultNodeMappingFunc = (node) => {
+    // Skip properties 'id', 'parentId' as they contain  internal stuff
+    // that callers are most likely not interested in. Otherwise they can supply
+    // a custom `NodeMappingFunc` that does return those properties as well.
+    const { id, parentId, ...resultNode } = node; // eslint-disable-line @typescript-eslint/no-unused-vars
+    return resultNode;
+};
+exports.defaultNodeMappingFunc = defaultNodeMappingFunc;
+/**
+ * A class that constructs simple node hierarchy from specified
+ * imodel and ruleset.
+ *
+ * @public
+ */
+class HierarchyBuilder {
+    /** Constructor */
+    constructor(props) {
+        var _a;
+        this._iModel = props.imodel;
+        this._nodeMappingFunc = (_a = props.nodeMappingFunc) !== null && _a !== void 0 ? _a : exports.defaultNodeMappingFunc;
+    }
+    async createSubHierarchy(nodes, dataProvider) {
+        const hierarchy = [];
+        for (const node of nodes) {
+            // istanbul ignore next: for some reason coverage tool thinks the below statement is conditional and one of branches is not covered...
+            const { [presentation_components_1.PRESENTATION_TREE_NODE_KEY]: key, ...nodeNoKey } = node; // eslint-disable-line @typescript-eslint/no-unused-vars
+            const nodeIndex = hierarchy.push(this._nodeMappingFunc(nodeNoKey)) - 1;
+            const childNodes = await dataProvider.getNodes(node);
+            if (childNodes.length > 0)
+                hierarchy[nodeIndex].children = await this.createSubHierarchy(childNodes, dataProvider);
+        }
+        return hierarchy;
+    }
+    async doCreateHierarchy(rulesetId) {
+        const dataProvider = new presentation_components_1.PresentationTreeDataProvider({ imodel: this._iModel, ruleset: rulesetId });
+        const rootNodes = await dataProvider.getNodes();
+        return this.createSubHierarchy(rootNodes, dataProvider);
+    }
+    /**
+     * Create a hierarchy using the supplied presentation ruleset.
+     * @param rulesetOrId Either a [[Ruleset]] object or a ruleset id.
+     */
+    async createHierarchy(rulesetOrId) {
+        if (typeof rulesetOrId === "string")
+            return this.doCreateHierarchy(rulesetOrId);
+        return (0, core_bentley_1.using)(await presentation_frontend_1.Presentation.presentation.rulesets().add(rulesetOrId), async (ruleset) => {
+            return this.doCreateHierarchy(ruleset.id);
+        });
+    }
+}
+exports.HierarchyBuilder = HierarchyBuilder;
 //# sourceMappingURL=HierarchyBuilder.js.map

package/lib/cjs/presentation-testing/HierarchyBuilder.js.map

@@ -1 +1 @@
-{"version":3,"file":"HierarchyBuilder.js","sourceRoot":"","sources":["../../../src/presentation-testing/HierarchyBuilder.ts"],"names":[],"mappings":";;;AAAA;;;+FAG+F;AAC/F;;GAEG;AACH,sDAA4C;AAG5C,4EAA0G;AAC1G,wEAA4D;AA+B5D;;;GAGG;AACI,MAAM,sBAAsB,GAAoB,CAAC,IAAkB,EAAE,EAAE;IAC5E,mEAAmE;IACnE,4EAA4E;IAC5E,wEAAwE;IACxE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,UAAU,EAAE,GAAG,IAAI,CAAC,CAAC,wDAAwD;IACtG,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AANW,QAAA,sBAAsB,0BAMjC;AAiBF;;;;;GAKG;AACH,MAAa,gBAAgB;IAI3B,kBAAkB;IAClB,YAAY,KAA4B;;QACtC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,gBAAgB,GAAG,MAAA,KAAK,CAAC,eAAe,mCAAI,8BAAsB,CAAC;IAC1E,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,KAAqB,EAAE,YAA0C;QAChG,MAAM,SAAS,GAAoB,EAAE,CAAC;QACtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;YACxB,sIAAsI;YACtI,MAAM,EAAE,CAAC,oDAA0B,CAAC,EAAE,GAAG,EAAE,GAAG,SAAS,EAAE,GAAI,IAA6D,CAAC,CAAC,wDAAwD;YACpL,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC;YACvE,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC;gBACvB,SAAS,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;SAC3F;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QAC/C,MAAM,YAAY,GAAG,IAAI,sDAA4B,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QACpG,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,QAAQ,EAAE,CAAC;QAChD,OAAO,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,eAAe,CAAC,WAA6B;QACxD,IAAI,OAAO,WAAW,KAAK,QAAQ;YACjC,OAAO,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAE7C,OAAO,IAAA,oBAAK,EAAC,MAAM,oCAAY,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,OAA0B,EAAE,EAAE;YAC7G,OAAO,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAzCD,4CAyCC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\n/** @packageDocumentation\r\n * @module Hierarchies\r\n */\r\nimport { using } from \"@itwin/core-bentley\";\r\nimport { IModelConnection } from \"@itwin/core-frontend\";\r\nimport { Omit, RegisteredRuleset, Ruleset } from \"@itwin/presentation-common\";\r\nimport { PRESENTATION_TREE_NODE_KEY, PresentationTreeDataProvider } from \"@itwin/presentation-components\";\r\nimport { Presentation } from \"@itwin/presentation-frontend\";\r\nimport { TreeNodeItem } from \"@itwin/components-react\";\r\n\r\n/**\r\n * Structure that describes a Node with any indexed properties\r\n * except `children`.\r\n *\r\n * @public\r\n */\r\nexport interface MappedNode {\r\n  /** Indexer for all properties in this data structure */\r\n  [index: string]: any;\r\n  /** Prohibited property */\r\n  children?: never;\r\n}\r\n\r\n/**\r\n * Node in a hierarchy.\r\n * @public\r\n */\r\nexport interface HierarchyNode extends Omit<MappedNode, \"children\"> {\r\n  /** Children of this node */\r\n  children?: HierarchyNode[];\r\n}\r\n\r\n/**\r\n * A function that converts `TreeNodeItem` into a new custom object.\r\n * @public\r\n */\r\nexport type NodeMappingFunc = (node: TreeNodeItem) => MappedNode;\r\n\r\n/**\r\n * Default [[NodeMappingFunc]] implementation that outputs the whole `TreeNodeItem` object.\r\n * @public\r\n */\r\nexport const defaultNodeMappingFunc: NodeMappingFunc = (node: TreeNodeItem) => {\r\n  // Skip properties 'id', 'parentId' as they contain  internal stuff\r\n  // that callers are most likely not interested in. Otherwise they can supply\r\n  // a custom `NodeMappingFunc` that does return those properties as well.\r\n  const { id, parentId, ...resultNode } = node; // eslint-disable-line @typescript-eslint/no-unused-vars\r\n  return resultNode;\r\n};\r\n\r\n/**\r\n * Properties for creating a `HierarchyBuilder` instance.\r\n * @public\r\n */\r\nexport interface HierarchyBuilderProps {\r\n  /** The iModel to pull data from */\r\n  imodel: IModelConnection;\r\n  /**\r\n   * A function that maps node to something that the user of\r\n   * this API is interested in. E.g. custom implementation may skip some unimportant\r\n   * node properties to make resulting object smaller and easier to read.\r\n   */\r\n  nodeMappingFunc?: NodeMappingFunc;\r\n}\r\n\r\n/**\r\n * A class that constructs simple node hierarchy from specified\r\n * imodel and ruleset.\r\n *\r\n * @public\r\n */\r\nexport class HierarchyBuilder {\r\n  private readonly _iModel: IModelConnection;\r\n  private readonly _nodeMappingFunc: NodeMappingFunc;\r\n\r\n  /** Constructor */\r\n  constructor(props: HierarchyBuilderProps) {\r\n    this._iModel = props.imodel;\r\n    this._nodeMappingFunc = props.nodeMappingFunc ?? defaultNodeMappingFunc;\r\n  }\r\n\r\n  private async createSubHierarchy(nodes: TreeNodeItem[], dataProvider: PresentationTreeDataProvider) {\r\n    const hierarchy: HierarchyNode[] = [];\r\n    for (const node of nodes) {\r\n      // istanbul ignore next: for some reason coverage tool thinks the below statement is conditional and one of branches is not covered...\r\n      const { [PRESENTATION_TREE_NODE_KEY]: key, ...nodeNoKey } = (node as TreeNodeItem & { [PRESENTATION_TREE_NODE_KEY]: any }); // eslint-disable-line @typescript-eslint/no-unused-vars\r\n      const nodeIndex = hierarchy.push(this._nodeMappingFunc(nodeNoKey)) - 1;\r\n      const childNodes = await dataProvider.getNodes(node);\r\n      if (childNodes.length > 0)\r\n        hierarchy[nodeIndex].children = await this.createSubHierarchy(childNodes, dataProvider);\r\n    }\r\n    return hierarchy;\r\n  }\r\n\r\n  private async doCreateHierarchy(rulesetId: string): Promise<HierarchyNode[]> {\r\n    const dataProvider = new PresentationTreeDataProvider({ imodel: this._iModel, ruleset: rulesetId });\r\n    const rootNodes = await dataProvider.getNodes();\r\n    return this.createSubHierarchy(rootNodes, dataProvider);\r\n  }\r\n\r\n  /**\r\n   * Create a hierarchy using the supplied presentation ruleset.\r\n   * @param rulesetOrId Either a [[Ruleset]] object or a ruleset id.\r\n   */\r\n  public async createHierarchy(rulesetOrId: Ruleset | string): Promise<HierarchyNode[]> {\r\n    if (typeof rulesetOrId === \"string\")\r\n      return this.doCreateHierarchy(rulesetOrId);\r\n\r\n    return using(await Presentation.presentation.rulesets().add(rulesetOrId), async (ruleset: RegisteredRuleset) => {\r\n      return this.doCreateHierarchy(ruleset.id);\r\n    });\r\n  }\r\n}\r\n"]}
+{"version":3,"file":"HierarchyBuilder.js","sourceRoot":"","sources":["../../../src/presentation-testing/HierarchyBuilder.ts"],"names":[],"mappings":";;;AAAA;;;+FAG+F;AAC/F;;GAEG;AACH,sDAA4C;AAG5C,4EAA0G;AAC1G,wEAA4D;AA+B5D;;;GAGG;AACI,MAAM,sBAAsB,GAAoB,CAAC,IAAkB,EAAE,EAAE;IAC5E,mEAAmE;IACnE,4EAA4E;IAC5E,wEAAwE;IACxE,MAAM,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,UAAU,EAAE,GAAG,IAAI,CAAC,CAAC,wDAAwD;IACtG,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AANW,QAAA,sBAAsB,0BAMjC;AAiBF;;;;;GAKG;AACH,MAAa,gBAAgB;IAI3B,kBAAkB;IAClB,YAAY,KAA4B;;QACtC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,gBAAgB,GAAG,MAAA,KAAK,CAAC,eAAe,mCAAI,8BAAsB,CAAC;IAC1E,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,KAAqB,EAAE,YAA0C;QAChG,MAAM,SAAS,GAAoB,EAAE,CAAC;QACtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;YACxB,sIAAsI;YACtI,MAAM,EAAE,CAAC,oDAA0B,CAAC,EAAE,GAAG,EAAE,GAAG,SAAS,EAAE,GAAI,IAA6D,CAAC,CAAC,wDAAwD;YACpL,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC;YACvE,MAAM,UAAU,GAAG,MAAM,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACrD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC;gBACvB,SAAS,CAAC,SAAS,CAAC,CAAC,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;SAC3F;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QAC/C,MAAM,YAAY,GAAG,IAAI,sDAA4B,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;QACpG,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,QAAQ,EAAE,CAAC;QAChD,OAAO,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;IAC1D,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,eAAe,CAAC,WAA6B;QACxD,IAAI,OAAO,WAAW,KAAK,QAAQ;YACjC,OAAO,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAE7C,OAAO,IAAA,oBAAK,EAAC,MAAM,oCAAY,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,OAA0B,EAAE,EAAE;YAC7G,OAAO,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAzCD,4CAyCC","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/** @packageDocumentation\n * @module Hierarchies\n */\nimport { using } from \"@itwin/core-bentley\";\nimport { IModelConnection } from \"@itwin/core-frontend\";\nimport { Omit, RegisteredRuleset, Ruleset } from \"@itwin/presentation-common\";\nimport { PRESENTATION_TREE_NODE_KEY, PresentationTreeDataProvider } from \"@itwin/presentation-components\";\nimport { Presentation } from \"@itwin/presentation-frontend\";\nimport { TreeNodeItem } from \"@itwin/components-react\";\n\n/**\n * Structure that describes a Node with any indexed properties\n * except `children`.\n *\n * @public\n */\nexport interface MappedNode {\n  /** Indexer for all properties in this data structure */\n  [index: string]: any;\n  /** Prohibited property */\n  children?: never;\n}\n\n/**\n * Node in a hierarchy.\n * @public\n */\nexport interface HierarchyNode extends Omit<MappedNode, \"children\"> {\n  /** Children of this node */\n  children?: HierarchyNode[];\n}\n\n/**\n * A function that converts `TreeNodeItem` into a new custom object.\n * @public\n */\nexport type NodeMappingFunc = (node: TreeNodeItem) => MappedNode;\n\n/**\n * Default [[NodeMappingFunc]] implementation that outputs the whole `TreeNodeItem` object.\n * @public\n */\nexport const defaultNodeMappingFunc: NodeMappingFunc = (node: TreeNodeItem) => {\n  // Skip properties 'id', 'parentId' as they contain  internal stuff\n  // that callers are most likely not interested in. Otherwise they can supply\n  // a custom `NodeMappingFunc` that does return those properties as well.\n  const { id, parentId, ...resultNode } = node; // eslint-disable-line @typescript-eslint/no-unused-vars\n  return resultNode;\n};\n\n/**\n * Properties for creating a `HierarchyBuilder` instance.\n * @public\n */\nexport interface HierarchyBuilderProps {\n  /** The iModel to pull data from */\n  imodel: IModelConnection;\n  /**\n   * A function that maps node to something that the user of\n   * this API is interested in. E.g. custom implementation may skip some unimportant\n   * node properties to make resulting object smaller and easier to read.\n   */\n  nodeMappingFunc?: NodeMappingFunc;\n}\n\n/**\n * A class that constructs simple node hierarchy from specified\n * imodel and ruleset.\n *\n * @public\n */\nexport class HierarchyBuilder {\n  private readonly _iModel: IModelConnection;\n  private readonly _nodeMappingFunc: NodeMappingFunc;\n\n  /** Constructor */\n  constructor(props: HierarchyBuilderProps) {\n    this._iModel = props.imodel;\n    this._nodeMappingFunc = props.nodeMappingFunc ?? defaultNodeMappingFunc;\n  }\n\n  private async createSubHierarchy(nodes: TreeNodeItem[], dataProvider: PresentationTreeDataProvider) {\n    const hierarchy: HierarchyNode[] = [];\n    for (const node of nodes) {\n      // istanbul ignore next: for some reason coverage tool thinks the below statement is conditional and one of branches is not covered...\n      const { [PRESENTATION_TREE_NODE_KEY]: key, ...nodeNoKey } = (node as TreeNodeItem & { [PRESENTATION_TREE_NODE_KEY]: any }); // eslint-disable-line @typescript-eslint/no-unused-vars\n      const nodeIndex = hierarchy.push(this._nodeMappingFunc(nodeNoKey)) - 1;\n      const childNodes = await dataProvider.getNodes(node);\n      if (childNodes.length > 0)\n        hierarchy[nodeIndex].children = await this.createSubHierarchy(childNodes, dataProvider);\n    }\n    return hierarchy;\n  }\n\n  private async doCreateHierarchy(rulesetId: string): Promise<HierarchyNode[]> {\n    const dataProvider = new PresentationTreeDataProvider({ imodel: this._iModel, ruleset: rulesetId });\n    const rootNodes = await dataProvider.getNodes();\n    return this.createSubHierarchy(rootNodes, dataProvider);\n  }\n\n  /**\n   * Create a hierarchy using the supplied presentation ruleset.\n   * @param rulesetOrId Either a [[Ruleset]] object or a ruleset id.\n   */\n  public async createHierarchy(rulesetOrId: Ruleset | string): Promise<HierarchyNode[]> {\n    if (typeof rulesetOrId === \"string\")\n      return this.doCreateHierarchy(rulesetOrId);\n\n    return using(await Presentation.presentation.rulesets().add(rulesetOrId), async (ruleset: RegisteredRuleset) => {\n      return this.doCreateHierarchy(ruleset.id);\n    });\n  }\n}\n"]}

package/lib/cjs/presentation-testing/IModelBuilder.d.ts

@@ -0,0 +1,46 @@
+/** @packageDocumentation
+ * @module IModel
+ */
+import { Element, IModelDb } from "@itwin/core-backend";
+import { Id64String } from "@itwin/core-bentley";
+import { Code, CodeProps, LocalFileName, RelatedElement } from "@itwin/core-common";
+export declare class IModelBuilder {
+    /**
+     * Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel.
+     * @return [modeledElementId, modelId]
+     */
+    static createAndInsertPhysicalPartitionAndModel(testImodel: IModelDb, newModelCode: CodeProps, privateModel?: boolean, parent?: Id64String): Id64String[];
+    /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */
+    static createAndInsertPhysicalPartition(testDb: IModelDb, newModelCode: CodeProps, parentId?: Id64String): Id64String;
+    /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */
+    static createAndInsertPhysicalModel(testDb: IModelDb, modeledElementRef: RelatedElement, privateModel?: boolean): Id64String;
+    /**
+     * Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel.
+     * @return [modeledElementId, modelId]
+     */
+    static createAndInsertDrawingPartitionAndModel(testImodel: IModelDb, newModelCode: CodeProps, privateModel?: boolean, parent?: Id64String): Id64String[];
+    /** Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel. */
+    static createAndInsertDrawingPartition(testDb: IModelDb, newModelCode: CodeProps, parentId?: Id64String): Id64String;
+    /** Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel. */
+    static createAndInsertDrawingModel(testDb: IModelDb, modeledElementRef: RelatedElement, privateModel?: boolean): Id64String;
+    static createDrawingGraphic(testImodel: IModelDb, modelId: Id64String, categoryId: Id64String, elemCode?: Code): Element;
+    static createPhysicalObject(testImodel: IModelDb, modelId: Id64String, categoryId: Id64String, elemCode?: Code): Element;
+    static getUniqueModelCode(testDb: IModelDb, newModelCodeBase: string): Code;
+    /** Generate a name for an iModel that's unique using the baseName provided and appending a new GUID.  */
+    static generateUniqueName(baseName: string): string;
+    /** Prepare for an output file by:
+     * - Resolving the output file name under the known test output directory
+     * - Making directories as necessary
+     * - Removing a previous copy of the output file
+     * @param subDirName Sub-directory under known test output directory. Should match the name of the test file minus the .test.ts file extension.
+     * @param fileName Name of output fille
+     */
+    static prepareOutputFile(subDirName: string, fileName: string): LocalFileName;
+}
+export declare class KnownTestLocations {
+    /** The directory where test assets are stored. Keep in mind that the test is playing the role of the app. */
+    static get assetsDir(): string;
+    /** The directory where tests can write. */
+    static get outputDir(): string;
+}
+//# sourceMappingURL=IModelBuilder.d.ts.map

package/lib/cjs/presentation-testing/IModelBuilder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"IModelBuilder.d.ts","sourceRoot":"","sources":["../../../src/presentation-testing/IModelBuilder.ts"],"names":[],"mappings":"AAIA;;GAEG;AAEH,OAAO,EAAyB,OAAO,EAAE,QAAQ,EAA2G,MAAM,qBAAqB,CAAC;AACxL,OAAO,EAAc,UAAU,EAAmB,MAAM,qBAAqB,CAAC;AAC9E,OAAO,EAAE,IAAI,EAAE,SAAS,EAAiD,aAAa,EAAwB,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAKzJ,qBAAa,aAAa;IACxB;;;OAGG;WACW,wCAAwC,CAAC,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,GAAE,OAAe,EAAE,MAAM,CAAC,EAAE,UAAU,GAAG,UAAU,EAAE;IAOvK,8GAA8G;WAChG,gCAAgC,CAAC,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,UAAU,GAAG,UAAU;IAc5H,8GAA8G;WAChG,4BAA4B,CAAC,MAAM,EAAE,QAAQ,EAAE,iBAAiB,EAAE,cAAc,EAAE,YAAY,GAAE,OAAe,GAAG,UAAU;IAS1I;;;OAGG;WACW,uCAAuC,CAAC,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,YAAY,GAAE,OAAe,EAAE,MAAM,CAAC,EAAE,UAAU,GAAG,UAAU,EAAE;IAOtK,4GAA4G;WAC9F,+BAA+B,CAAC,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,QAAQ,CAAC,EAAE,UAAU,GAAG,UAAU;IAc3H,4GAA4G;WAC9F,2BAA2B,CAAC,MAAM,EAAE,QAAQ,EAAE,iBAAiB,EAAE,cAAc,EAAE,YAAY,GAAE,OAAe,GAAG,UAAU;WAU3H,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,IAAI,GAAG,OAAO;WAWjH,oBAAoB,CAAC,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,CAAC,EAAE,IAAI,GAAG,OAAO;WAUjH,kBAAkB,CAAC,MAAM,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,GAAG,IAAI;IAalF,yGAAyG;WAC3F,kBAAkB,CAAC,QAAQ,EAAE,MAAM;IAIjD;;;;;;OAMG;WACW,iBAAiB,CAAC,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,aAAa;CAcrF;AAED,qBAAa,kBAAkB;IAE7B,6GAA6G;IAC7G,WAAkB,SAAS,IAAI,MAAM,CAEpC;IAED,2CAA2C;IAC3C,WAAkB,SAAS,IAAI,MAAM,CAOpC;CACF"}

package/lib/cjs/presentation-testing/IModelBuilder.js

@@ -0,0 +1,171 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+/** @packageDocumentation
+ * @module IModel
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KnownTestLocations = exports.IModelBuilder = void 0;
+const core_backend_1 = require("@itwin/core-backend");
+const core_bentley_1 = require("@itwin/core-bentley");
+const core_common_1 = require("@itwin/core-common");
+const chai_1 = require("chai");
+const os_1 = require("os");
+const path_1 = __importStar(require("path"));
+class IModelBuilder {
+    /**
+     * Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel.
+     * @return [modeledElementId, modelId]
+     */
+    static createAndInsertPhysicalPartitionAndModel(testImodel, newModelCode, privateModel = false, parent) {
+        const eid = IModelBuilder.createAndInsertPhysicalPartition(testImodel, newModelCode, parent);
+        const modeledElementRef = new core_common_1.RelatedElement({ id: eid });
+        const mid = IModelBuilder.createAndInsertPhysicalModel(testImodel, modeledElementRef, privateModel);
+        return [eid, mid];
+    }
+    /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */
+    static createAndInsertPhysicalPartition(testDb, newModelCode, parentId) {
+        const model = parentId ? testDb.elements.getElement(parentId).model : core_common_1.IModel.repositoryModelId;
+        const parent = new core_backend_1.SubjectOwnsPartitionElements(parentId || core_common_1.IModel.rootSubjectId);
+        const modeledElementProps = {
+            classFullName: core_backend_1.PhysicalPartition.classFullName,
+            parent,
+            model,
+            code: newModelCode,
+        };
+        const modeledElement = testDb.elements.createElement(modeledElementProps);
+        return testDb.elements.insertElement(modeledElement.toJSON());
+    }
+    /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */
+    static createAndInsertPhysicalModel(testDb, modeledElementRef, privateModel = false) {
+        const newModel = testDb.models.createModel({ modeledElement: modeledElementRef, classFullName: core_backend_1.PhysicalModel.classFullName, isPrivate: privateModel });
+        const newModelId = newModel.id = testDb.models.insertModel(newModel.toJSON());
+        chai_1.assert.isTrue(core_bentley_1.Id64.isValidId64(newModelId));
+        chai_1.assert.isTrue(core_bentley_1.Id64.isValidId64(newModel.id));
+        chai_1.assert.deepEqual(newModelId, newModel.id);
+        return newModelId;
+    }
+    /**
+     * Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel.
+     * @return [modeledElementId, modelId]
+     */
+    static createAndInsertDrawingPartitionAndModel(testImodel, newModelCode, privateModel = false, parent) {
+        const eid = IModelBuilder.createAndInsertDrawingPartition(testImodel, newModelCode, parent);
+        const modeledElementRef = new core_common_1.RelatedElement({ id: eid });
+        const mid = IModelBuilder.createAndInsertDrawingModel(testImodel, modeledElementRef, privateModel);
+        return [eid, mid];
+    }
+    /** Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel. */
+    static createAndInsertDrawingPartition(testDb, newModelCode, parentId) {
+        const model = parentId ? testDb.elements.getElement(parentId).model : core_common_1.IModel.repositoryModelId;
+        const parent = new core_backend_1.SubjectOwnsPartitionElements(parentId || core_common_1.IModel.rootSubjectId);
+        const modeledElementProps = {
+            classFullName: core_backend_1.Drawing.classFullName,
+            parent,
+            model,
+            code: newModelCode,
+        };
+        const modeledElement = testDb.elements.createElement(modeledElementProps);
+        return testDb.elements.insertElement(modeledElement.toJSON());
+    }
+    /** Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel. */
+    static createAndInsertDrawingModel(testDb, modeledElementRef, privateModel = false) {
+        const newModel = testDb.models.createModel({ modeledElement: modeledElementRef, classFullName: core_backend_1.DrawingModel.classFullName, isPrivate: privateModel });
+        const newModelId = newModel.id = testDb.models.insertModel(newModel.toJSON());
+        chai_1.assert.isTrue(core_bentley_1.Id64.isValidId64(newModelId));
+        chai_1.assert.isTrue(core_bentley_1.Id64.isValidId64(newModel.id));
+        chai_1.assert.deepEqual(newModelId, newModel.id);
+        return newModelId;
+    }
+    // Create a DrawingObject. (Does not insert it.)
+    static createDrawingGraphic(testImodel, modelId, categoryId, elemCode) {
+        const elementProps = {
+            classFullName: "BisCore:DrawingGraphic",
+            model: modelId,
+            category: categoryId,
+            code: elemCode ? elemCode : core_common_1.Code.createEmpty(),
+        };
+        return testImodel.elements.createElement(elementProps);
+    }
+    // Create a PhysicalObject. (Does not insert it.)
+    static createPhysicalObject(testImodel, modelId, categoryId, elemCode) {
+        const elementProps = {
+            classFullName: "Generic:PhysicalObject",
+            model: modelId,
+            category: categoryId,
+            code: elemCode ? elemCode : core_common_1.Code.createEmpty(),
+        };
+        return testImodel.elements.createElement(elementProps);
+    }
+    static getUniqueModelCode(testDb, newModelCodeBase) {
+        let newModelCode = newModelCodeBase;
+        let iter = 0;
+        while (true) {
+            const modelCode = core_backend_1.InformationPartitionElement.createCode(testDb, core_common_1.IModel.rootSubjectId, newModelCode);
+            if (testDb.elements.queryElementIdByCode(modelCode) === undefined)
+                return modelCode;
+            newModelCode = newModelCodeBase + iter;
+            ++iter;
+        }
+    }
+    /** Generate a name for an iModel that's unique using the baseName provided and appending a new GUID.  */
+    static generateUniqueName(baseName) {
+        return `${baseName} - ${core_bentley_1.Guid.createValue()}`;
+    }
+    /** Prepare for an output file by:
+     * - Resolving the output file name under the known test output directory
+     * - Making directories as necessary
+     * - Removing a previous copy of the output file
+     * @param subDirName Sub-directory under known test output directory. Should match the name of the test file minus the .test.ts file extension.
+     * @param fileName Name of output fille
+     */
+    static prepareOutputFile(subDirName, fileName) {
+        if (!core_backend_1.IModelJsFs.existsSync(KnownTestLocations.outputDir))
+            core_backend_1.IModelJsFs.mkdirSync(KnownTestLocations.outputDir);
+        const outputDir = path_1.default.join(KnownTestLocations.outputDir, subDirName);
+        if (!core_backend_1.IModelJsFs.existsSync(outputDir))
+            core_backend_1.IModelJsFs.mkdirSync(outputDir);
+        const outputFile = path_1.default.join(outputDir, fileName);
+        if (core_backend_1.IModelJsFs.existsSync(outputFile))
+            core_backend_1.IModelJsFs.unlinkSync(outputFile);
+        return outputFile;
+    }
+}
+exports.IModelBuilder = IModelBuilder;
+class KnownTestLocations {
+    /** The directory where test assets are stored. Keep in mind that the test is playing the role of the app. */
+    static get assetsDir() {
+        return (0, path_1.join)(__dirname, "assets");
+    }
+    /** The directory where tests can write. */
+    static get outputDir() {
+        if (core_bentley_1.ProcessDetector.isMobileAppBackend) {
+            return (0, path_1.join)((0, os_1.tmpdir)(), "output");
+        }
+        // Assume that we are running in nodejs
+        return (0, path_1.join)(__dirname, "output");
+    }
+}
+exports.KnownTestLocations = KnownTestLocations;
+//# sourceMappingURL=IModelBuilder.js.map

package/lib/cjs/presentation-testing/IModelBuilder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"IModelBuilder.js","sourceRoot":"","sources":["../../../src/presentation-testing/IModelBuilder.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;AAC/F;;GAEG;;;;;;;;;;;;;;;;;;;;;;AAEH,sDAAwL;AACxL,sDAA8E;AAC9E,oDAAyJ;AACzJ,+BAA8B;AAC9B,2BAA4B;AAC5B,6CAAkC;AAElC,MAAa,aAAa;IACxB;;;OAGG;IACI,MAAM,CAAC,wCAAwC,CAAC,UAAoB,EAAE,YAAuB,EAAE,eAAwB,KAAK,EAAE,MAAmB;QACtJ,MAAM,GAAG,GAAG,aAAa,CAAC,gCAAgC,CAAC,UAAU,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAC7F,MAAM,iBAAiB,GAAG,IAAI,4BAAc,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1D,MAAM,GAAG,GAAG,aAAa,CAAC,4BAA4B,CAAC,UAAU,EAAE,iBAAiB,EAAE,YAAY,CAAC,CAAC;QACpG,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACpB,CAAC;IAED,8GAA8G;IACvG,MAAM,CAAC,gCAAgC,CAAC,MAAgB,EAAE,YAAuB,EAAE,QAAqB;QAC7G,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,oBAAM,CAAC,iBAAiB,CAAC;QAC/F,MAAM,MAAM,GAAG,IAAI,2CAA4B,CAAC,QAAQ,IAAI,oBAAM,CAAC,aAAa,CAAC,CAAC;QAElF,MAAM,mBAAmB,GAAiB;YACxC,aAAa,EAAE,gCAAiB,CAAC,aAAa;YAC9C,MAAM;YACN,KAAK;YACL,IAAI,EAAE,YAAY;SACnB,CAAC;QACF,MAAM,cAAc,GAAY,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,mBAAmB,CAAC,CAAC;QACnF,OAAO,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,8GAA8G;IACvG,MAAM,CAAC,4BAA4B,CAAC,MAAgB,EAAE,iBAAiC,EAAE,eAAwB,KAAK;QAC3H,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,iBAAiB,EAAE,aAAa,EAAE,4BAAa,CAAC,aAAa,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;QACvJ,MAAM,UAAU,GAAG,QAAQ,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9E,aAAM,CAAC,MAAM,CAAC,mBAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC;QAC5C,aAAM,CAAC,MAAM,CAAC,mBAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7C,aAAM,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC1C,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,uCAAuC,CAAC,UAAoB,EAAE,YAAuB,EAAE,eAAwB,KAAK,EAAE,MAAmB;QACrJ,MAAM,GAAG,GAAG,aAAa,CAAC,+BAA+B,CAAC,UAAU,EAAE,YAAY,EAAE,MAAM,CAAC,CAAC;QAC5F,MAAM,iBAAiB,GAAG,IAAI,4BAAc,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1D,MAAM,GAAG,GAAG,aAAa,CAAC,2BAA2B,CAAC,UAAU,EAAE,iBAAiB,EAAE,YAAY,CAAC,CAAC;QACnG,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACpB,CAAC;IAED,4GAA4G;IACrG,MAAM,CAAC,+BAA+B,CAAC,MAAgB,EAAE,YAAuB,EAAE,QAAqB;QAC5G,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,oBAAM,CAAC,iBAAiB,CAAC;QAC/F,MAAM,MAAM,GAAG,IAAI,2CAA4B,CAAC,QAAQ,IAAI,oBAAM,CAAC,aAAa,CAAC,CAAC;QAElF,MAAM,mBAAmB,GAAiB;YACxC,aAAa,EAAE,sBAAO,CAAC,aAAa;YACpC,MAAM;YACN,KAAK;YACL,IAAI,EAAE,YAAY;SACnB,CAAC;QACF,MAAM,cAAc,GAAY,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,mBAAmB,CAAC,CAAC;QACnF,OAAO,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC;IAChE,CAAC;IAED,4GAA4G;IACrG,MAAM,CAAC,2BAA2B,CAAC,MAAgB,EAAE,iBAAiC,EAAE,eAAwB,KAAK;QAC1H,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,iBAAiB,EAAE,aAAa,EAAE,2BAAY,CAAC,aAAa,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC,CAAC;QACtJ,MAAM,UAAU,GAAG,QAAQ,CAAC,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9E,aAAM,CAAC,MAAM,CAAC,mBAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC;QAC5C,aAAM,CAAC,MAAM,CAAC,mBAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7C,aAAM,CAAC,SAAS,CAAC,UAAU,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC1C,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,gDAAgD;IACzC,MAAM,CAAC,oBAAoB,CAAC,UAAoB,EAAE,OAAmB,EAAE,UAAsB,EAAE,QAAe;QACnH,MAAM,YAAY,GAA4B;YAC5C,aAAa,EAAE,wBAAwB;YACvC,KAAK,EAAE,OAAO;YACd,QAAQ,EAAE,UAAU;YACpB,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,kBAAI,CAAC,WAAW,EAAE;SAC/C,CAAC;QACF,OAAO,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;IACzD,CAAC;IAED,iDAAiD;IAC1C,MAAM,CAAC,oBAAoB,CAAC,UAAoB,EAAE,OAAmB,EAAE,UAAsB,EAAE,QAAe;QACnH,MAAM,YAAY,GAAyB;YACzC,aAAa,EAAE,wBAAwB;YACvC,KAAK,EAAE,OAAO;YACd,QAAQ,EAAE,UAAU;YACpB,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,kBAAI,CAAC,WAAW,EAAE;SAC/C,CAAC;QACF,OAAO,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;IACzD,CAAC;IAEM,MAAM,CAAC,kBAAkB,CAAC,MAAgB,EAAE,gBAAwB;QACzE,IAAI,YAAY,GAAW,gBAAgB,CAAC;QAC5C,IAAI,IAAI,GAAW,CAAC,CAAC;QACrB,OAAO,IAAI,EAAE;YACX,MAAM,SAAS,GAAG,0CAA2B,CAAC,UAAU,CAAC,MAAM,EAAE,oBAAM,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;YACrG,IAAI,MAAM,CAAC,QAAQ,CAAC,oBAAoB,CAAC,SAAS,CAAC,KAAK,SAAS;gBAC/D,OAAO,SAAS,CAAC;YAEnB,YAAY,GAAG,gBAAgB,GAAG,IAAI,CAAC;YACvC,EAAE,IAAI,CAAC;SACR;IACH,CAAC;IAED,yGAAyG;IAClG,MAAM,CAAC,kBAAkB,CAAC,QAAgB;QAC/C,OAAO,GAAG,QAAQ,MAAM,mBAAI,CAAC,WAAW,EAAE,EAAE,CAAC;IAC/C,CAAC;IAED;;;;;;OAMG;IACI,MAAM,CAAC,iBAAiB,CAAC,UAAkB,EAAE,QAAgB;QAClE,IAAI,CAAC,yBAAU,CAAC,UAAU,CAAC,kBAAkB,CAAC,SAAS,CAAC;YACtD,yBAAU,CAAC,SAAS,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAErD,MAAM,SAAS,GAAG,cAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QACtE,IAAI,CAAC,yBAAU,CAAC,UAAU,CAAC,SAAS,CAAC;YACnC,yBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAElC,MAAM,UAAU,GAAG,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAClD,IAAI,yBAAU,CAAC,UAAU,CAAC,UAAU,CAAC;YACnC,yBAAU,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QAEpC,OAAO,UAAU,CAAC;IACpB,CAAC;CACF;AAtID,sCAsIC;AAED,MAAa,kBAAkB;IAE7B,6GAA6G;IACtG,MAAM,KAAK,SAAS;QACzB,OAAO,IAAA,WAAI,EAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IACnC,CAAC;IAED,2CAA2C;IACpC,MAAM,KAAK,SAAS;QACzB,IAAI,8BAAe,CAAC,kBAAkB,EAAE;YACtC,OAAO,IAAA,WAAI,EAAC,IAAA,WAAM,GAAE,EAAE,QAAQ,CAAC,CAAC;SACjC;QAED,uCAAuC;QACvC,OAAO,IAAA,WAAI,EAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IACnC,CAAC;CACF;AAhBD,gDAgBC","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/** @packageDocumentation\n * @module IModel\n */\n\nimport { Drawing, DrawingModel, Element, IModelDb, IModelJsFs, InformationPartitionElement, PhysicalModel, PhysicalPartition, SubjectOwnsPartitionElements } from \"@itwin/core-backend\";\nimport { Guid, Id64, Id64String, ProcessDetector } from \"@itwin/core-bentley\";\nimport { Code, CodeProps, ElementProps, GeometricElement2dProps, IModel, LocalFileName, PhysicalElementProps, RelatedElement } from \"@itwin/core-common\";\nimport { assert } from \"chai\";\nimport { tmpdir } from \"os\";\nimport path, { join } from \"path\";\n\nexport class IModelBuilder {\n  /**\n   * Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel.\n   * @return [modeledElementId, modelId]\n   */\n  public static createAndInsertPhysicalPartitionAndModel(testImodel: IModelDb, newModelCode: CodeProps, privateModel: boolean = false, parent?: Id64String): Id64String[] {\n    const eid = IModelBuilder.createAndInsertPhysicalPartition(testImodel, newModelCode, parent);\n    const modeledElementRef = new RelatedElement({ id: eid });\n    const mid = IModelBuilder.createAndInsertPhysicalModel(testImodel, modeledElementRef, privateModel);\n    return [eid, mid];\n  }\n\n  /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */\n  public static createAndInsertPhysicalPartition(testDb: IModelDb, newModelCode: CodeProps, parentId?: Id64String): Id64String {\n    const model = parentId ? testDb.elements.getElement(parentId).model : IModel.repositoryModelId;\n    const parent = new SubjectOwnsPartitionElements(parentId || IModel.rootSubjectId);\n\n    const modeledElementProps: ElementProps = {\n      classFullName: PhysicalPartition.classFullName,\n      parent,\n      model,\n      code: newModelCode,\n    };\n    const modeledElement: Element = testDb.elements.createElement(modeledElementProps);\n    return testDb.elements.insertElement(modeledElement.toJSON());\n  }\n\n  /** Create and insert a PhysicalPartition element (in the repositoryModel) and an associated PhysicalModel. */\n  public static createAndInsertPhysicalModel(testDb: IModelDb, modeledElementRef: RelatedElement, privateModel: boolean = false): Id64String {\n    const newModel = testDb.models.createModel({ modeledElement: modeledElementRef, classFullName: PhysicalModel.classFullName, isPrivate: privateModel });\n    const newModelId = newModel.id = testDb.models.insertModel(newModel.toJSON());\n    assert.isTrue(Id64.isValidId64(newModelId));\n    assert.isTrue(Id64.isValidId64(newModel.id));\n    assert.deepEqual(newModelId, newModel.id);\n    return newModelId;\n  }\n\n  /**\n   * Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel.\n   * @return [modeledElementId, modelId]\n   */\n  public static createAndInsertDrawingPartitionAndModel(testImodel: IModelDb, newModelCode: CodeProps, privateModel: boolean = false, parent?: Id64String): Id64String[] {\n    const eid = IModelBuilder.createAndInsertDrawingPartition(testImodel, newModelCode, parent);\n    const modeledElementRef = new RelatedElement({ id: eid });\n    const mid = IModelBuilder.createAndInsertDrawingModel(testImodel, modeledElementRef, privateModel);\n    return [eid, mid];\n  }\n\n  /** Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel. */\n  public static createAndInsertDrawingPartition(testDb: IModelDb, newModelCode: CodeProps, parentId?: Id64String): Id64String {\n    const model = parentId ? testDb.elements.getElement(parentId).model : IModel.repositoryModelId;\n    const parent = new SubjectOwnsPartitionElements(parentId || IModel.rootSubjectId);\n\n    const modeledElementProps: ElementProps = {\n      classFullName: Drawing.classFullName,\n      parent,\n      model,\n      code: newModelCode,\n    };\n    const modeledElement: Element = testDb.elements.createElement(modeledElementProps);\n    return testDb.elements.insertElement(modeledElement.toJSON());\n  }\n\n  /** Create and insert a DrawingPartition element (in the repositoryModel) and an associated DrawingModel. */\n  public static createAndInsertDrawingModel(testDb: IModelDb, modeledElementRef: RelatedElement, privateModel: boolean = false): Id64String {\n    const newModel = testDb.models.createModel({ modeledElement: modeledElementRef, classFullName: DrawingModel.classFullName, isPrivate: privateModel });\n    const newModelId = newModel.id = testDb.models.insertModel(newModel.toJSON());\n    assert.isTrue(Id64.isValidId64(newModelId));\n    assert.isTrue(Id64.isValidId64(newModel.id));\n    assert.deepEqual(newModelId, newModel.id);\n    return newModelId;\n  }\n\n  // Create a DrawingObject. (Does not insert it.)\n  public static createDrawingGraphic(testImodel: IModelDb, modelId: Id64String, categoryId: Id64String, elemCode?: Code): Element {\n    const elementProps: GeometricElement2dProps = {\n      classFullName: \"BisCore:DrawingGraphic\",\n      model: modelId,\n      category: categoryId,\n      code: elemCode ? elemCode : Code.createEmpty(),\n    };\n    return testImodel.elements.createElement(elementProps);\n  }\n\n  // Create a PhysicalObject. (Does not insert it.)\n  public static createPhysicalObject(testImodel: IModelDb, modelId: Id64String, categoryId: Id64String, elemCode?: Code): Element {\n    const elementProps: PhysicalElementProps = {\n      classFullName: \"Generic:PhysicalObject\",\n      model: modelId,\n      category: categoryId,\n      code: elemCode ? elemCode : Code.createEmpty(),\n    };\n    return testImodel.elements.createElement(elementProps);\n  }\n\n  public static getUniqueModelCode(testDb: IModelDb, newModelCodeBase: string): Code {\n    let newModelCode: string = newModelCodeBase;\n    let iter: number = 0;\n    while (true) {\n      const modelCode = InformationPartitionElement.createCode(testDb, IModel.rootSubjectId, newModelCode);\n      if (testDb.elements.queryElementIdByCode(modelCode) === undefined)\n        return modelCode;\n\n      newModelCode = newModelCodeBase + iter;\n      ++iter;\n    }\n  }\n\n  /** Generate a name for an iModel that's unique using the baseName provided and appending a new GUID.  */\n  public static generateUniqueName(baseName: string) {\n    return `${baseName} - ${Guid.createValue()}`;\n  }\n\n  /** Prepare for an output file by:\n   * - Resolving the output file name under the known test output directory\n   * - Making directories as necessary\n   * - Removing a previous copy of the output file\n   * @param subDirName Sub-directory under known test output directory. Should match the name of the test file minus the .test.ts file extension.\n   * @param fileName Name of output fille\n   */\n  public static prepareOutputFile(subDirName: string, fileName: string): LocalFileName {\n    if (!IModelJsFs.existsSync(KnownTestLocations.outputDir))\n      IModelJsFs.mkdirSync(KnownTestLocations.outputDir);\n\n    const outputDir = path.join(KnownTestLocations.outputDir, subDirName);\n    if (!IModelJsFs.existsSync(outputDir))\n      IModelJsFs.mkdirSync(outputDir);\n\n    const outputFile = path.join(outputDir, fileName);\n    if (IModelJsFs.existsSync(outputFile))\n      IModelJsFs.unlinkSync(outputFile);\n\n    return outputFile;\n  }\n}\n\nexport class KnownTestLocations {\n\n  /** The directory where test assets are stored. Keep in mind that the test is playing the role of the app. */\n  public static get assetsDir(): string {\n    return join(__dirname, \"assets\");\n  }\n\n  /** The directory where tests can write. */\n  public static get outputDir(): string {\n    if (ProcessDetector.isMobileAppBackend) {\n      return join(tmpdir(), \"output\");\n    }\n\n    // Assume that we are running in nodejs\n    return join(__dirname, \"output\");\n  }\n}\n"]}

package/lib/cjs/presentation-testing/IModelTestUtility.d.ts

@@ -0,0 +1,25 @@
+import { GuidString } from "@itwin/core-bentley";
+import { Code, ElementProps } from "@itwin/core-common";
+/**
+ * Test utility to push an iModel.
+ *
+ * @internal
+ */
+export declare class IModelTestUtility {
+    iTwinId: GuidString;
+    iModelId: string;
+    outputFile: string;
+    private _iModel;
+    private _modelId;
+    createIModel(): void;
+    addPhysicalModel(): string;
+    addDrawingModel(): string;
+    addSpatialCategory(elementProps: ElementProps): string;
+    addDrawingCategory(elementProps: ElementProps): string;
+    addPhysicalObject(categoryId: string): void;
+    addDrawingGraphic(categoryId: string): void;
+    getSpatialCategoryCode(codeValue: string): Code;
+    getDrawingCategoryCode(codeValue: string): Code;
+    closeIModel(): void;
+}
+//# sourceMappingURL=IModelTestUtility.d.ts.map

package/lib/cjs/presentation-testing/IModelTestUtility.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"IModelTestUtility.d.ts","sourceRoot":"","sources":["../../../src/presentation-testing/IModelTestUtility.ts"],"names":[],"mappings":"AAQA,OAAO,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACjD,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGxD;;;;GAIG;AACH,qBAAa,iBAAiB;IACrB,OAAO,EAAG,UAAU,CAAC;IACrB,QAAQ,EAAG,MAAM,CAAC;IAClB,UAAU,EAAG,MAAM,CAAC;IAC3B,OAAO,CAAC,OAAO,CAAY;IAC3B,OAAO,CAAC,QAAQ,CAAU;IAEnB,YAAY,IAAI,IAAI;IAMpB,gBAAgB,IAAI,MAAM;IAM1B,eAAe,IAAI,MAAM;IAMzB,kBAAkB,CAAC,YAAY,EAAE,YAAY,GAAG,MAAM;IAMtD,kBAAkB,CAAC,YAAY,EAAE,YAAY,GAAG,MAAM;IAMtD,iBAAiB,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAK3C,iBAAiB,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAK3C,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAI/C,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAI/C,WAAW,IAAI,IAAI;CAG3B"}

package/lib/cjs/presentation-testing/IModelTestUtility.js

@@ -0,0 +1,63 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IModelTestUtility = void 0;
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+/** @packageDocumentation
+ * @module TestUtilities
+ */
+const core_backend_1 = require("@itwin/core-backend");
+const IModelBuilder_1 = require("./IModelBuilder");
+/**
+ * Test utility to push an iModel.
+ *
+ * @internal
+ */
+class IModelTestUtility {
+    createIModel() {
+        this.outputFile = IModelBuilder_1.IModelBuilder.prepareOutputFile("IModelTest", IModelBuilder_1.IModelBuilder.generateUniqueName("IModelTest.bim"));
+        this._iModel = core_backend_1.SnapshotDb.createEmpty(this.outputFile, { rootSubject: { name: "IModelTest" } });
+        this.iModelId = this._iModel.iModelId;
+    }
+    addPhysicalModel() {
+        [, this._modelId] = IModelBuilder_1.IModelBuilder.createAndInsertPhysicalPartitionAndModel(this._iModel, IModelBuilder_1.IModelBuilder.getUniqueModelCode(this._iModel, "TestPhysicalModel"), false);
+        this._iModel.saveChanges("Added test model");
+        return this._modelId;
+    }
+    addDrawingModel() {
+        [, this._modelId] = IModelBuilder_1.IModelBuilder.createAndInsertDrawingPartitionAndModel(this._iModel, IModelBuilder_1.IModelBuilder.getUniqueModelCode(this._iModel, "TestDrawingModel"), false);
+        this._iModel.saveChanges("Added test model");
+        return this._modelId;
+    }
+    addSpatialCategory(elementProps) {
+        const categoryId = this._iModel.elements.insertElement(new core_backend_1.SpatialCategory(elementProps, this._iModel).toJSON());
+        this._iModel.saveChanges("Added spatial category");
+        return categoryId;
+    }
+    addDrawingCategory(elementProps) {
+        const categoryId = this._iModel.elements.insertElement(new core_backend_1.DrawingCategory(elementProps, this._iModel).toJSON());
+        this._iModel.saveChanges("Added drawing category");
+        return categoryId;
+    }
+    addPhysicalObject(categoryId) {
+        this._iModel.elements.insertElement(IModelBuilder_1.IModelBuilder.createPhysicalObject(this._iModel, this._modelId, categoryId).toJSON());
+        this._iModel.saveChanges("Added physical object");
+    }
+    addDrawingGraphic(categoryId) {
+        this._iModel.elements.insertElement(IModelBuilder_1.IModelBuilder.createDrawingGraphic(this._iModel, this._modelId, categoryId).toJSON());
+        this._iModel.saveChanges("Added drawing graphic");
+    }
+    getSpatialCategoryCode(codeValue) {
+        return core_backend_1.SpatialCategory.createCode(this._iModel, this._modelId, codeValue);
+    }
+    getDrawingCategoryCode(codeValue) {
+        return core_backend_1.DrawingCategory.createCode(this._iModel, this._modelId, codeValue);
+    }
+    closeIModel() {
+        this._iModel.close();
+    }
+}
+exports.IModelTestUtility = IModelTestUtility;
+//# sourceMappingURL=IModelTestUtility.js.map

package/lib/cjs/presentation-testing/IModelTestUtility.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"IModelTestUtility.js","sourceRoot":"","sources":["../../../src/presentation-testing/IModelTestUtility.ts"],"names":[],"mappings":";;;AAAA;;;+FAG+F;AAC/F;;GAEG;AACH,sDAA6F;AAG7F,mDAAgD;AAEhD;;;;GAIG;AACH,MAAa,iBAAiB;IAOrB,YAAY;QACjB,IAAI,CAAC,UAAU,GAAG,6BAAa,CAAC,iBAAiB,CAAC,YAAY,EAAE,6BAAa,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC;QACpH,IAAI,CAAC,OAAO,GAAG,yBAAU,CAAC,WAAW,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,EAAE,CAAC,CAAC;QAChG,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;IACxC,CAAC;IAEM,gBAAgB;QACrB,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,6BAAa,CAAC,wCAAwC,CAAC,IAAI,CAAC,OAAO,EAAE,6BAAa,CAAC,kBAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,mBAAmB,CAAC,EAAE,KAAK,CAAC,CAAC;QACrK,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;QAC7C,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAEM,eAAe;QACpB,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,6BAAa,CAAC,uCAAuC,CAAC,IAAI,CAAC,OAAO,EAAE,6BAAa,CAAC,kBAAkB,CAAC,IAAI,CAAC,OAAO,EAAE,kBAAkB,CAAC,EAAE,KAAK,CAAC,CAAC;QACnK,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;QAC7C,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAEM,kBAAkB,CAAC,YAA0B;QAClD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,8BAAe,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACjH,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,wBAAwB,CAAC,CAAC;QACnD,OAAO,UAAU,CAAC;IACpB,CAAC;IAEM,kBAAkB,CAAC,YAA0B;QAClD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,8BAAe,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QACjH,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,wBAAwB,CAAC,CAAC;QACnD,OAAO,UAAU,CAAC;IACpB,CAAC;IAEM,iBAAiB,CAAC,UAAkB;QACzC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,6BAAa,CAAC,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1H,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,uBAAuB,CAAC,CAAC;IACpD,CAAC;IAEM,iBAAiB,CAAC,UAAkB;QACzC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,6BAAa,CAAC,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1H,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,uBAAuB,CAAC,CAAC;IACpD,CAAC;IAEM,sBAAsB,CAAC,SAAiB;QAC7C,OAAO,8BAAe,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC5E,CAAC;IAEM,sBAAsB,CAAC,SAAiB;QAC7C,OAAO,8BAAe,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAC5E,CAAC;IAEM,WAAW;QAChB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;CACF;AA1DD,8CA0DC","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/** @packageDocumentation\n * @module TestUtilities\n */\nimport { DrawingCategory, IModelDb, SnapshotDb, SpatialCategory } from \"@itwin/core-backend\";\nimport { GuidString } from \"@itwin/core-bentley\";\nimport { Code, ElementProps } from \"@itwin/core-common\";\nimport { IModelBuilder } from \"./IModelBuilder\";\n\n/**\n * Test utility to push an iModel.\n *\n * @internal\n */\nexport class IModelTestUtility {\n  public iTwinId!: GuidString;\n  public iModelId!: string;\n  public outputFile!: string;\n  private _iModel!: IModelDb;\n  private _modelId!: string;\n\n  public createIModel(): void {\n    this.outputFile = IModelBuilder.prepareOutputFile(\"IModelTest\", IModelBuilder.generateUniqueName(\"IModelTest.bim\"));\n    this._iModel = SnapshotDb.createEmpty(this.outputFile, { rootSubject: { name: \"IModelTest\" } });\n    this.iModelId = this._iModel.iModelId;\n  }\n\n  public addPhysicalModel(): string {\n    [, this._modelId] = IModelBuilder.createAndInsertPhysicalPartitionAndModel(this._iModel, IModelBuilder.getUniqueModelCode(this._iModel, \"TestPhysicalModel\"), false);\n    this._iModel.saveChanges(\"Added test model\");\n    return this._modelId;\n  }\n\n  public addDrawingModel(): string {\n    [, this._modelId] = IModelBuilder.createAndInsertDrawingPartitionAndModel(this._iModel, IModelBuilder.getUniqueModelCode(this._iModel, \"TestDrawingModel\"), false);\n    this._iModel.saveChanges(\"Added test model\");\n    return this._modelId;\n  }\n\n  public addSpatialCategory(elementProps: ElementProps): string {\n    const categoryId = this._iModel.elements.insertElement(new SpatialCategory(elementProps, this._iModel).toJSON());\n    this._iModel.saveChanges(\"Added spatial category\");\n    return categoryId;\n  }\n\n  public addDrawingCategory(elementProps: ElementProps): string {\n    const categoryId = this._iModel.elements.insertElement(new DrawingCategory(elementProps, this._iModel).toJSON());\n    this._iModel.saveChanges(\"Added drawing category\");\n    return categoryId;\n  }\n\n  public addPhysicalObject(categoryId: string): void {\n    this._iModel.elements.insertElement(IModelBuilder.createPhysicalObject(this._iModel, this._modelId, categoryId).toJSON());\n    this._iModel.saveChanges(\"Added physical object\");\n  }\n\n  public addDrawingGraphic(categoryId: string): void {\n    this._iModel.elements.insertElement(IModelBuilder.createDrawingGraphic(this._iModel, this._modelId, categoryId).toJSON());\n    this._iModel.saveChanges(\"Added drawing graphic\");\n  }\n\n  public getSpatialCategoryCode(codeValue: string): Code {\n    return SpatialCategory.createCode(this._iModel, this._modelId, codeValue);\n  }\n\n  public getDrawingCategoryCode(codeValue: string): Code {\n    return DrawingCategory.createCode(this._iModel, this._modelId, codeValue);\n  }\n\n  public closeIModel(): void {\n    this._iModel.close();\n  }\n}\n"]}

package/lib/cjs/presentation-testing.d.ts

@@ -1,22 +1,29 @@
-/**
- * @module Hierarchies
- *
- * @docs-group-description Hierarchies
- * Types for testing hierarchies.
- */
-export * from "./presentation-testing/HierarchyBuilder";
-/**
- * @module Content
- *
- * @docs-group-description Content
- * Types for testing content.
- */
-export * from "./presentation-testing/ContentBuilder";
-/**
- * @module Helpers
- *
- * @docs-group-description Helpers
- * Various test helpers.
- */
-export * from "./presentation-testing/Helpers";
+/**
+ * @module Hierarchies
+ *
+ * @docs-group-description Hierarchies
+ * Types for testing hierarchies.
+ */
+export * from "./presentation-testing/HierarchyBuilder";
+/**
+ * @module Content
+ *
+ * @docs-group-description Content
+ * Types for testing content.
+ */
+export * from "./presentation-testing/ContentBuilder";
+/**
+ * @module Helpers
+ *
+ * @docs-group-description Helpers
+ * Various test helpers.
+ */
+export * from "./presentation-testing/Helpers";
+/**
+ * @module TestUtilities
+ *
+ * @docs-group-description TestUtilities
+ * Various test helpers.
+ */
+export * from "./presentation-testing/IModelTestUtility";
 //# sourceMappingURL=presentation-testing.d.ts.map

package/lib/cjs/presentation-testing.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"presentation-testing.d.ts","sourceRoot":"","sources":["../../src/presentation-testing.ts"],"names":[],"mappings":"AAIA;;;;;GAKG;AACH,cAAc,yCAAyC,CAAC;AAExD;;;;;GAKG;AACH,cAAc,uCAAuC,CAAC;AAEtD;;;;;GAKG;AACH,cAAc,gCAAgC,CAAC"}
+{"version":3,"file":"presentation-testing.d.ts","sourceRoot":"","sources":["../../src/presentation-testing.ts"],"names":[],"mappings":"AAIA;;;;;GAKG;AACH,cAAc,yCAAyC,CAAC;AAExD;;;;;GAKG;AACH,cAAc,uCAAuC,CAAC;AAEtD;;;;;GAKG;AACH,cAAc,gCAAgC,CAAC;AAE/C;;;;;GAKG;AACH,cAAc,0CAA0C,CAAC"}