/** * Given a set of billing events, add corresponding blocking (overdue) billing events. * * @param billingEvents the original list of billing events to update (without overdue events) */ public void insertBlockingEvents(final SortedSet<BillingEvent> billingEvents, final InternalTenantContext context) { if (billingEvents.size() <= 0) { return; } final Account account = billingEvents.first().getAccount(); final Hashtable<UUID, List<SubscriptionBase>> bundleMap = createBundleSubscriptionMap(billingEvents); final SortedSet<BillingEvent> billingEventsToAdd = new TreeSet<BillingEvent>(); final SortedSet<BillingEvent> billingEventsToRemove = new TreeSet<BillingEvent>(); final List<BlockingState> blockingEvents = blockingApi.getBlockingAllForAccount(context); final List<DisabledDuration> blockingDurations = createBlockingDurations(blockingEvents); for (final UUID bundleId : bundleMap.keySet()) { for (final SubscriptionBase subscription : bundleMap.get(bundleId)) { billingEventsToAdd.addAll(createNewEvents(blockingDurations, billingEvents, account, subscription)); billingEventsToRemove.addAll(eventsToRemove(blockingDurations, billingEvents, subscription)); } } for (final BillingEvent eventToAdd : billingEventsToAdd) { billingEvents.add(eventToAdd); } for (final BillingEvent eventToRemove : billingEventsToRemove) { billingEvents.remove(eventToRemove); } }
@Test(groups = "fast") public void testCreateNewEventsClosedFollow() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, now.plusDays(2))); billingEvents.add(createRealEvent(now.plusDays(3), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 0); }
@Test(groups = "fast") public void testCreateNewEventsOpenFollow() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, null)); billingEvents.add(createRealEvent(now.plusDays(1), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 0); }
@Test(groups = "fast") public void testCreateNewEventsClosedBetweenFollow() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, now.plusDays(2))); billingEvents.add(createRealEvent(now.plusDays(1), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 1); assertEquals(results.last().getEffectiveDate(), now.plusDays(2)); assertEquals(results.last().getRecurringPrice(), billingEvents.first().getRecurringPrice()); assertEquals(results.last().getTransitionType(), SubscriptionBaseTransitionType.END_BILLING_DISABLED); }
@Test(groups = "fast") public void testCreateNewEventsClosedBetwn() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, now.plusDays(2))); billingEvents.add(createRealEvent(now.plusDays(1), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 1); assertEquals(results.last().getEffectiveDate(), now.plusDays(2)); assertEquals(results.last().getRecurringPrice(), billingEvents.first().getRecurringPrice()); assertEquals(results.last().getTransitionType(), SubscriptionBaseTransitionType.END_BILLING_DISABLED); }
@Test(groups = "fast") public void testCreateNewEventsOpenPrev() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, null)); billingEvents.add(createRealEvent(now.minusDays(1), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 1); assertEquals(results.first().getEffectiveDate(), now); assertNull(results.first().getFixedPrice()); assertNull(results.first().getRecurringPrice()); assertEquals(results.first().getBillingPeriod(), BillingPeriod.NO_BILLING_PERIOD); assertEquals(results.first().getTransitionType(), SubscriptionBaseTransitionType.START_BILLING_DISABLED); }
@Test(groups = "fast") public void testCreateNewEventsOpenPrevFollow() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, null)); billingEvents.add(createRealEvent(now.minusDays(1), subscription1)); billingEvents.add(createRealEvent(now.plusDays(1), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 1); assertEquals(results.first().getEffectiveDate(), now); assertNull(results.first().getFixedPrice()); assertNull(results.first().getRecurringPrice()); assertEquals(results.first().getBillingPeriod(), BillingPeriod.NO_BILLING_PERIOD); assertEquals(results.first().getTransitionType(), SubscriptionBaseTransitionType.START_BILLING_DISABLED); }
@Test(groups = "fast") public void testCreateNewEventsClosedPrevBetw() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, now.plusDays(2))); billingEvents.add(createRealEvent(now.minusDays(1), subscription1)); billingEvents.add(createRealEvent(now.plusDays(1), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 2); assertEquals(results.first().getEffectiveDate(), now); assertNull(results.first().getFixedPrice()); assertNull(results.first().getRecurringPrice()); assertEquals(results.first().getBillingPeriod(), BillingPeriod.NO_BILLING_PERIOD); assertEquals(results.first().getTransitionType(), SubscriptionBaseTransitionType.START_BILLING_DISABLED); assertEquals(results.last().getEffectiveDate(), now.plusDays(2)); assertEquals(results.last().getRecurringPrice(), billingEvents.first().getRecurringPrice()); assertEquals(results.last().getTransitionType(), SubscriptionBaseTransitionType.END_BILLING_DISABLED); }
@Test(groups = "fast") public void testCreateNewEventsClosedPrev() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, now.plusDays(2))); billingEvents.add(createRealEvent(now.minusDays(1), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 2); assertEquals(results.first().getEffectiveDate(), now); assertNull(results.first().getFixedPrice()); assertNull(results.first().getRecurringPrice()); assertEquals(results.first().getBillingPeriod(), BillingPeriod.NO_BILLING_PERIOD); assertEquals(results.first().getTransitionType(), SubscriptionBaseTransitionType.START_BILLING_DISABLED); assertEquals(results.last().getEffectiveDate(), now.plusDays(2)); assertEquals(results.last().getRecurringPrice(), billingEvents.first().getRecurringPrice()); assertEquals(results.last().getTransitionType(), SubscriptionBaseTransitionType.END_BILLING_DISABLED); }
@Test(groups = "fast") public void testCreateNewEventsClosedPrevBetwNext() { final DateTime now = clock.getUTCNow(); final List<DisabledDuration> disabledDuration = new ArrayList<BlockingCalculator.DisabledDuration>(); final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>(); disabledDuration.add(new DisabledDuration(now, now.plusDays(2))); billingEvents.add(createRealEvent(now.minusDays(1), subscription1)); billingEvents.add(createRealEvent(now.plusDays(1), subscription1)); billingEvents.add(createRealEvent(now.plusDays(3), subscription1)); final SortedSet<BillingEvent> results = blockingCalculator.createNewEvents(disabledDuration, billingEvents, account, subscription1); assertEquals(results.size(), 2); assertEquals(results.first().getEffectiveDate(), now); assertNull(results.first().getFixedPrice()); assertNull(results.first().getRecurringPrice()); assertEquals(results.first().getBillingPeriod(), BillingPeriod.NO_BILLING_PERIOD); assertEquals(results.first().getTransitionType(), SubscriptionBaseTransitionType.START_BILLING_DISABLED); assertEquals(results.last().getEffectiveDate(), now.plusDays(2)); assertEquals(results.last().getRecurringPrice(), billingEvents.first().getRecurringPrice()); assertEquals(results.last().getTransitionType(), SubscriptionBaseTransitionType.END_BILLING_DISABLED); }