use of com.vmware.vim.binding.vim.host.Summary.ConfigSummary in project flowgate by vmware.
the class VCDataServiceTest method testFeedHostMetaData.
@Test
public void testFeedHostMetaData() {
HostSystem host = Mockito.mock(HostSystem.class);
Capability capability = Mockito.mock(Capability.class);
when(host.getCapability()).thenReturn(capability);
when(capability.isMaintenanceModeSupported()).thenReturn(true);
when(capability.isRebootSupported()).thenReturn(true);
when(capability.getMaxRunningVMs()).thenReturn(10);
when(capability.getMaxSupportedVcpus()).thenReturn(10);
when(capability.getMaxRegisteredVMs()).thenReturn(10);
RuntimeInfo runtimeInfo = Mockito.mock(RuntimeInfo.class);
when(host.getRuntime()).thenReturn(runtimeInfo);
when(runtimeInfo.getConnectionState()).thenReturn(ConnectionState.connected);
when(runtimeInfo.getPowerState()).thenReturn(PowerState.poweredOff);
Calendar time = Mockito.mock(Calendar.class);
when(runtimeInfo.getBootTime()).thenReturn(time);
Summary summary = Mockito.mock(Summary.class);
when(host.getSummary()).thenReturn(summary);
when(summary.isRebootRequired()).thenReturn(true);
QuickStats quickStats = Mockito.mock(QuickStats.class);
when(summary.getQuickStats()).thenReturn(quickStats);
when(quickStats.getUptime()).thenReturn(10);
AboutInfo aboutInfo = Mockito.mock(AboutInfo.class);
ConfigSummary configSummary = Mockito.mock(ConfigSummary.class);
when(summary.getConfig()).thenReturn(configSummary);
when(configSummary.getProduct()).thenReturn(aboutInfo);
when(aboutInfo.getBuild()).thenReturn("build");
when(aboutInfo.getFullName()).thenReturn("fullname");
when(aboutInfo.getLicenseProductName()).thenReturn("productname");
when(aboutInfo.getVersion()).thenReturn("version");
when(aboutInfo.getLicenseProductVersion()).thenReturn("productversion");
HardwareSummary hardwareSummary = Mockito.mock(HardwareSummary.class);
when(summary.getHardware()).thenReturn(hardwareSummary);
when(hardwareSummary.getModel()).thenReturn("model");
when(hardwareSummary.getVendor()).thenReturn("model");
when(hardwareSummary.getNumCpuCores()).thenReturn((short) 2);
when(hardwareSummary.getNumCpuPkgs()).thenReturn((short) 2);
when(hardwareSummary.getNumCpuThreads()).thenReturn((short) 2);
when(hardwareSummary.getCpuMhz()).thenReturn(2);
when(hardwareSummary.getMemorySize()).thenReturn(2L);
NetworkInfo networkInfo = Mockito.mock(NetworkInfo.class);
ConfigInfo configInfo = Mockito.mock(ConfigInfo.class);
when(host.getConfig()).thenReturn(configInfo);
when(configInfo.getNetwork()).thenReturn(networkInfo);
LinkSpeedDuplex linkSpeedDuplex = Mockito.mock(LinkSpeedDuplex.class);
when(linkSpeedDuplex.isDuplex()).thenReturn(true);
when(linkSpeedDuplex.getSpeedMb()).thenReturn(2);
PhysicalNic[] physicalNics = new PhysicalNic[2];
physicalNics[0] = Mockito.mock(PhysicalNic.class);
when(physicalNics[0].getMac()).thenReturn("mac");
when(physicalNics[0].getDriver()).thenReturn("driver");
when(physicalNics[0].getLinkSpeed()).thenReturn(linkSpeedDuplex);
when(physicalNics[0].getDevice()).thenReturn("device");
physicalNics[1] = Mockito.mock(PhysicalNic.class);
when(physicalNics[1].getMac()).thenReturn("mac1");
when(physicalNics[1].getDriver()).thenReturn("driver1");
when(physicalNics[1].getLinkSpeed()).thenReturn(linkSpeedDuplex);
when(physicalNics[1].getDevice()).thenReturn("device1");
when(networkInfo.getPnic()).thenReturn(physicalNics);
ConnectInfo connectInfo = Mockito.mock(ConnectInfo.class);
DatastoreInfo[] datastores = { Mockito.mock(DatastoreInfo.class), Mockito.mock(DatastoreInfo.class) };
com.vmware.vim.binding.vim.Datastore.Summary connectInfoSummary = Mockito.mock(com.vmware.vim.binding.vim.Datastore.Summary.class);
when(connectInfoSummary.getCapacity()).thenReturn(2L);
when(datastores[0].getSummary()).thenReturn(connectInfoSummary);
ManagedObjectReference mor1 = Mockito.mock(ManagedObjectReference.class);
when(datastores[1].getSummary()).thenReturn(connectInfoSummary);
when(connectInfoSummary.getDatastore()).thenReturn(mor1);
when(mor1.getValue()).thenReturn("datastoreid");
when(host.queryConnectionInfo()).thenReturn(connectInfo);
when(connectInfo.getDatastore()).thenReturn(datastores);
com.vmware.vim.binding.vim.vsan.host.ConfigInfo vsanConfigInfo = Mockito.mock(com.vmware.vim.binding.vim.vsan.host.ConfigInfo.class);
when(configInfo.getVsanHostConfig()).thenReturn(vsanConfigInfo);
when(vsanConfigInfo.getEnabled()).thenReturn(true);
when(capability.getVsanSupported()).thenReturn(true);
Collection<ClusterComputeResource> clusterComputeResources = new ArrayList<>();
ClusterComputeResource cluster = Mockito.mock(ClusterComputeResource.class);
clusterComputeResources.add(cluster);
clusterComputeResources.add(cluster);
when(vsphereClient.getAllClusterComputeResource()).thenReturn(clusterComputeResources);
ManagedObjectReference[] mors = new ManagedObjectReference[2];
mors[0] = Mockito.mock(ManagedObjectReference.class);
mors[1] = Mockito.mock(ManagedObjectReference.class);
when(mors[0].getValue()).thenReturn("hostMobId");
when(mors[1].getValue()).thenReturn("hostMobId");
when(cluster.getHost()).thenReturn(mors);
ConfigInfoEx ci = Mockito.mock(ConfigInfoEx.class);
when(cluster.getConfigurationEx()).thenReturn(ci);
when(cluster.getName()).thenReturn("cluster");
DpmConfigInfo dpmConfigInfo = Mockito.mock(DpmConfigInfo.class);
when(dpmConfigInfo.getEnabled()).thenReturn(true);
when(ci.getDpmConfigInfo()).thenReturn(dpmConfigInfo);
DrsConfigInfo drsConfigInfo = Mockito.mock(DrsConfigInfo.class);
when(drsConfigInfo.getDefaultVmBehavior()).thenReturn(DrsBehavior.fullyAutomated);
when(ci.getDrsConfig()).thenReturn(drsConfigInfo);
com.vmware.vim.binding.vim.ComputeResource.Summary computeResourceSummary = Mockito.mock(com.vmware.vim.binding.vim.ComputeResource.Summary.class);
when(computeResourceSummary.getNumEffectiveHosts()).thenReturn(2);
when(computeResourceSummary.getNumHosts()).thenReturn(2);
when(computeResourceSummary.getTotalCpu()).thenReturn(2);
when(computeResourceSummary.getNumCpuCores()).thenReturn((short) 2);
when(computeResourceSummary.getNumCpuThreads()).thenReturn((short) 2);
when(computeResourceSummary.getTotalMemory()).thenReturn(2L);
when(cluster.getSummary()).thenReturn(computeResourceSummary);
com.vmware.vim.binding.vim.vsan.cluster.ConfigInfo clusterConfigInfo = Mockito.mock(com.vmware.vim.binding.vim.vsan.cluster.ConfigInfo.class);
when(ci.getVsanConfigInfo()).thenReturn(clusterConfigInfo);
when(clusterConfigInfo.getEnabled()).thenReturn(true);
DpmHostConfigInfo[] dpmHostConfigInfos = new DpmHostConfigInfo[2];
dpmHostConfigInfos[0] = Mockito.mock(DpmHostConfigInfo.class);
dpmHostConfigInfos[1] = Mockito.mock(DpmHostConfigInfo.class);
ManagedObjectReference mor2 = Mockito.mock(ManagedObjectReference.class);
when(dpmHostConfigInfos[0].getKey()).thenReturn(mor2);
when(mor2.getValue()).thenReturn("hostMobId");
when(dpmHostConfigInfos[1].getKey()).thenReturn(mor2);
when(mor2.getValue()).thenReturn("hostMobId");
when(ci.getDpmHostConfig()).thenReturn(dpmHostConfigInfos);
ManagedObjectReference mor3 = Mockito.mock(ManagedObjectReference.class);
when(cluster._getRef()).thenReturn(mor3);
when(mor3.getValue()).thenReturn("clusterMobid");
ManagedObjectReference mor4 = Mockito.mock(ManagedObjectReference.class);
when(cluster.getParent()).thenReturn(mor4);
when(mor4.getValue()).thenReturn("InstanceId");
when(host.getName()).thenReturn("hostname");
HostInfo hostInfo = new HostInfo();
List<HostNic> hostNics = new ArrayList<HostNic>();
HostNic hostNic1 = new HostNic();
hostNic1.setDriver("driver1");
hostNic1.setMacAddress("mac1");
hostNic1.setDuplex(true);
hostNic1.setLinkSpeedMb(2);
hostNic1.setName("nic1");
HostNic hostNic2 = new HostNic();
hostNic2.setDriver("driver2");
hostNic2.setMacAddress("mac2");
hostNic2.setDuplex(true);
hostNic2.setLinkSpeedMb(2);
hostNic2.setName("nic2");
hostNics.add(hostNic1);
hostNics.add(hostNic2);
hostInfo.setHostNics(hostNics);
boolean needUpdate = service.feedHostMetaData(host, hostInfo);
TestCase.assertEquals(needUpdate, true);
}
Aggregations