|
5 | 5 | process_validation_task_by_id, |
6 | 6 | perform_ro_crate_validation, |
7 | 7 | return_ro_crate_validation, |
8 | | - process_validation_task_by_metadata |
| 8 | + process_validation_task_by_metadata, |
| 9 | + check_ro_crate_exists |
9 | 10 | ) |
10 | 11 |
|
11 | 12 | from app.utils.minio_utils import InvalidAPIUsage |
@@ -319,3 +320,31 @@ def test_return_validation_raises_error(mock_get_status): |
319 | 320 |
|
320 | 321 | assert "MinIO S3 Error" in str(exc_info.value.message) |
321 | 322 | mock_get_status.assert_called_once_with("crate789") |
| 323 | + |
| 324 | + |
| 325 | +# Test function: check_ro_crate_exists |
| 326 | + |
| 327 | +@mock.patch("app.tasks.validation_tasks.get_minio_client_and_bucket", return_value=("mock_client", "mock_bucket")) |
| 328 | +@mock.patch("app.tasks.validation_tasks.find_rocrate_object_on_minio", return_value="crate123") |
| 329 | +def test_ro_crate_exists( |
| 330 | + mock_find_rocrate, |
| 331 | + mock_get_client |
| 332 | +): |
| 333 | + result = check_ro_crate_exists("crate123") |
| 334 | + |
| 335 | + mock_get_client.assert_called_once() |
| 336 | + mock_find_rocrate.assert_called_once_with("crate123", "mock_client", "mock_bucket", storage_path='') |
| 337 | + assert result is True |
| 338 | + |
| 339 | + |
| 340 | +@mock.patch("app.tasks.validation_tasks.get_minio_client_and_bucket", return_value=("mock_client", "mock_bucket")) |
| 341 | +@mock.patch("app.tasks.validation_tasks.find_rocrate_object_on_minio", return_value=False) |
| 342 | +def test_ro_crate_does_not_exist( |
| 343 | + mock_find_rocrate, |
| 344 | + mock_get_client |
| 345 | +): |
| 346 | + result = check_ro_crate_exists("crate12z") |
| 347 | + |
| 348 | + mock_get_client.assert_called_once() |
| 349 | + mock_find_rocrate.assert_called_once_with("crate12z", "mock_client", "mock_bucket", storage_path='') |
| 350 | + assert result is False |
0 commit comments