Examples with DefaultInvoice org.killbill.billing.invoice.model.DefaultInvoice used on opensource projects

Example 41 with DefaultInvoice

use of org.killbill.billing.invoice.model.DefaultInvoice in project killbill by killbill.

the class TestInvoiceDao method testAccountBalanceWithRefundAndCBAInternal.

private void testAccountBalanceWithRefundAndCBAInternal(final boolean withAdjustment, final BigDecimal refundAmount, final BigDecimal expectedFinalBalance) throws InvoiceApiException, EntityPersistenceException {
    final UUID accountId = account.getId();
    final UUID bundleId = UUID.randomUUID();
    final LocalDate targetDate1 = new LocalDate(2011, 10, 6);
    final Invoice invoice1 = new DefaultInvoice(accountId, clock.getUTCToday(), targetDate1, Currency.USD);
    invoiceUtil.createInvoice(invoice1, context);
    final LocalDate startDate = new LocalDate(2011, 3, 1);
    final LocalDate endDate = startDate.plusMonths(1);
    final BigDecimal amount1 = new BigDecimal("5.0");
    final BigDecimal rate1 = new BigDecimal("20.0");
    final BigDecimal rate2 = new BigDecimal("10.0");
    // Fixed Item
    final FixedPriceInvoiceItem item1 = new FixedPriceInvoiceItem(invoice1.getId(), accountId, bundleId, UUID.randomUUID(), "test product", "test plan", "test phase A", null, startDate, amount1, Currency.USD);
    invoiceUtil.createInvoiceItem(item1, context);
    BigDecimal balance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(balance.compareTo(new BigDecimal("5.00")), 0);
    // Recurring item
    final RecurringInvoiceItem item2 = new RecurringInvoiceItem(invoice1.getId(), accountId, bundleId, UUID.randomUUID(), "test product", "test plan", "test phase B", null, startDate, endDate, rate1, rate1, Currency.USD);
    invoiceUtil.createInvoiceItem(item2, context);
    balance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(balance.compareTo(new BigDecimal("25.00")), 0);
    // Pay the whole thing
    final UUID paymentId = UUID.randomUUID();
    final BigDecimal payment1 = amount1.add(rate1);
    final InvoicePayment payment = new DefaultInvoicePayment(InvoicePaymentType.ATTEMPT, paymentId, invoice1.getId(), new DateTime(), payment1, Currency.USD, Currency.USD, null, true);
    invoiceUtil.createPayment(payment, context);
    balance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(balance.compareTo(new BigDecimal("0.00")), 0);
    // Repair previous item with rate 2
    final RepairAdjInvoiceItem item2Repair = new RepairAdjInvoiceItem(invoice1.getId(), accountId, startDate, endDate, rate1.negate(), Currency.USD, item2.getId());
    final RecurringInvoiceItem item2Replace = new RecurringInvoiceItem(invoice1.getId(), accountId, bundleId, UUID.randomUUID(), "test product", "test plan", "test phase B", null, startDate, endDate, rate2, rate2, Currency.USD);
    invoiceUtil.createInvoiceItem(item2Repair, context);
    invoiceUtil.createInvoiceItem(item2Replace, context);
    balance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(balance.compareTo(new BigDecimal("-10.00")), 0);
    // CBA
    final CreditBalanceAdjInvoiceItem cbaItem = new CreditBalanceAdjInvoiceItem(invoice1.getId(), accountId, new LocalDate(), balance.negate(), Currency.USD);
    invoiceUtil.createInvoiceItem(cbaItem, context);
    balance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(balance.compareTo(new BigDecimal("-10.00")), 0);
    BigDecimal cba = invoiceDao.getAccountCBA(accountId, context);
    assertEquals(cba.compareTo(new BigDecimal("10.00")), 0);
    // PARTIAL REFUND on the payment
    final Map<UUID, BigDecimal> invoiceItemIdsWithAmounts = new HashMap<UUID, BigDecimal>();
    if (withAdjustment) {
        invoiceItemIdsWithAmounts.put(item2Replace.getId(), refundAmount);
    }
    invoiceDao.createRefund(paymentId, UUID.randomUUID(), refundAmount, withAdjustment, invoiceItemIdsWithAmounts, UUID.randomUUID().toString(), true, context);
    balance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(balance.compareTo(expectedFinalBalance), 0);
    cba = invoiceDao.getAccountCBA(accountId, context);
    final BigDecimal expectedCba = balance.compareTo(BigDecimal.ZERO) < 0 ? balance.negate() : BigDecimal.ZERO;
    assertEquals(cba.compareTo(expectedCba), 0);
}

