mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-04-13 15:35:08 +00:00
[PR #9838/a1781d09 backport][stable-10] Unit tests: make set_module_args() a context manager, and remove copies of it in some tests (#9839)
Unit tests: make set_module_args() a context manager, and remove copies of it in some tests (#9838)
Make set_module_args() a context manager, and remove copies of set_module_args().
Prepares for Data Tagging.
(cherry picked from commit a1781d09dd)
Co-authored-by: Felix Fontein <felix@fontein.de>
This commit is contained in:
patchback[bot]
GitHub
parent
cd0ca389ed
commit
7d45b678e4
84 changed files with 4043 additions and 4302 deletions
|
|
@ -30,45 +30,46 @@ class TestCirconusAnnotation(ModuleTestCase):
|
|||
def test_without_required_parameters(self):
|
||||
"""Failure must occurs when all parameters are missing"""
|
||||
with self.assertRaises(AnsibleFailJson):
|
||||
set_module_args({})
|
||||
self.module.main()
|
||||
with set_module_args({}):
|
||||
self.module.main()
|
||||
|
||||
def test_add_annotation(self):
|
||||
"""Check that result is changed"""
|
||||
set_module_args({
|
||||
with set_module_args({
|
||||
'category': 'test category',
|
||||
'description': 'test description',
|
||||
'title': 'test title',
|
||||
'api_key': str(uuid.uuid4()),
|
||||
})
|
||||
}):
|
||||
|
||||
cid = '/annotation/100000'
|
||||
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',
|
||||
}
|
||||
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'}
|
||||
return self.build_response(request, resp)
|
||||
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',
|
||||
}
|
||||
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'}
|
||||
return self.build_response(request, resp)
|
||||
|
||||
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)
|
||||
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.assertEqual(send.call_count, 1)
|
||||
|
||||
def test_add_annotation_unicode(self):
|
||||
|
|
@ -76,78 +77,80 @@ class TestCirconusAnnotation(ModuleTestCase):
|
|||
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."""
|
||||
set_module_args({
|
||||
with set_module_args({
|
||||
'category': 'new catégorÿ',
|
||||
'description': 'test description',
|
||||
'title': 'test title',
|
||||
'api_key': str(uuid.uuid4()),
|
||||
})
|
||||
}):
|
||||
|
||||
cid = '/annotation/100000'
|
||||
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',
|
||||
# use res['annotation']['category'].encode('latin1').decode('utf8')
|
||||
'category': u'new cat\xc3\xa9gor\xc3\xbf',
|
||||
'description': 'test description',
|
||||
'rel_metrics': [],
|
||||
'start': 1502236927,
|
||||
'stop': 1502236927,
|
||||
'title': 'test title',
|
||||
}
|
||||
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',
|
||||
# use res['annotation']['category'].encode('latin1').decode('utf8')
|
||||
'category': u'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')
|
||||
resp = HTTPResponse(body=io.BytesIO(raw), preload_content=False)
|
||||
resp.status = 200
|
||||
resp.reason = 'OK'
|
||||
resp.headers = {'X-Circonus-API-Version': '2.00'}
|
||||
return self.build_response(request, resp)
|
||||
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'}
|
||||
return self.build_response(request, resp)
|
||||
|
||||
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)
|
||||
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.assertEqual(send.call_count, 1)
|
||||
|
||||
def test_auth_failure(self):
|
||||
"""Check that an error is raised when authentication failed"""
|
||||
set_module_args({
|
||||
with set_module_args({
|
||||
'category': 'test category',
|
||||
'description': 'test description',
|
||||
'title': 'test title',
|
||||
'api_key': str(uuid.uuid4()),
|
||||
})
|
||||
}):
|
||||
|
||||
cid = '/annotation/100000'
|
||||
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',
|
||||
}
|
||||
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'}
|
||||
return self.build_response(request, resp)
|
||||
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',
|
||||
}
|
||||
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'}
|
||||
return self.build_response(request, resp)
|
||||
|
||||
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']))
|
||||
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.assertEqual(send.call_count, 1)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue