def test_stop_host(self):
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.client.stopVirtualMachine.return_value = {"jobid": "qwe321"}
manager.stop_host('host-id', True)
manager.client.stopVirtualMachine.assert_called_with({'id': 'host-id', 'forced': 'true'})
manager.client.wait_for_job.assert_called_with('qwe321', 100)
def test_create(self):
self.config.update({
"CLOUDSTACK_TEMPLATE_ID": "abc123",
"CLOUDSTACK_SERVICE_OFFERING_ID": "qwe123",
"CLOUDSTACK_ZONE_ID": "zone1",
"CLOUDSTACK_PROJECT_ID": "project-123",
"CLOUDSTACK_NETWORK_IDS": "net-123",
"CLOUDSTACK_GROUP": "feaas",
})
client_mock = mock.Mock()
client_mock.deployVirtualMachine.return_value = {"id": "abc123",
"jobid": "qwe321"}
vm = {"id": "abc123", "nic": [{"ipaddress": "10.0.0.1"}]}
client_mock.listVirtualMachines.return_value = {"virtualmachine": [vm]}
manager = cloudstack.CloudStackManager(self.config)
manager.client = client_mock
host = manager.create_host('xxx')
self.assertEqual("abc123", host.id)
self.assertEqual("10.0.0.1", host.dns_name)
create_data = {
"group": "feaas",
"displayname": "feaas_xxx",
"templateid": "abc123",
"zoneid": "zone1",
"serviceofferingid": "qwe123",
"networkids": "net-123",
"projectid": "project-123",
}
client_mock.deployVirtualMachine.assert_called_with(create_data)
client_mock.wait_for_job.assert_called_with('qwe321', 100)
def test_create_timeout(self):
self.config.update({
"CLOUDSTACK_TEMPLATE_ID": "abc123",
"CLOUDSTACK_SERVICE_OFFERING_ID": "qwe123",
"CLOUDSTACK_ZONE_ID": "zone1",
"CLOUDSTACK_GROUP": "feaas",
"CLOUDSTACK_MAX_TRIES": 1,
})
client_mock = mock.Mock()
client_mock.deployVirtualMachine.return_value = {"id": "abc123",
"jobid": "qwe321"}
vm = {"id": "abc123", "nic": [{"ipaddress": "10.0.0.1"}]}
client_mock.listVirtualMachines.return_value = {"virtualmachine": [vm]}
client_mock.wait_for_job.side_effect = cloudstack_client.MaxTryWaitingForJobError(1, 'qwe321')
manager = cloudstack.CloudStackManager(self.config)
manager.client = client_mock
with self.assertRaises(cloudstack_client.MaxTryWaitingForJobError) as cm:
manager.create_host()
exc = cm.exception
self.assertEqual(1, exc.max_tries)
self.assertEqual("qwe321", exc.job_id)
self.assertEqual("exceeded 1 tries waiting for job qwe321", str(exc))
create_data = {
"group": "feaas",
"displayname": "feaas",
"templateid": "abc123",
"zoneid": "zone1",
"serviceofferingid": "qwe123",
}
client_mock.wait_for_job.assert_called_with('qwe321', 1)
client_mock.deployVirtualMachine.assert_called_with(create_data)
def test_create_invalid_response(self):
self.config.update({
"CLOUDSTACK_TEMPLATE_ID": "abc123",
"CLOUDSTACK_SERVICE_OFFERING_ID": "qwe123",
"CLOUDSTACK_ZONE_ID": "zone1",
"CLOUDSTACK_GROUP": "feaas",
})
client_mock = mock.Mock()
client_mock.deployVirtualMachine.return_value = {"error": "xxx"}
vm = {"id": "abc123", "nic": [{"ipaddress": "10.0.0.1"}]}
client_mock.listVirtualMachines.return_value = {"virtualmachine": [vm]}
manager = cloudstack.CloudStackManager(self.config)
manager.client = client_mock
with self.assertRaises(cloudstack.CloudStackException) as ctx:
manager.create_host()
exc = ctx.exception
self.assertRegexpMatches(
str(exc), r"unexpected response from deployVirtualMachine\({.+}\)"
", expected jobid key, got: {'error': 'xxx'}")
create_data = {
"group": "feaas",
"displayname": "feaas",
"templateid": "abc123",
"zoneid": "zone1",
"serviceofferingid": "qwe123",
}
client_mock.deployVirtualMachine.assert_called_with(create_data)
def test_restore_host(self):
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.restore_host('host-id')
manager.client.make_request.assert_called_with('restoreVirtualMachine',
{'virtualmachineid': 'host-id'},
response_key='restorevmresponse')
def test_init_no_secret_key(self):
with self.assertRaises(config.MissConfigurationError) as cm:
cloudstack.CloudStackManager({
"CLOUDSTACK_API_URL": "something",
"CLOUDSTACK_API_KEY": "not_secret",
})
exc = cm.exception
self.assertEqual(("env var CLOUDSTACK_SECRET_KEY is required",),
exc.args)
def test_scale_host_ignore_same_offering(self):
self.config.update({
"CLOUDSTACK_SERVICE_OFFERING_ID": "large"
})
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.client.scaleVirtualMachine.return_value = {"jobid": "qwe321"}
manager.client.listVirtualMachines.return_value = {"virtualmachine": [{"serviceofferingid": "large"}]}
manager.scale_host('host-id')
manager.client.scaleVirtualMachine.assert_not_called()
manager.client.wait_for_job.assert_not_called()
def test_scale_host_not_found(self):
self.config.update({
"CLOUDSTACK_SERVICE_OFFERING_ID": "large"
})
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.client.listVirtualMachines.side_effect = Exception("Host not found")
with self.assertRaises(Exception):
manager.scale_host('host-id')
manager.client.scaleVirtualMachine.assert_not_called()
manager.client.wait_for_job.assert_not_called()
def test_tag_vm_ignore_duplicated_tags(self):
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.client.listTags.return_value = {"tag": [{"key": "foo", "value": "bar"},
{"key": "bleh", "value": "blah"},
{"key": "duh", "value": "dah"}]}
manager.tag_vm(['foo:bar', 'bleh:blah', 'duh:dah', 's:s'], 'host-id', 'project-id')
create_tags = {'tags[0].key': 's', 'tags[0].value': 's',
'resourceids': 'host-id', 'resourcetype': 'UserVm', 'projectid': 'project-id'}
manager.client.deleteTags.assert_not_called()
manager.client.createTags.assert_called_with(create_tags)
def test_scale_host(self):
self.config.update({
"CLOUDSTACK_SERVICE_OFFERING_ID": "large",
"CLOUDSTACK_PROJECT_ID": "project-id"
})
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.client.scaleVirtualMachine.return_value = {"jobid": "qwe321"}
manager.client.listVirtualMachines.return_value = {"virtualmachine": [{"serviceofferingid": "small"}]}
manager.scale_host('host-id')
manager.client.listVirtualMachines.assert_called_with({"id": "host-id", "projectid": "project-id"})
manager.client.scaleVirtualMachine.assert_called_with({"id": "host-id", "serviceofferingid": "large"})
manager.client.wait_for_job.assert_called_with('qwe321', 100)
def test_restore_host(self):
client_mock = mock.Mock()
client_mock.make_request.return_value = {"id": "abc123",
"jobid": "qwe321"}
vm = {"id": "host-id", "nic": [{"ipaddress": "10.0.0.1"}]}
client_mock.listVirtualMachines.return_value = {"virtualmachine": [vm]}
manager = cloudstack.CloudStackManager(self.config)
manager.client = client_mock
manager.tag_vm = mock.Mock()
manager.restore_host('host-id')
manager.client.make_request.assert_called_with('restoreVirtualMachine',
{'virtualmachineid': 'host-id'},
response_key='restorevmresponse')
manager.client.wait_for_job.assert_called_with('qwe321', 100)
manager.tag_vm.assert_not_called()
def test_create_with_tags(self):
self.config.update({
"CLOUDSTACK_TEMPLATE_ID": "abc123",
"CLOUDSTACK_SERVICE_OFFERING_ID": "qwe123",
"CLOUDSTACK_ZONE_ID": "zone1",
"CLOUDSTACK_PROJECT_ID": "project-123",
"CLOUDSTACK_NETWORK_IDS": "net-123",
"CLOUDSTACK_GROUP": "feaas",
"HOST_TAGS": "Name:something,monitor:1,wait=wat,syslog:logging:5140",
})
client_mock = mock.Mock()
client_mock.deployVirtualMachine.return_value = {"id": "abc123",
"jobid": "qwe321"}
vm = {"id": "abc123", "nic": [{"ipaddress": "10.0.0.1"}]}
client_mock.listVirtualMachines.return_value = {"virtualmachine": [vm]}
client_mock.listTags.return_value = {}
manager = cloudstack.CloudStackManager(self.config)
manager.client = client_mock
host = manager.create_host('xxx')
self.assertEqual("abc123", host.id)
self.assertEqual("10.0.0.1", host.dns_name)
create_data = {
"group": "feaas",
"displayname": "feaas_xxx",
"templateid": "abc123",
"zoneid": "zone1",
"serviceofferingid": "qwe123",
"networkids": "net-123",
"projectid": "project-123",
}
client_mock.deployVirtualMachine.assert_called_with(create_data)
client_mock.wait_for_job.assert_called_with('qwe321', 100)
tag_data = {
"resourcetype": "UserVm",
"resourceids": "abc123",
"projectid": "project-123",
"tags[0].key": "Name",
"tags[0].value": "something",
"tags[1].key": "monitor",
"tags[1].value": "1",
"tags[2].key": "syslog",
"tags[2].value": "logging:5140",
}
client_mock.createTags.assert_called_with(tag_data)
def test_tag_vm_replacing_tags(self):
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.client.deleteTags.return_value = {"jobid": "1"}
manager.client.listTags.return_value = {"tag": [{"key": "foo", "value": "bar"},
{"key": "bleh", "value": "blah"},
{"key": "duh", "value": "dah"}]}
manager.tag_vm(['bleh:xxx', 'test1:test', 'duh:aaaa'], 'host-id', 'project-id')
delete_calls = [mock.call({'resourcetype': 'UserVm', 'resourceids': 'host-id',
'tags[0].key': 'bleh', 'tags[0].value': 'blah',
'tags[1].key': 'duh', 'tags[1].value': 'dah',
'projectid': 'project-id'})]
manager.client.deleteTags.assert_has_calls(delete_calls)
create_tags = {'tags[0].key': 'bleh', 'tags[0].value': 'xxx', 'tags[1].key': 'test1',
'tags[1].value': 'test', 'tags[2].key': 'duh', 'tags[2].value': 'aaaa',
'resourceids': 'host-id', 'resourcetype': 'UserVm', 'projectid': 'project-id'}
manager.client.createTags.assert_called_with(create_tags)
def test_restore_host_fail_and_rollback_tags(self):
client_mock = mock.Mock()
client_mock.make_request.return_value = {"id": "abc123",
"jobid": "qwe321"}
self.config.update({
"HOST_TAGS": "blah:bleh,monitor:1,wait:wat",
"CLOUDSTACK_PROJECT_ID": "project-base",
"CLOUDSTACK_TEMPLATE_ID": "1234"
})
client_mock.listVirtualMachines.return_value = Exception("fail")
manager = cloudstack.CloudStackManager(self.config)
manager.client = client_mock
manager.client.listTags.return_value = {"tag": [{"key": "foo", "value": "bar"},
{"key": "bleh", "value": "blah"},
{"key": "duh", "value": "dah"}]}
manager.tag_vm = mock.Mock()
with self.assertRaises(cloudstack.CloudStackException):
manager.restore_host('host-id', True, True)
manager.client.make_request.assert_called_with('restoreVirtualMachine',
{'virtualmachineid': 'host-id', 'templateid': '1234'},
response_key='restorevmresponse')
manager.tag_vm.assert_called_with(['foo:bar', 'bleh:blah', 'duh:dah'], 'host-id', 'project-base')
def test_restore_host_with_tags_and_reset_template(self):
client_mock = mock.Mock()
client_mock.make_request.return_value = {"id": "abc123",
"jobid": "qwe321"}
vm = {"id": "host-id", "nic": [{"ipaddress": "10.0.0.1"}]}
client_mock.listVirtualMachines.return_value = {"virtualmachine": [vm]}
self.config.update({
"HOST_TAGS": "blah:bleh,monitor:1,wait:wat",
"CLOUDSTACK_PROJECT_ID": "project-base",
"CLOUDSTACK_TEMPLATE_ID": "1234"
})
client_mock.listVirtualMachines.return_value = {"virtualmachine": [vm]}
manager = cloudstack.CloudStackManager(self.config)
manager.client = client_mock
manager.client.listTags.return_value = {"tag": [{"key": "foo", "value": "bar"},
{"key": "bleh", "value": "blah"},
{"key": "duh", "value": "dah"}]}
manager.tag_vm = mock.Mock()
manager.restore_host('host-id', True, True)
manager.client.make_request.assert_called_with('restoreVirtualMachine',
{'virtualmachineid': 'host-id', 'templateid': '1234'},
response_key='restorevmresponse')
manager.tag_vm.assert_called_with(['blah:bleh', 'monitor:1', 'wait:wat'], 'host-id', 'project-base')
def test_scale_host_offering_not_defined_error(self):
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
with self.assertRaises(Exception):
manager.scale_host('host-id')
manager.client.scaleVirtualMachine.assert_not_called()
def test_init(self):
client = cloudstack.CloudStackManager(self.config)
self.assertEqual(client.client.api_url, self.config["CLOUDSTACK_API_URL"])
self.assertEqual(client.client.api_key, self.config["CLOUDSTACK_API_KEY"])
self.assertEqual(client.client.secret, self.config["CLOUDSTACK_SECRET_KEY"])
def test_init_no_api_url(self):
with self.assertRaises(config.MissConfigurationError) as cm:
cloudstack.CloudStackManager()
exc = cm.exception
self.assertEqual(("env var CLOUDSTACK_API_URL is required",),
exc.args)
def test_start_host(self):
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.start_host('host-id')
manager.client.startVirtualMachine.assert_called_with({'id': 'host-id'})
def test_stop_host(self):
manager = cloudstack.CloudStackManager(self.config)
manager.client = mock.Mock()
manager.stop_host('host-id', True)
manager.client.stopVirtualMachine.assert_called_with({'id': 'host-id', 'forced': True})