Example 42 with DefaultInvoice

use of org.killbill.billing.invoice.model.DefaultInvoice in project killbill by killbill.

the class TestInvoiceDao method testInvoiceCreditInternal.

private void testInvoiceCreditInternal(final BigDecimal creditAmount, final BigDecimal expectedBalance, final boolean expectCBA) throws EntityPersistenceException, InvoiceApiException {
    final UUID accountId = account.getId();
    final UUID bundleId = UUID.randomUUID();
    // Create one invoice with a fixed invoice item
    final LocalDate targetDate = new LocalDate(2011, 2, 15);
    final Invoice invoice1 = new DefaultInvoice(accountId, clock.getUTCToday(), targetDate, Currency.USD);
    invoiceUtil.createInvoice(invoice1, context);
    final LocalDate startDate = new LocalDate(2011, 3, 1);
    final BigDecimal amount1 = new BigDecimal("5.0");
    // Fixed Item
    final FixedPriceInvoiceItem item1 = new FixedPriceInvoiceItem(invoice1.getId(), accountId, bundleId, UUID.randomUUID(), "test product", "test plan", "test phase A", null, startDate, amount1, Currency.USD);
    invoiceUtil.createInvoiceItem(item1, context);
    // Create the credit item
    final LocalDate effectiveDate = new LocalDate(2011, 3, 1);
    createCredit(accountId, invoice1.getId(), effectiveDate, creditAmount, false);
    final List<InvoiceModelDao> invoices = invoiceDao.getAllInvoicesByAccount(false, context);
    assertEquals(invoices.size(), 1);
    final InvoiceModelDao invoice = invoices.get(0);
    assertTrue(InvoiceModelDaoHelper.getRawBalanceForRegularInvoice(invoice).compareTo(expectedBalance) == 0);
    final List<InvoiceItemModelDao> invoiceItems = invoice.getInvoiceItems();
    assertEquals(invoiceItems.size(), expectCBA ? 3 : 2);
    boolean foundCredit = false;
    boolean foundCBA = false;
    for (final InvoiceItemModelDao cur : invoiceItems) {
        if (cur.getType() == InvoiceItemType.CREDIT_ADJ) {
            foundCredit = true;
            assertTrue(cur.getAmount().compareTo(creditAmount.negate()) == 0);
        } else if (cur.getType() == InvoiceItemType.CBA_ADJ) {
            foundCBA = true;
        }
    }
    assertEquals(foundCBA, expectCBA);
    assertTrue(foundCredit);
}

Example 43 with DefaultInvoice

use of org.killbill.billing.invoice.model.DefaultInvoice in project killbill by killbill.

the class TestInvoiceDao method testRefundWithRepairAndInvoiceItemAdjustmentInternal.

