From 10225cf26095efe82713136ddde3330e7afc6d10 Mon Sep 17 00:00:00 2001
From: Nejc Habjan <hab.nejc@gmail.com>
Date: Sun, 7 Mar 2021 18:05:23 +0100
Subject: [PATCH] test(objects): add tests for resource state events

---
 gitlab/tests/conftest.py                      |  10 ++
 .../objects/test_resource_state_events.py     | 105 ++++++++++++++++++
 2 files changed, 115 insertions(+)
 create mode 100644 gitlab/tests/objects/test_resource_state_events.py

diff --git a/gitlab/tests/conftest.py b/gitlab/tests/conftest.py
index fc8312f34..74fb858fa 100644
--- a/gitlab/tests/conftest.py
+++ b/gitlab/tests/conftest.py
@@ -52,6 +52,16 @@ def project(gl):
     return gl.projects.get(1, lazy=True)
 
 
+@pytest.fixture
+def project_issue(project):
+    return project.issues.get(1, lazy=True)
+
+
+@pytest.fixture
+def project_merge_request(project):
+    return project.mergerequests.get(1, lazy=True)
+
+
 @pytest.fixture
 def release(project, tag_name):
     return project.releases.get(tag_name, lazy=True)
diff --git a/gitlab/tests/objects/test_resource_state_events.py b/gitlab/tests/objects/test_resource_state_events.py
new file mode 100644
index 000000000..01c18870f
--- /dev/null
+++ b/gitlab/tests/objects/test_resource_state_events.py
@@ -0,0 +1,105 @@
+"""
+GitLab API: https://docs.gitlab.com/ee/api/resource_state_events.html
+"""
+
+import pytest
+import responses
+
+from gitlab.v4.objects import (
+    ProjectIssueResourceStateEvent,
+    ProjectMergeRequestResourceStateEvent,
+)
+
+
+issue_event_content = {"id": 1, "resource_type": "Issue"}
+mr_event_content = {"id": 1, "resource_type": "MergeRequest"}
+
+
+@pytest.fixture()
+def resp_list_project_issue_state_events():
+    with responses.RequestsMock() as rsps:
+        rsps.add(
+            method=responses.GET,
+            url="http://localhost/api/v4/projects/1/issues/1/resource_state_events",
+            json=[issue_event_content],
+            content_type="application/json",
+            status=200,
+        )
+        yield rsps
+
+
+@pytest.fixture()
+def resp_get_project_issue_state_event():
+    with responses.RequestsMock() as rsps:
+        rsps.add(
+            method=responses.GET,
+            url="http://localhost/api/v4/projects/1/issues/1/resource_state_events/1",
+            json=issue_event_content,
+            content_type="application/json",
+            status=200,
+        )
+        yield rsps
+
+
+@pytest.fixture()
+def resp_list_merge_request_state_events():
+    with responses.RequestsMock() as rsps:
+        rsps.add(
+            method=responses.GET,
+            url="http://localhost/api/v4/projects/1/merge_requests/1/resource_state_events",
+            json=[mr_event_content],
+            content_type="application/json",
+            status=200,
+        )
+        yield rsps
+
+
+@pytest.fixture()
+def resp_get_merge_request_state_event():
+    with responses.RequestsMock() as rsps:
+        rsps.add(
+            method=responses.GET,
+            url="http://localhost/api/v4/projects/1/merge_requests/1/resource_state_events/1",
+            json=mr_event_content,
+            content_type="application/json",
+            status=200,
+        )
+        yield rsps
+
+
+def test_list_project_issue_state_events(
+    project_issue, resp_list_project_issue_state_events
+):
+    state_events = project_issue.resourcestateevents.list()
+    assert isinstance(state_events, list)
+
+    state_event = state_events[0]
+    assert isinstance(state_event, ProjectIssueResourceStateEvent)
+    assert state_event.resource_type == "Issue"
+
+
+def test_get_project_issue_state_event(
+    project_issue, resp_get_project_issue_state_event
+):
+    state_event = project_issue.resourcestateevents.get(1)
+    assert isinstance(state_event, ProjectIssueResourceStateEvent)
+    assert state_event.resource_type == "Issue"
+
+
+def test_list_merge_request_state_events(
+    project_merge_request, resp_list_merge_request_state_events
+):
+    state_events = project_merge_request.resourcestateevents.list()
+    assert isinstance(state_events, list)
+
+    state_event = state_events[0]
+    assert isinstance(state_event, ProjectMergeRequestResourceStateEvent)
+    assert state_event.resource_type == "MergeRequest"
+
+
+def test_get_merge_request_state_event(
+    project_merge_request, resp_get_merge_request_state_event
+):
+    state_event = project_merge_request.resourcestateevents.get(1)
+    assert isinstance(state_event, ProjectMergeRequestResourceStateEvent)
+    assert state_event.resource_type == "MergeRequest"