def test_get_libvirt_networks(self, _fetch_network_data, openReadOnly): conn = openReadOnly.return_value conn.listAllNetworks.return_value = ["net1", "net2"] networks = libvirt_nets.get_libvirt_networks() self.assertEqual([_fetch_network_data.return_value, _fetch_network_data.return_value], networks) openReadOnly.assertCalledWith() conn.listAllNetworks.assertCalledWith() _fetch_network_data.assertCalledWith("net1") _fetch_network_data.assertCalledWith("net2") conn.close.assertCalledWith()
def test_get_libvirt_networks(self, _fetch_network_data, openReadOnly): conn = openReadOnly.return_value conn.listAllNetworks.return_value = ['net1', 'net2'] networks = libvirt_nets.get_libvirt_networks() self.assertEqual([ _fetch_network_data.return_value, _fetch_network_data.return_value ], networks) openReadOnly.assertCalledWith() conn.listAllNetworks.assertCalledWith() _fetch_network_data.assertCalledWith('net1') _fetch_network_data.assertCalledWith('net2') conn.close.assertCalledWith()
def process_nic_mappings(args): if not _has_nic_mapping_args(args): return args LOG.info('Looking for NIC configurations in the image...') nics = _fetch_nics_from_image(args) LOG.info('NICs found: %s', ', '.join([n['name'] for n in nics])) for nic in nics: LOG.debug('NIC %s: %s', nic['name'], str(nic)) networks = libvirt_nets.get_libvirt_networks() mapped_nics = nics if _auto_nic_mappings_enabled(args): mapped_nics = _map_nics_auto(nics, networks) manual_mappings = _parse_manual_nic_mappings(args) mapped_nics = _map_nics_manual(mapped_nics, manual_mappings) # TODO(jistr): check mappings' sanity return _convert_nic_mappings_args(args, mapped_nics)