Search in sources :

Example 6 with Bundle

use of org.killbill.billing.client.model.gen.Bundle in project killbill by killbill.

the class TestBundle method assertSubscriptionState.

private void assertSubscriptionState(final Bundles bundles, final UUID bundleId, final EntitlementState expectedState) {
    final Bundle bundle = Iterables.tryFind(bundles, new Predicate<Bundle>() {

        @Override
        public boolean apply(final Bundle input) {
            return input.getBundleId().equals(bundleId);
        }
    }).orNull();
    assertNotNull(bundle);
    assertEquals(bundle.getSubscriptions().get(0).getState(), expectedState);
}
Also used : Bundle(org.killbill.billing.client.model.gen.Bundle) Predicate(com.google.common.base.Predicate)

Example 7 with Bundle

use of org.killbill.billing.client.model.gen.Bundle in project killbill by killbill.

the class TestBundle method testBundleFromAccount.

@Test(groups = "slow", description = "Can retrieve account bundles")
public void testBundleFromAccount() throws Exception {
    final Account accountJson = createAccount();
    createSubscription(accountJson.getAccountId(), "156567", "Shotgun", ProductCategory.BASE, BillingPeriod.MONTHLY);
    createSubscription(accountJson.getAccountId(), "265658", "Shotgun", ProductCategory.BASE, BillingPeriod.MONTHLY);
    final List<Bundle> objFromJson = accountApi.getAccountBundles(accountJson.getAccountId(), null, null, requestOptions);
    Assert.assertEquals(objFromJson.size(), 2);
}
Also used : Account(org.killbill.billing.client.model.gen.Account) Bundle(org.killbill.billing.client.model.gen.Bundle) Test(org.testng.annotations.Test)

Example 8 with Bundle

use of org.killbill.billing.client.model.gen.Bundle in project killbill by killbill.

the class TestBundle method testBundleOk.

@Test(groups = "slow", description = "Can retrieve bundles by external key")
public void testBundleOk() throws Exception {
    final Account accountJson = createAccount();
    createSubscription(accountJson.getAccountId(), "123467", "Shotgun", ProductCategory.BASE, BillingPeriod.MONTHLY);
    // Retrieves by external key
    final List<Bundle> objFromJson = accountApi.getAccountBundles(accountJson.getAccountId(), "123467", null, requestOptions);
    Assert.assertEquals(objFromJson.size(), 1);
}
Also used : Account(org.killbill.billing.client.model.gen.Account) Bundle(org.killbill.billing.client.model.gen.Bundle) Test(org.testng.annotations.Test)

Example 9 with Bundle

use of org.killbill.billing.client.model.gen.Bundle in project killbill by killbill.

the class TestEntitlement method testCreateEntitlementsWithAddOnsThenCloseAccountWithItemAdjustment.

