def test02_get_vm_neg(self):
"""Negative test case for attempting to retrieve non-existent VM instance."""
# Construct mock return object.
json_body = '{"ok": false}'
resp = _HttpResponse(404, json_body)
# Patch
with patch.object(service, '_make_request', return_value=resp) as mock_func:
with self.assertRaises(RuntimeError) as cm:
service.get_vm(self.vmpooler_url, self.template_name, self.auth_token)
excep = cm.exception
self.assertEqual(excep.msg,
'Could not retrieve template! Invalid template name provided!')
def test02_get_vm_neg(self):
"""Negative test case for attempting to retrieve non-existent VM instance."""
# Construct mock return object.
json_body = '{"ok": false}'
resp = _HttpResponse(404, json_body)
# Patch
with patch.object(service, '_make_request',
return_value=resp) as mock_func:
with self.assertRaises(RuntimeError) as cm:
service.get_vm(self.vmpooler_hostname, self.template_name,
self.auth_token)
excep = cm.exception
self.assertEqual(
excep.msg,
'Could not retrieve template! Invalid template name provided!'
)
def test01_get_vm(self):
"""Happy path test to verify retrieving VM instances from the pooler."""
# Construct mock return object.
json_body = """
{{
"ok": true,
"{0}": {{
"hostname": "{1}",
"ok": true
}},
"domain": "delivery.puppetlabs.net"
}}""".format(self.template_name, self.hostname)
resp = _HttpResponse(200, json_body)
# Patch
with patch.object(service, '_make_request', return_value=resp) as mock_func:
self.assertEqual(service.get_vm(self.vmpooler_url,
self.template_name,
self.auth_token),
self.hostname)
def test01_get_vm(self):
"""Happy path test to verify retrieving VM instances from the pooler."""
# Construct mock return object.
json_body = """
{{
"ok": true,
"{0}": {{
"hostname": "{1}",
"ok": true
}},
"domain": "delivery.puppetlabs.net"
}}""".format(self.template_name, self.hostname)
resp = _HttpResponse(200, json_body)
# Patch
with patch.object(service, '_make_request',
return_value=resp) as mock_func:
self.assertEqual(
service.get_vm(self.vmpooler_hostname, self.template_name,
self.auth_token), self.hostname)
