mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
225 lines
8.5 KiB
Python
225 lines
8.5 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
||
|
|
|
||
|
|
from __future__ import (absolute_import, division, print_function)
|
||
|
|
|
||
|
|
__metaclass__ = type
|
||
|
|
|
||
|
|
from ansible_collections.community.general.plugins.modules.monitoring.datadog import datadog_downtime
|
||
|
|
from ansible_collections.community.general.tests.unit.compat.mock import MagicMock, patch
|
||
|
|
from ansible_collections.community.general.tests.unit.plugins.modules.utils import (
|
||
|
|
AnsibleExitJson, AnsibleFailJson, ModuleTestCase, set_module_args
|
||
|
|
)
|
||
|
|
|
||
|
|
from pytest import importorskip
|
||
|
|
|
||
|
|
# Skip this test if python 2 so datadog_api_client cannot be installed
|
||
|
|
datadog_api_client = importorskip("datadog_api_client")
|
||
|
|
Downtime = datadog_api_client.v1.model.downtime.Downtime
|
||
|
|
DowntimeRecurrence = datadog_api_client.v1.model.downtime_recurrence.DowntimeRecurrence
|
||
|
|
|
||
|
|
|
||
|
|
class TestDatadogDowntime(ModuleTestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super(TestDatadogDowntime, self).setUp()
|
||
|
|
self.module = datadog_downtime
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
super(TestDatadogDowntime, self).tearDown()
|
||
|
|
|
||
|
|
def test_without_required_parameters(self):
|
||
|
|
"""Failure must occurs when all parameters are missing"""
|
||
|
|
with self.assertRaises(AnsibleFailJson):
|
||
|
|
set_module_args({})
|
||
|
|
self.module.main()
|
||
|
|
|
||
|
|
@patch("ansible_collections.community.general.plugins.modules.monitoring.datadog.datadog_downtime.DowntimesApi")
|
||
|
|
def test_create_downtime_when_no_id(self, downtimes_api_mock):
|
||
|
|
set_module_args({
|
||
|
|
"monitor_tags": ["foo:bar"],
|
||
|
|
"scope": ["*"],
|
||
|
|
"monitor_id": 12345,
|
||
|
|
"downtime_message": "Message",
|
||
|
|
"start": 1111,
|
||
|
|
"end": 2222,
|
||
|
|
"timezone": "UTC",
|
||
|
|
"rrule": "rrule",
|
||
|
|
"api_key": "an_api_key",
|
||
|
|
"app_key": "an_app_key",
|
||
|
|
})
|
||
|
|
|
||
|
|
downtime = Downtime()
|
||
|
|
downtime.monitor_tags = ["foo:bar"]
|
||
|
|
downtime.scope = ["*"]
|
||
|
|
downtime.monitor_id = 12345
|
||
|
|
downtime.message = "Message"
|
||
|
|
downtime.start = 1111
|
||
|
|
downtime.end = 2222
|
||
|
|
downtime.timezone = "UTC"
|
||
|
|
downtime.recurrence = DowntimeRecurrence(
|
||
|
|
rrule="rrule"
|
||
|
|
)
|
||
|
|
|
||
|
|
create_downtime_mock = MagicMock(return_value=Downtime(id=12345))
|
||
|
|
downtimes_api_mock.return_value = MagicMock(create_downtime=create_downtime_mock)
|
||
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
||
|
|
self.module.main()
|
||
|
|
self.assertTrue(result.exception.args[0]['changed'])
|
||
|
|
self.assertEqual(result.exception.args[0]['downtime']['id'], 12345)
|
||
|
|
create_downtime_mock.assert_called_once_with(downtime)
|
||
|
|
|
||
|
|
@patch("ansible_collections.community.general.plugins.modules.monitoring.datadog.datadog_downtime.DowntimesApi")
|
||
|
|
def test_create_downtime_when_id_and_disabled(self, downtimes_api_mock):
|
||
|
|
set_module_args({
|
||
|
|
"id": 1212,
|
||
|
|
"monitor_tags": ["foo:bar"],
|
||
|
|
"scope": ["*"],
|
||
|
|
"monitor_id": 12345,
|
||
|
|
"downtime_message": "Message",
|
||
|
|
"start": 1111,
|
||
|
|
"end": 2222,
|
||
|
|
"timezone": "UTC",
|
||
|
|
"rrule": "rrule",
|
||
|
|
"api_key": "an_api_key",
|
||
|
|
"app_key": "an_app_key",
|
||
|
|
})
|
||
|
|
|
||
|
|
downtime = Downtime()
|
||
|
|
downtime.monitor_tags = ["foo:bar"]
|
||
|
|
downtime.scope = ["*"]
|
||
|
|
downtime.monitor_id = 12345
|
||
|
|
downtime.message = "Message"
|
||
|
|
downtime.start = 1111
|
||
|
|
downtime.end = 2222
|
||
|
|
downtime.timezone = "UTC"
|
||
|
|
downtime.recurrence = DowntimeRecurrence(
|
||
|
|
rrule="rrule"
|
||
|
|
)
|
||
|
|
|
||
|
|
create_downtime_mock = MagicMock(return_value=Downtime(id=12345))
|
||
|
|
get_downtime_mock = MagicMock(return_value=Downtime(id=1212, disabled=True))
|
||
|
|
downtimes_api_mock.return_value = MagicMock(
|
||
|
|
create_downtime=create_downtime_mock, get_downtime=get_downtime_mock
|
||
|
|
)
|
||
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
||
|
|
self.module.main()
|
||
|
|
self.assertTrue(result.exception.args[0]['changed'])
|
||
|
|
self.assertEqual(result.exception.args[0]['downtime']['id'], 12345)
|
||
|
|
create_downtime_mock.assert_called_once_with(downtime)
|
||
|
|
get_downtime_mock.assert_called_once_with(1212)
|
||
|
|
|
||
|
|
@patch("ansible_collections.community.general.plugins.modules.monitoring.datadog.datadog_downtime.DowntimesApi")
|
||
|
|
def test_update_downtime_when_not_disabled(self, downtimes_api_mock):
|
||
|
|
set_module_args({
|
||
|
|
"id": 1212,
|
||
|
|
"monitor_tags": ["foo:bar"],
|
||
|
|
"scope": ["*"],
|
||
|
|
"monitor_id": 12345,
|
||
|
|
"downtime_message": "Message",
|
||
|
|
"start": 1111,
|
||
|
|
"end": 2222,
|
||
|
|
"timezone": "UTC",
|
||
|
|
"rrule": "rrule",
|
||
|
|
"api_key": "an_api_key",
|
||
|
|
"app_key": "an_app_key",
|
||
|
|
})
|
||
|
|
|
||
|
|
downtime = Downtime()
|
||
|
|
downtime.monitor_tags = ["foo:bar"]
|
||
|
|
downtime.scope = ["*"]
|
||
|
|
downtime.monitor_id = 12345
|
||
|
|
downtime.message = "Message"
|
||
|
|
downtime.start = 1111
|
||
|
|
downtime.end = 2222
|
||
|
|
downtime.timezone = "UTC"
|
||
|
|
downtime.recurrence = DowntimeRecurrence(
|
||
|
|
rrule="rrule"
|
||
|
|
)
|
||
|
|
|
||
|
|
update_downtime_mock = MagicMock(return_value=Downtime(id=1212))
|
||
|
|
get_downtime_mock = MagicMock(return_value=Downtime(id=1212, disabled=False))
|
||
|
|
downtimes_api_mock.return_value = MagicMock(
|
||
|
|
update_downtime=update_downtime_mock, get_downtime=get_downtime_mock
|
||
|
|
)
|
||
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
||
|
|
self.module.main()
|
||
|
|
self.assertTrue(result.exception.args[0]['changed'])
|
||
|
|
self.assertEqual(result.exception.args[0]['downtime']['id'], 1212)
|
||
|
|
update_downtime_mock.assert_called_once_with(1212, downtime)
|
||
|
|
get_downtime_mock.assert_called_once_with(1212)
|
||
|
|
|
||
|
|
@patch("ansible_collections.community.general.plugins.modules.monitoring.datadog.datadog_downtime.DowntimesApi")
|
||
|
|
def test_update_downtime_no_change(self, downtimes_api_mock):
|
||
|
|
set_module_args({
|
||
|
|
"id": 1212,
|
||
|
|
"monitor_tags": ["foo:bar"],
|
||
|
|
"scope": ["*"],
|
||
|
|
"monitor_id": 12345,
|
||
|
|
"downtime_message": "Message",
|
||
|
|
"start": 1111,
|
||
|
|
"end": 2222,
|
||
|
|
"timezone": "UTC",
|
||
|
|
"rrule": "rrule",
|
||
|
|
"api_key": "an_api_key",
|
||
|
|
"app_key": "an_app_key",
|
||
|
|
})
|
||
|
|
|
||
|
|
downtime = Downtime()
|
||
|
|
downtime.monitor_tags = ["foo:bar"]
|
||
|
|
downtime.scope = ["*"]
|
||
|
|
downtime.monitor_id = 12345
|
||
|
|
downtime.message = "Message"
|
||
|
|
downtime.start = 1111
|
||
|
|
downtime.end = 2222
|
||
|
|
downtime.timezone = "UTC"
|
||
|
|
downtime.recurrence = DowntimeRecurrence(
|
||
|
|
rrule="rrule"
|
||
|
|
)
|
||
|
|
|
||
|
|
downtime_get = Downtime()
|
||
|
|
downtime_get.id = 1212
|
||
|
|
downtime_get.disabled = False
|
||
|
|
downtime_get.monitor_tags = ["foo:bar"]
|
||
|
|
downtime_get.scope = ["*"]
|
||
|
|
downtime_get.monitor_id = 12345
|
||
|
|
downtime_get.message = "Message"
|
||
|
|
downtime_get.start = 1111
|
||
|
|
downtime_get.end = 2222
|
||
|
|
downtime_get.timezone = "UTC"
|
||
|
|
downtime_get.recurrence = DowntimeRecurrence(
|
||
|
|
rrule="rrule"
|
||
|
|
)
|
||
|
|
|
||
|
|
update_downtime_mock = MagicMock(return_value=downtime_get)
|
||
|
|
get_downtime_mock = MagicMock(return_value=downtime_get)
|
||
|
|
downtimes_api_mock.return_value = MagicMock(
|
||
|
|
update_downtime=update_downtime_mock, get_downtime=get_downtime_mock
|
||
|
|
)
|
||
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
||
|
|
self.module.main()
|
||
|
|
self.assertFalse(result.exception.args[0]['changed'])
|
||
|
|
self.assertEqual(result.exception.args[0]['downtime']['id'], 1212)
|
||
|
|
update_downtime_mock.assert_called_once_with(1212, downtime)
|
||
|
|
get_downtime_mock.assert_called_once_with(1212)
|
||
|
|
|
||
|
|
@patch("ansible_collections.community.general.plugins.modules.monitoring.datadog.datadog_downtime.DowntimesApi")
|
||
|
|
def test_delete_downtime(self, downtimes_api_mock):
|
||
|
|
set_module_args({
|
||
|
|
"id": 1212,
|
||
|
|
"state": "absent",
|
||
|
|
"api_key": "an_api_key",
|
||
|
|
"app_key": "an_app_key",
|
||
|
|
})
|
||
|
|
|
||
|
|
cancel_downtime_mock = MagicMock()
|
||
|
|
get_downtime_mock = MagicMock(return_value=Downtime(id=1212))
|
||
|
|
downtimes_api_mock.return_value = MagicMock(
|
||
|
|
get_downtime=get_downtime_mock,
|
||
|
|
cancel_downtime=cancel_downtime_mock
|
||
|
|
)
|
||
|
|
with self.assertRaises(AnsibleExitJson) as result:
|
||
|
|
self.module.main()
|
||
|
|
self.assertTrue(result.exception.args[0]['changed'])
|
||
|
|
cancel_downtime_mock.assert_called_once_with(1212)
|