SingleDetailImportContextTest.java 文件源码

@Test
public void excludesZeroAssetQuantity() throws Exception {
    final ByteArrayInputStream inputStream = new ByteArrayInputStream("".getBytes());
    final TransactionCategory category = new TransactionCategory();
    when(importFile.parse(inputStream)).thenReturn(singletonList(ImmutableListMultimap.of(amountField, "10", categoryField, CATEGORY_CODE, sharesField, "0.0")));
    ImportContext context = new SingleDetailImportContext(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().compareTo(BigDecimal.TEN)).isEqualTo(0);
    assertThat(transactions.get(0).getDetails().get(0).getAssetQuantity()).isNull();
}