Implement GithubCommit.get_prs

Signed-off-by: Cristian Le <[email protected]>

Author: LecrisUT

ogr/services/github/commit.py

@@ -1,12 +1,16 @@
 # Copyright Contributors to the Packit project.
 # SPDX-License-Identifier: MIT
+from typing import Optional
+
 import github
 from github.Commit import Commit as _GitCommit
 
+from ogr.abstract import PullRequest
 from ogr.exceptions import GithubAPIException
 from ogr.services import github as ogr_github
 from ogr.services.base import BaseGitCommit
 from ogr.services.github.changes import GithubCommitChanges
+from ogr.services.github.pull_request import GithubPullRequest
 
 
 class GithubCommit(BaseGitCommit):
@@ -23,6 +27,12 @@ def changes(self) -> GithubCommitChanges:
             self._changes = GithubCommitChanges(self)
         return self._changes
 
+    def get_prs(self) -> Optional[list[PullRequest]]:
+        raw_prs = list(self._raw_commit.get_pulls())
+        if not raw_prs:
+            return None
+        return [GithubPullRequest(pr, self.project) for pr in raw_prs]
+
     @staticmethod
     def get(project: "ogr_github.GithubProject", sha: str) -> "GithubCommit":
         try:

tests/integration/github/test_commit.py

@@ -22,3 +22,16 @@ def test_changes(self):
         commit = self.hello_world_project.get_commit("f06fed9")
         changes = commit.changes
         assert list(changes.files) == ["LICENSE", "README.md"]
+
+    def test_get_prs(self):
+        # Commit with PR associated
+        commit_with_one_pr = self.hello_world_project.get_commit("0840e2d")
+        prs = commit_with_one_pr.get_prs()
+        assert prs
+        assert len(prs) == 1
+        assert prs[0].id == 556
+        # Commit with no PR
+        commit_without_pr = self.hello_world_project.get_commit("f2c98da")
+        prs = commit_without_pr.get_prs()
+        assert not prs
+        # No test data for commit with multiple PRs