Skip to content

Commit 495a85f

Browse files
committed
fix: mock return of test
1 parent d57c968 commit 495a85f

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

backend/tests/test_audit_service.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,11 @@ def test_process_audit_question_success(
2323
mock_cursor.description = [("user_email",), ("operation",)]
2424
mock_cursor.fetchall.return_value = [("test@example.com", "insert")]
2525

26-
mock_summarize.return_value = "Summary of results."
26+
# Mock Summary Response
27+
mock_response = MagicMock()
28+
mock_response.summary = "Summary of results."
29+
mock_response.visualization = None
30+
mock_summarize.return_value = mock_response
2731

2832
# Execute
2933
result = process_audit_question("Show me inserts")

0 commit comments

Comments
 (0)