Merge pull request #6 from AllenInstitute/feature/update-batch-data-s…

…ync-models

feature/update batch data sync models

src/aibs_informatics_aws_utils/efs/mount_point.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import re
+from functools import cache
 
 __all__ = [
     "MountPointConfiguration",
@@ -375,6 +376,7 @@ def __repr__(self) -> str:
         )
 
 
+@cache
 def detect_mount_points() -> List[MountPointConfiguration]:
     mount_points: List[MountPointConfiguration] = []
 

test/aibs_informatics_aws_utils/data_sync/test_file_system.py

@@ -27,6 +27,7 @@
     S3FileSystem,
 )
 from aibs_informatics_aws_utils.efs import MountPointConfiguration
+from aibs_informatics_aws_utils.efs.mount_point import detect_mount_points
 
 
 def any_s3_uri(bucket: str = "bucket", key: str = "key") -> S3URI:
@@ -282,6 +283,10 @@ def assertLocalFileSystem_partition(
 
 @mock_sts
 class EFSFileSystemTests(EFSTestsBase):
+    def setUp(self) -> None:
+        super().setUp()
+        detect_mount_points.cache_clear()
+
     def setUpEFSFileSystem(
         self, name: str, access_point_path: Optional[Union[str, Path]] = None
     ) -> Tuple[Path, EFSPath]:

test/aibs_informatics_aws_utils/efs/test_mount_point.py

@@ -21,6 +21,10 @@
 
 @mock_sts
 class MountPointConfigurationTests(EFSTestsBase):
+    def setUp(self) -> None:
+        super().setUp()
+        detect_mount_points.cache_clear()
+
     def setUpEFS(self, *access_points: Tuple[str, Path], file_system_name: Optional[str] = None):
         self.create_file_system(file_system_name)
         for access_point_name, access_point_path in access_points: