use of org.openlmis.stockmanagement.testutils.ObjectReferenceDtoDataBuilder in project openlmis-stockmanagement by OpenLMIS.
the class StockCardSummariesV2DtoBuilderTest method shouldBuildStockCardSummariesWithMultipleStockCardsForOrderable.
@Test
public void shouldBuildStockCardSummariesWithMultipleStockCardsForOrderable() throws Exception {
List<StockCard> stockCards = asList(stockCard, stockCard1, stockCard2);
LocalDate asOfDate = LocalDate.now();
List<StockCardSummaryV2Dto> result = builder.build(asList(orderable1, orderable2, orderable3), stockCards, fulfillMap, asOfDate);
StockCardSummaryV2Dto summary1 = new StockCardSummaryV2Dto(new ObjectReferenceDtoDataBuilder().withPath(ORDERABLES).withId(orderable1.getId()).build(), asSet(new CanFulfillForMeEntryDtoDataBuilder().buildWithStockCardAndOrderable(stockCard1, orderable3, asOfDate), new CanFulfillForMeEntryDtoDataBuilder().buildWithStockCardAndOrderable(stockCard2, orderable3, asOfDate), new CanFulfillForMeEntryDtoDataBuilder().buildWithStockCardAndOrderable(stockCard, orderable1, asOfDate)));
StockCardSummaryV2Dto summary2 = new StockCardSummaryV2Dto(new ObjectReferenceDtoDataBuilder().withPath(ORDERABLES).withId(orderable2.getId()).build(), asSet(new CanFulfillForMeEntryDtoDataBuilder().buildWithStockCardAndOrderable(stockCard, orderable1, asOfDate)));
StockCardSummaryV2Dto summary3 = new StockCardSummaryV2Dto(new ObjectReferenceDtoDataBuilder().withPath(ORDERABLES).withId(orderable3.getId()).build(), asSet(new CanFulfillForMeEntryDtoDataBuilder().buildWithStockCardAndOrderable(stockCard1, orderable3, asOfDate), new CanFulfillForMeEntryDtoDataBuilder().buildWithStockCardAndOrderable(stockCard2, orderable3, asOfDate)));
assertEquals(3, result.size());
assertThat(result, hasItems(summary1, summary2, summary3));
}
Aggregations