file-based: implement get schema method/proerty in unit tests

airbytehq · Feb 11, 2025 · 791241c · 791241c
1 parent 428c669
commit 791241c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/unit_tests/sources/file_based/in_memory_files_source.py b/unit_tests/sources/file_based/in_memory_files_source.py
@@ -155,6 +155,10 @@ def load_identity_groups(self, logger: logging.Logger) -> Iterable[Dict[str, Any
     def file_permissions_schema(self) -> Dict[str, Any]:
         return {"type": "object", "properties": {}}
 
+    @property
+    def identities_schema(self) -> Dict[str, Any]:
+        return {"type": "object", "properties": {}}
+
     def open_file(
         self, file: RemoteFile, mode: FileReadMode, encoding: Optional[str], logger: logging.Logger
     ) -> IOBase:

diff --git a/unit_tests/sources/file_based/test_file_based_stream_reader.py b/unit_tests/sources/file_based/test_file_based_stream_reader.py
@@ -94,6 +94,10 @@ def load_identity_groups(self, logger: logging.Logger) -> Iterable[Dict[str, Any
     def file_permissions_schema(self) -> Dict[str, Any]:
         return {"type": "object", "properties": {}}
 
+    @property
+    def identities_schema(self) -> Dict[str, Any]:
+        return {"type": "object", "properties": {}}
+
 
 class TestSpec(AbstractFileBasedSpec):
     @classmethod