Merge pull request #1055 from nejch/feat/commit-gpg-signature

feat: add support for commit GPG signature

3 files changed, 124 insertions, 44 deletions

diff --git a/gitlab/tests/objects/test_commits.py b/gitlab/tests/objects/test_commits.py
new file mode 100644
index 0000000..23a4285
--- /dev/null
+++ b/gitlab/tests/objects/test_commits.py
@@ -0,0 +1,106 @@
+from httmock import urlmatch, response, with_httmock
+
+from .test_projects import headers, TestProject
+
+
+@urlmatch(
+    scheme="http",
+    netloc="localhost",
+    path="/api/v4/projects/1/repository/commits/6b2257ea",
+    method="get",
+)
+def resp_get_commit(url, request):
+    """Mock for commit GET response."""
+    content = """{
+    "id": "6b2257eabcec3db1f59dafbd84935e3caea04235",
+    "short_id": "6b2257ea",
+    "title": "Initial commit"
+    }"""
+    content = content.encode("utf-8")
+    return response(200, content, headers, None, 5, request)
+
+
+@urlmatch(
+    scheme="http", path="/api/v4/projects/1/repository/commits", method="post",
+)
+def resp_create_commit(url, request):
+    """Mock for commit create POST response."""
+    content = """{
+    "id": "ed899a2f4b50b4370feeea94676502b42383c746",
+    "short_id": "ed899a2f",
+    "title": "Commit message"
+    }"""
+    content = content.encode("utf-8")
+    return response(200, content, headers, None, 5, request)
+
+
+@urlmatch(
+    scheme="http", path="/api/v4/projects/1/repository/commits/6b2257ea", method="post",
+)
+def resp_revert_commit(url, request):
+    """Mock for commit revert POST response."""
+    content = """{
+    "id": "8b090c1b79a14f2bd9e8a738f717824ff53aebad",
+    "short_id": "8b090c1b",
+    "title":"Revert \\"Initial commit\\""
+    }"""
+    content = content.encode("utf-8")
+    return response(200, content, headers, None, 5, request)
+
+
+@urlmatch(
+    scheme="http",
+    netloc="localhost",
+    path="/api/v4/projects/1/repository/commits/6b2257ea/signature",
+    method="get",
+)
+def resp_get_commit_gpg_signature(url, request):
+    """Mock for commit GPG signature GET response."""
+    content = """{
+    "gpg_key_id": 1,
+    "gpg_key_primary_keyid": "8254AAB3FBD54AC9",
+    "gpg_key_user_name": "John Doe",
+    "gpg_key_user_email": "johndoe@example.com",
+    "verification_status": "verified",
+    "gpg_key_subkey_id": null
+    }"""
+    content = content.encode("utf-8")
+    return response(200, content, headers, None, 5, request)
+
+
+class TestCommit(TestProject):
+    """
+    Base class for commit tests. Inherits from TestProject,
+    since currently all commit methods are under projects.
+    """
+
+    @with_httmock(resp_get_commit)
+    def test_get_commit(self):
+        commit = self.project.commits.get("6b2257ea")
+        self.assertEqual(commit.short_id, "6b2257ea")
+        self.assertEqual(commit.title, "Initial commit")
+
+    @with_httmock(resp_create_commit)
+    def test_create_commit(self):
+        data = {
+            "branch": "master",
+            "commit_message": "Commit message",
+            "actions": [{"action": "create", "file_path": "README", "content": "",}],
+        }
+        commit = self.project.commits.create(data)
+        self.assertEqual(commit.short_id, "ed899a2f")
+        self.assertEqual(commit.title, data["commit_message"])
+
+    @with_httmock(resp_revert_commit)
+    def test_revert_commit(self):
+        commit = self.project.commits.get("6b2257ea", lazy=True)
+        revert_commit = commit.revert(branch="master")
+        self.assertEqual(revert_commit["short_id"], "8b090c1b")
+        self.assertEqual(revert_commit["title"], 'Revert "Initial commit"')
+
+    @with_httmock(resp_get_commit_gpg_signature)
+    def test_get_commit_gpg_signature(self):
+        commit = self.project.commits.get("6b2257ea", lazy=True)
+        signature = commit.signature()
+        self.assertEqual(signature["gpg_key_primary_keyid"], "8254AAB3FBD54AC9")
+        self.assertEqual(signature["verification_status"], "verified")
diff --git a/gitlab/tests/test_gitlab.py b/gitlab/tests/test_gitlab.py
index 591f166..d104c7d 100644
--- a/gitlab/tests/test_gitlab.py
+++ b/gitlab/tests/test_gitlab.py
@@ -843,50 +843,6 @@ class TestGitlab(unittest.TestCase):
             self.gl.users.get(1, lazy=True).activate()
             self.gl.users.get(1, lazy=True).deactivate()
 
