diff options
Diffstat (limited to 'qpid/java/broker-core/src/test/java/org/apache/qpid/server/store/TestMemoryMessageStore.java')
-rw-r--r-- | qpid/java/broker-core/src/test/java/org/apache/qpid/server/store/TestMemoryMessageStore.java | 21 |
1 files changed, 18 insertions, 3 deletions
diff --git a/qpid/java/broker-core/src/test/java/org/apache/qpid/server/store/TestMemoryMessageStore.java b/qpid/java/broker-core/src/test/java/org/apache/qpid/server/store/TestMemoryMessageStore.java index 32df355c07..bfa4e1d52e 100644 --- a/qpid/java/broker-core/src/test/java/org/apache/qpid/server/store/TestMemoryMessageStore.java +++ b/qpid/java/broker-core/src/test/java/org/apache/qpid/server/store/TestMemoryMessageStore.java @@ -20,15 +20,30 @@ */ package org.apache.qpid.server.store; +import java.util.concurrent.atomic.AtomicInteger; + +import org.apache.qpid.server.store.handler.MessageHandler; + /** A simple message store that stores the messages in a thread-safe structure in memory. */ public class TestMemoryMessageStore extends AbstractMemoryMessageStore { public static final String TYPE = "TestMemory"; - @Override - public String getStoreType() + public int getMessageCount() { - return TYPE; + final AtomicInteger counter = new AtomicInteger(); + visitMessages(new MessageHandler() + { + + @Override + public boolean handle(StoredMessage<?> storedMessage) + { + counter.incrementAndGet(); + return true; + } + }); + return counter.get(); } + } |