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

Example 1 with UsagePriceOverride

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

the class DefaultPriceOverride method getResolvedUsageOverrides.

public List<UsagePriceOverride> getResolvedUsageOverrides(Usage[] usages, List<UsagePriceOverride> usagePriceOverrides) throws CatalogApiException {
    List<UsagePriceOverride> resolvedUsageOverrides = new ArrayList<UsagePriceOverride>();
    for (final Usage curUsage : usages) {
        final UsagePriceOverride curOverride = Iterables.tryFind(usagePriceOverrides, new Predicate<UsagePriceOverride>() {

            @Override
            public boolean apply(final UsagePriceOverride input) {
                return input.getName() != null && input.getName().equals(curUsage.getName());
            }
        }).orNull();
        if (curOverride != null) {
            List<TierPriceOverride> tierPriceOverrides = getResolvedTierOverrides(curUsage.getTiers(), curOverride.getTierPriceOverrides());
            resolvedUsageOverrides.add(new DefaultUsagePriceOverride(curUsage.getName(), curUsage.getUsageType(), tierPriceOverrides));
        } else {
            resolvedUsageOverrides.add(null);
        }
    }
    return resolvedUsageOverrides;
}

Example 2 with UsagePriceOverride

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

the class TestDefaultPriceOverride method testOverrideOneOutOfTwoTieredBlockPrices.

@Test(groups = "slow")
public void testOverrideOneOutOfTwoTieredBlockPrices() throws Exception {
    final StandaloneCatalog catalog = getCatalog("UsageExperimental.xml");
    final Plan plan = catalog.findPlan("chocolate-monthly");
    final List<PlanPhasePriceOverride> overrides = new ArrayList<PlanPhasePriceOverride>();
    final List<TieredBlockPriceOverride> tieredBlockPriceOverrides = new ArrayList<TieredBlockPriceOverride>();
    tieredBlockPriceOverrides.add(new DefaultTieredBlockPriceOverride("chocolate-videos", new Double("1"), new BigDecimal("0.75"), Currency.USD, new Double("10000")));
    final List<TierPriceOverride> tierPriceOverrides = new ArrayList<TierPriceOverride>();
    tierPriceOverrides.add(new DefaultTierPriceOverride(tieredBlockPriceOverrides));
    final List<UsagePriceOverride> usagePriceOverrides = new ArrayList<UsagePriceOverride>();
    usagePriceOverrides.add(new DefaultUsagePriceOverride("chocolate-monthly-videos", UsageType.CONSUMABLE, tierPriceOverrides));
    final PlanPhasePriceOverride phase = new DefaultPlanPhasePriceOverride(plan.getFinalPhase().getName(), Currency.USD, null, null, usagePriceOverrides);
    overrides.add(phase);
    // Overriding only the tieredblockprice for unit - 'chocolate-videos' with size = 1 and max = 10000 from $1 to $0.75
    final DefaultPlan overriddenPlan = priceOverride.getOrCreateOverriddenPlan(catalog, plan, new DateTime(catalog.getEffectiveDate()), overrides, internalCallContext);
    final String[] parts = priceOverridePattern.getPlanParts(overriddenPlan.getName());
    assertEquals(parts[0], plan.getName());
    assertEquals(overriddenPlan.getProduct().getName(), plan.getProduct().getName());
    assertEquals(overriddenPlan.getRecurringBillingPeriod(), plan.getRecurringBillingPeriod());
    if (plan.getEffectiveDateForExistingSubscriptions() != null) {
        assertEquals(overriddenPlan.getEffectiveDateForExistingSubscriptions().compareTo(plan.getEffectiveDateForExistingSubscriptions()), 0);
    }
    assertNotEquals(overriddenPlan.getFinalPhase().getName(), plan.getFinalPhase().getName());
    for (int i = 0; i < overriddenPlan.getFinalPhase().getUsages().length; i++) {
        final DefaultUsage initialUsage = (DefaultUsage) plan.getFinalPhase().getUsages()[i];
        final DefaultUsage newUsage = (DefaultUsage) overriddenPlan.getFinalPhase().getUsages()[i];
        assertEquals(newUsage.getName(), initialUsage.getName());
        assertEquals(newUsage.getUsageType(), initialUsage.getUsageType());
        assertEquals(newUsage.getBillingPeriod(), initialUsage.getBillingPeriod());
        assertEquals(newUsage.getBillingPeriod(), initialUsage.getBillingPeriod());
        assertEquals(newUsage.getTiers().length, initialUsage.getTiers().length);
        for (int j = 0; j < newUsage.getTiers().length; j++) {
            final DefaultTier initialTier = (DefaultTier) initialUsage.getTiers()[j];
            final DefaultTier newTier = (DefaultTier) newUsage.getTiers()[j];
            assertEquals(newTier.getTieredBlocks().length, initialTier.getTieredBlocks().length);
            for (int k = 0; k < newTier.getTieredBlocks().length; k++) {
                final DefaultTieredBlock initialTieredBlock = (DefaultTieredBlock) initialTier.getTieredBlocks()[k];
                final DefaultTieredBlock newTieredBlock = (DefaultTieredBlock) newTier.getTieredBlocks()[k];
                final TieredBlockPriceOverride override = Iterables.tryFind(tieredBlockPriceOverrides, new Predicate<TieredBlockPriceOverride>() {

                    @Override
                    public boolean apply(final TieredBlockPriceOverride input) {
                        return input.getUnitName().equals(initialTieredBlock.getUnit().getName()) && Double.compare(input.getSize(), initialTieredBlock.getSize()) == 0 && Double.compare(input.getMax(), initialTieredBlock.getMax()) == 0;
                    }
                }).orNull();
                assertEquals(newTieredBlock.getUnit().getName(), initialTieredBlock.getUnit().getName());
                assertEquals(newTieredBlock.getMax(), initialTieredBlock.getMax());
                assertEquals(newTieredBlock.getSize(), initialTieredBlock.getSize());
                assertTieredBlockInternationalPrice(newTieredBlock.getPrice(), initialTieredBlock.getPrice(), override);
            }
        }
    }
}

