GroupedDetailImportContextTest.java 文件源码

@Test
public void mergesDetailsForCategory() throws Exception {
    final ByteArrayInputStream inputStream = new ByteArrayInputStream("".getBytes());
    final TransactionCategory category = new TransactionCategory(CATEGORY_CODE);
    when(importFile.parse(inputStream)).thenReturn(Arrays.asList(
            ImmutableListMultimap.of(dateField, "01/15/1990", amountField, "10", categoryField, CATEGORY_CODE, sharesField, "2"),
            ImmutableListMultimap.of(dateField, "01/15/1990", amountField, "1", categoryField, CATEGORY_CODE, sharesField, "20")));
    ImportContext context = new GroupedDetailImportContext(importFile, payeeMapper, securityMapper, categoryMapper);
    when(categoryMapper.get(CATEGORY_CODE)).thenReturn(category);

    List<Transaction> transactions = context.parseTransactions(inputStream);

    assertThat(transactions).hasSize(1);
    assertThat(transactions.get(0).getDetails()).hasSize(1);
    assertThat(transactions.get(0).getDetails().get(0).getCategory()).isSameAs(category);
    assertThat(transactions.get(0).getDetails().get(0).getAmount().toString()).isEqualTo("11");
    assertThat(transactions.get(0).getDetails().get(0).getAssetQuantity().toString()).isEqualTo("-22");
}