Skip to content

Commit 5eaaad4

Browse files
committed
fix breaking tests
1 parent 4ff0b34 commit 5eaaad4

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/test_dataset.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,10 @@ async def test_download_files(python_dataset):
136136

137137
minio_mock.download_file.return_value = Path("/path/to/downloaded_file")
138138
minio_mock.get_signed_url.return_value = Path("http://example.com/signed_url")
139+
minio_mock.list_bucket.return_value = [
140+
Mock(filename="file1.txt"),
141+
Mock(filename="file2.txt"),
142+
]
139143

140144
progress_mock = Mock()
141145
python_dataset.minio_polling_interval = 0
@@ -161,6 +165,10 @@ async def test_download_files_with_signed_urls(python_dataset):
161165
python_dataset.configuration = config
162166
minio_mock.download_file.return_value = "/path/to/downloaded_file"
163167
minio_mock.get_signed_url.return_value = "http://example.com/signed_url"
168+
minio_mock.list_bucket.return_value = [
169+
Mock(filename="file1.txt"),
170+
Mock(filename="file2.txt"),
171+
]
164172
progress_mock = Mock()
165173

166174
python_dataset.servicex = AsyncMock()

0 commit comments

Comments
 (0)