From 7bf952b4781925c06d3941530d58b9ba100969ac Mon Sep 17 00:00:00 2001
From: Utkarsh Sharma <utkarsharma2@gmail.com>
Date: Fri, 19 May 2023 16:46:18 +0530
Subject: [PATCH] Add Change event V2 API support for pagerduty (#31222)

* Add Change event V2 API
---
 .../pagerduty/hooks/pagerduty_events.py       | 43 +++++++++++++++++++
 .../pagerduty/hooks/test_pagerduty_events.py  |  8 ++++
 2 files changed, 51 insertions(+)

diff --git a/airflow/providers/pagerduty/hooks/pagerduty_events.py b/airflow/providers/pagerduty/hooks/pagerduty_events.py
index 47a3094722b52..c86790bbd75c0 100644
--- a/airflow/providers/pagerduty/hooks/pagerduty_events.py
+++ b/airflow/providers/pagerduty/hooks/pagerduty_events.py
@@ -18,6 +18,7 @@
 """Hook for sending or receiving data from PagerDuty as well as creating PagerDuty incidents."""
 from __future__ import annotations
 
+from datetime import datetime
 from typing import Any
 
 import pdpyras
@@ -152,6 +153,48 @@ def create_event(
         resp.raise_for_status()
         return resp.json()
 
+    def create_change_event(
+        self,
+        summary: str,
+        source: str = "airflow",
+        custom_details: Any | None = None,
+        timestamp: datetime | None = None,
+        links: list[Any] | None = None,
+    ) -> dict:
+        """
+        Create change event for service integration.
+
+        :param summary: Summary for the event
+        :param source: Specific human-readable unique identifier, such as a
+            hostname, for the system having the problem.
+        :param custom_details: Free-form details from the event. Can be a dictionary or a string.
+            If a dictionary is passed it will show up in PagerDuty as a table.
+        :param timestamp: The time at which the emitting tool detected or generated the event.
+        :param links: List of links to include. Each dictionary in the list accepts the following keys:
+            `href`: URL of the link to be attached.
+            `text`: [Optional] Plain text that describes the purpose of the link, and can be used as the
+            link's text.
+        :return: PagerDuty Change Events API v2 response.
+        """
+        payload = {
+            "summary": summary,
+        }
+        if custom_details is not None:
+            payload["custom_details"] = custom_details
+
+        if timestamp is not None:
+            payload["timestamp"] = timestamp.isoformat()
+
+        if source is not None:
+            payload["source"] = source
+
+        data: dict[str, Any] = {"payload": payload}
+        if links is not None:
+            data["links"] = links
+
+        session = pdpyras.ChangeEventsAPISession(self.integration_key)
+        return session.send_change_event(payload=payload, links=links)
+
     def test_connection(self):
         try:
             session = pdpyras.EventsAPISession(self.integration_key)
diff --git a/tests/providers/pagerduty/hooks/test_pagerduty_events.py b/tests/providers/pagerduty/hooks/test_pagerduty_events.py
index 3c68ba8247954..7ba86d25dbc0d 100644
--- a/tests/providers/pagerduty/hooks/test_pagerduty_events.py
+++ b/tests/providers/pagerduty/hooks/test_pagerduty_events.py
@@ -54,3 +54,11 @@ def test_create_event(self, requests_mock, events_connections):
             severity="error",
         )
         assert resp == mock_response_body
+
+    def test_create_change_event(self, requests_mock, events_connections):
+        hook = PagerdutyEventsHook(pagerduty_events_conn_id=DEFAULT_CONN_ID)
+        change_event_id = "change_event_id"
+        mock_response_body = {"id": change_event_id}
+        requests_mock.post("https://events.pagerduty.com/v2/change/enqueue", json=mock_response_body)
+        resp = hook.create_change_event(summary="test", source="airflow")
+        assert resp == change_event_id