fix: adds missing status check methods for merge requests

SachinKSingh28 · SachinKSingh28 · commit 61ba6bd03af9 · 2025-02-08T00:10:00.000Z
diff --git a/docs/gl_objects/status_checks.rst b/docs/gl_objects/status_checks.rst
@@ -55,3 +55,18 @@ Delete an external status check::
 
     status_check.delete(status_check_id)
 
+List external status check for a project merge request::
+
+    merge_request = project.mergerequests.get(1)
+
+    merge_request.status_checks.list()
+
+Set external status check for a project merge request::
+
+    merge_request = project.mergerequests.get(1)
+
+    merge_request.status_check_response.update({
+        "external_status_check_id": status_check_id,
+        "status": "passed",
+        "sha": merge_request.sha
+    })
diff --git a/gitlab/v4/objects/merge_requests.py b/gitlab/v4/objects/merge_requests.py
@@ -46,7 +46,10 @@
 from .notes import ProjectMergeRequestNoteManager  # noqa: F401
 from .pipelines import ProjectMergeRequestPipelineManager  # noqa: F401
 from .reviewers import ProjectMergeRequestReviewerDetailManager
-from .status_checks import ProjectMergeRequestStatusCheckManager
+from .status_checks import (
+    ProjectMergeRequestStatusCheckManager,
+    ProjectMergeRequestStatusCheckResponseManager,
+)
 
 __all__ = [
     "MergeRequest",
@@ -171,6 +174,7 @@ class ProjectMergeRequest(
     resourcestateevents: ProjectMergeRequestResourceStateEventManager
     reviewer_details: ProjectMergeRequestReviewerDetailManager
     status_checks: ProjectMergeRequestStatusCheckManager
+    status_check_response: ProjectMergeRequestStatusCheckResponseManager
 
     @cli.register_custom_action(cls_names="ProjectMergeRequest")
     @exc.on_http_error(exc.GitlabMROnBuildSuccessError)
diff --git a/gitlab/v4/objects/status_checks.py b/gitlab/v4/objects/status_checks.py
@@ -1,4 +1,6 @@
-from gitlab.base import RESTObject
+from typing import Any, Dict, Optional
+
+from gitlab.base import RESTManager, RESTObject
 from gitlab.mixins import (
     CreateMixin,
     DeleteMixin,
@@ -27,6 +29,7 @@ class ProjectExternalStatusCheckManager(
     CreateMixin[ProjectExternalStatusCheck],
     UpdateMixin[ProjectExternalStatusCheck],
     DeleteMixin[ProjectExternalStatusCheck],
+    RESTManager[ProjectExternalStatusCheck],
 ):
     _path = "/projects/{project_id}/external_status_checks"
     _obj_cls = ProjectExternalStatusCheck
@@ -41,15 +44,41 @@ class ProjectExternalStatusCheckManager(
     _types = {"protected_branch_ids": ArrayAttribute}
 
 
-class ProjectMergeRequestStatusCheck(SaveMixin, RESTObject):
+class ProjectMergeRequestStatusCheckResponse(SaveMixin, RESTObject):
     pass
 
 
-class ProjectMergeRequestStatusCheckManager(ListMixin[ProjectMergeRequestStatusCheck]):
-    _path = "/projects/{project_id}/merge_requests/{merge_request_iid}/status_checks"
-    _obj_cls = ProjectMergeRequestStatusCheck
-    _from_parent_attrs = {"project_id": "project_id", "merge_request_iid": "iid"}
+class ProjectMergeRequestStatusCheckResponseManager(
+    UpdateMixin[ProjectMergeRequestStatusCheckResponse],
+    RESTManager[ProjectMergeRequestStatusCheckResponse],
+):
+    _path = "/projects/{project_id}/merge_requests/{mr_iid}/status_check_responses"
+    _obj_cls = ProjectMergeRequestStatusCheckResponse
+    _from_parent_attrs = {"project_id": "project_id", "mr_iid": "iid"}
     _update_attrs = RequiredOptional(
         required=("sha", "external_status_check_id", "status")
     )
     _update_method = UpdateMethod.POST
+
+    def update(  # type: ignore[override]
+        self, new_data: Optional[Dict[str, Any]] = None, **kwargs: Any
+    ) -> Dict[str, Any]:
+        """Update a Label on the server.
+
+        Args:
+            **kwargs: Extra options to send to the server (e.g. sudo)
+        """
+        return super().update(id=None, new_data=new_data, **kwargs)
+
+
+class ProjectMergeRequestStatusCheck(RESTObject):
+    pass
+
+
+class ProjectMergeRequestStatusCheckManager(
+    ListMixin[ProjectMergeRequestStatusCheck],
+    RESTManager[ProjectMergeRequestStatusCheck],
+):
+    _path = "/projects/{project_id}/merge_requests/{mr_iid}/status_checks"
+    _obj_cls = ProjectMergeRequestStatusCheck
+    _from_parent_attrs = {"project_id": "project_id", "mr_iid": "iid"}
diff --git a/tests/unit/objects/test_status_checks.py b/tests/unit/objects/test_status_checks.py
@@ -5,6 +5,48 @@
 import pytest
 import responses
 
+mr_content = {
+    "id": 1,
+    "iid": 1,
+    "project_id": 3,
+    "title": "test1",
+    "description": "fixed login page css paddings",
+    "state": "merged",
+    "merged_by": {
+        "id": 87854,
+        "name": "Douwe Maan",
+        "username": "DouweM",
+        "state": "active",
+        "avatar_url": "https://gitlab.example.com/uploads/-/system/user/avatar/87854/avatar.png",
+        "web_url": "https://gitlab.com/DouweM",
+    },
+    "reviewers": [
+        {
+            "id": 2,
+            "name": "Sam Bauch",
+            "username": "kenyatta_oconnell",
+            "state": "active",
+            "avatar_url": "https://www.gravatar.com/avatar/956c92487c6f6f7616b536927e22c9a0?s=80&d=identicon",
+            "web_url": "http://gitlab.example.com//kenyatta_oconnell",
+        }
+    ],
+}
+
+external_status_checks_content = [
+    {
+        "id": 2,
+        "name": "Service 2",
+        "external_url": "https://gitlab.example.com/test-endpoint-2",
+        "status": "passed",
+    },
+    {
+        "id": 1,
+        "name": "Service 1",
+        "external_url": "https://gitlab.example.com/test-endpoint-1",
+        "status": "pending",
+    },
+]
+
 
 @pytest.fixture
 def external_status_check():
@@ -107,6 +149,26 @@ def resp_delete_external_status_checks():
         yield rsps
 
 
+@pytest.fixture
+def resp_list_merge_requests_status_checks():
+    with responses.RequestsMock() as rsps:
+        rsps.add(
+            method=responses.GET,
+            url="http://localhost/api/v4/projects/1/merge_requests/1",
+            json=mr_content,
+            content_type="application/json",
+            status=200,
+        )
+        rsps.add(
+            method=responses.GET,
+            url="http://localhost/api/v4/projects/1/merge_requests/1/status_checks",
+            json=external_status_checks_content,
+            content_type="application/json",
+            status=200,
+        )
+        yield rsps
+
+
 def test_list_external_status_checks(gl, resp_list_external_status_checks):
     status_checks = gl.projects.get(1, lazy=True).external_status_checks.list()
     assert len(status_checks) == 1
@@ -125,3 +187,9 @@ def test_delete_external_status_checks(gl, resp_delete_external_status_checks):
     gl.projects.get(1, lazy=True).external_status_checks.delete(1)
     status_checks = gl.projects.get(1, lazy=True).external_status_checks.list()
     assert len(status_checks) == 0
+
+
+def test_get_merge_request_status_checks(gl, resp_list_merge_requests_status_checks):
+    merge_request = gl.projects.get(1, lazy=True).mergerequests.get(1)
+    status_checks = merge_request.status_checks.list()
+    assert len(status_checks) == 2
