def test_create_event_with_default_routing_key(self, mock_event_create): hook = PagerdutyHook(pagerduty_conn_id=DEFAULT_CONN_ID) mock_event_create.return_value = { "status": "success", "message": "Event processed", "dedup_key": "samplekeyhere", } resp = hook.create_event( summary="test", source="airflow_test", severity="error", custom_details='{"foo": "bar"}', ) self.assertEqual(resp["status"], "success") mock_event_create.assert_called_once_with(api_key="pagerduty_token", data={ "routing_key": "route", "event_action": "trigger", "payload": { "severity": "error", "source": "airflow_test", "summary": "test", "custom_details": '{"foo": "bar"}', }, })
def test_create_event(self, m): hook = PagerdutyHook(pagerduty_conn_id=DEFAULT_CONN_ID) mock_response_body = { "status": "success", "message": "Event processed", "dedup_key": "samplekeyhere", } m.post('https://events.pagerduty.com/v2/enqueue', json=mock_response_body) resp = hook.create_event( routing_key="different_key", summary="test", source="airflow_test", severity="error", ) self.assertEqual(resp, mock_response_body)