From cd2f0030cf21db511fb66ff40dc911f701df15a9 Mon Sep 17 00:00:00 2001
From: Mads Rasmussen <madsr@hey.com>
Date: Thu, 17 Nov 2022 13:45:41 +0100
Subject: [PATCH] add tree methods to data type data

---
 .../src/core/mocks/data/data-type.data.ts            | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/Umbraco.Web.UI.Client/src/core/mocks/data/data-type.data.ts b/src/Umbraco.Web.UI.Client/src/core/mocks/data/data-type.data.ts
index d131329499..bd844f9760 100644
--- a/src/Umbraco.Web.UI.Client/src/core/mocks/data/data-type.data.ts
+++ b/src/Umbraco.Web.UI.Client/src/core/mocks/data/data-type.data.ts
@@ -98,6 +98,18 @@ class UmbDataTypeData extends UmbEntityData<DataTypeDetails> {
 	constructor() {
 		super(data);
 	}
+
+	getTreeRoot(): Array<FolderTreeItem> {
+		return this.data.filter((item) => item.parentKey === null);
+	}
+
+	getTreeItemChildren(key: string): Array<FolderTreeItem> {
+		return this.data.filter((item) => item.parentKey === key);
+	}
+
+	getTreeItem(keys: Array<string>): Array<FolderTreeItem> {
+		return this.data.filter((item) => keys.includes(item.key ?? ''));
+	}
 }
 
 export const umbDataTypeData = new UmbDataTypeData();