mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-05-07 22:02:50 +00:00
Reformat everything.
This commit is contained in:
@@ -12,88 +12,81 @@ import json
|
||||
|
||||
class PagerDutyTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.pd = pagerduty.PagerDutyRequest(module=pagerduty, name='name', user='user', token='token')
|
||||
self.pd = pagerduty.PagerDutyRequest(module=pagerduty, name="name", user="user", token="token")
|
||||
|
||||
def _assert_ongoing_maintenance_windows(self, module, url, headers):
|
||||
self.assertEqual('https://api.pagerduty.com/maintenance_windows?filter=ongoing', url)
|
||||
return object(), {'status': 200}
|
||||
self.assertEqual("https://api.pagerduty.com/maintenance_windows?filter=ongoing", url)
|
||||
return object(), {"status": 200}
|
||||
|
||||
def _assert_ongoing_window_with_v1_compatible_header(self, module, url, headers, data=None, method=None):
|
||||
self.assertEqual(
|
||||
'application/vnd.pagerduty+json;version=2',
|
||||
headers.get('Accept'),
|
||||
'Accept:application/vnd.pagerduty+json;version=2 HTTP header not found'
|
||||
"application/vnd.pagerduty+json;version=2",
|
||||
headers.get("Accept"),
|
||||
"Accept:application/vnd.pagerduty+json;version=2 HTTP header not found",
|
||||
)
|
||||
return object(), {'status': 200}
|
||||
return object(), {"status": 200}
|
||||
|
||||
def _assert_create_a_maintenance_window_url(self, module, url, headers, data=None, method=None):
|
||||
self.assertEqual('https://api.pagerduty.com/maintenance_windows', url)
|
||||
return object(), {'status': 201}
|
||||
self.assertEqual("https://api.pagerduty.com/maintenance_windows", url)
|
||||
return object(), {"status": 201}
|
||||
|
||||
def _assert_create_a_maintenance_window_http_method(self, module, url, headers, data=None, method=None):
|
||||
self.assertEqual('POST', method)
|
||||
return object(), {'status': 201}
|
||||
self.assertEqual("POST", method)
|
||||
return object(), {"status": 201}
|
||||
|
||||
def _assert_create_a_maintenance_window_from_header(self, module, url, headers, data=None, method=None):
|
||||
self.assertEqual(
|
||||
'requester_id',
|
||||
headers.get('From'),
|
||||
'From:requester_id HTTP header not found'
|
||||
)
|
||||
return object(), {'status': 201}
|
||||
self.assertEqual("requester_id", headers.get("From"), "From:requester_id HTTP header not found")
|
||||
return object(), {"status": 201}
|
||||
|
||||
def _assert_create_window_with_v1_compatible_header(self, module, url, headers, data=None, method=None):
|
||||
self.assertEqual(
|
||||
'application/vnd.pagerduty+json;version=2',
|
||||
headers.get('Accept'),
|
||||
'Accept:application/vnd.pagerduty+json;version=2 HTTP header not found'
|
||||
"application/vnd.pagerduty+json;version=2",
|
||||
headers.get("Accept"),
|
||||
"Accept:application/vnd.pagerduty+json;version=2 HTTP header not found",
|
||||
)
|
||||
return object(), {'status': 201}
|
||||
return object(), {"status": 201}
|
||||
|
||||
def _assert_create_window_payload(self, module, url, headers, data=None, method=None):
|
||||
payload = json.loads(data)
|
||||
window_data = payload['maintenance_window']
|
||||
self.assertTrue('start_time' in window_data, '"start_time" is required attribute')
|
||||
self.assertTrue('end_time' in window_data, '"end_time" is required attribute')
|
||||
self.assertTrue('services' in window_data, '"services" is required attribute')
|
||||
return object(), {'status': 201}
|
||||
window_data = payload["maintenance_window"]
|
||||
self.assertTrue("start_time" in window_data, '"start_time" is required attribute')
|
||||
self.assertTrue("end_time" in window_data, '"end_time" is required attribute')
|
||||
self.assertTrue("services" in window_data, '"services" is required attribute')
|
||||
return object(), {"status": 201}
|
||||
|
||||
def _assert_create_window_single_service(self, module, url, headers, data=None, method=None):
|
||||
payload = json.loads(data)
|
||||
window_data = payload['maintenance_window']
|
||||
services = window_data['services']
|
||||
self.assertEqual(
|
||||
[{'id': 'service_id', 'type': 'service_reference'}],
|
||||
services
|
||||
)
|
||||
return object(), {'status': 201}
|
||||
window_data = payload["maintenance_window"]
|
||||
services = window_data["services"]
|
||||
self.assertEqual([{"id": "service_id", "type": "service_reference"}], services)
|
||||
return object(), {"status": 201}
|
||||
|
||||
def _assert_create_window_multiple_service(self, module, url, headers, data=None, method=None):
|
||||
payload = json.loads(data)
|
||||
window_data = payload['maintenance_window']
|
||||
services = window_data['services']
|
||||
window_data = payload["maintenance_window"]
|
||||
services = window_data["services"]
|
||||
print(services)
|
||||
self.assertEqual(
|
||||
[
|
||||
{'id': 'service_id_1', 'type': 'service_reference'},
|
||||
{'id': 'service_id_2', 'type': 'service_reference'},
|
||||
{'id': 'service_id_3', 'type': 'service_reference'},
|
||||
{"id": "service_id_1", "type": "service_reference"},
|
||||
{"id": "service_id_2", "type": "service_reference"},
|
||||
{"id": "service_id_3", "type": "service_reference"},
|
||||
],
|
||||
services
|
||||
services,
|
||||
)
|
||||
return object(), {'status': 201}
|
||||
return object(), {"status": 201}
|
||||
|
||||
def _assert_absent_maintenance_window_url(self, module, url, headers, method=None):
|
||||
self.assertEqual('https://api.pagerduty.com/maintenance_windows/window_id', url)
|
||||
return object(), {'status': 204}
|
||||
self.assertEqual("https://api.pagerduty.com/maintenance_windows/window_id", url)
|
||||
return object(), {"status": 204}
|
||||
|
||||
def _assert_absent_window_with_v1_compatible_header(self, module, url, headers, method=None):
|
||||
self.assertEqual(
|
||||
'application/vnd.pagerduty+json;version=2',
|
||||
headers.get('Accept'),
|
||||
'Accept:application/vnd.pagerduty+json;version=2 HTTP header not found'
|
||||
"application/vnd.pagerduty+json;version=2",
|
||||
headers.get("Accept"),
|
||||
"Accept:application/vnd.pagerduty+json;version=2 HTTP header not found",
|
||||
)
|
||||
return object(), {'status': 204}
|
||||
return object(), {"status": 204}
|
||||
|
||||
def test_ongoing_maintenance_windos_url(self):
|
||||
self.pd.ongoing(http_call=self._assert_ongoing_maintenance_windows)
|
||||
@@ -102,28 +95,41 @@ class PagerDutyTest(unittest.TestCase):
|
||||
self.pd.ongoing(http_call=self._assert_ongoing_window_with_v1_compatible_header)
|
||||
|
||||
def test_create_maintenance_window_url(self):
|
||||
self.pd.create('requester_id', 'service', 1, 0, 'desc', http_call=self._assert_create_a_maintenance_window_url)
|
||||
self.pd.create("requester_id", "service", 1, 0, "desc", http_call=self._assert_create_a_maintenance_window_url)
|
||||
|
||||
def test_create_maintenance_window_http_method(self):
|
||||
self.pd.create('requester_id', 'service', 1, 0, 'desc', http_call=self._assert_create_a_maintenance_window_http_method)
|
||||
self.pd.create(
|
||||
"requester_id", "service", 1, 0, "desc", http_call=self._assert_create_a_maintenance_window_http_method
|
||||
)
|
||||
|
||||
def test_create_maintenance_from_header(self):
|
||||
self.pd.create('requester_id', 'service', 1, 0, 'desc', http_call=self._assert_create_a_maintenance_window_from_header)
|
||||
self.pd.create(
|
||||
"requester_id", "service", 1, 0, "desc", http_call=self._assert_create_a_maintenance_window_from_header
|
||||
)
|
||||
|
||||
def test_create_maintenance_compatibility_header(self):
|
||||
self.pd.create('requester_id', 'service', 1, 0, 'desc', http_call=self._assert_create_window_with_v1_compatible_header)
|
||||
self.pd.create(
|
||||
"requester_id", "service", 1, 0, "desc", http_call=self._assert_create_window_with_v1_compatible_header
|
||||
)
|
||||
|
||||
def test_create_maintenance_request_payload(self):
|
||||
self.pd.create('requester_id', 'service', 1, 0, 'desc', http_call=self._assert_create_window_payload)
|
||||
self.pd.create("requester_id", "service", 1, 0, "desc", http_call=self._assert_create_window_payload)
|
||||
|
||||
def test_create_maintenance_for_single_service(self):
|
||||
self.pd.create('requester_id', 'service_id', 1, 0, 'desc', http_call=self._assert_create_window_single_service)
|
||||
self.pd.create("requester_id", "service_id", 1, 0, "desc", http_call=self._assert_create_window_single_service)
|
||||
|
||||
def test_create_maintenance_for_multiple_services(self):
|
||||
self.pd.create('requester_id', ['service_id_1', 'service_id_2', 'service_id_3'], 1, 0, 'desc', http_call=self._assert_create_window_multiple_service)
|
||||
self.pd.create(
|
||||
"requester_id",
|
||||
["service_id_1", "service_id_2", "service_id_3"],
|
||||
1,
|
||||
0,
|
||||
"desc",
|
||||
http_call=self._assert_create_window_multiple_service,
|
||||
)
|
||||
|
||||
def test_absent_maintenance_window_url(self):
|
||||
self.pd.absent('window_id', http_call=self._assert_absent_maintenance_window_url)
|
||||
self.pd.absent("window_id", http_call=self._assert_absent_maintenance_window_url)
|
||||
|
||||
def test_absent_maintenance_compatibility_header(self):
|
||||
self.pd.absent('window_id', http_call=self._assert_absent_window_with_v1_compatible_header)
|
||||
self.pd.absent("window_id", http_call=self._assert_absent_window_with_v1_compatible_header)
|
||||
|
||||
Reference in New Issue
Block a user