Skip to content

Commit 806bb81

Browse files
Fix lint error
1 parent a396ee4 commit 806bb81

2 files changed

Lines changed: 48 additions & 16 deletions

File tree

backend/main.py

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,9 @@ async def lifespan(app: FastAPI):
6161
"name": "PictoPy Postman Collection",
6262
"url": "https://www.postman.com/aossie-pictopy/pictopy/overview",
6363
},
64-
servers=[{"url": "http://localhost:8000", "description": "Local Development server"}],
64+
servers=[
65+
{"url": "http://localhost:8000", "description": "Local Development server"}
66+
],
6567
)
6668

6769
app.logger = CustomizeLogger.make_logger("app/logging_config.json")
@@ -80,7 +82,9 @@ def generate_openapi_json():
8082
openapi_schema["info"]["contact"] = app.contact
8183

8284
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
83-
openapi_path = os.path.join(project_root, "docs", "backend", "backend_python", "openapi.json")
85+
openapi_path = os.path.join(
86+
project_root, "docs", "backend", "backend_python", "openapi.json"
87+
)
8488

8589
os.makedirs(os.path.dirname(openapi_path), exist_ok=True)
8690

@@ -110,8 +114,12 @@ async def root():
110114
app.include_router(folders_router, prefix="/folders", tags=["Folders"])
111115
app.include_router(albums_router, prefix="/albums", tags=["Albums"])
112116
app.include_router(images_router, prefix="/images", tags=["Images"])
113-
app.include_router(face_clusters_router, prefix="/face-clusters", tags=["Face Clusters"])
114-
app.include_router(user_preferences_router, prefix="/user-preferences", tags=["User Preferences"])
117+
app.include_router(
118+
face_clusters_router, prefix="/face-clusters", tags=["Face Clusters"]
119+
)
120+
app.include_router(
121+
user_preferences_router, prefix="/user-preferences", tags=["User Preferences"]
122+
)
115123

116124

117125
# Entry point for running with: python3 main.py

backend/tests/test_folders.py

