Signed-off-by: Lucio Correia <luciojhc@linux.vnet.ibm.com> --- mockmodel.py | 41 ++++++++++++++++++++++++++++++++++++++++- 1 file changed, 40 insertions(+), 1 deletion(-) diff --git a/mockmodel.py b/mockmodel.py index 142e81f..ae5cbc1 100644 --- a/mockmodel.py +++ b/mockmodel.py @@ -26,21 +26,25 @@ from collections import defaultdict from lxml import objectify from lxml.builder import E -from wok.exception import NotFoundError, OperationFailed +from wok.exception import InvalidParameter, NotFoundError, OperationFailed from wok.objectstore import ObjectStore from wok.utils import add_task from wok.xmlutils.utils import xml_item_update from wok.plugins.kimchi import imageinfo +from wok.plugins.kimchi import netinfo from wok.plugins.kimchi import osinfo from wok.plugins.kimchi.model import cpuinfo from wok.plugins.kimchi.model.groups import PAMGroupsModel from wok.plugins.kimchi.model.host import DeviceModel from wok.plugins.kimchi.model.host import DevicesModel +from wok.plugins.kimchi.model.interfaces import InterfaceModel +from wok.plugins.kimchi.model.interfaces import InterfacesModel from wok.plugins.kimchi.model.libvirtstoragepool import IscsiPoolDef from wok.plugins.kimchi.model.libvirtstoragepool import NetfsPoolDef from wok.plugins.kimchi.model.libvirtstoragepool import StoragePoolDef from wok.plugins.kimchi.model.model import Model +from wok.plugins.kimchi.model.networks import NetworksModel from wok.plugins.kimchi.model.storagepools import StoragePoolModel from wok.plugins.kimchi.model.storagepools import StoragePoolsModel from wok.plugins.kimchi.model.storagevolumes import StorageVolumeModel @@ -123,6 +127,11 @@ class MockModel(Model): DeviceModel.is_device_3D_controller = \ self._mock_device_is_device_3D_controller DevicesModel.get_list = self._mock_devices_get_list + InterfacesModel.get_list = self._mock_interfaces_get_list + InterfaceModel.lookup = self._mock_interface_lookup + NetworksModel._set_network_vepa = self.networks_set_network_vepa + self.networks = NetworksModel(conn=self.conn, + objstore_loc=objstore_loc) StoragePoolsModel._check_lvm = self._check_lvm StoragePoolModel._update_lvm_disks = self._update_lvm_disks StoragePoolModel._pool_used_by_template = self._pool_used_by_template @@ -350,6 +359,36 @@ class MockModel(Model): def _mock_device_lookup(self, dev_name): return self._mock_devices.devices[dev_name] + def _mock_interfaces_get_list(self): + return list(set(netinfo.all_favored_interfaces()) - + set(self.networks.get_all_networks_interfaces())) + \ + ['eth-mock0'] + + def _mock_interface_lookup(self, name): + if name == 'eth-mock0': + return {'name': "eth-mock0", + 'type': "nic", + 'status': "active", + 'link_detected': "yes", + 'ipaddr': "", + 'netmask': ""} + + try: + return netinfo.get_interface_info(name) + except ValueError: + raise NotFoundError("KCHIFACE0001E", {'name': name}) + + def networks_set_network_vepa(self, params): + for iface in params['interfaces']: + if iface == 'eth-mock0': + continue + + if ('vlan_id' in params or not (netinfo.is_bare_nic(iface) or + netinfo.is_bonding(iface))): + raise InvalidParameter('KCHNET0028E', {'name': iface}) + + params['forward'] = {'mode': 'vepa', 'devs': params['interfaces']} + def _mock_partitions_get_list(self): return self._mock_partitions.partitions.keys() -- 1.9.1