From 18a8a877b02cef7a18f0021aa3287a83d22198a5 Mon Sep 17 00:00:00 2001 From: businkv Date: Sat, 29 Aug 2026 20:03:27 +0300 Subject: [PATCH 1/2] =?UTF-8?q?=D0=B4=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?= =?UTF-8?q?=D0=BD=D1=8B=20=D1=82=D0=B5=D1=81=D1=82=20=D0=BD=D0=B0=20=D0=BE?= =?UTF-8?q?=D0=B1=D1=89=D0=B8=D0=B9=20=D0=BB=D0=B8=D0=BC=D0=B8=D1=82=20?= =?UTF-8?q?=D0=BA=D0=BE=D0=BC=D0=BC=D0=B5=D0=BD=D1=82=D0=B0=D1=80=D0=B8?= =?UTF-8?q?=D0=B5=D0=B2=20=D0=BE=D1=82=20=D0=BE=D0=B4=D0=BD=D0=BE=D0=B3?= =?UTF-8?q?=D0=BE=20=D0=BF=D0=BE=D0=BB=D1=8C=D0=B7=D0=BE=D0=B2=D0=B0=D1=82?= =?UTF-8?q?=D0=B5=D0=BB=D1=8F=20=D0=B8=20=D1=82=D0=B5=D1=81=D1=82=20=D0=BD?= =?UTF-8?q?=D0=B0=20=D0=BB=D0=B8=D0=BC=D0=B8=D1=82=20=D0=BA=D0=BE=D0=BC?= =?UTF-8?q?=D0=BC=D0=B5=D0=BD=D1=82=D0=B0=D1=80=D0=B8=D0=B5=D0=B2=20=D0=B4?= =?UTF-8?q?=D0=BB=D1=8F=20=D0=BE=D0=B4=D0=BD=D0=BE=D0=B3=D0=BE=20=D0=BB?= =?UTF-8?q?=D0=B5=D0=BA=D1=82=D0=BE=D1=80=D0=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_routes/test_comment.py | 109 ++++++++++++++++++++++++++++++ 1 file changed, 109 insertions(+) diff --git a/tests/test_routes/test_comment.py b/tests/test_routes/test_comment.py index ba563bb8..16a2e990 100644 --- a/tests/test_routes/test_comment.py +++ b/tests/test_routes/test_comment.py @@ -792,3 +792,112 @@ def test_post_like(client, dbsession, comment): dbsession.refresh(comment) assert comment.like_count == 0 assert comment.dislike_count == 0 + + +def test_comment_lecturer_limit( + client, + lecturers, + authlib_user, + mocker, +): + """ + Тест лимита на одного лектора + """ + new_user = authlib_user.copy() + new_user["id"] = 99999 + + achive_get_url = settings.API_URL + f"achievement/user/{new_user.get('id')}" + achive_post_url = ( + settings.API_URL + f"achievement/achievement/{settings.FIRST_COMMENT_ACHIEVEMENT_ID}/reciever/{new_user.get('id')}" + ) + mock_aiohttp_session = aiohttp_mock( + authlib_user_id=new_user.get("id"), + aiohttp_response_status=status.HTTP_200_OK, + achievement_id=settings.FIRST_COMMENT_ACHIEVEMENT_ID, + get_url=achive_get_url, + post_url=achive_post_url, + ) + mocker.patch("aiohttp.ClientSession", return_value=mock_aiohttp_session) + + lecturer_id = lecturers[0].id + body = { + "subject": "Subject", + "text": "Text", + "mark_kindness": 1, + "mark_freebie": 0, + "mark_clarity": 0, + } + + for _ in range(settings.COMMENT_TO_LECTURER_LIMIT - 1): + response = client.post(url, json=body, params={"lecturer_id": lecturer_id}) + assert response.status_code == status.HTTP_200_OK + + response_5 = client.post(url, json=body, params={"lecturer_id": lecturer_id}) + assert response_5.status_code == status.HTTP_200_OK + + response_6 = client.post(url, json=body, params={"lecturer_id": lecturer_id}) + assert response_6.status_code == status.HTTP_429_TOO_MANY_REQUESTS + + +def test_comment_total_limit( + client, + dbsession, + mocker, +): + """ + Тест общего лимита комментариев пользователя за период + """ + dbsession.query(LecturerUserComment).delete() + dbsession.query(Comment).delete() + dbsession.commit() + + from rating_api.models import Lecturer + + extra_lecturers = [] + for i in range(5): + lecturer = Lecturer( + id=200 + i, + first_name=f"total_fname{i}", + last_name=f"total_lname{i}", + middle_name=f"total_mname{i}", + timetable_id=5000 + i, + ) + dbsession.add(lecturer) + extra_lecturers.append(lecturer) + dbsession.commit() + + for lecturer in extra_lecturers: + dbsession.refresh(lecturer) + + new_user = {"id": 99999, "email": "test@example.com"} + + achive_get_url = settings.API_URL + f"achievement/user/{new_user.get('id')}" + achive_post_url = ( + settings.API_URL + f"achievement/achievement/{settings.FIRST_COMMENT_ACHIEVEMENT_ID}/reciever/{new_user.get('id')}" + ) + mock_aiohttp_session = aiohttp_mock( + authlib_user_id=new_user.get("id"), + aiohttp_response_status=status.HTTP_200_OK, + achievement_id=settings.FIRST_COMMENT_ACHIEVEMENT_ID, + get_url=achive_get_url, + post_url=achive_post_url, + ) + mocker.patch("aiohttp.ClientSession", return_value=mock_aiohttp_session) + + body = { + "subject": "TestSubject", + "text": "TestText", + "mark_kindness": 1, + "mark_freebie": 0, + "mark_clarity": 0, + } + + # По 4 коммента каждому из 5 лекторов = 20 + for lecturer in extra_lecturers: + for _ in range(4): + response = client.post(url, json=body, params={"lecturer_id": lecturer.id}) + assert response.status_code == status.HTTP_200_OK + + # 21й - превышение лимита + response_21 = client.post(url, json=body, params={"lecturer_id": extra_lecturers[0].id}) + assert response_21.status_code == status.HTTP_429_TOO_MANY_REQUESTS \ No newline at end of file From 7ff4de032d88a66fca8c28aec85ad4710dfb19d2 Mon Sep 17 00:00:00 2001 From: businkv Date: Sat, 29 Aug 2026 20:06:46 +0300 Subject: [PATCH 2/2] =?UTF-8?q?=D1=84=D0=BE=D1=80=D0=BC=D0=B0=D1=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_routes/test_comment.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/test_routes/test_comment.py b/tests/test_routes/test_comment.py index 16a2e990..6aa6be20 100644 --- a/tests/test_routes/test_comment.py +++ b/tests/test_routes/test_comment.py @@ -808,7 +808,8 @@ def test_comment_lecturer_limit( achive_get_url = settings.API_URL + f"achievement/user/{new_user.get('id')}" achive_post_url = ( - settings.API_URL + f"achievement/achievement/{settings.FIRST_COMMENT_ACHIEVEMENT_ID}/reciever/{new_user.get('id')}" + settings.API_URL + + f"achievement/achievement/{settings.FIRST_COMMENT_ACHIEVEMENT_ID}/reciever/{new_user.get('id')}" ) mock_aiohttp_session = aiohttp_mock( authlib_user_id=new_user.get("id"), @@ -873,7 +874,8 @@ def test_comment_total_limit( achive_get_url = settings.API_URL + f"achievement/user/{new_user.get('id')}" achive_post_url = ( - settings.API_URL + f"achievement/achievement/{settings.FIRST_COMMENT_ACHIEVEMENT_ID}/reciever/{new_user.get('id')}" + settings.API_URL + + f"achievement/achievement/{settings.FIRST_COMMENT_ACHIEVEMENT_ID}/reciever/{new_user.get('id')}" ) mock_aiohttp_session = aiohttp_mock( authlib_user_id=new_user.get("id"), @@ -900,4 +902,4 @@ def test_comment_total_limit( # 21й - превышение лимита response_21 = client.post(url, json=body, params={"lecturer_id": extra_lecturers[0].id}) - assert response_21.status_code == status.HTTP_429_TOO_MANY_REQUESTS \ No newline at end of file + assert response_21.status_code == status.HTTP_429_TOO_MANY_REQUESTS