mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
0d598c3542
* Fix copyright lines.
* Add default copyright line.
(cherry picked from commit 424d706f92
)
Co-authored-by: Felix Fontein <felix@fontein.de>
46 lines
2.3 KiB
Python
46 lines
2.3 KiB
Python
# Copyright (c) Ansible project
|
|
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
from ansible_collections.community.general.tests.unit.compat import unittest
|
|
from ansible_collections.community.general.plugins.modules.monitoring import pagerduty_alert
|
|
|
|
|
|
class PagerDutyAlertsTest(unittest.TestCase):
|
|
def _assert_incident_api(self, module, url, method, headers):
|
|
self.assertTrue('https://api.pagerduty.com/incidents' in url, 'url must contain REST API v2 network path')
|
|
self.assertTrue('service_ids%5B%5D=service_id' in url, 'url must contain service id to filter incidents')
|
|
self.assertTrue('sort_by=incident_number%3Adesc' in url, 'url should contain sorting parameter')
|
|
self.assertTrue('time_zone=UTC' in url, 'url should contain time zone parameter')
|
|
return Response(), {'status': 200}
|
|
|
|
def _assert_compatibility_header(self, module, url, method, headers):
|
|
self.assertDictContainsSubset(
|
|
{'Accept': 'application/vnd.pagerduty+json;version=2'},
|
|
headers,
|
|
'Accept:application/vnd.pagerduty+json;version=2 HTTP header not found'
|
|
)
|
|
return Response(), {'status': 200}
|
|
|
|
def _assert_incident_key(self, module, url, method, headers):
|
|
self.assertTrue('incident_key=incident_key_value' in url, 'url must contain incident key')
|
|
return Response(), {'status': 200}
|
|
|
|
def test_incident_url(self):
|
|
pagerduty_alert.check(None, 'name', 'state', 'service_id', 'integration_key', 'api_key', http_call=self._assert_incident_api)
|
|
|
|
def test_compatibility_header(self):
|
|
pagerduty_alert.check(None, 'name', 'state', 'service_id', 'integration_key', 'api_key', http_call=self._assert_compatibility_header)
|
|
|
|
def test_incident_key_in_url_when_it_is_given(self):
|
|
pagerduty_alert.check(
|
|
None, 'name', 'state', 'service_id', 'integration_key', 'api_key', incident_key='incident_key_value', http_call=self._assert_incident_key
|
|
)
|
|
|
|
|
|
class Response(object):
|
|
def read(self):
|
|
return '{"incidents":[{"id": "incident_id", "status": "triggered"}]}'
|