Skip to content

Commit 7ee17b5

Browse files
try fix
1 parent 0d9b6bf commit 7ee17b5

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

api/tests/unit_tests/controllers/service_api/app/test_file_preview.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def test_validate_file_ownership_file_not_found(self, file_preview_api):
100100

101101
with patch("controllers.service_api.app.file_preview.db") as mock_db:
102102
# Mock MessageFile not found
103-
mock_db.session.query.return_value.where.return_value.first.return_value = None
103+
mock_db.session.query.return_value.filter_by.return_value.first.return_value = None
104104

105105
# Execute and assert exception
106106
with pytest.raises(FileNotFoundError) as exc_info:

api/tests/unit_tests/core/app/apps/advanced_chat/test_app_runner_conversation_variables.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def track_add_all(items):
129129
):
130130
# Setup mocks
131131
mock_session_class.return_value.__enter__.return_value = mock_session
132-
mock_db.session.query.return_value.where.return_value.first.return_value = MagicMock() # App exists
132+
mock_db.session.query.return_value.filter_by.return_value.first.return_value = MagicMock() # App exists
133133
mock_db.engine = MagicMock()
134134

135135
# Mock graph initialization
@@ -262,7 +262,7 @@ def track_add_all(items):
262262
):
263263
# Setup mocks
264264
mock_session_class.return_value.__enter__.return_value = mock_session
265-
mock_db.session.query.return_value.where.return_value.first.return_value = MagicMock() # App exists
265+
mock_db.session.query.return_value.filter_by.return_value.first.return_value = MagicMock() # App exists
266266
mock_db.engine = MagicMock()
267267

268268
# Mock ConversationVariable.from_variable to return mock objects
@@ -400,7 +400,7 @@ def test_all_variables_exist_no_changes(self):
400400
):
401401
# Setup mocks
402402
mock_session_class.return_value.__enter__.return_value = mock_session
403-
mock_db.session.query.return_value.where.return_value.first.return_value = MagicMock() # App exists
403+
mock_db.session.query.return_value.filter_by.return_value.first.return_value = MagicMock() # App exists
404404
mock_db.engine = MagicMock()
405405

406406
# Mock graph initialization

api/tests/unit_tests/services/test_account_service.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1381,7 +1381,7 @@ def test_get_invitation_if_token_valid_success(self, mock_db_dependencies, mock_
13811381

13821382
# Mock database queries - complex query mocking
13831383
mock_query1 = MagicMock()
1384-
mock_query1.where.return_value.first.return_value = mock_tenant
1384+
mock_query1.filter_by.return_value.first.return_value = mock_tenant
13851385

13861386
mock_query2 = MagicMock()
13871387
mock_query2.join.return_value.filter_by.return_value.first.return_value = (mock_account, "normal")

0 commit comments

Comments
 (0)