浏览代码

fix: revert the removal of _export() in the MockRepository

trgiangdo 11 月之前
父节点
当前提交
b06e8ac82b
共有 1 个文件被更改,包括 4 次插入1 次删除
  1. 4 1
      tests/core/_manager/test_manager.py

+ 4 - 1
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 @@ class MockRepository(_AbstractRepository):  # type: ignore
     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