Skip to content

Commit 05d4e61

Browse files
committed
minio and validation_tasks tests updated
1 parent 2a69aa0 commit 05d4e61

2 files changed

Lines changed: 27 additions & 27 deletions

File tree

tests/test_minio.py

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ def test_rocrate_found_as_directory(mock_get_list):
110110
minio_client = MagicMock()
111111

112112
from app.utils.minio_utils import find_rocrate_object_on_minio
113-
result = find_rocrate_object_on_minio("rocrate123", minio_client, "bucket", storage_path="my/path")
113+
result = find_rocrate_object_on_minio("rocrate123", minio_client, "bucket", root_path="my/path")
114114
assert result == obj
115115

116116

@@ -162,7 +162,7 @@ def test_storage_path_provided(mock_get_list):
162162
minio_client = MagicMock()
163163

164164
from app.utils.minio_utils import find_rocrate_object_on_minio
165-
result = find_rocrate_object_on_minio("rocrate789", minio_client, "bucket", storage_path="data")
165+
result = find_rocrate_object_on_minio("rocrate789", minio_client, "bucket", root_path="data")
166166
assert result == obj
167167

168168

@@ -177,7 +177,7 @@ def test_validation_object_found_with_storage_path(mock_get_list):
177177

178178
from app.utils.minio_utils import find_validation_object_on_minio
179179
# Execute
180-
result = find_validation_object_on_minio("rocrate123", MagicMock(), "bucket", storage_path="my/storage")
180+
result = find_validation_object_on_minio("rocrate123", MagicMock(), "bucket", root_path="my/storage")
181181

182182
# Assert
183183
assert result == obj
@@ -425,44 +425,44 @@ def test_update_validation_status_unexpected_error(mock_get_client):
425425
@patch("app.utils.minio_utils.find_rocrate_object_on_minio")
426426
@patch("app.utils.minio_utils.get_minio_client")
427427
def test_fetch_rocrate_zip(
428-
mock_get_client_and_bucket,
428+
mock_get_client,
429429
mock_find_object,
430430
mock_get_list,
431431
mock_download,
432432
tmp_path,
433433
):
434434
# Setup mocks
435-
mock_get_client_and_bucket.return_value = "minio_client"
435+
mock_get_client.return_value = "minio_client"
436436
rocrate_obj = DummyObject("some/path/rocrate123.zip", is_dir=False)
437437
mock_find_object.return_value = rocrate_obj
438438

439439
from app.utils.minio_utils import fetch_ro_crate_from_minio
440440

441441
with patch("app.utils.minio_utils.tempfile.mkdtemp", return_value=str(tmp_path)):
442442
# Execute
443-
result = fetch_ro_crate_from_minio("test_bucket", "rocrate123")
443+
result = fetch_ro_crate_from_minio("test_bucket", "rocrate123", "some/path")
444444

445-
# Assert
446-
expected_path = tmp_path / "rocrate123.zip"
447-
assert result == str(expected_path)
448-
mock_download.assert_called_once_with(
449-
"minio_client", "test_bucket",
450-
"some/path/rocrate123.zip", str(expected_path))
445+
# Assert
446+
expected_path = tmp_path / "rocrate123.zip"
447+
assert result == str(expected_path)
448+
mock_download.assert_called_once_with(
449+
"minio_client", "test_bucket",
450+
"some/path/rocrate123.zip", str(expected_path))
451451

452452

453453
@patch("app.utils.minio_utils.download_file_from_minio")
454454
@patch("app.utils.minio_utils.get_minio_object_list")
455455
@patch("app.utils.minio_utils.find_rocrate_object_on_minio")
456456
@patch("app.utils.minio_utils.get_minio_client")
457457
def test_fetch_rocrate_directory(
458-
mock_get_client_and_bucket,
458+
mock_get_client,
459459
mock_find_object,
460460
mock_get_list,
461461
mock_download,
462462
tmp_path,
463463
):
464464
# Setup mocks
465-
mock_get_client_and_bucket.return_value = "minio_client"
465+
mock_get_client.return_value = "minio_client"
466466
rocrate_obj = DummyObject("rocrates/rocrate124", is_dir=True)
467467
mock_find_object.return_value = rocrate_obj
468468

@@ -476,7 +476,7 @@ def test_fetch_rocrate_directory(
476476
]
477477

478478
# Execute
479-
result = fetch_ro_crate_from_minio("test_bucket", "rocrate124")
479+
result = fetch_ro_crate_from_minio("test_bucket", "rocrate124", "rocrates")
480480

