@@ -240,7 +240,7 @@ async def test_list_child_items_empty(self):
240
240
return_value = ItemSampleResult (items = [], total_count = 0 )
241
241
)
242
242
243
- result = await self .imp .list_child_items ("" )
243
+ result = await self .imp .list_child_items ("" , item_type = None )
244
244
245
245
self .imp .list_root_items .assert_awaited_once_with ("" )
246
246
self .assertEqual (len (result .items ), 0 )
@@ -265,7 +265,7 @@ async def test_list_child_items_dataverse(self):
265
265
item_type = ItemType .FOLDER ,
266
266
)
267
267
268
- result = await self .imp .list_child_items ("dataverse/123" )
268
+ result = await self .imp .list_child_items ("dataverse/123" , item_type = None )
269
269
270
270
self ._assert_get ("api/dataverses/123/contents" )
271
271
@@ -298,7 +298,35 @@ async def test_parse_invalid_dataset(self):
298
298
await self .imp .get_item_info ("dataset/invalid" )
299
299
300
300
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 )
302
302
303
303
self .assertEqual (len (result .items ), 0 )
304
304
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