@Before public void setUp() { networkDeviceHelper = new NetworkDeviceHelperImpl(interfaceDao, hostDeviceDao, hostNicVfsConfigDao, vdsDao); when(netDevice.getHostId()).thenReturn(HOST_ID); when(netDevice.getDeviceName()).thenReturn(NET_DEVICE_NAME); when(netDevice.getName()).thenReturn(NET_DEVICE_NAME); when(netDevice.getNetworkInterfaceName()).thenReturn(NIC_NAME); when(netDevice.getParentDeviceName()).thenReturn(PCI_DEVICE_NAME); when(pciDevice.getHostId()).thenReturn(HOST_ID); when(pciDevice.getDeviceName()).thenReturn(PCI_DEVICE_NAME); when(pciDevice.getName()).thenReturn(PCI_DEVICE_NAME); List<HostDevice> devices = new ArrayList<>(); devices.add(netDevice); devices.add(pciDevice); mockHostDevices(devices); when(nic.getId()).thenReturn(NIC_ID); when(nic.getName()).thenReturn(NIC_NAME); when(nic.getVdsId()).thenReturn(HOST_ID); when(interfaceDao.get(NIC_ID)).thenReturn(nic); when(nic.getName()).thenReturn(NIC_NAME); when(hostNicVfsConfig.getNicId()).thenReturn(NIC_ID); when(hostNicVfsConfigDao.getByNicId(NIC_ID)).thenReturn(hostNicVfsConfig); }
private VdsNetworkInterface mockNicForNetDevice(HostDevice netDeviceParam) { VdsNetworkInterface nic = new VdsNetworkInterface(); nic.setVdsId(netDeviceParam.getHostId()); nic.setName(netDeviceParam.getNetworkInterfaceName()); return nic; }
@Test public void getNicByNetDeviceNoNic() { VdsNetworkInterface newNic = new VdsNetworkInterface(); newNic.setName(netDevice.getNetworkInterfaceName() + "not"); mockNics(Collections.singletonList(newNic), false); assertNull(networkDeviceHelper.getNicByPciDevice(pciDevice)); }
@Test public void testGetVfMap() { final HostDevice pfNetDevice = new HostDevice(); final HostDevice pfPciDevice = new HostDevice(); final Guid pfNicId = Guid.newGuid(); final String pfNicName = "pf" + NIC_NAME; final String pfPciDeviceName = "pf" + PCI_DEVICE_NAME; pfNetDevice.setHostId(HOST_ID); pfNetDevice.setDeviceName("pf" + NET_DEVICE_NAME); pfNetDevice.setNetworkInterfaceName(pfNicName); pfNetDevice.setParentDeviceName(pfPciDeviceName); pfPciDevice.setHostId(HOST_ID); pfPciDevice.setDeviceName(pfPciDeviceName); pfPciDevice.setDeviceName(pfPciDeviceName); when(pciDevice.getParentPhysicalFunction()).thenReturn(pfPciDeviceName); mockHostDevices(Arrays.asList(pfNetDevice, pfPciDevice, new HostDevice())); when(nic.getVlanId()).thenReturn(null); final VdsNetworkInterface pfNic = new VdsNetworkInterface(); pfNic.setId(pfNicId); pfNic.setName(pfNetDevice.getNetworkInterfaceName()); final VdsNetworkInterface bondNic = new VdsNetworkInterface(); bondNic.setBonded(true); final VdsNetworkInterface vlanNic = new VdsNetworkInterface(); vlanNic.setVlanId(666); mockNics(Arrays.asList(pfNic, bondNic, vlanNic), true); mockHostSupportsSriov(true); final Map<Guid, Guid> actual = networkDeviceHelper.getVfMap(HOST_ID); assertEquals(1, actual.size()); assertThat(actual, hasEntry(NIC_ID, pfNicId)); }