mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-05-02 16:32:56 +00:00
Reformat everything.
This commit is contained in:
Felix Fontein
parent
3f2213791a
commit
340ff8586d
1008 changed files with 61301 additions and 58309 deletions
|
|
@ -13,11 +13,15 @@ from urllib3.response import HTTPResponse
|
|||
|
||||
from ansible.module_utils.common.text.converters import to_bytes
|
||||
from ansible_collections.community.general.plugins.modules import circonus_annotation
|
||||
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 TestCirconusAnnotation(ModuleTestCase):
|
||||
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
self.module = circonus_annotation
|
||||
|
|
@ -33,122 +37,125 @@ class TestCirconusAnnotation(ModuleTestCase):
|
|||
|
||||
def test_add_annotation(self):
|
||||
"""Check that result is changed"""
|
||||
with set_module_args({
|
||||
'category': 'test category',
|
||||
'description': 'test description',
|
||||
'title': 'test title',
|
||||
'api_key': str(uuid.uuid4()),
|
||||
}):
|
||||
|
||||
cid = '/annotation/100000'
|
||||
with set_module_args(
|
||||
{
|
||||
"category": "test category",
|
||||
"description": "test description",
|
||||
"title": "test title",
|
||||
"api_key": str(uuid.uuid4()),
|
||||
}
|
||||
):
|
||||
cid = "/annotation/100000"
|
||||
|
||||
def send(self, request, stream=False, timeout=None, verify=True, cert=None, proxies=None):
|
||||
data = {
|
||||
'_cid': cid,
|
||||
'_created': 1502146995,
|
||||
'_last_modified': 1502146995,
|
||||
'_last_modified_by': '/user/1000',
|
||||
'category': 'test category',
|
||||
'description': 'test description',
|
||||
'rel_metrics': [],
|
||||
'start': 1502145480,
|
||||
'stop': None,
|
||||
'title': 'test title',
|
||||
"_cid": cid,
|
||||
"_created": 1502146995,
|
||||
"_last_modified": 1502146995,
|
||||
"_last_modified_by": "/user/1000",
|
||||
"category": "test category",
|
||||
"description": "test description",
|
||||
"rel_metrics": [],
|
||||
"start": 1502145480,
|
||||
"stop": None,
|
||||
"title": "test title",
|
||||
}
|
||||
raw = to_bytes(json.dumps(data))
|
||||
resp = HTTPResponse(body=io.BytesIO(raw), preload_content=False)
|
||||
resp.status = 200
|
||||
resp.reason = 'OK'
|
||||
resp.headers = {'X-Circonus-API-Version': '2.00'}
|
||||
resp.reason = "OK"
|
||||
resp.headers = {"X-Circonus-API-Version": "2.00"}
|
||||
return self.build_response(request, resp)
|
||||
|
||||
with patch('requests.adapters.HTTPAdapter.send', autospec=True, side_effect=send) as send:
|
||||
with patch("requests.adapters.HTTPAdapter.send", autospec=True, side_effect=send) as send:
|
||||
with self.assertRaises(AnsibleExitJson) as result:
|
||||
self.module.main()
|
||||
|
||||
self.assertTrue(result.exception.args[0]['changed'])
|
||||
self.assertEqual(result.exception.args[0]['annotation']['_cid'], cid)
|
||||
self.assertTrue(result.exception.args[0]["changed"])
|
||||
self.assertEqual(result.exception.args[0]["annotation"]["_cid"], cid)
|
||||
self.assertEqual(send.call_count, 1)
|
||||
|
||||
def test_add_annotation_unicode(self):
|
||||
"""Check that result is changed.
|
||||
Note: it seems there is a bug which prevent to create an annotation
|
||||
with a non-ASCII category if this category already exists, in such
|
||||
case an Internal Server Error (500) occurs."""
|
||||
with set_module_args({
|
||||
'category': 'new catégorÿ',
|
||||
'description': 'test description',
|
||||
'title': 'test title',
|
||||
'api_key': str(uuid.uuid4()),
|
||||
}):
|
||||
|
||||
cid = '/annotation/100000'
|
||||
Note: it seems there is a bug which prevent to create an annotation
|
||||
with a non-ASCII category if this category already exists, in such
|
||||
case an Internal Server Error (500) occurs."""
|
||||
with set_module_args(
|
||||
{
|
||||
"category": "new catégorÿ",
|
||||
"description": "test description",
|
||||
"title": "test title",
|
||||
"api_key": str(uuid.uuid4()),
|
||||
}
|
||||
):
|
||||
cid = "/annotation/100000"
|
||||
|
||||
def send(self, request, stream=False, timeout=None, verify=True, cert=None, proxies=None):
|
||||
data = {
|
||||
'_cid': '/annotation/100000',
|
||||
'_created': 1502236928,
|
||||
'_last_modified': 1502236928,
|
||||
'_last_modified_by': '/user/1000',
|
||||
"_cid": "/annotation/100000",
|
||||
"_created": 1502236928,
|
||||
"_last_modified": 1502236928,
|
||||
"_last_modified_by": "/user/1000",
|
||||
# use res['annotation']['category'].encode('latin1').decode('utf8')
|
||||
'category': 'new cat\xc3\xa9gor\xc3\xbf',
|
||||
'description': 'test description',
|
||||
'rel_metrics': [],
|
||||
'start': 1502236927,
|
||||
'stop': 1502236927,
|
||||
'title': 'test title',
|
||||
"category": "new cat\xc3\xa9gor\xc3\xbf",
|
||||
"description": "test description",
|
||||
"rel_metrics": [],
|
||||
"start": 1502236927,
|
||||
"stop": 1502236927,
|
||||
"title": "test title",
|
||||
}
|
||||
|
||||
raw = to_bytes(json.dumps(data), encoding='latin1')
|
||||
raw = to_bytes(json.dumps(data), encoding="latin1")
|
||||
resp = HTTPResponse(body=io.BytesIO(raw), preload_content=False)
|
||||
resp.status = 200
|
||||
resp.reason = 'OK'
|
||||
resp.headers = {'X-Circonus-API-Version': '2.00'}
|
||||
resp.reason = "OK"
|
||||
resp.headers = {"X-Circonus-API-Version": "2.00"}
|
||||
return self.build_response(request, resp)
|
||||
|
||||
with patch('requests.adapters.HTTPAdapter.send', autospec=True, side_effect=send) as send:
|
||||
with patch("requests.adapters.HTTPAdapter.send", autospec=True, side_effect=send) as send:
|
||||
with self.assertRaises(AnsibleExitJson) as result:
|
||||
self.module.main()
|
||||
|
||||
self.assertTrue(result.exception.args[0]['changed'])
|
||||
self.assertEqual(result.exception.args[0]['annotation']['_cid'], cid)
|
||||
self.assertTrue(result.exception.args[0]["changed"])
|
||||
self.assertEqual(result.exception.args[0]["annotation"]["_cid"], cid)
|
||||
self.assertEqual(send.call_count, 1)
|
||||
|
||||
def test_auth_failure(self):
|
||||
"""Check that an error is raised when authentication failed"""
|
||||
with set_module_args({
|
||||
'category': 'test category',
|
||||
'description': 'test description',
|
||||
'title': 'test title',
|
||||
'api_key': str(uuid.uuid4()),
|
||||
}):
|
||||
|
||||
cid = '/annotation/100000'
|
||||
with set_module_args(
|
||||
{
|
||||
"category": "test category",
|
||||
"description": "test description",
|
||||
"title": "test title",
|
||||
"api_key": str(uuid.uuid4()),
|
||||
}
|
||||
):
|
||||
cid = "/annotation/100000"
|
||||
|
||||
def send(self, request, stream=False, timeout=None, verify=True, cert=None, proxies=None):
|
||||
data = {
|
||||
'_cid': cid,
|
||||
'_created': 1502146995,
|
||||
'_last_modified': 1502146995,
|
||||
'_last_modified_by': '/user/1000',
|
||||
'category': 'test category',
|
||||
'description': 'test description',
|
||||
'rel_metrics': [],
|
||||
'start': 1502145480,
|
||||
'stop': None,
|
||||
'title': 'test title',
|
||||
"_cid": cid,
|
||||
"_created": 1502146995,
|
||||
"_last_modified": 1502146995,
|
||||
"_last_modified_by": "/user/1000",
|
||||
"category": "test category",
|
||||
"description": "test description",
|
||||
"rel_metrics": [],
|
||||
"start": 1502145480,
|
||||
"stop": None,
|
||||
"title": "test title",
|
||||
}
|
||||
raw = to_bytes(json.dumps(data))
|
||||
resp = HTTPResponse(body=io.BytesIO(raw), preload_content=False)
|
||||
resp.status = 403
|
||||
resp.reason = 'Forbidden'
|
||||
resp.headers = {'X-Circonus-API-Version': '2.00'}
|
||||
resp.reason = "Forbidden"
|
||||
resp.headers = {"X-Circonus-API-Version": "2.00"}
|
||||
return self.build_response(request, resp)
|
||||
|
||||
with patch('requests.adapters.HTTPAdapter.send', autospec=True, side_effect=send) as send:
|
||||
with patch("requests.adapters.HTTPAdapter.send", autospec=True, side_effect=send) as send:
|
||||
with self.assertRaises(AnsibleFailJson) as result:
|
||||
self.module.main()
|
||||
|
||||
self.assertTrue(result.exception.args[0]['failed'])
|
||||
self.assertTrue(re.match(r'\b403\b', result.exception.args[0]['reason']))
|
||||
self.assertTrue(result.exception.args[0]["failed"])
|
||||
self.assertTrue(re.match(r"\b403\b", result.exception.args[0]["reason"]))
|
||||
self.assertEqual(send.call_count, 1)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue