ansible/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2026-04-26 13:42:49 +00:00
Code Activity

Reformat everything.

Browse source
This commit is contained in:
Felix Fontein 2025-11-01 12:08:41 +01:00
parent 3f2213791a
commit 340ff8586d
1008 changed files with 61301 additions and 58309 deletions
Download patch file Download diff file Expand all files Collapse all files

273
tests/unit/plugins/modules/test_alerta_customer.py
View file

@ -9,7 +9,12 @@ from unittest.mock import patch
import pytest
from ansible_collections.community.general.plugins.modules import alerta_customer
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import AnsibleExitJson, AnsibleFailJson, ModuleTestCase, set_module_args
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import (
AnsibleExitJson,
AnsibleFailJson,
ModuleTestCase,
set_module_args,
)
class MockedReponse:
@ -21,53 +26,62 @@ class MockedReponse:
def customer_response_page1():
server_response = json.dumps({"customers": [
server_response = json.dumps(
{
"customer": "admin",
"href": "http://localhost:8080/api/customer/d89664a7-9c87-4ab9-8be8-830e7e5f0616",
"id": "d89664a7-9c87-4ab9-8be8-830e7e5f0616",
"match": "admin@example.com"
},
{
"customer": "Developer",
"href": "http://localhost:8080/api/customer/188ed093-84cc-4f46-bf80-4c9127180d9c",
"id": "188ed093-84cc-4f46-bf80-4c9127180d9c",
"match": "dev@example.com"
}],
"more": True,
"page": 1,
"pageSize": 50,
"pages": 1,
"status": "ok",
"total": 2})
"customers": [
{
"customer": "admin",
"href": "http://localhost:8080/api/customer/d89664a7-9c87-4ab9-8be8-830e7e5f0616",
"id": "d89664a7-9c87-4ab9-8be8-830e7e5f0616",
"match": "admin@example.com",
},
{
"customer": "Developer",
"href": "http://localhost:8080/api/customer/188ed093-84cc-4f46-bf80-4c9127180d9c",
"id": "188ed093-84cc-4f46-bf80-4c9127180d9c",
"match": "dev@example.com",
},
],
"more": True,
"page": 1,
"pageSize": 50,
"pages": 1,
"status": "ok",
"total": 2,
}
)
return (MockedReponse(server_response), {"status": 200})
def customer_response_page2():
server_response = json.dumps({"customers": [
server_response = json.dumps(
{
"customer": "admin",
"href": "http://localhost:8080/api/customer/d89664a7-9c87-4ab9-8be8-830e7e5f0616",
"id": "d89664a7-9c87-4ab9-8be8-830e7e5f0616",
"match": "admin@example.com"
},
{
"customer": "Developer",
"href": "http://localhost:8080/api/customer/188ed093-84cc-4f46-bf80-4c9127180d9c",
"id": "188ed093-84cc-4f46-bf80-4c9127180d9c",
"match": "dev@example.com"
}],
"more": True,
"page": 2,
"pageSize": 50,
"pages": 2,
"status": "ok",
"total": 52})
"customers": [
{
"customer": "admin",
"href": "http://localhost:8080/api/customer/d89664a7-9c87-4ab9-8be8-830e7e5f0616",
"id": "d89664a7-9c87-4ab9-8be8-830e7e5f0616",
"match": "admin@example.com",
},
{
"customer": "Developer",
"href": "http://localhost:8080/api/customer/188ed093-84cc-4f46-bf80-4c9127180d9c",
"id": "188ed093-84cc-4f46-bf80-4c9127180d9c",
"match": "dev@example.com",
},
],
"more": True,
"page": 2,
"pageSize": 50,
"pages": 2,
"status": "ok",
"total": 52,
}
)
return (MockedReponse(server_response), {"status": 200})
class TestAlertaCustomerModule(ModuleTestCase):
def setUp(self):
super().setUp()
self.module = alerta_customer
@ -87,23 +101,23 @@ class TestAlertaCustomerModule(ModuleTestCase):
def test_without_content(self):
"""Failure if customer and match are missing"""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_username': "admin@example.com",
'api_password': "password"
}):
with set_module_args(
{"alerta_url": "http://localhost:8080", "api_username": "admin@example.com", "api_password": "password"}
):
with self.assertRaises(AnsibleFailJson):
self.module.main()
def test_successful_existing_customer_creation(self):
"""Test the customer creation (already exists)."""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_username': "admin@example.com",
'api_password': "password",
'customer': 'Developer',
'match': 'dev@example.com'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080",
"api_username": "admin@example.com",
"api_password": "password",
"customer": "Developer",
"match": "dev@example.com",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = customer_response_page1()
with self.assertRaises(AnsibleExitJson):
@ -112,97 +126,118 @@ class TestAlertaCustomerModule(ModuleTestCase):
def test_successful_customer_creation(self):
"""Test the customer creation."""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_username': "admin@example.com",
'api_password': "password",
'customer': 'Developer',
'match': 'dev2@example.com'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080",
"api_username": "admin@example.com",
"api_password": "password",
"customer": "Developer",
"match": "dev2@example.com",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = customer_response_page1()
with self.assertRaises(AnsibleExitJson):
self.module.main()
self.assertTrue(fetch_url_mock.call_count, 1)
call_data = json.loads(fetch_url_mock.call_args[1]['data'])
assert call_data['match'] == "dev2@example.com"
assert call_data['customer'] == "Developer"
call_data = json.loads(fetch_url_mock.call_args[1]["data"])
assert call_data["match"] == "dev2@example.com"
assert call_data["customer"] == "Developer"
def test_successful_customer_creation_key(self):
"""Test the customer creation using api_key."""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_key': "demo-key",
'customer': 'Developer',
'match': 'dev2@example.com'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080",
"api_key": "demo-key",
"customer": "Developer",
"match": "dev2@example.com",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = customer_response_page1()
with self.assertRaises(AnsibleExitJson):
self.module.main()
self.assertTrue(fetch_url_mock.call_count, 1)
call_data = json.loads(fetch_url_mock.call_args[1]['data'])
assert call_data['match'] == "dev2@example.com"
assert call_data['customer'] == "Developer"
call_data = json.loads(fetch_url_mock.call_args[1]["data"])
assert call_data["match"] == "dev2@example.com"
assert call_data["customer"] == "Developer"
def test_failed_not_found(self):
"""Test failure with wrong URL."""
with set_module_args({
'alerta_url': "http://localhost:8080/s",
'api_username': "admin@example.com",
'api_password': "password",
'customer': 'Developer',
'match': 'dev@example.com'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080/s",
"api_username": "admin@example.com",
"api_password": "password",
"customer": "Developer",
"match": "dev@example.com",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = (None, {"status": 404, 'msg': 'Not found for request GET on http://localhost:8080/a/api/customers'})
fetch_url_mock.return_value = (
None,
{"status": 404, "msg": "Not found for request GET on http://localhost:8080/a/api/customers"},
)
with self.assertRaises(AnsibleFailJson):
self.module.main()
def test_failed_forbidden(self):
"""Test failure with wrong user."""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_username': "dev@example.com",
'api_password': "password",
'customer': 'Developer',
'match': 'dev@example.com'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080",
"api_username": "dev@example.com",
"api_password": "password",
"customer": "Developer",
"match": "dev@example.com",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = (None, {"status": 403, 'msg': 'Permission Denied for GET on http://localhost:8080/api/customers'})
fetch_url_mock.return_value = (
None,
{"status": 403, "msg": "Permission Denied for GET on http://localhost:8080/api/customers"},
)
with self.assertRaises(AnsibleFailJson):
self.module.main()
def test_failed_unauthorized(self):
"""Test failure with wrong username or password."""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_username': "admin@example.com",
'api_password': "password_wrong",
'customer': 'Developer',
'match': 'dev@example.com'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080",
"api_username": "admin@example.com",
"api_password": "password_wrong",
"customer": "Developer",
"match": "dev@example.com",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = (None, {"status": 401, 'msg': 'Unauthorized to request GET on http://localhost:8080/api/customers'})
fetch_url_mock.return_value = (
None,
{"status": 401, "msg": "Unauthorized to request GET on http://localhost:8080/api/customers"},
)
with self.assertRaises(AnsibleFailJson):
self.module.main()
def test_successful_customer_deletion(self):
"""Test the customer deletion."""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_username': "admin@example.com",
'api_password': "password",
'customer': 'Developer',
'match': 'dev@example.com',
'state': 'absent'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080",
"api_username": "admin@example.com",
"api_password": "password",
"customer": "Developer",
"match": "dev@example.com",
"state": "absent",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = customer_response_page1()
with self.assertRaises(AnsibleExitJson):
@ -211,14 +246,16 @@ class TestAlertaCustomerModule(ModuleTestCase):
def test_successful_customer_deletion_page2(self):
"""Test the customer deletion on the second page."""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_username': "admin@example.com",
'api_password': "password",
'customer': 'Developer',
'match': 'dev@example.com',
'state': 'absent'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080",
"api_username": "admin@example.com",
"api_password": "password",
"customer": "Developer",
"match": "dev@example.com",
"state": "absent",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = customer_response_page2()
with self.assertRaises(AnsibleExitJson):
@ -227,14 +264,16 @@ class TestAlertaCustomerModule(ModuleTestCase):
def test_successful_nonexisting_customer_deletion(self):
"""Test the customer deletion (non existing)."""
with set_module_args({
'alerta_url': "http://localhost:8080",
'api_username': "admin@example.com",
'api_password': "password",
'customer': 'Billing',
'match': 'dev@example.com',
'state': 'absent'
}):
with set_module_args(
{
"alerta_url": "http://localhost:8080",
"api_username": "admin@example.com",
"api_password": "password",
"customer": "Billing",
"match": "dev@example.com",
"state": "absent",
}
):
with patch.object(alerta_customer, "fetch_url") as fetch_url_mock:
fetch_url_mock.return_value = customer_response_page1()
with self.assertRaises(AnsibleExitJson):