feat(objects): add support for group audit events API

nejch · nejch · commit 3193b8da2d51 · 2021-03-07T12:50:26.000+01:00
diff --git a/gitlab/tests/objects/test_audit_events.py b/gitlab/tests/objects/test_audit_events.py
@@ -8,7 +8,7 @@
 import pytest
 import responses
 
-from gitlab.v4.objects.audit_events import AuditEvent, ProjectAudit
+from gitlab.v4.objects.audit_events import AuditEvent, GroupAuditEvent, ProjectAudit
 
 id = 5
 
@@ -79,6 +79,19 @@ def test_get_instance_audit_events(gl, resp_get_audit_event):
     assert audit_event.id == id
 
 
+def test_list_group_audit_events(group, resp_list_audit_events):
+    audit_events = group.audit_events.list()
+    assert isinstance(audit_events, list)
+    assert isinstance(audit_events[0], GroupAuditEvent)
+    assert audit_events[0].id == id
+
+
+def test_get_group_audit_events(group, resp_get_audit_event):
+    audit_event = group.audit_events.get(id)
+    assert isinstance(audit_event, GroupAuditEvent)
+    assert audit_event.id == id
+
+
 def test_list_project_audit_events(project, resp_list_audit_events):
     audit_events = project.audit_events.list()
     assert isinstance(audit_events, list)
diff --git a/gitlab/v4/objects/audit_events.py b/gitlab/v4/objects/audit_events.py
@@ -9,6 +9,8 @@
 __all__ = [
     "AuditEvent",
     "AuditEventManager",
+    "GroupAuditEvent",
+    "GroupAuditEventManager",
     "ProjectAudit",
     "ProjectAuditManager",
 ]
@@ -24,6 +26,17 @@ class AuditEventManager(RetrieveMixin, RESTManager):
     _list_filters = ("created_after", "created_before", "entity_type", "entity_id")
 
 
+class GroupAuditEvent(RESTObject):
+    _id_attr = "id"
+
+
+class GroupAuditEventManager(RetrieveMixin, RESTManager):
+    _path = "/projects/%(project_id)s/audit_events"
+    _obj_cls = GroupAuditEvent
+    _from_parent_attrs = {"project_id": "id"}
+    _list_filters = ("created_after", "created_before")
+
+
 class ProjectAudit(RESTObject):
     _id_attr = "id"
 
diff --git a/gitlab/v4/objects/groups.py b/gitlab/v4/objects/groups.py
@@ -3,6 +3,7 @@
 from gitlab.base import RESTManager, RESTObject
 from gitlab.mixins import CRUDMixin, ListMixin, ObjectDeleteMixin, SaveMixin
 from .access_requests import GroupAccessRequestManager
+from .audit_events import GroupAuditEventManager
 from .badges import GroupBadgeManager
 from .boards import GroupBoardManager
 from .custom_attributes import GroupCustomAttributeManager
@@ -34,6 +35,7 @@ class Group(SaveMixin, ObjectDeleteMixin, RESTObject):
     _short_print_attr = "name"
     _managers = (
         ("accessrequests", "GroupAccessRequestManager"),
+        ("audit_events", "GroupAuditEventManager"),
         ("badges", "GroupBadgeManager"),
         ("boards", "GroupBoardManager"),
         ("customattributes", "GroupCustomAttributeManager"),