481481
# Assert
482482
expected_root = tmp_path / "rocrate124"
@@ -498,21 +498,21 @@ def test_fetch_rocrate_directory(
498498
@patch("app.utils.minio_utils.find_rocrate_object_on_minio")
499499
@patch("app.utils.minio_utils.get_minio_client")
500500
def test_fetch_rocrate_handles_empty_dir(
501-
mock_get_client_and_bucket,
501+
mock_get_client,
502502
mock_find_object,
503503
mock_get_list,
504504
mock_download,
505505
tmp_path,
506506
):
507-
mock_get_client_and_bucket.return_value = "minio_client"
507+
mock_get_client.return_value = "minio_client"
508508
rocrate_obj = DummyObject("rocrate456", is_dir=True)
509509
mock_find_object.return_value = rocrate_obj
510510
mock_get_list.return_value = []
511511

512512
from app.utils.minio_utils import fetch_ro_crate_from_minio
513513

514514
with patch("app.utils.minio_utils.tempfile.mkdtemp", return_value=str(tmp_path)):
515-
result = fetch_ro_crate_from_minio("test_bucket", "rocrate456")
515+
result = fetch_ro_crate_from_minio("test_bucket", "rocrate456", "")
516516

517517
expected_root = tmp_path / "rocrate456"
518518
assert result == str(expected_root)

tests/test_validation_tasks.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -38,11 +38,11 @@ def test_process_validation_zipfile_success(
3838
mock_validation_result.to_json.return_value = '{"status": "valid"}'
3939
mock_validate.return_value = mock_validation_result
4040

41-
process_validation_task_by_id("crate123", "profileA", "https://example.com/hook")
41+
process_validation_task_by_id("test_bucket", "crate123", "", "profileA", "https://example.com/hook")
4242

43-
mock_fetch.assert_called_once_with("crate123")
43+
mock_fetch.assert_called_once_with("test_bucket", "crate123", "")
4444
mock_validate.assert_called_once_with("/tmp/crate.zip", "profileA")
45-
mock_update.assert_called_once_with("crate123", '{"status": "valid"}')
45+
mock_update.assert_called_once_with("test_bucket", "crate123", "", '{"status": "valid"}')
4646
mock_webhook.assert_called_once_with("https://example.com/hook", '{"status": "valid"}')
4747
mock_remove.assert_called_once_with("/tmp/crate.zip")
4848

@@ -72,11 +72,11 @@ def test_process_validation_directory_success(
7272
mock_validation_result.to_json.return_value = '{"status": "valid"}'
7373
mock_validate.return_value = mock_validation_result
7474

75-
process_validation_task_by_id("crate123", "profileA", "https://example.com/hook")
75+
process_validation_task_by_id("test_bucket", "crate123", "", "profileA", "https://example.com/hook")
7676

77-
mock_fetch.assert_called_once_with("crate123")
77+
mock_fetch.assert_called_once_with("test_bucket", "crate123", "")
7878
mock_validate.assert_called_once_with("/tmp/crate123/", "profileA")
79-
mock_update.assert_called_once_with("crate123", '{"status": "valid"}')
79+
mock_update.assert_called_once_with("test_bucket", "crate123", "", '{"status": "valid"}')
8080
mock_webhook.assert_called_once_with("https://example.com/hook", '{"status": "valid"}')
8181
mock_rmtree.assert_called_once_with("/tmp/crate123/")
8282

@@ -100,7 +100,7 @@ def test_process_validation_fails_with_message(
100100
mock_fetch.return_value = "/tmp/crate.zip"
101101
mock_validate.return_value = "Validation failed"
102102

103-
process_validation_task_by_id("crate123", "profileA", "https://example.com/hook")
103+
process_validation_task_by_id("test_bucket", "crate123", "", "profileA", "https://example.com/hook")
104104

105105
mock_update.assert_not_called()
106106
mock_webhook.assert_called_once()
@@ -128,7 +128,7 @@ def test_process_validation_exception(
128128
):
129129
mock_fetch.return_value = "/tmp/crate.zip"
130130

131-
process_validation_task_by_id("crate123", "profileA", "https://example.com/hook")
131+
process_validation_task_by_id("test_bucket", "crate123", "", "profileA", "https://example.com/hook")
132132

133133
mock_update.assert_not_called()
134134
mock_webhook.assert_called_once()
@@ -150,7 +150,7 @@ def test_process_validation_fetch_error(
150150
mock_webhook,
151151
mock_exists
152152
):
153-
process_validation_task_by_id("crate123", "profileA", "https://example.com/hook")
153+
process_validation_task_by_id("test_bucket", "crate123", "", "profileA", "https://example.com/hook")
154154

155155
mock_validate.assert_not_called()
156156
mock_update.assert_not_called()

0 commit comments

Comments
 (0)