killbill-memoizeit

diff --git a/analytics/src/test/java/com/ning/billing/analytics/MockBusinessSubscriptionTransitionDao.java b/analytics/src/test/java/com/ning/billing/analytics/MockBusinessSubscriptionTransitionDao.java
new file mode 100644
index 0000000..d7e36b4
--- /dev/null
+++ b/analytics/src/test/java/com/ning/billing/analytics/MockBusinessSubscriptionTransitionDao.java
@@ -0,0 +1,52 @@
+/*
+ * Copyright 2010-2011 Ning, Inc.
+ *
+ * Ning licenses this file to you under the Apache License, version 2.0
+ * (the "License"); you may not use this file except in compliance with the
+ * License.  You may obtain a copy of the License at:
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ */
+
+package com.ning.billing.analytics;
+
+import com.ning.billing.analytics.dao.BusinessSubscriptionTransitionBinder;
+import com.ning.billing.analytics.dao.BusinessSubscriptionTransitionDao;
+import org.skife.jdbi.v2.sqlobject.Bind;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public class MockBusinessSubscriptionTransitionDao implements BusinessSubscriptionTransitionDao
+{
+    private final Map<String, List<BusinessSubscriptionTransition>> content = new HashMap<String, List<BusinessSubscriptionTransition>>();
+
+    @Override
+    public List<BusinessSubscriptionTransition> getTransitions(@Bind("event_key") final String key)
+    {
+        return content.get(key);
+    }
+
+    @Override
+    public int createTransition(@BusinessSubscriptionTransitionBinder final BusinessSubscriptionTransition transition)
+    {
+        if (content.get(transition.getKey()) == null) {
+            content.put(transition.getKey(), new ArrayList<BusinessSubscriptionTransition>());
+        }
+        content.get(transition.getKey()).add(transition);
+        return 1;
+    }
+
+    @Override
+    public void test()
+    {
+    }
+}

diff --git a/analytics/src/test/java/com/ning/billing/analytics/MockIAccountUserApi.java b/analytics/src/test/java/com/ning/billing/analytics/MockIAccountUserApi.java
new file mode 100644
index 0000000..64fe29f
--- /dev/null
+++ b/analytics/src/test/java/com/ning/billing/analytics/MockIAccountUserApi.java
@@ -0,0 +1,51 @@
+/*
+ * Copyright 2010-2011 Ning, Inc.
+ *
+ * Ning licenses this file to you under the Apache License, version 2.0
+ * (the "License"); you may not use this file except in compliance with the
+ * License.  You may obtain a copy of the License at:
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ */
+
+package com.ning.billing.analytics;
+
+import com.ning.billing.account.api.IAccount;
+import com.ning.billing.account.api.IAccountData;
+import com.ning.billing.account.api.IAccountUserApi;
+
+import java.util.List;
+import java.util.UUID;
+
+public class MockIAccountUserApi implements IAccountUserApi
+{
+    @Override
+    public IAccount createAccount(final IAccountData data)
+    {
+        throw new UnsupportedOperationException();
+    }
+
+    @Override
+    public IAccount getAccountByKey(final String key)
+    {
+        throw new UnsupportedOperationException();
+    }
+
+    @Override
+    public IAccount getAccountFromId(final UUID uid)
+    {
+        return null;
+    }
+
+    @Override
+    public List<IAccount> getAccounts()
+    {
+        throw new UnsupportedOperationException();
+    }
+}

