@Test
public final void shouldReturnWorkPlan() throws Exception {
// when
Entity resultEntity = workPlanService.getWorkPlan(1L);
// then
Assert.assertSame(workPlanDD, resultEntity.getDataDefinition());
Assert.assertEquals(workPlan.getId(), resultEntity.getId());
}
@Test
public final void shouldReturnOrdersById() throws Exception {
// given
Entity order1 = mock(Entity.class);
when(order1.getId()).thenReturn(1L);
Entity order2 = mock(Entity.class);
when(order2.getId()).thenReturn(2L);
Entity order3 = mock(Entity.class);
when(order3.getId()).thenReturn(3L);
@SuppressWarnings("unchecked")
Iterator<Long> iterator = mock(Iterator.class);
when(iterator.hasNext()).thenReturn(true, true, true, true, false);
when(iterator.next()).thenReturn(1L, 2L, 3L, 4L);
@SuppressWarnings("unchecked")
Set<Long> selectedOrderIds = mock(Set.class);
when(selectedOrderIds.iterator()).thenReturn(iterator);
when(selectedOrderIds.size()).thenReturn(4);
SearchCriteriaBuilder criteria = mock(SearchCriteriaBuilder.class);
when(criteria.add(Mockito.any(SearchCriterion.class))).thenReturn(criteria);
SearchResult result = mock(SearchResult.class);
when(criteria.list()).thenReturn(result);
when(result.getTotalNumberOfEntities()).thenReturn(3);
when(result.getEntities()).thenReturn(Lists.newArrayList(order1, order2, order3));
DataDefinition orderDD = mock(DataDefinition.class);
when(orderDD.find()).thenReturn(criteria);
when(dataDefinitionService.get(OrdersConstants.PLUGIN_IDENTIFIER, OrdersConstants.MODEL_ORDER))
.thenReturn(orderDD);
// when
List<Entity> resultList = workPlanService.getSelectedOrders(selectedOrderIds);
// then
Assert.assertEquals(3, resultList.size());
Assert.assertNotNull(resultList.get(0));
Assert.assertSame(1L, resultList.get(0).getId());
Assert.assertNotNull(resultList.get(1));
Assert.assertSame(2L, resultList.get(1).getId());
Assert.assertNotNull(resultList.get(2));
Assert.assertSame(3L, resultList.get(2).getId());
}
@Test
public final void shouldReturnEmptyListIfGivenIdsSetIsEmpty() throws Exception {
// given
@SuppressWarnings("unchecked")
Iterator<Long> iterator = mock(Iterator.class);
when(iterator.hasNext()).thenReturn(false);
@SuppressWarnings("unchecked")
Set<Long> selectedOrderIds = mock(Set.class);
when(selectedOrderIds.iterator()).thenReturn(iterator);
when(selectedOrderIds.size()).thenReturn(0);
when(selectedOrderIds.isEmpty()).thenReturn(true);
// when
List<Entity> resultList = workPlanService.getSelectedOrders(selectedOrderIds);
// then
Assert.assertEquals(0, resultList.size());
}
@Test
public final void shouldGenerateWorkPlanEntity() throws Exception {
// given
Entity emptyWorkPlan = mock(Entity.class);
when(workPlanDD.create()).thenReturn(emptyWorkPlan);
when(workPlanDD.save(emptyWorkPlan)).thenReturn(emptyWorkPlan);
when(emptyWorkPlan.getDataDefinition()).thenReturn(workPlanDD);
Entity order = mock(Entity.class);
@SuppressWarnings("unchecked")
Iterator<Entity> iterator = mock(Iterator.class);
when(iterator.hasNext()).thenReturn(true, true, true, false);
when(iterator.next()).thenReturn(order);
@SuppressWarnings("unchecked")
List<Entity> orders = mock(List.class);
when(orders.iterator()).thenReturn(iterator);
when(orders.size()).thenReturn(3);
when(orders.get(Mockito.anyInt())).thenReturn(order);
@SuppressWarnings("rawtypes")
ArgumentCaptor<List> listArgCaptor = ArgumentCaptor.forClass(List.class);
ArgumentCaptor<String> stringArgCaptor = ArgumentCaptor.forClass(String.class);
// when
workPlanService.generateWorkPlanEntity(orders);
// then
verify(emptyWorkPlan, times(1)).setField(Mockito.eq("orders"), listArgCaptor.capture());
@SuppressWarnings("unchecked")
List<Entity> resultOrders = listArgCaptor.getValue();
Assert.assertEquals(orders.size(), resultOrders.size());
Assert.assertSame(order, resultOrders.get(0));
Assert.assertSame(order, resultOrders.get(1));
Assert.assertSame(order, resultOrders.get(2));
verify(emptyWorkPlan, times(1)).setField(Mockito.eq("name"), stringArgCaptor.capture());
Assert.assertEquals(TRANSLATED_STRING, stringArgCaptor.getValue());
verify(emptyWorkPlan, times(1)).setField(Mockito.eq("type"), stringArgCaptor.capture());
Assert.assertEquals(WorkPlanType.NO_DISTINCTION.getStringValue(), stringArgCaptor.getValue());
}