Implement GithubCommit.get_prs

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

Author: LecrisUT

ogr/services/github/commit.py

@@ -10,6 +10,7 @@
 from ogr.exceptions import GithubAPIException, OgrNetworkError
 from ogr.services import github as ogr_github
 from ogr.services.base import BaseGitCommit
+from ogr.services.github.pull_request import GithubPullRequest
 
 
 class GithubCommit(BaseGitCommit):
@@ -26,6 +27,10 @@ def changes(self) -> "GithubCommitChanges":
             self._changes = GithubCommitChanges(self)
         return self._changes
 
+    def get_prs(self) -> Iterable[GithubPullRequest]:
+        for pr in self._raw_commit.get_pulls():
+            yield GithubPullRequest(pr, self.project)
+
     @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 = list(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