add tree methods

src/Umbraco.Web.UI.Client/src/core/mocks/data/document-type.data.ts

@@ -1,5 +1,6 @@
 import { Entity } from './entities';
 import { UmbEntityData } from './entity.data';
+import { DocumentTypeTreeItem } from '@umbraco-cms/backend-api';
 
 export interface DocumentTypeEntity extends Entity {
 	key: string;
@@ -42,6 +43,18 @@ class UmbDocumentTypeData extends UmbEntityData<DocumentTypeEntity> {
 	constructor() {
 		super(data);
 	}
+
+	getTreeRoot(): Array<DocumentTypeTreeItem> {
+		return this.data.filter((item) => item.parentKey === null);
+	}
+
+	getTreeItemChildren(key: string): Array<DocumentTypeTreeItem> {
+		return this.data.filter((item) => item.parentKey === key);
+	}
+
+	getTreeItem(keys: Array<string>): Array<DocumentTypeTreeItem> {
+		return this.data.filter((item) => keys.includes(item.key ?? ''));
+	}
 }
 
 export const umbDocumentTypeData = new UmbDocumentTypeData();