Example 3 with UsagePriceOverride

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

the class TestDefaultPriceOverride method testOverrideTwoOutOfTwoTieredBlockPrices.

@Test(groups = "slow")
public void testOverrideTwoOutOfTwoTieredBlockPrices() throws Exception {
    final StandaloneCatalog catalog = getCatalog("UsageExperimental.xml");
    final Plan plan = catalog.findPlan("chocolate-monthly");
    final List<PlanPhasePriceOverride> overrides = new ArrayList<PlanPhasePriceOverride>();
    final List<TieredBlockPriceOverride> tieredBlockPriceOverrides1 = new ArrayList<TieredBlockPriceOverride>();
    tieredBlockPriceOverrides1.add(new DefaultTieredBlockPriceOverride("chocolate-videos", new Double("1"), new BigDecimal("1.5"), Currency.USD, new Double("5")));
    final List<TieredBlockPriceOverride> tieredBlockPriceOverrides2 = new ArrayList<TieredBlockPriceOverride>();
    tieredBlockPriceOverrides2.add(new DefaultTieredBlockPriceOverride("chocolate-videos", new Double("1"), new BigDecimal("0.75"), Currency.USD, new Double("10000")));
    final List<TierPriceOverride> tierPriceOverrides = new ArrayList<TierPriceOverride>();
    tierPriceOverrides.add(new DefaultTierPriceOverride(tieredBlockPriceOverrides1));
    tierPriceOverrides.add(new DefaultTierPriceOverride(tieredBlockPriceOverrides2));
    final List<UsagePriceOverride> usagePriceOverrides = new ArrayList<UsagePriceOverride>();
    usagePriceOverrides.add(new DefaultUsagePriceOverride("chocolate-monthly-videos", UsageType.CONSUMABLE, tierPriceOverrides));
    final PlanPhasePriceOverride phase = new DefaultPlanPhasePriceOverride(plan.getFinalPhase().getName(), Currency.USD, null, new BigDecimal("35"), usagePriceOverrides);
    overrides.add(phase);
    /* Overriding phase recurring price from $30 to $35, tieredblockprice from $2 to $1.5 for unit- 'chocolate-videos' with size = 1 and max = 5 and
         also overriding tieredblockprice from $1 to $0.75 for unit - 'chocolate-videos' with size = 1 and max = 10000 */
    final DefaultPlan overriddenPlan = priceOverride.getOrCreateOverriddenPlan(catalog, plan, new DateTime(catalog.getEffectiveDate()), overrides, internalCallContext);
    final String[] parts = priceOverridePattern.getPlanParts(overriddenPlan.getName());
    assertEquals(parts[0], plan.getName());
    assertEquals(overriddenPlan.getProduct().getName(), plan.getProduct().getName());
    assertEquals(overriddenPlan.getRecurringBillingPeriod(), plan.getRecurringBillingPeriod());
    if (plan.getEffectiveDateForExistingSubscriptions() != null) {
        assertEquals(overriddenPlan.getEffectiveDateForExistingSubscriptions().compareTo(plan.getEffectiveDateForExistingSubscriptions()), 0);
    }
    assertNotEquals(overriddenPlan.getFinalPhase().getName(), plan.getFinalPhase().getName());
    final DefaultPlanPhase initialPhase = (DefaultPlanPhase) plan.getFinalPhase();
    final DefaultPlanPhase newPhase = (DefaultPlanPhase) overriddenPlan.getFinalPhase();
    final PlanPhasePriceOverride override = Iterables.tryFind(overrides, new Predicate<PlanPhasePriceOverride>() {

        @Override
        public boolean apply(final PlanPhasePriceOverride input) {
            return input.getPhaseName().equals(initialPhase.getName());
        }
    }).orNull();
    assertNotEquals(newPhase.getName(), initialPhase.getName());
    assertEquals(newPhase.getName(), overriddenPlan.getName() + "-" + initialPhase.getName().split("-")[initialPhase.getName().split("-").length - 1]);
    assertEquals(newPhase.getDuration(), initialPhase.getDuration());
    assertEquals(newPhase.getPhaseType(), initialPhase.getPhaseType());
    assertEquals(newPhase.getUsages().length, initialPhase.getUsages().length);
    if (initialPhase.getFixed() != null) {
        assertEquals(newPhase.getFixed().getType(), initialPhase.getFixed().getType());
        assertInternationalPrice(newPhase.getFixed().getPrice(), initialPhase.getFixed().getPrice(), override, true);
    }
    if (initialPhase.getRecurring() != null) {
        assertInternationalPrice(newPhase.getRecurring().getRecurringPrice(), initialPhase.getRecurring().getRecurringPrice(), override, false);
    }
    for (int i = 0; i < overriddenPlan.getFinalPhase().getUsages().length; i++) {
        final DefaultUsage initialUsage = (DefaultUsage) plan.getFinalPhase().getUsages()[i];
        final DefaultUsage newUsage = (DefaultUsage) overriddenPlan.getFinalPhase().getUsages()[i];
        assertEquals(newUsage.getName(), initialUsage.getName());
        assertEquals(newUsage.getUsageType(), initialUsage.getUsageType());
        assertEquals(newUsage.getBillingPeriod(), initialUsage.getBillingPeriod());
        assertEquals(newUsage.getBillingPeriod(), initialUsage.getBillingPeriod());
        assertEquals(newUsage.getTiers().length, initialUsage.getTiers().length);
        for (int j = 0; j < newUsage.getTiers().length; j++) {
            final DefaultTier initialTier = (DefaultTier) initialUsage.getTiers()[j];
            final DefaultTier newTier = (DefaultTier) newUsage.getTiers()[j];
            assertEquals(newTier.getTieredBlocks().length, initialTier.getTieredBlocks().length);
            for (int k = 0; k < newTier.getTieredBlocks().length; k++) {
                final DefaultTieredBlock initialTieredBlock = (DefaultTieredBlock) initialTier.getTieredBlocks()[k];
                final DefaultTieredBlock newTieredBlock = (DefaultTieredBlock) newTier.getTieredBlocks()[k];
                List<TieredBlockPriceOverride> tieredBlockPriceOverrides = new ArrayList<TieredBlockPriceOverride>();
                tieredBlockPriceOverrides.addAll(tieredBlockPriceOverrides1);
                tieredBlockPriceOverrides.addAll(tieredBlockPriceOverrides2);
                final TieredBlockPriceOverride tieredBlockPriceOverride = Iterables.tryFind(tieredBlockPriceOverrides, new Predicate<TieredBlockPriceOverride>() {

                    @Override
                    public boolean apply(final TieredBlockPriceOverride input) {
                        return input.getUnitName().equals(initialTieredBlock.getUnit().getName()) && Double.compare(input.getSize(), initialTieredBlock.getSize()) == 0 && Double.compare(input.getMax(), initialTieredBlock.getMax()) == 0;
                    }
                }).orNull();
                assertEquals(newTieredBlock.getUnit().getName(), initialTieredBlock.getUnit().getName());
                assertEquals(newTieredBlock.getMax(), initialTieredBlock.getMax());
                assertEquals(newTieredBlock.getSize(), initialTieredBlock.getSize());
                assertTieredBlockInternationalPrice(newTieredBlock.getPrice(), initialTieredBlock.getPrice(), tieredBlockPriceOverride);
            }
        }
    }
}

