class TestListNodePool(fakes.TestCCE):
_objs = fakes.FakeNodePool.create_multiple(3)
columns = ('ID', 'name', 'flavor', 'os', 'autoscaling', 'current_node',
'status')
data = []
for s in _objs:
flat_data = node_pool._flatten_node_pool(s)
data.append((
flat_data['id'],
flat_data['name'],
flat_data['flavor'],
flat_data['os'],
flat_data['autoscaling'],
flat_data['current_node'],
flat_data['status'],
))
def setUp(self):
super(TestListNodePool, self).setUp()
self.cmd = node_pool.ListCCENodePools(self.app, None)
self.client.node_pools = mock.Mock()
self.client.find_cluster = mock.Mock(return_value=cluster.Cluster(
id='cluster_id'))
def test_list_default(self):
arglist = ['cluster_id']
verifylist = [('cluster', 'cluster_id')]
# Verify cm is triggered with default parameters
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
# Set the response
self.client.node_pools.side_effect = [self._objs]
# Trigger the action
columns, data = self.cmd.take_action(parsed_args)
self.client.node_pools.assert_called_once_with('cluster_id')
self.assertEqual(self.columns, columns)
self.assertEqual(self.data, list(data))
def test_flatten(self):
obj = fakes.FakeNodePool.create_one()
flat_data = node_pool._flatten_node_pool(obj)
data = (
flat_data['id'],
flat_data['name'],
flat_data['flavor'],
flat_data['os'],
flat_data['current_node'],
flat_data['network_id'],
flat_data['root_volume_type'],
flat_data['root_volume_size'],
flat_data['data_volume_type'],
flat_data['data_volume_size'],
flat_data['autoscaling'],
flat_data['min_node_count'],
flat_data['max_node_count'],
flat_data['scale_down_cooldown_time'],
flat_data['priority'],
)
cmp_data = (
obj.id, obj.name, obj.spec.node_template_spec.flavor,
obj.spec.node_template_spec.os, obj.status.current_node,
obj.spec.node_template_spec.node_nic_spec.primary_nic.network_id,
obj.spec.node_template_spec.root_volume.type,
obj.spec.node_template_spec.root_volume.size,
obj.spec.node_template_spec.data_volumes[0].type,
obj.spec.node_template_spec.data_volumes[0].size,
obj.spec.autoscaling.enable, obj.spec.autoscaling.min_node_count,
obj.spec.autoscaling.max_node_count,
obj.spec.autoscaling.scale_down_cooldown_time,
obj.spec.autoscaling.priority)
self.assertEqual(data, cmp_data)
class TestCreateNodePool(fakes.TestCCE):
_obj = fakes.FakeNodePool.create_one()
columns = (
'ID',
'name',
'flavor',
'os',
'current_node',
'network_id',
'root_volume_type',
'root_volume_size',
'data_volume_type',
'data_volume_size',
'autoscaling',
'min_node_count',
'max_node_count',
'scale_down_cooldown_time',
'priority',
)
flat_data = node_pool._flatten_node_pool(_obj)
data = (
flat_data['id'],
flat_data['name'],
flat_data['flavor'],
flat_data['os'],
flat_data['current_node'],
flat_data['network_id'],
flat_data['root_volume_type'],
flat_data['root_volume_size'],
flat_data['data_volume_type'],
flat_data['data_volume_size'],
flat_data['autoscaling'],
flat_data['min_node_count'],
flat_data['max_node_count'],
flat_data['scale_down_cooldown_time'],
flat_data['priority'],
)
def setUp(self):
super(TestCreateNodePool, self).setUp()
self.cmd = node_pool.CreateCCENodePool(self.app, None)
self.app.client_manager.sdk_connection = mock.Mock()
self.cloud_client = self.app.client_manager.sdk_connection
self.cloud_client.create_cce_node_pool = mock.Mock()
self.client.find_cluster = mock.Mock(return_value=cluster.Cluster(
id='cluster_id'))
def test_create(self):
arglist = [
'cluster_name', 'pool_name', '--flavor', 'flav', '--network', 'nw',
'--os', 'CentOS', '--ssh-key', 'sshkey'
]
verifylist = [
('cluster', 'cluster_name'),
('name', 'pool_name'),
('autoscaling_enabled', False),
('az', 'random'),
('flavor', 'flav'),
('initial_node_count', 0),
('network', 'nw'),
('os', 'CentOS'),
('root_volume_size', 40),
('root_volume_type', 'SATA'),
('ssh_key', 'sshkey'),
]
# Verify cm is triggered with default parameters
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
# Set the response
self.cloud_client.create_cce_node_pool.side_effect = [self._obj]
# Trigger the action
columns, data = self.cmd.take_action(parsed_args)
self.cloud_client.create_cce_node_pool.assert_called_once_with(
availability_zone='random',
cluster='cluster_name',
flavor='flav',
name='pool_name',
network='nw',
os='CentOS',
root_volume_size=40,
root_volume_type='SATA',
ssh_key='sshkey')
self.assertEqual(self.columns, columns)
self.assertEqual(self.data, data)
class TestShowNodePool(fakes.TestCCE):
_obj = fakes.FakeNodePool.create_one()
columns = (
'ID',
'name',
'flavor',
'os',
'current_node',
'network_id',
'root_volume_type',
'root_volume_size',
'data_volume_type',
'data_volume_size',
'autoscaling',
'min_node_count',
'max_node_count',
'scale_down_cooldown_time',
'priority',
'status',
)
flat_data = node_pool._flatten_node_pool(_obj)
data = (flat_data['id'], flat_data['name'], flat_data['flavor'],
flat_data['os'], flat_data['current_node'],
flat_data['network_id'], flat_data['root_volume_type'],
flat_data['root_volume_size'], flat_data['data_volume_type'],
flat_data['data_volume_size'], flat_data['autoscaling'],
flat_data['min_node_count'], flat_data['max_node_count'],
flat_data['scale_down_cooldown_time'], flat_data['priority'],
flat_data['status'])
def setUp(self):
super(TestShowNodePool, self).setUp()
self.cmd = node_pool.ShowCCENodePool(self.app, None)
self.client.find_node_pool = mock.Mock()
self.client.find_cluster = mock.Mock(return_value=cluster.Cluster(
id='cluster_uuid'))
def test_show(self):
arglist = ['cluster_uuid', 'pool_id']
verifylist = [('cluster', 'cluster_uuid'), ('nodepool', 'pool_id')]
# Verify cm is triggered with default parameters
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
# Set the response
self.client.find_node_pool.side_effect = [self._obj]
# Trigger the action
columns, data = self.cmd.take_action(parsed_args)
self.client.find_node_pool.assert_called_once_with(
cluster='cluster_uuid', node_pool='pool_id')
self.assertEqual(self.columns, columns)
self.assertEqual(self.data, data)