-    def test_commit_revert(self):
-        @urlmatch(
-            scheme="http",
-            netloc="localhost",
-            path="/api/v4/projects/1/repository/commits/6b2257ea",
-            method="get",
-        )
-        def resp_get_commit(url, request):
-            headers = {"content-type": "application/json"}
-            content = """{
-            "id": "6b2257eabcec3db1f59dafbd84935e3caea04235",
-            "short_id": "6b2257ea",
-            "title": "Initial commit"
-            }"""
-            content = content.encode("utf-8")
-            return response(200, content, headers, None, 5, request)
-
-        @urlmatch(
-            scheme="http",
-            netloc="localhost",
-            path="/api/v4/projects/1/repository/commits/6b2257ea",
-            method="post",
-        )
-        def resp_revert_commit(url, request):
-            headers = {"content-type": "application/json"}
-            content = """{
-            "id": "8b090c1b79a14f2bd9e8a738f717824ff53aebad",
-            "short_id": "8b090c1b",
-            "title":"Revert \\"Initial commit\\""
-            }"""
-            content = content.encode("utf-8")
-            return response(200, content, headers, None, 5, request)
-
-        with HTTMock(resp_get_commit):
-            project = self.gl.projects.get(1, lazy=True)
-            commit = project.commits.get("6b2257ea")
-            self.assertEqual(commit.short_id, "6b2257ea")
-            self.assertEqual(commit.title, "Initial commit")
-
-        with HTTMock(resp_revert_commit):
-            revert_commit = commit.revert(branch="master")
-            self.assertEqual(revert_commit["short_id"], "8b090c1b")
-            self.assertEqual(revert_commit["title"], 'Revert "Initial commit"')
-
     def test_update_submodule(self):
         @urlmatch(
             scheme="http", netloc="localhost", path="/api/v4/projects/1$", method="get"
diff --git a/gitlab/v4/objects.py b/gitlab/v4/objects.py
index 9da9adf..96327b2 100644
--- a/gitlab/v4/objects.py
+++ b/gitlab/v4/objects.py
@@ -2172,6 +2172,24 @@ class ProjectCommit(RESTObject):
         post_data = {"branch": branch}
         return self.manager.gitlab.http_post(path, post_data=post_data, **kwargs)
 
+    @cli.register_custom_action("ProjectCommit")
+    @exc.on_http_error(exc.GitlabGetError)
+    def signature(self, **kwargs):
+        """Get the GPG signature of the commit.
+
+        Args:
+            **kwargs: Extra options to send to the server (e.g. sudo)
+
+        Raises:
+            GitlabAuthenticationError: If authentication is not correct
+            GitlabGetError: If the signature could not be retrieved
+
+        Returns:
+            dict: The commit's GPG signature data
+        """
+        path = "%s/%s/signature" % (self.manager.path, self.get_id())
+        return self.manager.gitlab.http_get(path, **kwargs)
+
 
 class ProjectCommitManager(RetrieveMixin, CreateMixin, RESTManager):
     _path = "/projects/%(project_id)s/repository/commits"