diff options
Diffstat (limited to 'java/systests/src/test/java/org/apache/qpid/server/store/TestableMemoryMessageStore.java')
-rw-r--r-- | java/systests/src/test/java/org/apache/qpid/server/store/TestableMemoryMessageStore.java | 22 |
1 files changed, 15 insertions, 7 deletions
diff --git a/java/systests/src/test/java/org/apache/qpid/server/store/TestableMemoryMessageStore.java b/java/systests/src/test/java/org/apache/qpid/server/store/TestableMemoryMessageStore.java index be7687a22c..9a649421dd 100644 --- a/java/systests/src/test/java/org/apache/qpid/server/store/TestableMemoryMessageStore.java +++ b/java/systests/src/test/java/org/apache/qpid/server/store/TestableMemoryMessageStore.java @@ -7,9 +7,9 @@ * 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 @@ -20,10 +20,12 @@ */ package org.apache.qpid.server.store; -import org.apache.qpid.server.queue.AMQMessage; +import org.apache.qpid.server.queue.MessageMetaData; +import org.apache.qpid.framing.ContentBody; -import java.util.concurrent.ConcurrentMap; import java.util.concurrent.ConcurrentHashMap; +import java.util.concurrent.ConcurrentMap; +import java.util.List; /** * Adds some extra methods to the memory message store for testing purposes. @@ -32,11 +34,17 @@ public class TestableMemoryMessageStore extends MemoryMessageStore { public TestableMemoryMessageStore() { - _messageMap = new ConcurrentHashMap<Long, AMQMessage>(); + _metaDataMap = new ConcurrentHashMap<Long, MessageMetaData>(); + _contentBodyMap = new ConcurrentHashMap<Long, List<ContentBody>>(); + } + + public ConcurrentMap<Long, MessageMetaData> getMessageMetaDataMap() + { + return _metaDataMap; } - public ConcurrentMap<Long, AMQMessage> getMessageMap() + public ConcurrentMap<Long, List<ContentBody>> getContentBodyMap() { - return _messageMap; + return _contentBodyMap; } } |