@Test(groups = "slow", description = "Create a bulk of base entitlement and addOns under the same transaction")
public void testCreateEntitlementsWithAddOnsThenCloseAccountWithItemAdjustment() throws Exception {
    final DateTime initialDate = new DateTime(2012, 4, 25, 0, 3, 42, 0);
    clock.setDeltaFromReality(initialDate.getMillis() - clock.getUTCNow().getMillis());
    final Account accountJson = createAccount();
    final Subscription base = new Subscription();
    base.setAccountId(accountJson.getAccountId());
    base.setProductName("Shotgun");
    base.setProductCategory(ProductCategory.BASE);
    base.setBillingPeriod(BillingPeriod.MONTHLY);
    base.setPriceList(PriceListSet.DEFAULT_PRICELIST_NAME);
    final Subscription addOn1 = new Subscription();
    addOn1.setAccountId(accountJson.getAccountId());
    addOn1.setProductName("Telescopic-Scope");
    addOn1.setProductCategory(ProductCategory.ADD_ON);
    addOn1.setBillingPeriod(BillingPeriod.MONTHLY);
    addOn1.setPriceList(PriceListSet.DEFAULT_PRICELIST_NAME);
    final Subscription addOn2 = new Subscription();
    addOn2.setAccountId(accountJson.getAccountId());
    addOn2.setProductName("Laser-Scope");
    addOn2.setProductCategory(ProductCategory.ADD_ON);
    addOn2.setBillingPeriod(BillingPeriod.MONTHLY);
    addOn2.setPriceList(PriceListSet.DEFAULT_PRICELIST_NAME);
    final List<Subscription> subscriptions = new ArrayList<Subscription>();
    subscriptions.add(base);
    subscriptions.add(addOn1);
    subscriptions.add(addOn2);
    final BulkSubscriptionsBundle bulkList = new BulkSubscriptionsBundle();
    bulkList.setBaseEntitlementAndAddOns(subscriptions);
    final BulkSubscriptionsBundles input = new BulkSubscriptionsBundles();
    input.add(bulkList);
    input.add(bulkList);
    callbackServlet.pushExpectedEvents(ExtBusEventType.ACCOUNT_CHANGE, ExtBusEventType.ENTITLEMENT_CREATION, ExtBusEventType.ENTITLEMENT_CREATION, ExtBusEventType.ENTITLEMENT_CREATION, ExtBusEventType.ENTITLEMENT_CREATION, ExtBusEventType.ENTITLEMENT_CREATION, ExtBusEventType.ENTITLEMENT_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.SUBSCRIPTION_CREATION, ExtBusEventType.INVOICE_CREATION, ExtBusEventType.INVOICE_PAYMENT_FAILED);
    final Bundles bundles = subscriptionApi.createSubscriptionsWithAddOns(input, null, null, false, false, null, true, DEFAULT_WAIT_COMPLETION_TIMEOUT_SEC, NULL_PLUGIN_PROPERTIES, requestOptions);
    callbackServlet.assertListenerStatus();
    assertNotNull(bundles);
    assertEquals(bundles.size(), 2);
    assertFalse(bundles.get(0).getExternalKey().equals(bundles.get(1).getExternalKey()));
    final List<Invoice> invoices = accountApi.getInvoicesForAccount(accountJson.getAccountId(), null, null, null, requestOptions);
    assertEquals(invoices.size(), 1);
    assertEquals(invoices.get(0).getBalance().compareTo(BigDecimal.ZERO), 1);
    assertEquals(invoiceApi.getInvoiceTags(invoices.get(0).getInvoiceId(), requestOptions).size(), 0);
    final Bundles accountBundles = accountApi.getAccountBundles(accountJson.getAccountId(), null, null, requestOptions);
    assertEquals(accountBundles.size(), 2);
    for (final Bundle bundle : accountBundles) {
        for (final Subscription subscription : bundle.getSubscriptions()) {
            assertEquals(subscription.getState(), EntitlementState.ACTIVE);
        }
    }
    clock.addDays(1);
    callbackServlet.assertListenerStatus();
    callbackServlet.pushExpectedEvents(ExtBusEventType.TAG_CREATION, ExtBusEventType.ENTITLEMENT_CANCEL, ExtBusEventType.ENTITLEMENT_CANCEL, ExtBusEventType.ENTITLEMENT_CANCEL, ExtBusEventType.ENTITLEMENT_CANCEL, ExtBusEventType.ENTITLEMENT_CANCEL, ExtBusEventType.ENTITLEMENT_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.SUBSCRIPTION_CANCEL, ExtBusEventType.INVOICE_ADJUSTMENT, ExtBusEventType.INVOICE_ADJUSTMENT, ExtBusEventType.INVOICE_ADJUSTMENT, ExtBusEventType.INVOICE_ADJUSTMENT, ExtBusEventType.BLOCKING_STATE);
    accountApi.closeAccount(accountJson.getAccountId(), true, false, true, true, requestOptions);
    callbackServlet.assertListenerStatus();
    final Bundles accountBundlesAfterClose = accountApi.getAccountBundles(accountJson.getAccountId(), null, null, requestOptions);
    assertEquals(accountBundlesAfterClose.size(), 2);
    for (final Bundle bundle : accountBundlesAfterClose) {
        for (final Subscription subscription : bundle.getSubscriptions()) {
            assertEquals(subscription.getState(), EntitlementState.CANCELLED);
        }
    }
    final List<Invoice> invoicesAfterClose = accountApi.getInvoicesForAccount(accountJson.getAccountId(), null, null, false, false, false, null, AuditLevel.NONE, requestOptions);
    assertEquals(invoicesAfterClose.size(), 1);
    assertEquals(invoicesAfterClose.get(0).getBalance().compareTo(BigDecimal.ZERO), 0);
    assertEquals(invoiceApi.getInvoiceTags(invoicesAfterClose.get(0).getInvoiceId(), requestOptions).size(), 0);
}
Also used : Account(org.killbill.billing.client.model.gen.Account) Invoice(org.killbill.billing.client.model.gen.Invoice) BulkSubscriptionsBundle(org.killbill.billing.client.model.gen.BulkSubscriptionsBundle) Bundle(org.killbill.billing.client.model.gen.Bundle) ArrayList(java.util.ArrayList) BulkSubscriptionsBundle(org.killbill.billing.client.model.gen.BulkSubscriptionsBundle) BulkSubscriptionsBundles(org.killbill.billing.client.model.BulkSubscriptionsBundles) Subscription(org.killbill.billing.client.model.gen.Subscription) DateTime(org.joda.time.DateTime) BulkSubscriptionsBundles(org.killbill.billing.client.model.BulkSubscriptionsBundles) Bundles(org.killbill.billing.client.model.Bundles) Test(org.testng.annotations.Test)

