From f3ea530ae8eddefd3afac44a90af0694fc58d10e Mon Sep 17 00:00:00 2001 From: "autofix-ci[bot]" <114827586+autofix-ci[bot]@users.noreply.github.com> Date: Thu, 7 May 2026 10:53:52 +0000 Subject: [PATCH] [autofix.ci] apply automated fixes --- .../factories/test_file_validation.py | 53 ++++++++++--------- 1 file changed, 29 insertions(+), 24 deletions(-) diff --git a/api/tests/unit_tests/factories/test_file_validation.py b/api/tests/unit_tests/factories/test_file_validation.py index a115e329f2..02e882a675 100644 --- a/api/tests/unit_tests/factories/test_file_validation.py +++ b/api/tests/unit_tests/factories/test_file_validation.py @@ -41,16 +41,12 @@ def _validate( ("video", ".mp4", [], [], True), ], ) -def test_bucket_semantics( - input_file_type, file_extension, allowed_file_types, allowed_file_extensions, expected -): +def test_bucket_semantics(input_file_type, file_extension, allowed_file_types, allowed_file_extensions, expected): config = FileUploadConfig( allowed_file_types=allowed_file_types, allowed_file_extensions=allowed_file_extensions, ) - assert _validate( - input_file_type=input_file_type, file_extension=file_extension, config=config - ) is expected + assert _validate(input_file_type=input_file_type, file_extension=file_extension, config=config) is expected @pytest.mark.parametrize("whitelist_entry", [".png", ".PNG", "png", "PNG", " .Png ", "PnG"]) @@ -81,12 +77,15 @@ def test_mixed_case_whitelist_replicating_real_user_config(): def test_tool_file_always_passes(): config = FileUploadConfig(allowed_file_types=[FileType.CUSTOM], allowed_file_extensions=[".pdf"]) - assert _validate( - input_file_type="image", - file_extension=".png", - file_transfer_method=FileTransferMethod.TOOL_FILE, - config=config, - ) is True + assert ( + _validate( + input_file_type="image", + file_extension=".png", + file_transfer_method=FileTransferMethod.TOOL_FILE, + config=config, + ) + is True + ) def test_transfer_method_gate_for_non_image(): @@ -94,18 +93,24 @@ def test_transfer_method_gate_for_non_image(): allowed_file_types=[FileType.DOCUMENT], allowed_file_upload_methods=[FileTransferMethod.LOCAL_FILE], ) - assert _validate( - input_file_type="document", - file_extension=".pdf", - file_transfer_method=FileTransferMethod.LOCAL_FILE, - config=config, - ) is True - assert _validate( - input_file_type="document", - file_extension=".pdf", - file_transfer_method=FileTransferMethod.REMOTE_URL, - config=config, - ) is False + assert ( + _validate( + input_file_type="document", + file_extension=".pdf", + file_transfer_method=FileTransferMethod.LOCAL_FILE, + config=config, + ) + is True + ) + assert ( + _validate( + input_file_type="document", + file_extension=".pdf", + file_transfer_method=FileTransferMethod.REMOTE_URL, + config=config, + ) + is False + ) def test_history_replay_matches_round_1_outcome_under_unchanged_config():