diff --git a/analytics/src/test/java/com/ning/billing/analytics/AnalyticsTestModule.java b/analytics/src/test/java/com/ning/billing/analytics/AnalyticsTestModule.java
index 3c3de7e..6ec95b9 100644
--- a/analytics/src/test/java/com/ning/billing/analytics/AnalyticsTestModule.java
+++ b/analytics/src/test/java/com/ning/billing/analytics/AnalyticsTestModule.java
@@ -22,6 +22,7 @@ import com.ning.billing.catalog.glue.CatalogModule;
import com.ning.billing.dbi.MysqlTestingHelper;
import com.ning.billing.entitlement.glue.EntitlementModule;
import com.ning.billing.util.glue.EventBusModule;
+import com.ning.billing.util.glue.NotificationQueueModule;
import com.ning.billing.util.glue.TagStoreModule;
import org.skife.jdbi.v2.DBI;
import org.skife.jdbi.v2.IDBI;
@@ -39,6 +40,7 @@ public class AnalyticsTestModule extends AnalyticsModule
install(new EventBusModule());
install(new EntitlementModule());
install(new TagStoreModule());
+ install(new NotificationQueueModule());
// Install the Dao layer
final MysqlTestingHelper helper = new MysqlTestingHelper();
diff --git a/entitlement/src/test/java/com/ning/billing/entitlement/api/billing/BrainDeadMockEntitlementDao.java b/entitlement/src/test/java/com/ning/billing/entitlement/api/billing/BrainDeadMockEntitlementDao.java
index f7d8186..94fd234 100644
--- a/entitlement/src/test/java/com/ning/billing/entitlement/api/billing/BrainDeadMockEntitlementDao.java
+++ b/entitlement/src/test/java/com/ning/billing/entitlement/api/billing/BrainDeadMockEntitlementDao.java
@@ -98,17 +98,6 @@ class BrainDeadMockEntitlementDao implements EntitlementDao {
throw new UnsupportedOperationException();
}
- @Override
- public List<EntitlementEvent> getEventsReady(UUID ownerId,
- int sequenceId) {
- throw new UnsupportedOperationException();
- }
-
- @Override
- public void clearEventsReady(UUID ownerId,
- Collection<EntitlementEvent> cleared) {
- throw new UnsupportedOperationException();
- }
@Override
public void createSubscription(SubscriptionData subscription,
@@ -142,7 +131,11 @@ class BrainDeadMockEntitlementDao implements EntitlementDao {
@Override
public void undoMigration(UUID accountId) {
- throw new UnsupportedOperationException();
+ throw new UnsupportedOperationException();
}
-
+
+ @Override
+ public EntitlementEvent getEventById(UUID eventId) {
+ throw new UnsupportedOperationException();
+ }
}
\ No newline at end of file