Example 10 with Bundle

use of org.killbill.billing.client.model.gen.Bundle in project killbill by killbill.

the class TestEntitlement method testEntitlementWithAddOnsWithWRITTEN_OFF.

@Test(groups = "slow", description = "Create a base entitlement and also addOns entitlements under the same bundle")
public void testEntitlementWithAddOnsWithWRITTEN_OFF() throws Exception {
    final DateTime initialDate = new DateTime(2012, 4, 25, 0, 3, 42, 0);
    clock.setDeltaFromReality(initialDate.getMillis() - clock.getUTCNow().getMillis());
    final String bundleExternalKey = "bundleKey12346542";
    final Account accountJson = createAccount();
    final Subscription base = new Subscription();
    base.setAccountId(accountJson.getAccountId());
    base.setBundleExternalKey(bundleExternalKey);
    base.setExternalKey("Base");
    base.setProductName("Shotgun");
    base.setProductCategory(ProductCategory.BASE);
    base.setBillingPeriod(BillingPeriod.MONTHLY);
    base.setPriceList(PriceListSet.DEFAULT_PRICELIST_NAME);
    final Subscription addOn1 = new Subscription();
    addOn1.setAccountId(accountJson.getAccountId());
    addOn1.setExternalKey("addOn1");
    addOn1.setProductName("Telescopic-Scope");
    addOn1.setProductCategory(ProductCategory.ADD_ON);
    addOn1.setBillingPeriod(BillingPeriod.MONTHLY);
    addOn1.setPriceList(PriceListSet.DEFAULT_PRICELIST_NAME);
    final Subscription addOn2 = new Subscription();
    addOn2.setAccountId(accountJson.getAccountId());
    addOn2.setExternalKey("addOn2");
    addOn2.setProductName("Laser-Scope");
    addOn2.setProductCategory(ProductCategory.ADD_ON);
    addOn2.setBillingPeriod(BillingPeriod.MONTHLY);
    addOn2.setPriceList(PriceListSet.DEFAULT_PRICELIST_NAME);
    final Subscriptions subscriptions = new Subscriptions();
    subscriptions.add(base);
    subscriptions.add(addOn1);
    subscriptions.add(addOn2);
    final Bundle bundle = subscriptionApi.createSubscriptionWithAddOns(subscriptions, null, null, null, null, null, true, DEFAULT_WAIT_COMPLETION_TIMEOUT_SEC, NULL_PLUGIN_PROPERTIES, requestOptions);
    assertNotNull(bundle);
    assertEquals(bundle.getExternalKey(), bundleExternalKey);
    assertEquals(bundle.getSubscriptions().size(), 3);
    int found = 0;
    for (int i = 0; i < 3; i++) {
        final Subscription cur = bundle.getSubscriptions().get(i);
        assertEquals(cur.getBundleExternalKey(), bundleExternalKey);
        if ("Shotgun".equals(cur.getProductName())) {
            assertEquals(cur.getExternalKey(), "Base");
            found++;
        } else if ("Telescopic-Scope".equals(cur.getProductName())) {
            assertEquals(cur.getExternalKey(), "addOn1");
            found++;
        } else if ("Laser-Scope".equals(cur.getProductName())) {
            assertEquals(cur.getExternalKey(), "addOn2");
            found++;
        }
    }
    assertEquals(found, 3);
    final List<Invoice> invoices = accountApi.getInvoicesForAccount(accountJson.getAccountId(), null, null, false, false, false, null, AuditLevel.NONE, requestOptions);
    assertEquals(invoices.size(), 1);
    assertEquals(invoices.get(0).getBalance().compareTo(BigDecimal.ZERO), 1);
    assertEquals(invoiceApi.getInvoiceTags(invoices.get(0).getInvoiceId(), requestOptions).size(), 0);
    final Bundles accountBundles = accountApi.getAccountBundles(accountJson.getAccountId(), null, null, requestOptions);
    assertEquals(accountBundles.size(), 1);
    for (final Subscription subscription : accountBundles.get(0).getSubscriptions()) {
        assertEquals(subscription.getState(), EntitlementState.ACTIVE);
    }
    accountApi.closeAccount(accountJson.getAccountId(), true, true, false, true, requestOptions);
    final Bundles accountBundlesAfterClose = accountApi.getAccountBundles(accountJson.getAccountId(), null, null, requestOptions);
    assertEquals(accountBundlesAfterClose.size(), 1);
    for (final Subscription subscription : accountBundlesAfterClose.get(0).getSubscriptions()) {
        assertEquals(subscription.getState(), EntitlementState.CANCELLED);
    }
    final List<Invoice> invoicesAfterClose = accountApi.getInvoicesForAccount(accountJson.getAccountId(), null, null, null, requestOptions);
    assertEquals(invoicesAfterClose.size(), 1);
    assertEquals(invoicesAfterClose.get(0).getBalance().compareTo(BigDecimal.ZERO), 0);
    assertEquals(invoiceApi.getInvoiceTags(invoicesAfterClose.get(0).getInvoiceId(), requestOptions).size(), 1);
}
Also used : Account(org.killbill.billing.client.model.gen.Account) Invoice(org.killbill.billing.client.model.gen.Invoice) Subscriptions(org.killbill.billing.client.model.Subscriptions) BulkSubscriptionsBundle(org.killbill.billing.client.model.gen.BulkSubscriptionsBundle) Bundle(org.killbill.billing.client.model.gen.Bundle) Subscription(org.killbill.billing.client.model.gen.Subscription) DateTime(org.joda.time.DateTime) BulkSubscriptionsBundles(org.killbill.billing.client.model.BulkSubscriptionsBundles) Bundles(org.killbill.billing.client.model.Bundles) Test(org.testng.annotations.Test)

