Skip to content

Commit c7730b6

Browse files
committed
Update tests for dataverse link imp
1 parent 50461c4 commit c7730b6

File tree

1 file changed

+31
-3
lines changed

1 file changed

+31
-3
lines changed

addon_imps/tests/link/test_dataverse.py

Lines changed: 31 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ async def test_list_child_items_empty(self):
240240
return_value=ItemSampleResult(items=[], total_count=0)
241241
)
242242

243-
result = await self.imp.list_child_items("")
243+
result = await self.imp.list_child_items("", item_type=None)
244244

245245
self.imp.list_root_items.assert_awaited_once_with("")
246246
self.assertEqual(len(result.items), 0)
@@ -265,7 +265,7 @@ async def test_list_child_items_dataverse(self):
265265
item_type=ItemType.FOLDER,
266266
)
267267

268-
result = await self.imp.list_child_items("dataverse/123")
268+
result = await self.imp.list_child_items("dataverse/123", item_type=None)
269269

270270
self._assert_get("api/dataverses/123/contents")
271271

@@ -298,7 +298,35 @@ async def test_parse_invalid_dataset(self):
298298
await self.imp.get_item_info("dataset/invalid")
299299

300300
async def test_list_child_items_non_dataverse(self):
301-
result = await self.imp.list_child_items("dataset/123")
301+
result = await self.imp.list_child_items("dataset/123", item_type=None)
302302

303303
self.assertEqual(len(result.items), 0)
304304
self.assertEqual(result.total_count, 0)
305+
306+
async def test_list_child_items_with_item_type(self):
307+
dataverse_contents = {
308+
"data": [
309+
{"type": "dataverse", "id": "456", "title": "Sub Dataverse"},
310+
{"type": "dataset", "id": "789", "title": "Dataset"},
311+
]
312+
}
313+
314+
self._patch_get(dataverse_contents)
315+
316+
with patch.object(
317+
self.imp, "_fetch_dataset", new_callable=AsyncMock
318+
) as mock_fetch_dataset:
319+
mock_fetch_dataset.return_value = ItemResult(
320+
item_id="dataset/789",
321+
item_name="Dataset",
322+
item_type=ItemType.FOLDER,
323+
)
324+
325+
result = await self.imp.list_child_items(
326+
"dataverse/123", item_type=ItemType.FOLDER
327+
)
328+
329+
self._assert_get("api/dataverses/123/contents")
330+
331+
for item in result.items:
332+
self.assertEqual(item.item_type, ItemType.FOLDER)

0 commit comments

Comments
 (0)