Example 4 with UsagePriceOverride

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

the class DefaultOverriddenPlanCache method getUsagePriceOverrides.

List<UsagePriceOverride> getUsagePriceOverrides(PlanPhase curPhase, CatalogOverridePhaseDefinitionModelDao overriddenPhase, final InternalTenantContext context) {
    final List<UsagePriceOverride> usagePriceOverrides = new ArrayList<UsagePriceOverride>();
    final List<CatalogOverrideUsageDefinitionModelDao> usageDefs = overrideDao.getOverriddenPhaseUsages(overriddenPhase.getRecordId(), context);
    for (int i = 0; i < curPhase.getUsages().length; i++) {
        final Usage curUsage = curPhase.getUsages()[i];
        final CatalogOverrideUsageDefinitionModelDao overriddenUsage = Iterables.tryFind(usageDefs, new Predicate<CatalogOverrideUsageDefinitionModelDao>() {

            @Override
            public boolean apply(final CatalogOverrideUsageDefinitionModelDao input) {
                return input.getParentUsageName().equals(curUsage.getName());
            }
        }).orNull();
        if (overriddenUsage != null) {
            List<TierPriceOverride> tierPriceOverrides = getTierPriceOverrides(curUsage, overriddenUsage, context);
            usagePriceOverrides.add(new DefaultUsagePriceOverride(overriddenUsage.getParentUsageName(), curUsage.getUsageType(), tierPriceOverrides));
        }
    }
    return usagePriceOverrides;
}