diff --git a/analytics/src/test/java/com/ning/billing/analytics/MockIEntitlementUserApi.java b/analytics/src/test/java/com/ning/billing/analytics/MockIEntitlementUserApi.java
new file mode 100644
index 0000000..4ed2e8c
--- /dev/null
+++ b/analytics/src/test/java/com/ning/billing/analytics/MockIEntitlementUserApi.java
@@ -0,0 +1,118 @@
+/*
+ * Copyright 2010-2011 Ning, Inc.
+ *
+ * Ning licenses this file to you under the Apache License, version 2.0
+ * (the "License"); you may not use this file except in compliance with the
+ * License.  You may obtain a copy of the License at:
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ */
+
+package com.ning.billing.analytics;
+
+import com.ning.billing.account.api.IAccount;
+import com.ning.billing.catalog.api.BillingPeriod;
+import com.ning.billing.entitlement.api.user.EntitlementUserApiException;
+import com.ning.billing.entitlement.api.user.IEntitlementUserApi;
+import com.ning.billing.entitlement.api.user.ISubscription;
+import com.ning.billing.entitlement.api.user.ISubscriptionBundle;
+import org.joda.time.DateTime;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.UUID;
+
+public class MockIEntitlementUserApi implements IEntitlementUserApi
+{
+    private final Map<UUID, String> subscriptionBundles = new HashMap<UUID, String>();
+
+    public MockIEntitlementUserApi(final UUID bundleUUID, final String key)
+    {
+        subscriptionBundles.put(bundleUUID, key);
+    }
+
+    @Override
+    public ISubscriptionBundle getBundleFromId(final UUID id)
+    {
+        final String key = subscriptionBundles.get(id);
+        if (key == null) {
+            return null;
+        }
+
+        return new ISubscriptionBundle()
+        {
+            @Override
+            public UUID getAccountId()
+            {
+                return UUID.randomUUID();
+            }
+
+            @Override
+            public UUID getId()
+            {
+                return id;
+            }
+
+            @Override
+            public DateTime getStartDate()
+            {
+                throw new UnsupportedOperationException();
+            }
+
+            @Override
+            public String getKey()
+            {
+                return key;
+            }
+
+            @Override
+            public void setPrivate(final String name, final String value)
+            {
+                throw new UnsupportedOperationException();
+            }
+
+            @Override
+            public String getPrivate(final String name)
+            {
+                throw new UnsupportedOperationException();
+            }
+        };
+    }
+
+    @Override
+    public ISubscription getSubscriptionFromId(final UUID id)
+    {
+        throw new UnsupportedOperationException();
+    }
+
+    @Override
+    public List<ISubscriptionBundle> getBundlesForAccount(final UUID accountId)
+    {
+        throw new UnsupportedOperationException();
+    }
+
+    @Override
+    public List<ISubscription> getSubscriptionsForBundle(final UUID bundleId)
+    {
+        throw new UnsupportedOperationException();
+    }
+
+    @Override
+    public ISubscriptionBundle createBundleForAccount(final IAccount account, final String bundleKey) throws EntitlementUserApiException
+    {
+        throw new UnsupportedOperationException();
+    }
+
+    @Override
+    public ISubscription createSubscription(final UUID bundleId, final String productName, final BillingPeriod term, final String planSet, final DateTime requestedDate) throws EntitlementUserApiException
+    {
+        throw new UnsupportedOperationException();
+    }
+}

