Examples with Catalog org.killbill.billing.catalog.api.Catalog used on opensource projects

Example 6 with Catalog

use of org.killbill.billing.catalog.api.Catalog in project killbill by killbill.

the class MockCatalogModule method configure.

@Override
protected void configure() {
    final Catalog catalog = Mockito.mock(Catalog.class);
    final CatalogService catalogService = Mockito.mock(CatalogService.class);
    try {
        Mockito.when(catalogService.getCurrentCatalog(Mockito.any(Boolean.class), Mockito.any(Boolean.class), Mockito.any(InternalCallContext.class))).thenReturn(new MockCatalog());
        Mockito.when(catalogService.getFullCatalog(Mockito.any(Boolean.class), Mockito.any(Boolean.class), Mockito.any(InternalCallContext.class))).thenReturn(catalog);
        bind(CatalogService.class).toInstance(catalogService);
    } catch (CatalogApiException e) {
        throw new RuntimeException(e);
    }
}

Example 7 with Catalog

use of org.killbill.billing.catalog.api.Catalog in project killbill by killbill.

the class JunctionTestSuiteWithEmbeddedDB method initCatalog.

private Catalog initCatalog(final CatalogService catalogService) throws Exception {
    ((DefaultCatalogService) catalogService).loadCatalog();
    final Catalog catalog = catalogService.getFullCatalog(true, true, internalCallContext);
    assertNotNull(catalog);
    return catalog;
}

Example 8 with Catalog

use of org.killbill.billing.catalog.api.Catalog in project killbill by killbill.

the class DefaultInternalBillingApi method addBillingEventsForSubscription.

private void addBillingEventsForSubscription(final ImmutableAccountData account, final List<SubscriptionBase> subscriptions, final SubscriptionBase baseSubscription, final boolean dryRunMode, final InternalCallContext context, final DefaultBillingEventSet result, final Set<UUID> skipSubscriptionsSet) throws AccountApiException, CatalogApiException, SubscriptionBaseApiException {
    // If dryRun is specified, we don't want to to update the account BCD value, so we initialize the flag updatedAccountBCD to true
    boolean updatedAccountBCD = dryRunMode;
    final Map<UUID, Integer> bcdCache = new HashMap<UUID, Integer>();
    int currentAccountBCD = accountApi.getBCD(account.getId(), context);
    for (final SubscriptionBase subscription : subscriptions) {
        // The subscription did not even start, so there is nothing to do yet, we can skip and avoid some NPE down the line when calculating the BCD
        if (subscription.getState() == EntitlementState.PENDING) {
            log.info("Skipping subscription id='{}', state = EntitlementState.PENDING", subscription.getId());
            continue;
        }
        final List<EffectiveSubscriptionInternalEvent> billingTransitions = subscriptionApi.getBillingTransitions(subscription, context);
        if (billingTransitions.isEmpty() || (billingTransitions.get(0).getTransitionType() != SubscriptionBaseTransitionType.CREATE && billingTransitions.get(0).getTransitionType() != SubscriptionBaseTransitionType.TRANSFER)) {
            log.warn("Skipping billing events for subscription " + subscription.getId() + ": Does not start with a valid CREATE transition");
            skipSubscriptionsSet.add(subscription.getId());
            return;
        }
        final Catalog catalog = catalogService.getFullCatalog(true, true, context);
        Integer overridenBCD = null;
        for (final EffectiveSubscriptionInternalEvent transition : billingTransitions) {
            //
            // A BCD_CHANGE transition defines a new billCycleDayLocal for the subscription and this overrides whatever computation
            // occurs below (which is based on billing alignment policy). Also multiple of those BCD_CHANGE transitions could occur,
            // to define different intervals with different billing cycle days.
            //
            overridenBCD = transition.getNextBillCycleDayLocal() != null ? transition.getNextBillCycleDayLocal() : overridenBCD;
            final int bcdLocal = overridenBCD != null ? overridenBCD : calculateBcdForTransition(catalog, bcdCache, baseSubscription, subscription, account, currentAccountBCD, transition);
            if (currentAccountBCD == 0 && !updatedAccountBCD) {
                log.info("Setting account BCD='{}', accountId='{}'", bcdLocal, account.getId());
                accountApi.updateBCD(account.getExternalKey(), bcdLocal, context);
                updatedAccountBCD = true;
            }
            final BillingEvent event = new DefaultBillingEvent(account, transition, subscription, bcdLocal, account.getCurrency(), catalog);
            result.add(event);
        }
    }
}

