def test__get_node_vif_ids_one_port(self): port1 = self._create_test_port(node_id=self.node.id, id=6, address='aa:bb:cc', uuid=utils.generate_uuid(), extra={'vif_port_id': 'test-vif-A'}, driver='fake') expected = {port1.uuid: 'test-vif-A'} with task_manager.acquire(self.context, self.node.uuid) as task: result = neutron.get_node_vif_ids(task) self.assertEqual(expected, result)
def get_ports_info_from_neutron(task): """ Get neutron port info from neutron about this task """ vifs = neutron.get_node_vif_ids(task) if not vifs: LOG.warning(_("No VIFs found for node %(node)s when attempting to " "update Neutron DHCP BOOT options."), {'node': task.node.uuid}) return failures = [] vif_ports_info = {} for port_id, port_vif in vifs.iteritems(): try: vif_ports_info[port_id] = get_vif_port_info(task,port_vif) except xcat_exception.FailedToGetInfoOnPort(port_id=port_vif): failures.append(port_vif) return vif_ports_info
def test__get_node_vif_ids_two_ports(self): port1 = self._create_test_port( node_id=self.node.id, id=6, address="aa:bb:cc", uuid=utils.generate_uuid(), extra={"vif_port_id": "test-vif-A"}, driver="fake", ) port2 = self._create_test_port( node_id=self.node.id, id=7, address="dd:ee:ff", uuid=utils.generate_uuid(), extra={"vif_port_id": "test-vif-B"}, driver="fake", ) expected = {port1.uuid: "test-vif-A", port2.uuid: "test-vif-B"} with task_manager.acquire(self.context, self.node.uuid) as task: result = neutron.get_node_vif_ids(task) self.assertEqual(expected, result)
def test_get_node_vif_ids_no_ports(self): expected = {} with task_manager.acquire(self.context, self.node.uuid) as task: result = neutron.get_node_vif_ids(task) self.assertEqual(expected, result)