diff --git a/analytics/src/test/java/com/ning/billing/analytics/TestAnalyticsListener.java b/analytics/src/test/java/com/ning/billing/analytics/TestAnalyticsListener.java
new file mode 100644
index 0000000..35c7a65
--- /dev/null
+++ b/analytics/src/test/java/com/ning/billing/analytics/TestAnalyticsListener.java
@@ -0,0 +1,217 @@
+/*
+ * Copyright 2010-2011 Ning, Inc.
+ *
+ * Ning licenses this file to you under the Apache License, version 2.0
+ * (the "License"); you may not use this file except in compliance with the
+ * License.  You may obtain a copy of the License at:
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+ * License for the specific language governing permissions and limitations
+ * under the License.
+ */
+
+package com.ning.billing.analytics;
+
+import com.ning.billing.catalog.api.IPlan;
+import com.ning.billing.catalog.api.IPlanPhase;
+import com.ning.billing.catalog.api.IProduct;
+import com.ning.billing.catalog.api.PhaseType;
+import com.ning.billing.catalog.api.ProductCategory;
+import com.ning.billing.entitlement.api.user.ISubscription;
+import com.ning.billing.entitlement.api.user.SubscriptionTransition;
+import com.ning.billing.entitlement.events.IEvent;
+import com.ning.billing.entitlement.events.user.ApiEventType;
+import org.joda.time.DateTime;
+import org.joda.time.DateTimeZone;
+import org.testng.Assert;
+import org.testng.annotations.BeforeMethod;
+import org.testng.annotations.Test;
+
+import java.util.UUID;
+
+public class TestAnalyticsListener
+{
+    private static final String KEY = "1234";
+
+    private final MockBusinessSubscriptionTransitionDao dao = new MockBusinessSubscriptionTransitionDao();
+    private final UUID subscriptionId = UUID.randomUUID();
+    private final UUID bundleUUID = UUID.randomUUID();
+    private final IProduct product = new MockProduct("platinium", "subscription", ProductCategory.BASE);
+    private final IPlan plan = new MockPlan("platinum-monthly", product);
+    private final IPlanPhase phase = new MockPhase(PhaseType.EVERGREEN, plan, MockDuration.UNLIMITED(), 25.95);
+    private final String priceList = "something";
+
+    private AnalyticsListener listener;
+
+    @BeforeMethod(alwaysRun = true)
+    public void setUp() throws Exception
+    {
+        listener = new AnalyticsListener(dao, new MockIEntitlementUserApi(bundleUUID, KEY), new MockIAccountUserApi());
+    }
+
+    @Test(groups = "fast")
+    public void testSubscriptionLifecycle() throws Exception
+    {
+        // Create a subscription
+        final DateTime effectiveTransitionTime = new DateTime(DateTimeZone.UTC);
+        final DateTime requestedTransitionTime = new DateTime(DateTimeZone.UTC);
+        final SubscriptionTransition firstTransition = createFirstSubscriptionTransition(requestedTransitionTime, effectiveTransitionTime);
+        final BusinessSubscriptionTransition firstBST = createExpectedFirstBST(requestedTransitionTime, effectiveTransitionTime);
+        listener.handleNotificationChange(firstTransition);
+        Assert.assertEquals(dao.getTransitions(KEY).size(), 1);
+        Assert.assertEquals(dao.getTransitions(KEY).get(0), firstBST);
+
+        // Pause it
+        final DateTime effectivePauseTransitionTime = new DateTime(DateTimeZone.UTC);
+        final DateTime requestedPauseTransitionTime = new DateTime(DateTimeZone.UTC);
+        final SubscriptionTransition pausedSubscriptionTransition = createPauseSubscriptionTransition(effectivePauseTransitionTime, requestedPauseTransitionTime, firstTransition.getNextState());
+        final BusinessSubscriptionTransition pausedBST = createExpectedPausedBST(requestedPauseTransitionTime, effectivePauseTransitionTime, firstBST.getNextSubscription());
+        listener.handleNotificationChange(pausedSubscriptionTransition);
+        Assert.assertEquals(dao.getTransitions(KEY).size(), 2);
+        Assert.assertEquals(dao.getTransitions(KEY).get(1), pausedBST);
+
+        // Un-Pause it
+        final DateTime effectiveResumeTransitionTime = new DateTime(DateTimeZone.UTC);
+        final DateTime requestedResumeTransitionTime = new DateTime(DateTimeZone.UTC);
+        final SubscriptionTransition resumedSubscriptionTransition = createResumeSubscriptionTransition(requestedResumeTransitionTime, effectiveResumeTransitionTime, pausedSubscriptionTransition.getNextState());
+        final BusinessSubscriptionTransition resumedBST = createExpectedResumedBST(requestedResumeTransitionTime, effectiveResumeTransitionTime, pausedBST.getNextSubscription());
+        listener.handleNotificationChange(resumedSubscriptionTransition);
+        Assert.assertEquals(dao.getTransitions(KEY).size(), 3);
+        Assert.assertEquals(dao.getTransitions(KEY).get(2), resumedBST);
+
+        // Cancel it
+        final DateTime effectiveCancelTransitionTime = new DateTime(DateTimeZone.UTC);
+        final DateTime requestedCancelTransitionTime = new DateTime(DateTimeZone.UTC);
+        listener.handleNotificationChange(createCancelSubscriptionTransition(requestedCancelTransitionTime, effectiveCancelTransitionTime, resumedSubscriptionTransition.getNextState()));
+        final BusinessSubscriptionTransition cancelledBST = createExpectedCancelledBST(requestedCancelTransitionTime, effectiveCancelTransitionTime, resumedBST.getNextSubscription());
+        Assert.assertEquals(dao.getTransitions(KEY).size(), 4);
+        Assert.assertEquals(dao.getTransitions(KEY).get(3), cancelledBST);
+    }
+
+    private BusinessSubscriptionTransition createExpectedFirstBST(final DateTime requestedTransitionTime, final DateTime effectiveTransitionTime)
+    {
+        final BusinessSubscriptionEvent event = BusinessSubscriptionEvent.subscriptionCreated(plan);
+        final ISubscription.SubscriptionState subscriptionState = ISubscription.SubscriptionState.ACTIVE;
+        final BusinessSubscription emptyBST = new BusinessSubscription(null, null, null, null, null, subscriptionId, bundleUUID);
+        return createExpectedBST(event, requestedTransitionTime, effectiveTransitionTime, emptyBST, subscriptionState);
+    }
+
+    private BusinessSubscriptionTransition createExpectedPausedBST(final DateTime requestedTransitionTime, final DateTime effectiveTransitionTime, final BusinessSubscription lastSubscription)
+    {
+        final BusinessSubscriptionEvent event = BusinessSubscriptionEvent.subscriptionPaused(plan);
+        final ISubscription.SubscriptionState subscriptionState = ISubscription.SubscriptionState.PAUSED;
+        return createExpectedBST(event, requestedTransitionTime, effectiveTransitionTime, lastSubscription, subscriptionState);
+    }
+
+    private BusinessSubscriptionTransition createExpectedResumedBST(final DateTime requestedTransitionTime, final DateTime effectiveTransitionTime, final BusinessSubscription lastSubscription)
+    {
+        final BusinessSubscriptionEvent event = BusinessSubscriptionEvent.subscriptionResumed(plan);
+        final ISubscription.SubscriptionState nextState = ISubscription.SubscriptionState.ACTIVE;
+        return createExpectedBST(event, requestedTransitionTime, effectiveTransitionTime, lastSubscription, nextState);
+    }
+
+    private BusinessSubscriptionTransition createExpectedCancelledBST(final DateTime requestedTransitionTime, final DateTime effectiveTransitionTime, final BusinessSubscription lastSubscription)
+    {
+        final BusinessSubscriptionEvent event = BusinessSubscriptionEvent.subscriptionCancelled(plan);
+        final ISubscription.SubscriptionState nextState = ISubscription.SubscriptionState.CANCELLED;
+        return createExpectedBST(event, requestedTransitionTime, effectiveTransitionTime, lastSubscription, nextState);
+    }
+
+    private BusinessSubscriptionTransition createExpectedBST(
+        final BusinessSubscriptionEvent eventType,
+        final DateTime requestedTransitionTime,
+        final DateTime effectiveTransitionTime,
+        final BusinessSubscription previousSubscription,
+        final ISubscription.SubscriptionState nextState
+    )
+    {
+        return new BusinessSubscriptionTransition(
+            KEY,
+            requestedTransitionTime,
+            eventType,
+            previousSubscription,
+            new BusinessSubscription(
+                plan,
+                phase,
+                null,
+                effectiveTransitionTime,
+                nextState,
+                subscriptionId,
+                bundleUUID
+            )
+        );
+    }
+
+    private SubscriptionTransition createFirstSubscriptionTransition(final DateTime requestedTransitionTime, final DateTime effectiveTransitionTime)
+    {
+        final ApiEventType eventType = ApiEventType.CREATE;
+        final ISubscription.SubscriptionState nextState = ISubscription.SubscriptionState.ACTIVE;
+        return new SubscriptionTransition(
+            subscriptionId,
+            bundleUUID,
+            IEvent.EventType.API_USER,
+            eventType,
+            requestedTransitionTime,
+            effectiveTransitionTime,
+            null,
+            null,
+            null,
+            null,
+            nextState,
+            plan,
+            phase,
+            priceList
+        );
+    }
+
+    private SubscriptionTransition createPauseSubscriptionTransition(final DateTime requestedTransitionTime, final DateTime effectiveTransitionTime, final ISubscription.SubscriptionState previousState)
+    {
+        final ApiEventType eventType = ApiEventType.PAUSE;
+        final ISubscription.SubscriptionState nextState = ISubscription.SubscriptionState.PAUSED;
+        return createSubscriptionTransition(eventType, requestedTransitionTime, effectiveTransitionTime, previousState, nextState);
+    }
+
+    private SubscriptionTransition createResumeSubscriptionTransition(final DateTime requestedTransitionTime, final DateTime effectiveTransitionTime, final ISubscription.SubscriptionState previousState)
+    {
+        final ApiEventType eventType = ApiEventType.RESUME;
+        final ISubscription.SubscriptionState nextState = ISubscription.SubscriptionState.ACTIVE;
+        return createSubscriptionTransition(eventType, requestedTransitionTime, effectiveTransitionTime, previousState, nextState);
+    }
+
+    private SubscriptionTransition createCancelSubscriptionTransition(final DateTime requestedTransitionTime, final DateTime effectiveTransitionTime, final ISubscription.SubscriptionState previousState)
+    {
+        final ApiEventType eventType = ApiEventType.CANCEL;
+        final ISubscription.SubscriptionState nextState = ISubscription.SubscriptionState.CANCELLED;
+        return createSubscriptionTransition(eventType, requestedTransitionTime, effectiveTransitionTime, previousState, nextState);
+    }
+
+    private SubscriptionTransition createSubscriptionTransition(
+        final ApiEventType eventType,
+        final DateTime requestedTransitionTime,
+        final DateTime effectiveTransitionTime,
+        final ISubscription.SubscriptionState previousState,
+        final ISubscription.SubscriptionState nextState
+    )
+    {
+        return new SubscriptionTransition(
+            subscriptionId,
+            bundleUUID,
+            IEvent.EventType.API_USER,
+            eventType,
+            requestedTransitionTime,
+            effectiveTransitionTime,
+            previousState,
+            plan,
+            phase,
+            priceList,
+            nextState,
+            plan,
+            phase,
+            priceList
+        );
+    }
+}
\ No newline at end of file