Example 9 with Catalog

use of org.killbill.billing.catalog.api.Catalog in project killbill by killbill.

the class PlanAligner method getTimedPhaseOnCreate.

private List<TimedPhase> getTimedPhaseOnCreate(final DateTime subscriptionStartDate, final DateTime bundleStartDate, final Plan plan, @Nullable final PhaseType initialPhase, final DateTime effectiveDate, final InternalTenantContext context) throws CatalogApiException, SubscriptionBaseApiException {
    final Catalog catalog = catalogService.getFullCatalog(true, true, context);
    final PlanSpecifier planSpecifier = new PlanSpecifier(plan.getName());
    final DateTime planStartDate;
    final PlanAlignmentCreate alignment = catalog.planCreateAlignment(planSpecifier, effectiveDate);
    switch(alignment) {
        case START_OF_SUBSCRIPTION:
            planStartDate = subscriptionStartDate;
            break;
        case START_OF_BUNDLE:
            planStartDate = bundleStartDate;
            break;
        default:
            throw new SubscriptionBaseError(String.format("Unknown PlanAlignmentCreate %s", alignment));
    }
    return getPhaseAlignments(plan, initialPhase, planStartDate);
}

Example 10 with Catalog

use of org.killbill.billing.catalog.api.Catalog in project killbill by killbill.

the class PlanAligner method getTimedPhaseOnChange.

private TimedPhase getTimedPhaseOnChange(final DateTime subscriptionStartDate, final DateTime bundleStartDate, final PlanPhase currentPhase, final Plan currentPlan, final Plan nextPlan, final DateTime effectiveDate, final DateTime lastOrCurrentChangeEffectiveDate, final PhaseType originalInitialPhase, @Nullable final PhaseType newPlanInitialPhaseType, final WhichPhase which, final InternalTenantContext context) throws CatalogApiException, SubscriptionBaseApiException {
    final Catalog catalog = catalogService.getFullCatalog(true, true, context);
    final PlanPhaseSpecifier fromPlanPhaseSpecifier = new PlanPhaseSpecifier(currentPlan.getName(), currentPhase.getPhaseType());
    final PlanSpecifier toPlanSpecifier = new PlanSpecifier(nextPlan.getName());
    final PhaseType initialPhase;
    final DateTime planStartDate;
    final PlanAlignmentChange alignment = catalog.planChangeAlignment(fromPlanPhaseSpecifier, toPlanSpecifier, effectiveDate);
    switch(alignment) {
        case START_OF_SUBSCRIPTION:
            planStartDate = subscriptionStartDate;
            initialPhase = newPlanInitialPhaseType != null ? newPlanInitialPhaseType : (isPlanContainPhaseType(nextPlan, originalInitialPhase) ? originalInitialPhase : null);
            break;
        case START_OF_BUNDLE:
            planStartDate = bundleStartDate;
            initialPhase = newPlanInitialPhaseType != null ? newPlanInitialPhaseType : (isPlanContainPhaseType(nextPlan, originalInitialPhase) ? originalInitialPhase : null);
            break;
        case CHANGE_OF_PLAN:
            planStartDate = lastOrCurrentChangeEffectiveDate;
            initialPhase = newPlanInitialPhaseType;
            break;
        case CHANGE_OF_PRICELIST:
            throw new SubscriptionBaseError(String.format("Not implemented yet %s", alignment));
        default:
            throw new SubscriptionBaseError(String.format("Unknown PlanAlignmentChange %s", alignment));
    }
    final List<TimedPhase> timedPhases = getPhaseAlignments(nextPlan, initialPhase, planStartDate);
    return getTimedPhase(timedPhases, effectiveDate, which);
}