private void testRefundWithRepairAndInvoiceItemAdjustmentInternal(final BigDecimal refundAmount) throws InvoiceApiException, EntityPersistenceException {
    final UUID accountId = account.getId();
    final UUID bundleId = UUID.randomUUID();
    final LocalDate targetDate1 = new LocalDate(2011, 10, 6);
    final Invoice invoice = new DefaultInvoice(accountId, clock.getUTCToday(), targetDate1, Currency.USD);
    invoiceUtil.createInvoice(invoice, context);
    final LocalDate startDate = new LocalDate(2011, 3, 1);
    final LocalDate endDate = startDate.plusMonths(1);
    final BigDecimal amount = new BigDecimal("20.0");
    // Recurring item
    final RecurringInvoiceItem item2 = new RecurringInvoiceItem(invoice.getId(), accountId, bundleId, UUID.randomUUID(), "test product", "test plan", "test phase B", null, startDate, endDate, amount, amount, Currency.USD);
    invoiceUtil.createInvoiceItem(item2, context);
    BigDecimal accountBalance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(accountBalance.compareTo(new BigDecimal("20.00")), 0);
    // Pay the whole thing
    final UUID paymentId = UUID.randomUUID();
    final BigDecimal payment1 = amount;
    final InvoicePayment payment = new DefaultInvoicePayment(InvoicePaymentType.ATTEMPT, paymentId, invoice.getId(), new DateTime(), payment1, Currency.USD, Currency.USD, null, true);
    invoiceUtil.createPayment(payment, context);
    accountBalance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(accountBalance.compareTo(new BigDecimal("0.00")), 0);
    // Repair the item (And add CBA item that should be generated)
    final InvoiceItem repairItem = new RepairAdjInvoiceItem(invoice.getId(), accountId, startDate, endDate, amount.negate(), Currency.USD, item2.getId());
    invoiceUtil.createInvoiceItem(repairItem, context);
    final InvoiceItem cbaItem = new CreditBalanceAdjInvoiceItem(invoice.getId(), accountId, startDate, amount, Currency.USD);
    invoiceUtil.createInvoiceItem(cbaItem, context);
    final Map<UUID, BigDecimal> itemAdjustment = new HashMap<UUID, BigDecimal>();
    // PAss a null value to let invoice calculate the amount to adjust
    itemAdjustment.put(item2.getId(), null);
    invoiceDao.createRefund(paymentId, UUID.randomUUID(), refundAmount, true, itemAdjustment, UUID.randomUUID().toString(), true, context);
    accountBalance = invoiceDao.getAccountBalance(accountId, context);
    final boolean partialRefund = refundAmount.compareTo(amount) < 0;
    final BigDecimal cba = invoiceDao.getAccountCBA(accountId, context);
    final InvoiceModelDao savedInvoice = invoiceDao.getById(invoice.getId(), context);
    final BigDecimal expectedCba = accountBalance.compareTo(BigDecimal.ZERO) < 0 ? accountBalance.negate() : BigDecimal.ZERO;
    assertEquals(cba.compareTo(expectedCba), 0);
    // Let's re-calculate them from invoice
    final BigDecimal balanceAfterRefund = invoiceDao.getAccountBalance(accountId, context);
    final BigDecimal cbaAfterRefund = invoiceDao.getAccountCBA(accountId, context);
    if (partialRefund) {
        // IB = 20 (rec) - 20 (repair) + 20 (cba) - (20 -7) = 7;  AB = IB - CBA = 7 - 20 = -13
        assertEquals(accountBalance.compareTo(new BigDecimal("-13.0")), 0);
        assertEquals(savedInvoice.getInvoiceItems().size(), 4);
        assertEquals(balanceAfterRefund.compareTo(new BigDecimal("-13.0")), 0);
        assertEquals(cbaAfterRefund.compareTo(expectedCba), 0);
    } else {
        assertEquals(accountBalance.compareTo(new BigDecimal("0.0")), 0);
        assertEquals(savedInvoice.getInvoiceItems().size(), 4);
        assertEquals(balanceAfterRefund.compareTo(BigDecimal.ZERO), 0);
        assertEquals(cbaAfterRefund.compareTo(expectedCba), 0);
    }
}

Example 44 with DefaultInvoice

use of org.killbill.billing.invoice.model.DefaultInvoice in project killbill by killbill.

the class TestInvoiceDao method testGetInvoicesBySubscriptionForFixedItems.