Lines changed: 36 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,9 @@ def test_add_folder_success(
163163
mock_add_folder_tree.assert_called_once()
164164

165165
@patch("app.routes.folders.db_folder_exists")
166-
def test_add_folder_already_exists(self, mock_folder_exists, client, temp_folder_structure):
166+
def test_add_folder_already_exists(
167+
self, mock_folder_exists, client, temp_folder_structure
168+
):
167169
"""Test adding folder that already exists in database."""
168170
mock_folder_exists.return_value = True
169171

@@ -231,7 +233,9 @@ def test_add_folder_with_parent_id(
231233
"""Test adding folder with specified parent_folder_id."""
232234

233235
mock_folder_exists.return_value = False
234-
mock_find_parent.return_value = None # Should not be called when parent_id provided
236+
mock_find_parent.return_value = (
237+
None # Should not be called when parent_id provided
238+
)
235239
mock_add_folder_tree.return_value = ("child-folder-id", {})
236240
mock_update_parent_ids.return_value = None
237241

@@ -254,7 +258,9 @@ def test_add_folder_with_parent_id(
254258

255259
@patch("app.routes.folders.folder_util_add_folder_tree")
256260
@patch("app.routes.folders.db_folder_exists")
257-
def test_add_folder_database_error(self, mock_folder_exists, mock_add_folder_tree, client, temp_folder_structure):
261+
def test_add_folder_database_error(
262+
self, mock_folder_exists, mock_add_folder_tree, client, temp_folder_structure
263+
):
258264
"""Test handling database errors during folder addition."""
259265
mock_folder_exists.return_value = False
260266
mock_add_folder_tree.side_effect = Exception("Database connection failed")
@@ -388,7 +394,9 @@ def test_enable_ai_tagging_database_error(self, mock_enable_batch, client):
388394
assert data["detail"]["error"] == "Internal server error"
389395

390396
@patch("app.routes.folders.db_enable_ai_tagging_batch")
391-
def test_enable_ai_tagging_background_processing_called(self, mock_enable_batch, client):
397+
def test_enable_ai_tagging_background_processing_called(
398+
self, mock_enable_batch, client
399+
):
392400
"""Test that background processing is triggered after enabling AI tagging."""
393401
mock_enable_batch.return_value = 2
394402

@@ -411,7 +419,9 @@ def test_disable_ai_tagging_success(self, mock_disable_batch, client):
411419
"""Test successfully disabling AI tagging for folders."""
412420
mock_disable_batch.return_value = 5 # 5 folders updated
413421

414-
request_data = {"folder_ids": ["folder-1", "folder-2", "folder-3", "folder-4", "folder-5"]}
422+
request_data = {
423+
"folder_ids": ["folder-1", "folder-2", "folder-3", "folder-4", "folder-5"]
424+
}
415425

416426
response = client.post("/folders/disable-ai-tagging", json=request_data)
417427

@@ -428,7 +438,9 @@ def test_disable_ai_tagging_success(self, mock_disable_batch, client):
428438
"folder-5",
429439
]
430440

431-
mock_disable_batch.assert_called_once_with(["folder-1", "folder-2", "folder-3", "folder-4", "folder-5"])
441+
mock_disable_batch.assert_called_once_with(
442+
["folder-1", "folder-2", "folder-3", "folder-4", "folder-5"]
443+
)
432444

433445
@patch("app.routes.folders.db_disable_ai_tagging_batch")
434446
def test_disable_ai_tagging_single_folder(self, mock_disable_batch, client):
@@ -479,7 +491,9 @@ def test_disable_ai_tagging_database_error(self, mock_disable_batch, client):
479491
assert data["detail"]["error"] == "Internal server error"
480492

481493
@patch("app.routes.folders.db_disable_ai_tagging_batch")
482-
def test_disable_ai_tagging_no_background_processing(self, mock_disable_batch, client):
494+
def test_disable_ai_tagging_no_background_processing(
495+
self, mock_disable_batch, client
496+
):
483497
"""Test that no background processing is triggered when disabling AI tagging."""
484498
mock_disable_batch.return_value = 2
485499

@@ -572,7 +586,9 @@ def test_delete_folders_database_error(self, mock_delete_batch, client):
572586
# ============================================================================
573587

574588
@patch("app.routes.folders.db_get_all_folder_details")
575-
def test_get_all_folders_success(self, mock_get_all_folders, client, sample_folder_details):
589+
def test_get_all_folders_success(
590+
self, mock_get_all_folders, client, sample_folder_details
591+
):
576592
"""Test successfully retrieving all folders."""
577593
mock_get_all_folders.return_value = sample_folder_details
578594

@@ -660,7 +676,9 @@ def test_disable_ai_tagging_no_folders_updated(self, mock_disable_batch, client)
660676
"""Test disabling AI tagging when no folders are actually updated."""
661677
mock_disable_batch.return_value = 0
662678

663-
request_data = {"folder_ids": ["non-existent-folder-1", "non-existent-folder-2"]}
679+
request_data = {
680+
"folder_ids": ["non-existent-folder-1", "non-existent-folder-2"]
681+
}
664682

665683
response = client.post("/folders/disable-ai-tagging", json=request_data)
666684

@@ -719,7 +737,9 @@ def test_complete_folder_workflow(
719737

720738
@patch("app.routes.folders.db_enable_ai_tagging_batch")
721739
@patch("app.routes.folders.db_disable_ai_tagging_batch")
722-
def test_ai_tagging_toggle_workflow(self, mock_disable_batch, mock_enable_batch, client):
740+
def test_ai_tagging_toggle_workflow(
741+
self, mock_disable_batch, mock_enable_batch, client
742+
):
723743
"""Test toggling AI tagging on and off for folders."""
724744
folder_ids = ["folder-1", "folder-2"]
725745

@@ -732,7 +752,9 @@ def test_ai_tagging_toggle_workflow(self, mock_disable_batch, mock_enable_batch,
732752
assert enable_response.json()["data"]["updated_count"] == 2
733753

734754
disable_request = {"folder_ids": folder_ids}
735-
disable_response = client.post("/folders/disable-ai-tagging", json=disable_request)
755+
disable_response = client.post(
756+
"/folders/disable-ai-tagging", json=disable_request
757+
)
736758
assert disable_response.status_code == 200
737759
assert disable_response.json()["data"]["updated_count"] == 2
738760

@@ -786,7 +808,9 @@ def mock_find_parent_side_effect(folder_path):
786808

787809
@patch("app.routes.folders.db_delete_folders_batch")
788810
@patch("app.routes.folders.db_enable_ai_tagging_batch")
789-
def test_complete_folder_lifecycle(self, mock_enable_batch, mock_delete_batch, client):
811+
def test_complete_folder_lifecycle(
812+
self, mock_enable_batch, mock_delete_batch, client
813+
):
790814
"""Test complete folder lifecycle: enable AI -> delete."""
791815
folder_ids = ["folder-1", "folder-2"]
792816

0 commit comments

Comments
 (0)