killbill-aplcache

diff --git a/util/src/test/java/com/ning/billing/util/entity/dao/MockEntityDaoBase.java b/util/src/test/java/com/ning/billing/util/entity/dao/MockEntityDaoBase.java
index 6d32cae..d59dc79 100644
--- a/util/src/test/java/com/ning/billing/util/entity/dao/MockEntityDaoBase.java
+++ b/util/src/test/java/com/ning/billing/util/entity/dao/MockEntityDaoBase.java
@@ -70,6 +70,11 @@ public class MockEntityDaoBase<T extends Entity, U extends BillingExceptionBase>
         return result;
     }
 
+    public void update(final T entity, final InternalCallContext context) {
+        final Long entityRecordId = getRecordId(entity.getId(), context);
+        entities.get(entity.getId()).put(entityRecordId, entity);
+    }
+
     public void delete(final T entity, final InternalCallContext context) {
         entities.remove(entity.getId());
     }