@Before
public void testSetUp() {
ComponentContext.initComponentsLifeCycle();
PlannerHostReservationVO reservationVO =
new PlannerHostReservationVO(200L, 1L, 2L, 3L, PlannerResourceUsage.Shared);
Mockito.when(_plannerHostReserveDao.persist(Matchers.any(PlannerHostReservationVO.class)))
.thenReturn(reservationVO);
Mockito.when(_plannerHostReserveDao.findById(Matchers.anyLong())).thenReturn(reservationVO);
Mockito.when(_affinityGroupVMMapDao.countAffinityGroupsForVm(Matchers.anyLong()))
.thenReturn(0L);
VMInstanceVO vm = new VMInstanceVO();
Mockito.when(vmProfile.getVirtualMachine()).thenReturn(vm);
Mockito.when(_dcDao.findById(Matchers.anyLong())).thenReturn(dc);
Mockito.when(dc.getId()).thenReturn(dataCenterId);
ClusterVO clusterVO = new ClusterVO();
clusterVO.setHypervisorType(HypervisorType.XenServer.toString());
Mockito.when(_clusterDao.findById(Matchers.anyLong())).thenReturn(clusterVO);
Mockito.when(_planner.getName()).thenReturn("FirstFitPlanner");
List<DeploymentPlanner> planners = new ArrayList<DeploymentPlanner>();
planners.add(_planner);
_dpm.setPlanners(planners);
}
@Override
protected void injectMockito() {
List<HostVO> hosts = new ArrayList<HostVO>();
hosts.add(this.host);
Mockito.when(
resourceMgr.listAllUpAndEnabledHosts(
(Type) Matchers.any(), Matchers.anyLong(), Matchers.anyLong(), Matchers.anyLong()))
.thenReturn(hosts);
RemoteHostEndPoint ep =
RemoteHostEndPoint.getHypervisorHostEndPoint(
this.host.getId(), this.host.getPrivateIpAddress(), this.host.getPublicIpAddress());
Mockito.when(epSelector.select(Matchers.any(DataObject.class), Matchers.any(DataObject.class)))
.thenReturn(ep);
Mockito.when(epSelector.select(Matchers.any(DataObject.class))).thenReturn(ep);
Mockito.when(epSelector.select(Matchers.any(DataStore.class))).thenReturn(ep);
}
@Override
protected void injectMockito() {
if (host == null) {
return;
}
List<HostVO> results = new ArrayList<HostVO>();
results.add(host);
Mockito.when(hostDao.listAll()).thenReturn(results);
Mockito.when(hostDao.findById(Matchers.anyLong())).thenReturn(host);
Mockito.when(hostDao.findHypervisorHostInCluster(Matchers.anyLong())).thenReturn(results);
List<EndPoint> eps = new ArrayList<EndPoint>();
eps.add(
RemoteHostEndPoint.getHypervisorHostEndPoint(
host.getId(), host.getPrivateIpAddress(), host.getPublicIpAddress()));
Mockito.when(selector.selectAll(Matchers.any(DataStore.class))).thenReturn(eps);
Mockito.when(selector.select(Matchers.any(DataObject.class))).thenReturn(eps.get(0));
Mockito.when(selector.select(Matchers.any(DataObject.class), Matchers.any(DataObject.class)))
.thenReturn(eps.get(0));
}
@Bean
public EntityManager entityManager() {
EntityManager mock = Mockito.mock(EntityManager.class);
try {
Mockito.when(mock.findById(Matchers.same(Account.class), Matchers.anyLong()))
.thenReturn(_accountDao.findById(Account.ACCOUNT_ID_SYSTEM));
Mockito.when(mock.findById(Matchers.same(User.class), Matchers.anyLong()))
.thenReturn(_userDao.findById(User.UID_SYSTEM));
Mockito.when(mock.findById(Matchers.same(NetworkOffering.class), Matchers.any(Long.class)))
.thenAnswer(
new Answer<NetworkOffering>() {
@Override
public NetworkOffering answer(final InvocationOnMock invocation) throws Throwable {
Long id = (Long) invocation.getArguments()[1];
return _networkOfferingDao.findById(id);
}
});
Mockito.when(mock.findById(Matchers.same(IpAddress.class), Matchers.any(Long.class)))
.thenAnswer(
new Answer<IpAddress>() {
@Override
public IpAddress answer(final InvocationOnMock invocation) throws Throwable {
Long id = (Long) invocation.getArguments()[1];
return _ipAddressDao.findById(id);
}
});
Mockito.when(mock.findById(Matchers.same(DataCenter.class), Matchers.any(Long.class)))
.thenAnswer(
new Answer<DataCenter>() {
@Override
public DataCenter answer(final InvocationOnMock invocation) throws Throwable {
Long id = (Long) invocation.getArguments()[1];
return _zoneDao.findById(id);
}
});
} catch (Exception e) {
e.printStackTrace();
}
CallContext.init(mock);
return mock;
}
protected void setUpDDMTemplateLocalServiceUtil() {
mockStatic(DDMTemplateLocalServiceUtil.class);
when(getTemplate(Matchers.anyLong()))
.then(
new Answer<DDMTemplate>() {
@Override
public DDMTemplate answer(InvocationOnMock invocationOnMock) throws Throwable {
Object[] args = invocationOnMock.getArguments();
Long templateId = (Long) args[0];
return _templates.get(templateId);
}
});
}
@SuppressWarnings("serial")
@Test
public void test() throws Exception {
_.setMapper(new ObjectMapper());
final ExecutorService es = Mockito.mock(ExecutorService.class);
Mockito.when(es.awaitTermination(Matchers.anyLong(), Matchers.any(TimeUnit.class))).thenReturn(true);
_.setExecutorService(es);
// _.setReceiptHandler(injector.getInstance(ReceiptHandler.class));
final ReceiptHandler rh = Mockito.mock(ReceiptHandler.class);
Mockito.when(rh.handle(Matchers.any(Body.class))).thenReturn(new Receipt() {
{
setExtracted_store_names(new LinkedList<GuessResult>() {
{
add(new GuessResult() {
{
name = "ALDI";
distance = 0.5d;
}
});
}
});
setId("999");
}
});
_.setReceiptHandler(rh);
final MessageQueue rq = Mockito.mock(MessageQueue.class);
final MessageQueue rs = Mockito.mock(MessageQueue.class);
final Message m1 = Mockito.mock(Message.class);
Mockito.when(m1.getMessageBody()).thenReturn("{\"receipt\":{\"url\":\"my_url\",\"id\":\"999\"}}");
Mockito.when(rq.receiveMessage()).thenReturn(m1, (Message) null);
_.setRequestQueue(rq);
_.setResponseQueue(rs);
final Message msg = _.popOrWait();
_.handleRequestMessage(msg);
Assert.assertEquals(m1, msg);
Mockito.verify(rq).deleteMessage(m1);
Mockito.verify(rs)
.sendMessage(
"{\"receipt\":{\"id\":\"999\",\"extracted_store_names\":[{\"name\":\"ALDI\",\"distance\":0.5,\"numOfErrors\":0,\"termLength\":0,\"serachTerm\":null,\"foundTerm\":[]}],\"extracted_tokens_list\":[]}}");
}