def test__get_hyp_version_for_host_empty_host_no_name(self):
host = {"hypervisorFullName": None}
with self.assertRaises(CurieTestException) as ar:
DiscoveryUtil._get_hyp_version_for_host(host)
self.assertIn("Cause: Cannot get hypervisor name from node: Unknown",
str(ar.exception))
def test__update_cluster_version_info_vmm(self, m_VmmClient, n_NtnxApiCli):
cluster_pb = proto_patch_encryption_support(CurieSettings.Cluster)()
mgmt_info = cluster_pb.cluster_management_server_info
mgmt_info.vmm_info.SetInParent()
software_info = cluster_pb.cluster_software_info
software_info.nutanix_info.SetInParent()
m_vmm_cli = m_VmmClient.return_value
m_vmm_cli.get_nodes.return_value = [{
"ips": ["1.2.3.4"],
"fqdn": "node1.somewhere.blah",
"name": "node1.somewhere.blah",
"id": "157bbf6f-010b-41c6-938b-2a3dc3fae7ca",
"bmc_port": "623",
"bmc_address": "1.2.3.5",
"overall_state": "OK",
"state": "Responding",
"version": "10.0.14393.351"
}, {
"ips": ["2.3.4.5"],
"fqdn": "node2.somewhere.blah",
"name": "node2.somewhere.blah",
"id": "4657f9f7-4027-4fc4-bc90-04c16188438d",
"bmc_port": "623",
"bmc_address": "2.3.4.6",
"overall_state": "OK",
"state": "Responding",
"version": "10.0.14393.351"
}, {
"ips": ["3.4.5.6"],
"fqdn": "node3.somewhere.blah",
"name": "node3.somewhere.blah",
"id": "a4b928cf-2d16-43a1-9139-f98d4cbd55d6",
"bmc_port": "623",
"bmc_address": "3.4.5.7",
"overall_state": "OK",
"state": "Responding",
"version": "10.0.14393.351"
}]
m_vmm_cli.get_vmm_version.return_value = "4.1.0.1"
m_ntnx_api = n_NtnxApiCli.return_value
cluster_inc_id = 12345
cluster_uuid = str(uuid.uuid4())
cluster_version = "el6-release-euphrates-5.0.2-stable-9d20638eb2ba1d3f84f213d5976fbcd412630c6d"
m_ntnx_api.get_nutanix_metadata.return_value = NutanixMetadata(
version=cluster_version,
cluster_uuid=cluster_uuid,
cluster_incarnation_id=cluster_inc_id)
DiscoveryUtil.update_cluster_version_info(cluster_pb)
self.assertEqual(cluster_pb.cluster_software_info.nutanix_info.version,
"5.0.2")
self.assertEqual(
cluster_pb.cluster_management_server_info.vmm_info.vmm_version,
"4.1.0.1")
self.assertEqual(
cluster_pb.cluster_hypervisor_info.hyperv_info.version,
["10.0.14393.351", "10.0.14393.351", "10.0.14393.351"])
def test_discover_clusters_vmm(self):
if self.type != "scvmm":
raise unittest.SkipTest("Test requires a VMM cluster (found '%s')" %
self.type)
resp_pb = DiscoverClustersV2Ret()
DiscoveryUtil.discover_clusters_vmm(
address=self.config["scvmm.address"],
username=self.config["scvmm.username"],
password=self.config["scvmm.password"],
ret=resp_pb)
inventory = self._DiscoverClustersV2Ret_to_inventory(resp_pb)
for cluster_id, cluster in inventory["clusters"].iteritems():
if cluster["name"] == self.config["scvmm.cluster"]:
break
else:
self.fail("Cluster with name '%s' not found in discovery response" %
self.config["scvmm.cluster"])
for network_id in cluster["networks"]:
network = inventory["networks"][network_id]
if network["name"] == self.config["scvmm.network"]:
break
else:
self.fail("Network with name '%s' not found with discovered cluster" %
self.config["scvmm.network"])
for container_id in cluster["containers"]:
if container_id == self.config["scvmm.share_path"]:
break
else:
self.fail("Share with path '%s' not found with discovered cluster" %
self.config["scvmm.share_path"])
datacenter = inventory["datacenters"][cluster["datacenter"]]
self.assertEqual("HyperV", datacenter["name"])
def test_discover_nodes_vmm(self):
if self.type != "scvmm":
raise unittest.SkipTest("Test requires a VMM cluster (found '%s')" %
self.type)
req_pb = DiscoverNodesV2Arg()
if self.config.get("oob.type") == "ipmi":
req_pb.oob_info.type = req_pb.oob_info.kIpmi
else:
self.assertEqual(req_pb.oob_info.type, req_pb.oob_info.kUnknownInterface)
req_pb.oob_info.conn_params.username = self.config["ipmi.username"]
req_pb.oob_info.conn_params.password = self.config["ipmi.password"]
req_pb.mgmt_server.type = req_pb.mgmt_server.kPrism
req_pb.mgmt_server.conn_params.address = self.config["scvmm.address"]
req_pb.mgmt_server.conn_params.username = self.config["scvmm.username"]
req_pb.mgmt_server.conn_params.password = self.config["scvmm.password"]
req_pb.cluster_collection.name = "HyperV"
cluster_pb = req_pb.cluster_collection.cluster_vec.add()
cluster_pb.name = self.config["scvmm.cluster"]
resp_pb = DiscoverNodesV2Ret()
DiscoveryUtil.discover_nodes_vmm(req_pb, resp_pb)
self.assertEqual(len(resp_pb.node_collection_vec), 1)
discovered_nodes = resp_pb.node_collection_vec[0].node_vec
self.assertEqual(len(discovered_nodes), len(self.config["nodes"]))
for discovered_node, expected_node in zip(discovered_nodes,
self.config["nodes"]):
self.assertEqual(discovered_node.hypervisor.type,
discovered_node.hypervisor.kHyperv)
self.assertNotIn(discovered_node.hypervisor.version, [None, ""])
self.assertEqual(discovered_node.oob_info.conn_params.address,
expected_node["ipmi_addr"])
if self.config.get("oob.type") == "ipmi":
self.assertEqual(discovered_node.oob_info.type,
discovered_node.oob_info.kIpmi)
else:
self.assertEqual(discovered_node.oob_info.type,
discovered_node.oob_info.kUnknownInterface)
if self.config.get("ipmi.vendor") == "supermicro":
self.assertEqual(discovered_node.oob_info.vendor,
discovered_node.oob_info.kSupermicro)
elif self.config.get("ipmi.vendor") == "dell":
self.assertEqual(discovered_node.oob_info.vendor,
discovered_node.oob_info.kDell)
else:
self.assertEqual(discovered_node.oob_info.vendor,
discovered_node.oob_info.kUnknownVendor)
def test_discover_clusters_vcenter(self):
if self.type != "vcenter":
raise unittest.SkipTest("Test requires a vCenter cluster (found '%s')" %
self.type)
resp_pb = DiscoverClustersV2Ret()
DiscoveryUtil.discover_clusters_vcenter(
address=self.config["vcenter.address"],
username=self.config["vcenter.username"],
password=self.config["vcenter.password"],
ret=resp_pb)
inventory = self._DiscoverClustersV2Ret_to_inventory(resp_pb)
for cluster_id, cluster in inventory["clusters"].iteritems():
if cluster["name"] == self.config["cluster.name"]:
break
else:
self.fail("Cluster with name '%s' not found in discovery response" %
self.config["cluster.name"])
datacenter_id = cluster["datacenter"]
datacenter = inventory["datacenters"][datacenter_id]
self.assertIn(self.config["cluster.name"], datacenter["clusters"])
for network_id in cluster["networks"]:
network = inventory["networks"][network_id]
if network["name"] == self.config["vcenter.network"]:
self.assertEqual(network["cluster"], self.config["cluster.name"])
self.assertEqual(network["datacenter"],
self.config["vcenter.datacenter"])
break
else:
self.fail("Network with name '%s' not found with discovered cluster" %
self.config["vcenter.network"])
for container_id in cluster["containers"]:
container = inventory["containers"][container_id]
if container["name"] == self.config["vcenter.datastore"]:
self.assertEqual(container["cluster"], self.config["cluster.name"])
self.assertEqual(container["datacenter"],
self.config["vcenter.datacenter"])
break
else:
self.fail("Datastore with name '%s' not found with discovered cluster" %
self.config["vcenter.datastore"])
def test_update_virtual_ip_prism(self, m_NutanixRestApiClient):
m_client = mock.MagicMock()
m_client.clusters_get.return_value = {
"name": "Mock-Cluster",
"clusterExternalIPAddress": "1.2.3.4",
}
m_NutanixRestApiClient.from_proto.return_value = m_client
cluster_pb = proto_patch_encryption_support(CurieSettings.Cluster)()
mgmt_info = cluster_pb.cluster_management_server_info
mgmt_info.prism_info.SetInParent()
software_info = cluster_pb.cluster_software_info
software_info.nutanix_info.SetInParent()
self.assertEqual(
"", cluster_pb.cluster_software_info.nutanix_info.prism_host)
DiscoveryUtil.update_cluster_virtual_ip(cluster_pb)
self.assertEqual(
"1.2.3.4",
cluster_pb.cluster_software_info.nutanix_info.prism_host)
def test_validate_oob_config(self, mock_ping, mock_status):
proto_patch_encryption_support(CurieSettings)
cluster_pb = CurieSettings.Cluster()
for ii in xrange(4):
node_pb = cluster_pb.cluster_nodes.add()
node_pb.CopyFrom(self._no_oob_node_proto)
node_pb.id = str(ii)
DiscoveryUtil.validate_oob_config(cluster_pb)
self.assertEqual(mock_ping.call_count, 0)
self.assertEqual(mock_status.call_count, 0)
cluster_pb = CurieSettings.Cluster()
for ii in xrange(4):
node_pb = cluster_pb.cluster_nodes.add()
node_pb.CopyFrom(self._ipmi_node_proto)
node_pb.id = str(ii)
mock_ping.return_value = True
DiscoveryUtil.validate_oob_config(cluster_pb)
self.assertEqual(mock_ping.call_count, len(cluster_pb.cluster_nodes))
self.assertEqual(mock_status.call_count, len(cluster_pb.cluster_nodes))
mock_ping.reset_mock()
mock_status.reset_mock()
mock_ping.side_effect = [True, False, True, True]
with self.assertRaises(CurieException):
DiscoveryUtil.validate_oob_config(cluster_pb)
# We expect that the first ping succeeds and then the second fails. There
# should be an exception after the second ping attempt. If ping fails, the
# expectations is then that the chassis status won't be called.
self.assertEqual(mock_ping.call_count, 2)
self.assertEqual(mock_status.call_count, 1)
mock_ping.reset_mock()
mock_status.reset_mock()
mock_ping.return_value = True
mock_ping.side_effect = None
mock_status.side_effect = [{},
CurieException(
CurieError.kOobAuthenticationError,
"AuthError"), {},
CurieException(CurieError.kInternalError,
"SomeOtherError")]
with self.assertRaises(CurieException):
DiscoveryUtil.validate_oob_config(cluster_pb)
self.assertEqual(mock_ping.call_count, 2)
self.assertEqual(mock_status.call_count, 2)
def test__get_hyp_version_for_host(self):
host = {
"hypervisorFullName": "Nutanix 20170726.42",
DiscoveryUtil.CE_HOST_ATTR_KEY: DiscoveryUtil.CE_HOST_ATTR_VAL
}
self.assertEqual(DiscoveryUtil._get_hyp_version_for_host(host),
"Nutanix CE 20170726.42")
host["hypervisorFullName"] = "20170726.42"
self.assertEqual(DiscoveryUtil._get_hyp_version_for_host(host),
"CE 20170726.42")
host["hypervisorFullName"] = "20170726.42"
host[DiscoveryUtil.CE_HOST_ATTR_KEY] = ""
self.assertEqual(DiscoveryUtil._get_hyp_version_for_host(host),
"20170726.42")
host["hypervisorFullName"] = "Nutanix %s" % host["hypervisorFullName"]
self.assertEqual(DiscoveryUtil._get_hyp_version_for_host(host),
"Nutanix 20170726.42")
def test_update_virtual_ip_vmm_no_cvms_found(self, m_VmmClient,
m_NutanixRestApiClient):
m_VmmClient.is_nutanix_cvm.side_effect = [False, False]
m_VmmClient.is_powered_on.side_effect = []
m_vmm_client = mock.MagicMock()
m_vmm_client.get_vms.return_value = [
{
"name": "FAKE-VM-A",
"ips": ["1.1.1.1"]
},
{
"name": "FAKE-ALSO-NOT-A-CVM",
"ips": ["1.1.1.2"]
},
]
m_VmmClient.return_value = m_vmm_client
m_nutanix_client = mock.MagicMock()
m_nutanix_client.clusters_get.return_value = {
"name": "Mock-Cluster",
"clusterExternalIPAddress": "1.2.3.4",
}
m_NutanixRestApiClient.return_value = m_nutanix_client
cluster_pb = proto_patch_encryption_support(CurieSettings.Cluster)()
mgmt_info = cluster_pb.cluster_management_server_info
mgmt_info.vmm_info.SetInParent()
software_info = cluster_pb.cluster_software_info
software_info.nutanix_info.SetInParent()
software_info.nutanix_info.prism_user = "******"
software_info.nutanix_info.prism_password = "******"
self.assertEqual(
"", cluster_pb.cluster_software_info.nutanix_info.prism_host)
with self.assertRaises(CurieTestException) as ar:
DiscoveryUtil.update_cluster_virtual_ip(cluster_pb)
self.assertIn(
"Cause: No Nutanix CVMs found.\n\n"
"Impact: The cluster virtual IP address can not be discovered.\n\n"
"Corrective Action: Please verify that the cluster contains Nutanix "
"CVMs, and that they are powered on.\n\n"
"Traceback: None", str(ar.exception))
def test_discover_clusters_prism(self):
if self.type != "prism":
raise unittest.SkipTest("Test requires a Prism cluster (found '%s')" %
self.type)
resp_pb = DiscoverClustersV2Ret()
DiscoveryUtil.discover_clusters_prism(
address=self.config["prism.address"],
username=self.config["prism.username"],
password=self.config["prism.password"],
ret=resp_pb)
inventory = self._DiscoverClustersV2Ret_to_inventory(resp_pb)
# Verify that the cluster name is found in the discovery
for cluster_id, cluster in inventory["clusters"].iteritems():
if cluster["name"] == self.config["cluster.name"]:
break
else:
self.fail("Cluster with name '%s' not found in discovery response" %
self.config["cluster.name"])
# Verify that the desired network is found in discovery
for network_id in cluster["networks"]:
network = inventory["networks"][network_id]
if network["name"] == self.config["prism.network"]:
break
else:
self.fail("Network with name '%s' not found with discovered cluster" %
self.config["prism.network"])
# Verify that the desired container is found in discovery
for container_id in cluster["containers"]:
container = inventory["containers"][container_id]
if container["name"] == self.config["prism.container"]:
break
else:
self.fail("Container with name '%s' not found with discovered cluster" %
self.config["prism.container"])
datacenter = inventory["datacenters"][cluster["datacenter"]]
self.assertEqual("Prism", datacenter["name"])
def test_update_virtual_ip_vmm_error_communicating_with_cvms(
self, m_VmmClient, m_NutanixRestApiClient):
m_VmmClient.is_nutanix_cvm.side_effect = [True, True]
m_VmmClient.is_powered_on.side_effect = [True, True]
m_vmm_client = mock.MagicMock()
m_vmm_client.get_vms.return_value = [
{
"name": "FAKE-CVM-A",
"ips": ["1.1.1.1"]
},
{
"name": "FAKE-CVM-B",
"ips": ["1.1.1.2"]
},
]
m_VmmClient.return_value = m_vmm_client
m_nutanix_client = mock.MagicMock()
m_nutanix_client.clusters_get.side_effect = IOError("Kaboom!")
m_NutanixRestApiClient.return_value = m_nutanix_client
cluster_pb = proto_patch_encryption_support(CurieSettings.Cluster)()
mgmt_info = cluster_pb.cluster_management_server_info
mgmt_info.vmm_info.SetInParent()
software_info = cluster_pb.cluster_software_info
software_info.nutanix_info.SetInParent()
software_info.nutanix_info.prism_user = "******"
software_info.nutanix_info.prism_password = "******"
self.assertEqual(
"", cluster_pb.cluster_software_info.nutanix_info.prism_host)
with self.assertRaises(CurieTestException) as ar:
DiscoveryUtil.update_cluster_virtual_ip(cluster_pb)
self.assertIn(
"Cause: Failed to query Prism on any Nutanix CVM.\n\n"
"Impact: The cluster virtual IP address can not be discovered.\n\n"
"Corrective Action: Please verify that the Nutanix CVMs on the cluster "
"are powered on, and that the network connectivity to the CVMs is "
"correct.\n\nTraceback (most recent call last):\n",
str(ar.exception))
self.assertIn("IOError: Kaboom!", ar.exception.traceback)
def test_update_virtual_ip_vmm_cvms(self, m_VmmClient,
m_NutanixRestApiClient):
m_VmmClient.is_nutanix_cvm.side_effect = [False, True]
m_VmmClient.is_powered_on.side_effect = [True]
m_vmm_client = mock.MagicMock()
m_vmm_client.get_vms.return_value = [
{
"name": "FAKE-VM-A",
"ips": ["1.1.1.1"]
},
{
"name": "FAKE-CVM",
"ips": ["1.1.1.2"]
},
]
m_VmmClient.return_value = m_vmm_client
m_nutanix_client = mock.MagicMock()
m_nutanix_client.clusters_get.return_value = {
"name": "Mock-Cluster",
"clusterExternalIPAddress": "1.2.3.4",
}
m_NutanixRestApiClient.return_value = m_nutanix_client
cluster_pb = proto_patch_encryption_support(CurieSettings.Cluster)()
mgmt_info = cluster_pb.cluster_management_server_info
mgmt_info.vmm_info.SetInParent()
software_info = cluster_pb.cluster_software_info
software_info.nutanix_info.SetInParent()
software_info.nutanix_info.prism_user = "******"
software_info.nutanix_info.prism_password = "******"
self.assertEqual(
"", cluster_pb.cluster_software_info.nutanix_info.prism_host)
DiscoveryUtil.update_cluster_virtual_ip(cluster_pb)
self.assertEqual(
"1.2.3.4",
cluster_pb.cluster_software_info.nutanix_info.prism_host)
m_NutanixRestApiClient.assert_has_calls([
mock.call("1.1.1.2", "fake_prism_user", "fake_prism_password"),
])
def test_dispatch(self):
cluster_pb = proto_patch_encryption_support(CurieSettings.Cluster)()
mgmt_info = cluster_pb.cluster_management_server_info
software_info = cluster_pb.cluster_software_info
hyp_info = cluster_pb.cluster_hypervisor_info
mgmt_info.prism_info.SetInParent()
with self.assertRaises(CurieException):
DiscoveryUtil.update_cluster_version_info(cluster_pb)
software_info.nutanix_info.SetInParent()
with self.assertRaises(CurieException):
DiscoveryUtil.update_cluster_version_info(cluster_pb)
hyp_info.ahv_info.SetInParent()
fq_update_prism = ("%s._update_cluster_version_info_prism" %
self.fq_disc_util_name)
with mock.patch(fq_update_prism) as mock_prism:
DiscoveryUtil.update_cluster_version_info(cluster_pb)
mock_prism.assert_called_once_with(cluster_pb)
mgmt_info.Clear()
software_info.Clear()
hyp_info.Clear()
mgmt_info.vcenter_info.SetInParent()
fq_update_vcenter = ("%s._update_cluster_version_info_vcenter" %
self.fq_disc_util_name)
with mock.patch(fq_update_vcenter) as mock_vcenter:
DiscoveryUtil.update_cluster_version_info(cluster_pb)
mock_vcenter.assert_called_once_with(cluster_pb)
mgmt_info.Clear()
mgmt_info.vmm_info.SetInParent()
fq_update_vmm = ("%s._update_cluster_version_info_vmm" %
self.fq_disc_util_name)
with mock.patch(fq_update_vmm) as mock_vmm:
DiscoveryUtil.update_cluster_version_info(cluster_pb)
mock_vmm.assert_called_once_with(cluster_pb)
mgmt_info.Clear()
with self.assertRaises(CurieException):
DiscoveryUtil.update_cluster_version_info(cluster_pb)
fq_update_vip = ("%s.update_cluster_virtual_ip" %
self.fq_disc_util_name)
with mock.patch(fq_update_vip) as mock_vip:
DiscoveryUtil.update_cluster_virtual_ip(cluster_pb)
mock_vip.assert_called_once_with(cluster_pb)