def test_get_cloud_name(self):
for module_name in self.get_cloudconnector_modulenames():
setattr(self.config_holder, CONFIGPARAM_CONNECTOR_MODULE_NAME, module_name)
setattr(self.config_holder, KEY_RUN_CATEGORY, RUN_CATEGORY_DEPLOYMENT)
cw = CloudWrapper(self.config_holder)
cw.initCloudConnector()
assert cw._get_cloud_service_name() == 'Test'
def test_get_cloud_name(self):
for module_name in self.get_cloudconnector_modulenames():
setattr(self.config_holder, CONFIGPARAM_CONNECTOR_MODULE_NAME,
module_name)
setattr(self.config_holder, KEY_RUN_CATEGORY,
RUN_CATEGORY_DEPLOYMENT)
cw = CloudWrapper(self.config_holder)
cw.initCloudConnector()
assert cw._get_cloud_service_name() == 'Test'
def createExecutor(configHolder):
cloudWrapper = CloudWrapper(configHolder)
category = cloudWrapper.getRunCategory()
configHolder.set(KEY_RUN_CATEGORY, category)
cloudWrapper.initCloudConnector(configHolder)
if category == RUN_CATEGORY_IMAGE:
return OrchestratorImageBuildExecutor(cloudWrapper, configHolder)
elif category == RUN_CATEGORY_DEPLOYMENT:
return OrchestratorDeploymentExecutor(cloudWrapper, configHolder)
else:
raise Exceptions.ClientError("Unknown category: %s" % category)
def test_putImageId(self):
self.configHolder.set(CONFIGPARAM_CONNECTOR_MODULE_NAME,
self.get_cloudconnector_modulename_by_cloudname('local'))
cw = CloudWrapper(self.configHolder)
cw.initCloudConnector()
cw.clientSlipStream.httpClient._call = Mock(return_value=('', ''))
cw._updateSlipStreamImage('module/Name', 'ABC')
cw.clientSlipStream.httpClient._call.assert_called_with(
'%s/module/Name/Test' % self.serviceurl,
'PUT', 'ABC', 'application/xml',
'application/xml')
def createExecutor(configHolder):
cloudWrapper = CloudWrapper(configHolder)
category = cloudWrapper.get_run_category()
configHolder.set(KEY_RUN_CATEGORY, category)
cloudWrapper.initCloudConnector(configHolder)
if category == RUN_CATEGORY_IMAGE:
return OrchestratorImageBuildExecutor(cloudWrapper, configHolder)
elif category == RUN_CATEGORY_DEPLOYMENT:
return OrchestratorDeploymentExecutor(cloudWrapper, configHolder)
else:
raise Exceptions.ClientError("Unknown category: %s" % category)
def test_put_image_id(self):
# pylint: disable=protected-access
self.config_holder.set(CONFIGPARAM_CONNECTOR_MODULE_NAME,
self.get_cloudconnector_modulename_by_cloudname('local'))
cw = CloudWrapper(self.config_holder)
cw.initCloudConnector()
cw._ss_client.httpClient._call = Mock(return_value=Mock())
cw._update_slipstream_image(NodeInstance({'image.resourceUri': 'module/Name'}), 'ABC')
cw._ss_client.httpClient._call.assert_called_with(
'%s/module/Name/Test' % self.serviceurl,
'PUT', 'ABC', 'application/xml',
'application/xml', retry=True)
def test_put_image_id(self):
# pylint: disable=protected-access
self.config_holder.set(
CONFIGPARAM_CONNECTOR_MODULE_NAME,
self.get_cloudconnector_modulename_by_cloudname('local'))
cw = CloudWrapper(self.config_holder)
cw.initCloudConnector()
cw._ss_client.httpClient._call = Mock(return_value=Mock())
cw._update_slipstream_image(
NodeInstance({'image.resourceUri': 'module/Name'}), 'ABC')
cw._ss_client.httpClient._call.assert_called_with(
'%s/module/Name/Test' % self.serviceurl,
'PUT',
'ABC',
'application/xml',
'application/xml',
retry=True)
def test_vertically_scalle_instances_nowait(self):
_scale_state = None
def _get_runtime_parameter(key):
if key.endswith(NodeDecorator.NODE_PROPERTY_SEPARATOR + NodeDecorator.SCALE_STATE_KEY):
return _scale_state
else:
return 'unknown'
node_instances = {
'n.1': NodeInstance({NodeDecorator.NODE_INSTANCE_NAME_KEY: 'n.1',
NodeDecorator.NODE_NAME_KEY: 'n',
NodeDecorator.SCALE_STATE_KEY: CloudWrapper.SCALE_STATE_RESIZING}),
'n.2': NodeInstance({NodeDecorator.NODE_INSTANCE_NAME_KEY: 'n.2',
NodeDecorator.NODE_NAME_KEY: 'n',
NodeDecorator.SCALE_STATE_KEY: CloudWrapper.SCALE_STATE_DISK_ATTACHING}),
'n.3': NodeInstance({NodeDecorator.NODE_INSTANCE_NAME_KEY: 'n.3',
NodeDecorator.NODE_NAME_KEY: 'n',
NodeDecorator.SCALE_STATE_KEY: CloudWrapper.SCALE_STATE_DISK_DETACHING}),
'm.1': NodeInstance({NodeDecorator.NODE_INSTANCE_NAME_KEY: 'm.1',
NodeDecorator.NODE_NAME_KEY: 'm',
NodeDecorator.SCALE_STATE_KEY: CloudWrapper.SCALE_STATE_OPERATIONAL})
}
self.config_holder.set('verboseLevel', 3)
setattr(self.config_holder, 'cloud', 'local')
setattr(self.config_holder, CONFIGPARAM_CONNECTOR_MODULE_NAME,
'slipstream.cloudconnectors.dummy.DummyClientCloud')
cw = CloudWrapper(self.config_holder)
cw._get_nodes_instances = Mock(return_value=node_instances)
cw.initCloudConnector(self.config_holder)
cw._set_runtime_parameter = Mock()
cw._get_user_timeout = Mock(return_value=2)
# No waiting.
cw._wait_pre_scale_done = Mock()
cw._wait_scale_state = Mock()
_scale_state = 'resizing'
cw._get_runtime_parameter = Mock(side_effect=_get_runtime_parameter)
cw._cloud_client.resize = Mock(wraps=cw._cloud_client.resize)
cw.vertically_scale_instances()
assert True == cw._cloud_client.resize.called
node_instance = cw._cloud_client.resize.call_args[0][0][0]
assert 'n.1' in node_instance.get_name()
assert cw._set_runtime_parameter.called_with('n.1:' + NodeDecorator.SCALE_IAAS_DONE, 'true')
_scale_state = 'disk_attaching'
cw._get_runtime_parameter = Mock(side_effect=_get_runtime_parameter)
cw._cloud_client.attach_disk = Mock(wraps=cw._cloud_client.attach_disk)
cw.vertically_scale_instances()
assert True == cw._cloud_client.attach_disk.called
node_instance = cw._cloud_client.attach_disk.call_args[0][0][0]
assert 'n.2' in node_instance.get_name()
assert cw._set_runtime_parameter.called_with('n.2:' + NodeDecorator.SCALE_IAAS_DONE, 'true')
_scale_state = 'disk_detaching'
cw._get_runtime_parameter = Mock(side_effect=_get_runtime_parameter)
cw._cloud_client.detach_disk = Mock(wraps=cw._cloud_client.detach_disk)
cw.vertically_scale_instances()
assert True == cw._cloud_client.detach_disk.called
node_instance = cw._cloud_client.detach_disk.call_args[0][0][0]
assert 'n.3' in node_instance.get_name()
assert cw._set_runtime_parameter.called_with('n.3:' + NodeDecorator.SCALE_IAAS_DONE, 'true')
def test_vertically_scalle_instances_nowait(self):
_scale_state = None
def _get_runtime_parameter(key):
if key.endswith(NodeDecorator.NODE_PROPERTY_SEPARATOR +
NodeDecorator.SCALE_STATE_KEY):
return _scale_state
else:
return 'unknown'
node_instances = {
'n.1':
NodeInstance({
NodeDecorator.NODE_INSTANCE_NAME_KEY:
'n.1',
NodeDecorator.NODE_NAME_KEY:
'n',
NodeDecorator.SCALE_STATE_KEY:
CloudWrapper.SCALE_STATE_RESIZING
}),
'n.2':
NodeInstance({
NodeDecorator.NODE_INSTANCE_NAME_KEY:
'n.2',
NodeDecorator.NODE_NAME_KEY:
'n',
NodeDecorator.SCALE_STATE_KEY:
CloudWrapper.SCALE_STATE_DISK_ATTACHING
}),
'n.3':
NodeInstance({
NodeDecorator.NODE_INSTANCE_NAME_KEY:
'n.3',
NodeDecorator.NODE_NAME_KEY:
'n',
NodeDecorator.SCALE_STATE_KEY:
CloudWrapper.SCALE_STATE_DISK_DETACHING
}),
'm.1':
NodeInstance({
NodeDecorator.NODE_INSTANCE_NAME_KEY:
'm.1',
NodeDecorator.NODE_NAME_KEY:
'm',
NodeDecorator.SCALE_STATE_KEY:
CloudWrapper.SCALE_STATE_OPERATIONAL
})
}
self.config_holder.set('verboseLevel', 3)
setattr(self.config_holder, 'cloud', 'local')
setattr(self.config_holder, CONFIGPARAM_CONNECTOR_MODULE_NAME,
'slipstream.cloudconnectors.dummy.DummyClientCloud')
cw = CloudWrapper(self.config_holder)
cw._get_nodes_instances = Mock(return_value=node_instances)
cw.initCloudConnector(self.config_holder)
cw._set_runtime_parameter = Mock()
cw._get_user_timeout = Mock(return_value=2)
# No waiting.
cw._wait_pre_scale_done = Mock()
cw._wait_scale_state = Mock()
_scale_state = 'resizing'
cw._get_runtime_parameter = Mock(side_effect=_get_runtime_parameter)
cw._cloud_client.resize = Mock(wraps=cw._cloud_client.resize)
cw.vertically_scale_instances()
assert True == cw._cloud_client.resize.called
node_instance = cw._cloud_client.resize.call_args[0][0][0]
assert 'n.1' in node_instance.get_name()
assert cw._set_runtime_parameter.called_with(
'n.1:' + NodeDecorator.SCALE_IAAS_DONE, 'true')
_scale_state = 'disk_attaching'
cw._get_runtime_parameter = Mock(side_effect=_get_runtime_parameter)
cw._cloud_client.attach_disk = Mock(wraps=cw._cloud_client.attach_disk)
cw.vertically_scale_instances()
assert True == cw._cloud_client.attach_disk.called
node_instance = cw._cloud_client.attach_disk.call_args[0][0][0]
assert 'n.2' in node_instance.get_name()
assert cw._set_runtime_parameter.called_with(
'n.2:' + NodeDecorator.SCALE_IAAS_DONE, 'true')
_scale_state = 'disk_detaching'
cw._get_runtime_parameter = Mock(side_effect=_get_runtime_parameter)
cw._cloud_client.detach_disk = Mock(wraps=cw._cloud_client.detach_disk)
cw.vertically_scale_instances()
assert True == cw._cloud_client.detach_disk.called
node_instance = cw._cloud_client.detach_disk.call_args[0][0][0]
assert 'n.3' in node_instance.get_name()
assert cw._set_runtime_parameter.called_with(
'n.3:' + NodeDecorator.SCALE_IAAS_DONE, 'true')