Fix yadocs source info response schema (#234)

Co-authored-by: Valeria Bulanova <[email protected]>

Author: vallbull

lib/dl_file_uploader_api_lib/dl_file_uploader_api_lib/views/sources.py

@@ -37,6 +37,8 @@
     GSheetsUserSourceDataSourceProperties,
     GSheetsUserSourceProperties,
     UserSourceProperties,
+    YaDocsUserSourceDataSourceProperties,
+    YaDocsUserSourceProperties,
 )
 from dl_file_uploader_task_interface.tasks import ParseFileTask
 
@@ -156,6 +158,13 @@ async def _make_source_resp(
             assert isinstance(source.user_source_dsrc_properties, GSheetsUserSourceDataSourceProperties)
             extra["sheet_id"] = source.user_source_dsrc_properties.sheet_id
 
+        if file_type == FileType.yadocs:
+            assert isinstance(user_source_properties, YaDocsUserSourceProperties)
+            extra["private_path"] = user_source_properties.private_path
+            extra["public_link"] = user_source_properties.public_link
+            assert isinstance(source.user_source_dsrc_properties, YaDocsUserSourceDataSourceProperties)
+            extra["sheet_id"] = source.user_source_dsrc_properties.sheet_id
+
         source_resp = dict(
             file_type=file_type,
             source_id=source.id,

lib/dl_file_uploader_api_lib/dl_file_uploader_api_lib_tests/ext/test_documents_api.py

@@ -42,6 +42,14 @@ async def test_yadocuments_public_file(
         assert sheet_src.status == FileProcessingStatus.ready
         assert sheet_src.error is None
 
+    for src in df.sources:
+        resp = await fu_client.make_request(ReqBuilder.source_info(file_id, src.id))
+        assert resp.status == 200
+        source = resp.json["source"]
+        assert source["public_link"]
+        assert source["sheet_id"]
+        assert source["is_valid"]
+
 
 @pytest.mark.asyncio
 async def test_yadocuments_private_file(