killbill-memoizeit

entitlement: fix TestUserApiAddOn Signed-off-by: Pierre-Alexandre

7/17/2013 10:23:03 PM

Details

diff --git a/entitlement/src/test/java/com/ning/billing/entitlement/api/user/TestUserApiAddOn.java b/entitlement/src/test/java/com/ning/billing/entitlement/api/user/TestUserApiAddOn.java
index a2b6ed1..e850a15 100644
--- a/entitlement/src/test/java/com/ning/billing/entitlement/api/user/TestUserApiAddOn.java
+++ b/entitlement/src/test/java/com/ning/billing/entitlement/api/user/TestUserApiAddOn.java
@@ -63,11 +63,11 @@ public class TestUserApiAddOn extends EntitlementTestSuiteWithEmbeddedDB {
             SubscriptionData aoSubscription = testUtil.createSubscription(bundle, aoProduct, aoTerm, aoPriceList);
             assertEquals(aoSubscription.getState(), SubscriptionState.ACTIVE);
 
+            testListener.pushExpectedEvent(NextEvent.CANCEL);
+
             final DateTime now = clock.getUTCNow();
             aoSubscription.cancel(now, callContext);
 
-            testListener.reset();
-            testListener.pushExpectedEvent(NextEvent.CANCEL);
             assertTrue(testListener.isCompleted(5000));
             aoSubscription = (SubscriptionData) entitlementApi.getSubscriptionFromId(aoSubscription.getId(), callContext);
             assertEquals(aoSubscription.getState(), SubscriptionState.CANCELLED);