From 5435d6c067449c3cb207addbd6a5cd10b297f1a9 Mon Sep 17 00:00:00 2001 From: Tim Pietrusky Date: Tue, 19 Mar 2024 17:26:08 +0100 Subject: [PATCH] test: added "subfolder" --- tests/test_rp_handler.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/test_rp_handler.py b/tests/test_rp_handler.py index be744866..47fab794 100644 --- a/tests/test_rp_handler.py +++ b/tests/test_rp_handler.py @@ -131,7 +131,9 @@ def test_bucket_endpoint_not_configured(self, mock_upload_image, mock_exists): mock_exists.return_value = True mock_upload_image.return_value = "simulated_uploaded/image.png" - outputs = {"node_id": {"images": [{"filename": "ComfyUI_00001_.png"}]}} + outputs = { + "node_id": {"images": [{"filename": "ComfyUI_00001_.png", "subfolder": ""}]} + } job_id = "123" result = rp_handler.process_output_images(outputs, job_id) @@ -155,7 +157,9 @@ def test_bucket_endpoint_configured(self, mock_upload_image, mock_exists): mock_upload_image.return_value = "http://example.com/uploaded/image.png" # Define the outputs and job_id for the test - outputs = {"node_id": {"images": [{"filename": "ComfyUI_00001_.png"}]}} + outputs = { + "node_id": {"images": [{"filename": "ComfyUI_00001_.png", "subfolder": ""}]} + } job_id = "123" # Call the function under test @@ -188,7 +192,9 @@ def test_bucket_image_upload_fails_env_vars_wrong_or_missing( # When AWS credentials are wrong or missing, upload_image should return 'simulated_uploaded/...' mock_upload_image.return_value = "simulated_uploaded/image.png" - outputs = {"node_id": {"images": [{"filename": "ComfyUI_00001_.png"}]}} + outputs = { + "node_id": {"images": [{"filename": "ComfyUI_00001_.png", "subfolder": ""}]} + } job_id = "123" result = rp_handler.process_output_images(outputs, job_id)