killbill-memoizeit

junction: fix testInsertBlockingEvents The mock object

8/27/2012 4:38:16 PM

Details

diff --git a/junction/src/test/java/com/ning/billing/junction/plumbing/billing/TestBlockingCalculator.java b/junction/src/test/java/com/ning/billing/junction/plumbing/billing/TestBlockingCalculator.java
index 466900d..dc535de 100644
--- a/junction/src/test/java/com/ning/billing/junction/plumbing/billing/TestBlockingCalculator.java
+++ b/junction/src/test/java/com/ning/billing/junction/plumbing/billing/TestBlockingCalculator.java
@@ -122,16 +122,14 @@ public class TestBlockingCalculator extends JunctionTestSuite {
     // S2 --B--[-------]--------------------------
     // S3 ------------------D---------------------
     @Test(groups = "fast")
-    public void testInsertBlockingEvents() {
+    public void testInsertBlockingEventsForBundle() {
         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));
         final BillingEvent A = createRealEvent(now.minusDays(1).minusHours(1), subscription1);
         final BillingEvent B = createRealEvent(now.minusDays(1), subscription2);
         final BillingEvent C = createRealEvent(now.plusDays(1), subscription2);
         final BillingEvent D = createRealEvent(now.plusDays(3), subscription3);
+        final SortedSet<BillingEvent> billingEvents = new TreeSet<BillingEvent>();
         billingEvents.add(A);
         billingEvents.add(B);
         billingEvents.add(C);
@@ -141,7 +139,7 @@ public class TestBlockingCalculator extends JunctionTestSuite {
         blockingStates.add(new DefaultBlockingState(bundleId1, DISABLED_BUNDLE, Blockable.Type.SUBSCRIPTION_BUNDLE, "test", true, true, true, now));
         blockingStates.add(new DefaultBlockingState(bundleId1, CLEAR_BUNDLE, Blockable.Type.SUBSCRIPTION_BUNDLE, "test", false, false, false, now.plusDays(2)));
 
-        Mockito.when(blockingApi.getBlockingHistory(Mockito.<Blockable>anyObject())).thenReturn(blockingStates);
+        Mockito.when(blockingApi.getBlockingHistory(bundleId1)).thenReturn(blockingStates);
 
         odc.insertBlockingEvents(billingEvents);