diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/utils/file-system/file-system-tree.manager.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/utils/file-system/file-system-tree.manager.ts index f1f2fe291a..cad195d8f9 100644 --- a/src/Umbraco.Web.UI.Client/src/mocks/data/utils/file-system/file-system-tree.manager.ts +++ b/src/Umbraco.Web.UI.Client/src/mocks/data/utils/file-system/file-system-tree.manager.ts @@ -15,9 +15,7 @@ export class UmbMockFileSystemTreeManager item.parent === null); - const paged = pagedResult(items, skip, take); - const treeItems = paged.items.map((item) => createFileSystemTreeItem(item)); - return { items: treeItems, total: paged.total }; + return this.#pagedTreeResult({ items, skip, take }); } getChildrenOf({ parentPath, skip = 0, take = 100 }: { parentPath: string; skip?: number; take?: number }): { @@ -25,8 +23,19 @@ export class UmbMockFileSystemTreeManager item.parent?.path === parentPath); + return this.#pagedTreeResult({ items, skip, take }); + } + + #pagedTreeResult({ items, skip, take }: { items: Array; skip: number; take: number }) { const paged = pagedResult(items, skip, take); const treeItems = paged.items.map((item) => createFileSystemTreeItem(item)); - return { items: treeItems, total: paged.total }; + const treeItemsHasChildren = treeItems.map((item) => { + const children = this.#db.getAll().filter((child) => child.parent?.path === item.path); + return { + ...item, + hasChildren: children.length > 0, + }; + }); + return { items: treeItemsHasChildren, total: paged.total }; } }