def setUp(self):
super(TestBackupRestore, self).setUp()
self.controller = backup_restore.Controller()
self.req = webob.Request.blank('/')
self.req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
def setUp(self):
super(TestOsInstall, self).setUp()
self.req = webob.Request.blank('/')
self.req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
self._log_handler.reset()
def test_add_version_patch(self, mock_do_request):
version_patch_meta = set_version_meta()
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
def fake_do_request(method, path, **params):
res = mock.Mock()
if method == 'POST':
post_result = {
u'version_patch': {u'status': u'used',
u'name': u'name3',
u'deleted': False, u'checksum': None,
u'created_at': u'2016-07-12',
u'description': None,
u'status': u'used',
u'deleted_at': None,
u'size': None}}
res.read.return_value = jsonutils.dumps(post_result)
return res
mock_do_request.side_effect = fake_do_request
add_version = self.controller.add_version_patch(req,
version_patch_meta)
self.assertEqual("name3",
add_version['version_patch_meta']['name'])
def setUp(self):
super(DeployServerFuncs, self).setUp()
self.controller = deploy_server.Controller()
self.req = webob.Request.blank('/')
self.req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
def test_begin_install_os_with_write_ipmi(self, mock_log, mock_update_host,
mock_set_boot_pxe,
mock_install_os_for_baremetal,
mock_set_power_reset):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
cluster_id = '1'
class InstallMock():
def __init__(self):
pass
def install_os_for_baremetal(self, host_detail):
pass
def set_power_reset(self, host_detail, impi_result_flag):
pass
hosts_detail = [{'id': '1'}]
mock_log.side_effect = self._log_handler
mock_update_host.return_value = {}
mock_set_boot_pxe.return_value = True
mock_install_os_for_baremetal.side_effect = \
InstallMock().install_os_for_baremetal
mock_set_power_reset.side_effect = InstallMock().set_power_reset
install.OSInstall(req, cluster_id, False).\
_begin_install_os(hosts_detail, cluster_id)
self.assertTrue(mock_install_os_for_baremetal.called)
self.assertTrue(mock_set_power_reset.called)
def test_add_network_invalid_ip_ranges(self, get_cluster, get_networks,
add_network,
fake_valid_network_range):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
ip_ranges = [
{
'start': '12.18.1.5',
'cidr': '12.18.1.1/24',
'end': '12.18.1.5',
'gateway': '12.18.1.2'
},
{
'start': '112.18.1.15',
'cidr': '112.18.1.1/24',
'end': '112.18.1.15',
'gateway': '112.18.1.5'
},
]
network_meta = {'name': 'PUBLICAPI',
'network_type': 'PUBLICAPI',
'ip_ranges': ip_ranges
}
return_network = {'name': 'MANAGEMENT1',
'network_type': 'MANAGEMENT',
'custom_name': 'management1'}
fake_valid_network_range.return_value = True
self.assertRaises(
exc.HTTPForbidden, self.controller.add_network, req, network_meta)
def test_delete_version_patch(self, mock_do_request):
version_patch_id = "34811a0e-a69f-4dd3-bbfb-66792a979e99"
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
def fake_do_request(method, path, **params):
res = mock.Mock()
if method == "GET":
get_result = {
"version_patch": {"status": "unused",
"name": "ssh2.exe",
"checksum": "60489112c277862fbf",
"size": 1089536,
"id": "34811a0e-a69f-2a979e99",
"description": "azsad"}}
res.read.return_value = jsonutils.dumps(get_result)
return res
elif method == "DELETE":
result = {
"version_patch": {"status": "unused",
"name": "ssh2.exe",
"checksum": "60489112c277a18147b",
"created_at": "2016-07-12",
"size": 1089536,
"updated_at": "2016-07-12",
"id": "34811a0e-a69f-4dd3",
"description": "azsdadsad"}}
res.read.return_value = jsonutils.dumps(result)
return res
mock_do_request.side_effect = fake_do_request
version = self.controller.delete_version_patch(req, version_patch_id)
self.assertEqual(200, version.status_code)
def test_add_network_publicapi_net(self, get_cluster, get_networks,
add_network, fake_valid_network_range,
fake_valid_ip_ranges):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
ip_ranges = [
{
'start': '112.18.1.5',
'cidr': '112.18.1.1/24',
'end': '112.18.1.5',
'gateway': '112.18.1.2'
},
{
'start': '112.18.1.15',
'cidr': '112.18.1.1/24',
'end': '112.18.1.15',
'gateway': '112.18.1.1'
},
]
network_meta = {'name': 'PUBLICAPI',
'network_type': 'PUBLICAPI',
'ip_ranges': ip_ranges,
'cidr': '112.18.1.1/24'
}
return_network = {'name': 'PUBLICAPI',
'network_type': 'PUBLICAPI'}
get_cluster.return_value = []
get_networks.return_value = []
add_network.return_value = return_network
fake_valid_network_range.return_value = True
network = self.controller.add_network(req, network_meta)
self.assertEqual('PUBLICAPI',
network['network_meta']['name'])
def test_add_network_with_exist_custom_name(self,
get_cluster,
get_networks):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
network_meta = {'cluster_id': '1',
'network_type': 'MANAGEMENT',
'custom_name': 'management1',
'name': 'MANAGEMENT1'}
cluster = {'id': '1',
'deleted': 0}
networks_with_same_custom_name = [{'name': 'PUBLICAPI',
'network_type': 'PUBLICAPI',
'custom_name': None},
{'name': 'physnet1',
'network_type': 'DATAPLANE'},
{'name': 'STORAGE',
'network_type': 'STORAGE',
'custom_name': 'management1'}]
get_cluster.return_value = cluster
get_networks.return_value = networks_with_same_custom_name
self.assertRaises(webob.exc.HTTPConflict, self.controller.add_network,
req, network_meta)
def test_add_network_with_no_exist_custom_name(self,
get_cluster,
get_networks,
add_network):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
network_meta = {'cluster_id': '1',
'network_type': 'MANAGEMENT',
'custom_name': 'management1',
'name': 'MANAGEMENT1'}
cluster = {'id': '1',
'deleted': 0}
networks = [{'name': 'PUBLICAPI',
'network_type': 'PUBLICAPI',
'custom_name': None},
{'name': 'physnet1',
'network_type': 'DATAPLANE',
'custom_name': 'physnet2'}]
return_network = {'name': 'MANAGEMENT1',
'network_type': 'MANAGEMENT',
'custom_name': 'management1'}
get_cluster.return_value = cluster
get_networks.return_value = networks
add_network.return_value = return_network
network = self.controller.add_network(req, network_meta)
self.assertEqual('management1',
network['network_meta']['custom_name'])
def test_list_deploy_server(self, mock_do_get_pxe_nic, mock_do_get_nics,
mock_do_get_all_networks):
def mock_get_all_networks(*args, **kwargs):
return [{"ip": "99.99.1.5"}]
def mock_get_nics(*args, **kwargs):
return set(['docker0', 'eth1', 'eth0'])
def mock_get_pxe_nic(*args, **kwargs):
return "bond0"
mock_do_get_all_networks.side_effect = mock_get_all_networks
mock_do_get_nics.side_effect = mock_get_nics
mock_do_get_pxe_nic.side_effect = mock_get_pxe_nic
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
actual = {
'deploy_servers': [{
'ip': '99.99.1.5',
'nics': set(['docker0', 'eth0', 'eth1']),
'pxe_nic': 'bond0'
}]
}
deploy_servers = self.controller.list_deploy_server(req)
self.assertEqual(actual, deploy_servers)
def setUp(self):
super(TestDiskArray, self).setUp()
self.controller = disk_array.Controller()
self.req = webob.Request.blank('/')
self.req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
def setUp(self):
super(TestInstall, self).setUp()
self.req = webob.Request.blank('/')
self.req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
self.installer = install.KOLLAInstallTask(
self.req, '8ad27e36-f3e2-48b4-84b8-5b676c6fabde')
def test_update_network_invalid_cidr(self, get_network_meta,
get_cluster_meta,
get_networks_detail,
fake_is_dataplane_in_use,
fake_update_network,
fake_valid_ip_ranges):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
ip_ranges = [
{
'start': '112.18.1.5',
'cidr': '112.18.1.1/24',
'end': '112.18.1.5',
'gateway': '112.18.1.2'
},
{
'start': '112.18.1.15',
'cidr': '112.18.1.1/24',
'end': '112.18.1.15',
'gateway': '112.18.1.1'
},
]
network_id = 'cf531581-a283-41dd-9e4e-4b98454d54e7'
network_meta = {'cluster_id': '1',
'name': 'PUBLICAPI',
'network_type': 'PUBLICAPI',
'ip_ranges': ip_ranges,
'cidr': None}
orig_network_meta = {'cidr': None,
'gateway': '112.18.1.1',
'cluster_id': '1',
'vlan_id': None,
'deleted': False,
'id': 'cf531581-a283-41dd-9e4e-4b98454d54e7',
'network_type': 'PUBLICAPI',
'type': 'default'}
cluster_meta = {'id': '1', 'deleted': False, }
networks_detail = [{'cluster_id': '1',
'gateway': '112.18.1.1',
'vlan_id': None,
'id': 'cf531581-a283-41dd-9e4e-4b98454d54e7',
'ip_ranges': [
{
'end': '112.18.1.16',
'start': '112.18.1.17'}],
'name': 'physnet1',
'network_type': 'PUBLICAPI',
'type': 'default'}]
get_network_meta.return_value = orig_network_meta
get_cluster_meta.return_value = cluster_meta
get_networks_detail.return_value = networks_detail
fake_update_network.return_value = network_meta
fake_is_dataplane_in_use.return_value = False
self.assertRaises(exc.HTTPForbidden, self.controller.update_network,
req, network_id, network_meta)
def test_add_version_patch_with_no_name(self):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
version_meta = {}
version_meta["description"] = "111"
self.assertRaises(ValueError, self.controller.add_version_patch, req,
version_meta)
def test_update_network_with_no_exist_custom_name(self,
get_network_meta,
get_cluster_meta,
get_networks_detail,
update_network):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
network_id = '1'
network_meta = {'custom_name': 'publicapi1'}
orig_network_meta = {'deleted': False,
'cluster_id': '1',
'id': '1',
'network_type': 'PUBLICAPI',
'cidr': '192.168.1.1/24',
'vlan_id': None,
'custom_name': 'publicapi',
'gateway': None}
cluster_meta = {'id': '1',
'deleted': False}
networks_detail = [{'deleted': False,
'cluster_id': '1',
'id': '1',
'network_type': 'PUBLICAPI',
'cidr': '192.168.1.1/24',
'vlan_id': None,
'custom_name': 'publicapi'},
{'deleted': False,
'cluster_id': '1',
'id': '2',
'network_type': 'MANAGEMENT',
'cidr': '192.168.1.1/24',
'vlan_id': None,
'custom_name': 'management'},
{'deleted': False,
'cluster_id': '1',
'id': '3',
'network_type': 'DATAPLANE',
'cidr': None,
'vlan_id': None,
'custom_name': 'physnet'}]
update_network_meta = {'deleted': False,
'cluster_id': '1',
'id': '1',
'network_type': 'PUBLICAPI',
'cidr': '192.168.1.1/24',
'vlan_id': None,
'custom_name': 'publicapi1'}
get_network_meta.return_value = orig_network_meta
get_cluster_meta.return_value = cluster_meta
get_networks_detail.return_value = networks_detail
update_network.return_value = update_network_meta
network_meta = self.controller.update_network(req, network_id,
network_meta)
self.assertEqual('publicapi1',
network_meta['network_meta']['custom_name'])
def test_export_db_to_json(self):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
template = {'cluster_name': 'cluster1', 'description': 'desc'}
registry.get_clusters_detail = mock.Mock(return_value={})
self.assertRaises(HTTPForbidden, self.controller.export_db_to_json,
req, template)
def test_list_host_patch_history(self, mock_list_history):
history_meta = {'patch_name': 'test'}
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
mock_list_history.return_value = history_meta
historys = self.controller.list_host_patch_history(req)
self.assertEqual({'patch_history_meta': history_meta}, historys)
def test_get_services_disk(self):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
role = {'id': '1', 'name': 'COMPUTER'}
registry.list_service_disk_metadata = mock.Mock(return_value=[])
service_disk = self.controller._get_services_disk(req, role)
self.assertEqual([], service_disk)
def test_delete_template(self):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
template_id = "123"
registry.delete_template_metadata = mock.Mock(return_value={""})
template = self.controller.delete_template(req, template_id)
self.assertEqual(200, template.status_code)
def test_import_optical_switchs_to_db(self):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
optical_switchs = [{'role_id': 'CONTROLLER_HA'}]
roles = [{'id': '2', 'name': 'CONTROLLER_LB'}]
self.assertRaises(HTTPBadRequest,
self.controller._import_optical_switchs_to_db, req,
optical_switchs, roles)
def test_add_template(self):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
template = {"template_name": "test"}
registry.add_template_metadata = mock.Mock(
return_value={"template_name": "test"})
template_meta = self.controller.add_template(req, template)
self.assertEqual("test", template_meta['template']['template_name'])
def test_list_host_patch_history_with_except(self, mock_list_history):
def mock_history(*args, **kwargs):
raise webob.exc.HTTPBadRequest
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
mock_list_history.side_effect = mock_history
self.assertRaises(webob.exc.HTTPBadRequest,
self.controller.list_host_patch_history, req)
def test_get_template_lists(self):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
self.controller._get_query_params = mock.Mock(
return_value={'filter': {}})
template = {"template_name": "test"}
registry.template_lists_metadata = mock.Mock(return_value=[])
templates = self.controller.get_template_lists(req)
self.assertEqual([], templates['template'])
def test_get_assigned_networks_by_network_id_exception(
self, mock_get_assigned_networks_by_network_id):
id = '3'
self.req = webob.Request.blank('/')
self.req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
mock_get_assigned_networks_by_network_id.side_effect = \
fake_network_id(id)
self.assertRaises(exception.Conflict,
self.controller.get_assigned_networks_by_network_id,
self.req, id)
def test_get_assigned_networks_by_network_id_notfound(
self, mock_get_assigned_networks_by_network_id):
id = '2'
self.req = webob.Request.blank('/')
self.req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
mock_get_assigned_networks_by_network_id.side_effect = \
fake_network_id(id)
self.assertRaises(webob.exc.HTTPNotFound,
self.controller.get_assigned_networks_by_network_id,
self.req, id)
def test_add_network(self, mock_add_network_metadata):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenamet')
network_meta = {'network_type': 'DATAPLANE',
'gateway': '192.168.1.1', 'name': 'pysnet1'}
mock_add_network_metadata.return_value = network_meta
#registry.add_network_metadata = mock.Mock(return_value=network_meta)
add_network = self.controller.add_network(req, network_meta)
self.assertEqual(network_meta['network_type'],
add_network['network_meta']['network_type'])
def test_update_db_host_status_with_os_patch_id(self, mock_do_update_host):
def mock_update_host(*args, **kwargs):
return host_status
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True, user='******',
tenant='fake tenant')
host_id = '123'
version_patch_id = "456"
host_status = {"os_progress": 100, "os_status": "active",
"version_patch_id": version_patch_id}
mock_do_update_host.side_effect = mock_update_host
host_info = common.update_db_host_status(req, host_id, host_status)
self.assertEqual("456", host_info['version_patch_id'])
def test_update_db_host_status(self, mock_do_update_host):
def mock_update_host(*args, **kwargs):
return host_status
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True, user='******',
tenant='fake tenant')
host_id = '123'
host_status = {"os_progress": 100, "os_status": "active",
"messages": "test"}
mock_do_update_host.side_effect = mock_update_host
host_info = common.update_db_host_status(req, host_id, host_status)
self.assertEqual(100, host_info['os_progress'])
self.assertEqual("active", host_info['os_status'])
self.assertEqual("test", host_info['messages'])
def test_upgrade_os_failed(self, mock_os_thread_in, mock_get_host_detail,
mock_do_update_db_status):
req = webob.Request.blank('/')
req.context = RequestContext(is_admin=True,
user='******',
tenant='fake tenant')
version_id = "1"
version_patch_id = ""
update_file = "test"
hosts_list = [{'10.43.177.1': {'id': "1"}}]
update_object = "vplat"
mock_do_update_db_status.return_value = {}
mock_get_host_detail.return_value = {'os_status': 'updating'}
mock_os_thread_in.return_value = {}
install.upgrade_os(req, version_id, version_patch_id, update_file,
hosts_list, update_object)