@Test(groups = "slow")
public void testGetInvoicesBySubscriptionForFixedItems() throws EntityPersistenceException {
    final UUID accountId = account.getId();
    final UUID bundleId = UUID.randomUUID();
    final UUID subscriptionId1 = UUID.randomUUID();
    final BigDecimal rate1 = new BigDecimal("17.0");
    final UUID subscriptionId2 = UUID.randomUUID();
    final BigDecimal rate2 = new BigDecimal("42.0");
    final UUID subscriptionId3 = UUID.randomUUID();
    final BigDecimal rate3 = new BigDecimal("3.0");
    final UUID subscriptionId4 = UUID.randomUUID();
    final BigDecimal rate4 = new BigDecimal("12.0");
    final LocalDate targetDate = new LocalDate(2011, 5, 23);
    // Create invoice 1 (subscriptions 1-4)
    final Invoice invoice1 = new DefaultInvoice(accountId, clock.getUTCToday(), targetDate, Currency.USD);
    invoiceUtil.createInvoice(invoice1, context);
    final UUID invoiceId1 = invoice1.getId();
    LocalDate startDate = new LocalDate(2011, 3, 1);
    LocalDate endDate = startDate.plusMonths(1);
    final FixedPriceInvoiceItem item1 = new FixedPriceInvoiceItem(invoiceId1, accountId, bundleId, subscriptionId1, "test product", "test plan", "test A", null, startDate, rate1, Currency.USD);
    invoiceUtil.createInvoiceItem(item1, context);
    final FixedPriceInvoiceItem item2 = new FixedPriceInvoiceItem(invoiceId1, accountId, bundleId, subscriptionId2, "test product", "test plan", "test B", null, startDate, rate2, Currency.USD);
    invoiceUtil.createInvoiceItem(item2, context);
    final FixedPriceInvoiceItem item3 = new FixedPriceInvoiceItem(invoiceId1, accountId, bundleId, subscriptionId3, "test product", "test plan", "test C", null, startDate, rate3, Currency.USD);
    invoiceUtil.createInvoiceItem(item3, context);
    final FixedPriceInvoiceItem item4 = new FixedPriceInvoiceItem(invoiceId1, accountId, bundleId, subscriptionId4, "test product", "test plan", "test D", null, startDate, rate4, Currency.USD);
    invoiceUtil.createInvoiceItem(item4, context);
    // create invoice 2 (subscriptions 1-3)
    final DefaultInvoice invoice2 = new DefaultInvoice(accountId, clock.getUTCToday(), targetDate, Currency.USD);
    invoiceUtil.createInvoice(invoice2, context);
    final UUID invoiceId2 = invoice2.getId();
    startDate = endDate;
    final FixedPriceInvoiceItem item5 = new FixedPriceInvoiceItem(invoiceId2, accountId, bundleId, subscriptionId1, "test product", "test plan", "test phase A", null, startDate, rate1, Currency.USD);
    invoiceUtil.createInvoiceItem(item5, context);
    final FixedPriceInvoiceItem item6 = new FixedPriceInvoiceItem(invoiceId2, accountId, bundleId, subscriptionId2, "test product", "test plan", "test phase B", null, startDate, rate2, Currency.USD);
    invoiceUtil.createInvoiceItem(item6, context);
    final FixedPriceInvoiceItem item7 = new FixedPriceInvoiceItem(invoiceId2, accountId, bundleId, subscriptionId3, "test product", "test plan", "test phase C", null, startDate, rate3, Currency.USD);
    invoiceUtil.createInvoiceItem(item7, context);
    // check that each subscription returns the correct number of invoices
    final List<InvoiceModelDao> items1 = invoiceDao.getInvoicesBySubscription(subscriptionId1, context);
    assertEquals(items1.size(), 2);
    final List<InvoiceModelDao> items2 = invoiceDao.getInvoicesBySubscription(subscriptionId2, context);
    assertEquals(items2.size(), 2);
    final List<InvoiceModelDao> items3 = invoiceDao.getInvoicesBySubscription(subscriptionId3, context);
    assertEquals(items3.size(), 2);
    final List<InvoiceModelDao> items4 = invoiceDao.getInvoicesBySubscription(subscriptionId4, context);
    assertEquals(items4.size(), 1);
}

Example 45 with DefaultInvoice

use of org.killbill.billing.invoice.model.DefaultInvoice in project killbill by killbill.

the class TestInvoiceDao method testAccountBalanceWithNoPayments.

@Test(groups = "slow")
public void testAccountBalanceWithNoPayments() throws EntityPersistenceException {
    final UUID accountId = account.getId();
    final UUID bundleId = UUID.randomUUID();
    final LocalDate targetDate1 = new LocalDate(2011, 10, 6);
    final Invoice invoice1 = new DefaultInvoice(accountId, clock.getUTCToday(), targetDate1, Currency.USD);
    invoiceUtil.createInvoice(invoice1, context);
    final LocalDate startDate = new LocalDate(2011, 3, 1);
    final LocalDate endDate = startDate.plusMonths(1);
    final BigDecimal rate1 = new BigDecimal("17.0");
    final BigDecimal rate2 = new BigDecimal("42.0");
    final RecurringInvoiceItem item1 = new RecurringInvoiceItem(invoice1.getId(), accountId, bundleId, UUID.randomUUID(), "test product", "test plan", "test phase A", null, startDate, endDate, rate1, rate1, Currency.USD);
    invoiceUtil.createInvoiceItem(item1, context);
    final RecurringInvoiceItem item2 = new RecurringInvoiceItem(invoice1.getId(), accountId, bundleId, UUID.randomUUID(), "test product", "test plan", "test phase B", null, startDate, endDate, rate2, rate2, Currency.USD);
    invoiceUtil.createInvoiceItem(item2, context);
    final BigDecimal balance = invoiceDao.getAccountBalance(accountId, context);
    assertEquals(balance.compareTo(rate1.add(rate2)), 0);
}