Merge pull request #1363 from python-gitlab/feat/all-audit-events

Feat: cover all audit events

1 files changed, 37 insertions, 7 deletions

diff --git a/gitlab/tests/objects/test_audit_events.py b/gitlab/tests/objects/test_audit_events.py
index 75bc11c..aba778b 100644
--- a/gitlab/tests/objects/test_audit_events.py
+++ b/gitlab/tests/objects/test_audit_events.py
@@ -8,7 +8,11 @@ import re
 import pytest
 import responses
 
-from gitlab.v4.objects.audit_events import ProjectAudit
+from gitlab.v4.objects.audit_events import (
+    AuditEvent,
+    GroupAuditEvent,
+    ProjectAuditEvent,
+)
 
 id = 5
 
@@ -32,11 +36,11 @@ audit_events_content = {
 }
 
 audit_events_url = re.compile(
-    r"http://localhost/api/v4/((groups|projects)/1/)audit_events"
+    r"http://localhost/api/v4/((groups|projects)/1/)?audit_events"
 )
 
 audit_events_url_id = re.compile(
-    rf"http://localhost/api/v4/((groups|projects)/1/)audit_events/{id}"
+    rf"http://localhost/api/v4/((groups|projects)/1/)?audit_events/{id}"
 )
 
 
@@ -54,7 +58,7 @@ def resp_list_audit_events():
 
 
 @pytest.fixture
-def resp_get_variable():
+def resp_get_audit_event():
     with responses.RequestsMock() as rsps:
         rsps.add(
             method=responses.GET,
@@ -66,14 +70,40 @@ def resp_get_variable():
         yield rsps
 
 
+def test_list_instance_audit_events(gl, resp_list_audit_events):
+    audit_events = gl.audit_events.list()
+    assert isinstance(audit_events, list)
+    assert isinstance(audit_events[0], AuditEvent)
+    assert audit_events[0].id == id
+
+
+def test_get_instance_audit_events(gl, resp_get_audit_event):
+    audit_event = gl.audit_events.get(id)
+    assert isinstance(audit_event, AuditEvent)
+    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)
-    assert isinstance(audit_events[0], ProjectAudit)
+    assert isinstance(audit_events[0], ProjectAuditEvent)
     assert audit_events[0].id == id
 
 
-def test_get_project_audit_events(project, resp_get_variable):
+def test_get_project_audit_events(project, resp_get_audit_event):
     audit_event = project.audit_events.get(id)
-    assert isinstance(audit_event, ProjectAudit)
+    assert isinstance(audit_event, ProjectAuditEvent)
     assert audit_event.id == id