From b06e8ac82b75e1c98f26be43b2adabbc06b290dc Mon Sep 17 00:00:00 2001 From: trgiangdo Date: Wed, 19 Jun 2024 22:38:53 +0700 Subject: [PATCH] fix: revert the removal of _export() in the MockRepository --- tests/core/_manager/test_manager.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/core/_manager/test_manager.py b/tests/core/_manager/test_manager.py index 7f5b5f6844..c0c570435a 100644 --- a/tests/core/_manager/test_manager.py +++ b/tests/core/_manager/test_manager.py @@ -12,7 +12,7 @@ import dataclasses import pathlib from dataclasses import dataclass -from typing import Any, Dict, Iterable, List, Optional +from typing import Any, Dict, Iterable, List, Optional, Union from taipy.config.config import Config from taipy.core._manager._manager import _Manager @@ -94,6 +94,9 @@ def _delete_by(self, attribute: str, value: str): def _search(self, attribute: str, value: Any, filters: Optional[List[Dict]] = None) -> List[MockEntity]: return self.repo._search(attribute, value, filters) + def _export(self, entity_id: str, folder_path: Union[str, pathlib.Path]): + return self.repo._export(self, entity_id, folder_path) + @property def _storage_folder(self) -> pathlib.Path: return pathlib.Path(Config.core.storage_folder) # type: ignore