From 25d6587ebe526e8b6f59aa6dcda2955ccc56134c Mon Sep 17 00:00:00 2001 From: "ievgenii.v" Date: Thu, 30 Oct 2025 16:24:07 +0200 Subject: [PATCH] feature: improve tests --- rhodecode/apps/repository/tests/test_ai_code_review.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/rhodecode/apps/repository/tests/test_ai_code_review.py b/rhodecode/apps/repository/tests/test_ai_code_review.py index 25c0d1bb..4c785091 100644 --- a/rhodecode/apps/repository/tests/test_ai_code_review.py +++ b/rhodecode/apps/repository/tests/test_ai_code_review.py @@ -53,7 +53,7 @@ class TestAiCodeReview: session = response.get_session_from_response() err_msg = session.peek_flash('error') - success_message = session.peek_flash() + success_message = session.peek_flash('success') assert response.status_code == 200 assert response.json_body == {'response': True, 'redirect_url': redirect_url} @@ -80,7 +80,12 @@ class TestAiCodeReview: ai_user = User.get_ai_user() + session = response.get_session_from_response() + success_message = session.peek_flash('success') + assert response.status_code == 200 assert run_task_mock.call_count == 1 assert len(existing_reviewers) == 1 - assert existing_reviewers[0].user_id == ai_user.user_id \ No newline at end of file + assert existing_reviewers[0].user_id == ai_user.user_id + assert len(success_message) == 1 + assert success_message[0] == "AI code review has started. The review will be ready in about 1–3 minutes."