Example 5 with UsagePriceOverride

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

the class DefaultOverriddenPlanCache method createOverrides.

private PlanPhasePriceOverride[] createOverrides(final Plan defaultPlan, final List<CatalogOverridePhaseDefinitionModelDao> phaseDefs, final InternalTenantContext context) {
    final PlanPhasePriceOverride[] result = new PlanPhasePriceOverride[defaultPlan.getAllPhases().length];
    for (int i = 0; i < defaultPlan.getAllPhases().length; i++) {
        final PlanPhase curPhase = defaultPlan.getAllPhases()[i];
        final CatalogOverridePhaseDefinitionModelDao overriddenPhase = Iterables.tryFind(phaseDefs, new Predicate<CatalogOverridePhaseDefinitionModelDao>() {

            @Override
            public boolean apply(final CatalogOverridePhaseDefinitionModelDao input) {
                return input.getParentPhaseName().equals(curPhase.getName());
            }
        }).orNull();
        if (overriddenPhase != null) {
            List<UsagePriceOverride> usagePriceOverrides = getUsagePriceOverrides(curPhase, overriddenPhase, context);
            result[i] = new DefaultPlanPhasePriceOverride(curPhase.getName(), Currency.valueOf(overriddenPhase.getCurrency()), overriddenPhase.getFixedPrice(), overriddenPhase.getRecurringPrice(), usagePriceOverrides);
        } else {
            result[i] = null;
        }
    }
    return result;
}