Aggregations

Bundle (org.killbill.billing.client.model.gen.Bundle)10 Account (org.killbill.billing.client.model.gen.Account)9 Test (org.testng.annotations.Test)9 Subscription (org.killbill.billing.client.model.gen.Subscription)6 Bundles (org.killbill.billing.client.model.Bundles)5 DateTime (org.joda.time.DateTime)4 Predicate (com.google.common.base.Predicate)3 Subscriptions (org.killbill.billing.client.model.Subscriptions)3 UUID (java.util.UUID)2 BillingPeriod (org.killbill.billing.catalog.api.BillingPeriod)2 BulkSubscriptionsBundles (org.killbill.billing.client.model.BulkSubscriptionsBundles)2 BulkSubscriptionsBundle (org.killbill.billing.client.model.gen.BulkSubscriptionsBundle)2 Invoice (org.killbill.billing.client.model.gen.Invoice)2 SubscriptionUsageRecord (org.killbill.billing.client.model.gen.SubscriptionUsageRecord)2 UnitUsageRecord (org.killbill.billing.client.model.gen.UnitUsageRecord)2 UsageRecord (org.killbill.billing.client.model.gen.UsageRecord)2 ArrayList (java.util.ArrayList)1 KillBillClientException (org.killbill.billing.client.KillBillClientException)1 BlockingStates (org.killbill.billing.client.model.BlockingStates)1 Invoices (org.killbill.billing.client.model.Invoices)1