From d6185b45d0e4c2bee964b7878facd94ba45caa1e Mon Sep 17 00:00:00 2001
From: Mads Rasmussen <madsr@hey.com>
Date: Fri, 5 Apr 2024 12:30:40 +0200
Subject: [PATCH] add tree mapper

---
 .../src/mocks/data/template/template.db.ts      | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/template/template.db.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/template/template.db.ts
index e0394582b7..59719a4c8c 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/data/template/template.db.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/data/template/template.db.ts
@@ -1,15 +1,17 @@
 import { UmbEntityMockDbBase } from '../utils/entity/entity-base.js';
 import { UmbMockEntityTreeManager } from '../utils/entity/entity-tree.manager.js';
-import { folderTreeItemMapper } from '../utils.js';
 import { UmbMockEntityItemManager } from '../utils/entity/entity-item.manager.js';
 import type { UmbMockTemplateModel } from './template.data.js';
 import { data } from './template.data.js';
 import { UmbMockTemplateDetailManager } from './template-detail.manager.js';
 import { UmbMockTemplateQueryManager } from './template-query.manager.js';
-import type { TemplateItemResponseModel } from '@umbraco-cms/backoffice/external/backend-api';
+import type {
+	NamedEntityTreeItemResponseModel,
+	TemplateItemResponseModel,
+} from '@umbraco-cms/backoffice/external/backend-api';
 
 class UmbTemplateMockDB extends UmbEntityMockDbBase<UmbMockTemplateModel> {
-	tree = new UmbMockEntityTreeManager<UmbMockTemplateModel>(this, folderTreeItemMapper);
+	tree = new UmbMockEntityTreeManager<UmbMockTemplateModel>(this, treeItemMapper);
 	item = new UmbMockEntityItemManager<UmbMockTemplateModel>(this, itemMapper);
 	detail = new UmbMockTemplateDetailManager(this);
 	query = new UmbMockTemplateQueryManager(this);
@@ -19,6 +21,15 @@ class UmbTemplateMockDB extends UmbEntityMockDbBase<UmbMockTemplateModel> {
 	}
 }
 
+const treeItemMapper = (model: UmbMockTemplateModel): NamedEntityTreeItemResponseModel => {
+	return {
+		name: model.name,
+		hasChildren: model.hasChildren,
+		id: model.id,
+		parent: model.parent,
+	};
+};
+
 const itemMapper = (item: UmbMockTemplateModel): TemplateItemResponseModel => {
 	return {
 		id: item.id,