Add test_connection method to PagerdutyHook (#31344)

* Add test_connection method to PagerdutyHook * update the error message
apache · May 18, 2023 · 8bce0ce · 8bce0ce
1 parent 8871a9e
commit 8bce0ce
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/airflow/providers/pagerduty/hooks/pagerduty.py b/airflow/providers/pagerduty/hooks/pagerduty.py
@@ -159,3 +159,11 @@ def create_event(
             images=images,
             links=links,
         )
+
+    def test_connection(self):
+        try:
+            session = pdpyras.APISession(self.token)
+            session.list_all("services", params={"query": "some_non_existing_service"})
+        except Exception:
+            return False, "connection test failed, invalid token"
+        return True, "connection tested successfully"