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'}) expected = {port1.uuid: 'test-vif-A'} with task_manager.acquire(self.context, self.node.uuid) as task: result = pxe._get_node_vif_ids(task) self.assertEqual(expected, result)
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"}, ) 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"}, ) expected = {port1.uuid: "test-vif-A", port2.uuid: "test-vif-B"} with task_manager.acquire(self.context, self.node.uuid) as task: result = pxe._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 = pxe._get_node_vif_ids(task) self.assertEqual(expected, result)