From bdde2ab7a11c5587929d437903f55a8d6bb15f70 Mon Sep 17 00:00:00 2001 From: Kenneth Kehl <@kkehl@flexion.us> Date: Mon, 7 Jul 2025 08:36:55 -0700 Subject: [PATCH] more tests --- tests/app/test_commands.py | 21 ++++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) diff --git a/tests/app/test_commands.py b/tests/app/test_commands.py index 4f0037fb4..2c3390b79 100644 --- a/tests/app/test_commands.py +++ b/tests/app/test_commands.py @@ -1,6 +1,6 @@ import os from datetime import datetime, timedelta -from unittest.mock import MagicMock, mock_open +from unittest.mock import MagicMock, mock_open, patch import pytest from sqlalchemy import func, select @@ -26,6 +26,7 @@ from app.commands import ( populate_organizations_from_file, process_row_from_job, promote_user_to_platform_admin, + purge_csv_bucket, purge_functional_test_data, update_jobs_archived_flag, ) @@ -732,3 +733,21 @@ def test_clear_templates_from_cache(mocker): [mocker.call(p) for p in expected_patterns], any_order=True ) mock_logger.info.assert_called_once_with("Number of templates deleted from cache 9") + + +@patch("app.commands.s3.purge_bucket") +@patch("app.commands.current_app") +def test_purge_csv_bucket(mock_current_app, mock_purge_bucket): + mock_current_app.config = { + "CSV_UPLOAD_BUCKET": { + "bucket": "test-bucket", + "access_key_id": "FAKE_ACCESS_KEY", + "secret_access_key": "FAKE_SECRET_KEY", # pragma: allowlist secret + "region": "us-north-1", + } + } + purge_csv_bucket() + + mock_purge_bucket.assert_called_once_with( + "test-bucket", "FAKE_ACCESS_KEY", "FAKE_SECRET_KEY", "us-north-1" + )