summaryrefslogtreecommitdiff
path: root/qpid/java/broker/src/test/java/org/apache/qpid/server/store/MessageStoreQuotaEventsTestBase.java
diff options
context:
space:
mode:
Diffstat (limited to 'qpid/java/broker/src/test/java/org/apache/qpid/server/store/MessageStoreQuotaEventsTestBase.java')
-rw-r--r--qpid/java/broker/src/test/java/org/apache/qpid/server/store/MessageStoreQuotaEventsTestBase.java191
1 files changed, 0 insertions, 191 deletions
diff --git a/qpid/java/broker/src/test/java/org/apache/qpid/server/store/MessageStoreQuotaEventsTestBase.java b/qpid/java/broker/src/test/java/org/apache/qpid/server/store/MessageStoreQuotaEventsTestBase.java
deleted file mode 100644
index b23890b10c..0000000000
--- a/qpid/java/broker/src/test/java/org/apache/qpid/server/store/MessageStoreQuotaEventsTestBase.java
+++ /dev/null
@@ -1,191 +0,0 @@
-/*
- *
- * Licensed to the Apache Software Foundation (ASF) under one
- * or more contributor license agreements. See the NOTICE file
- * distributed with this work for additional information
- * regarding copyright ownership. The ASF 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 org.apache.qpid.server.store;
-
-import java.io.File;
-import java.nio.ByteBuffer;
-import java.util.ArrayList;
-import java.util.List;
-import java.util.UUID;
-
-import org.apache.log4j.Logger;
-import org.apache.qpid.server.message.EnqueableMessage;
-import org.apache.qpid.server.plugin.MessageMetaDataType;
-import org.apache.qpid.server.model.VirtualHost;
-import org.apache.qpid.test.utils.QpidTestCase;
-import org.apache.qpid.util.FileUtils;
-
-import static org.mockito.Matchers.eq;
-import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.when;
-
-public abstract class MessageStoreQuotaEventsTestBase extends QpidTestCase implements EventListener, TransactionLogResource
-{
- private static final Logger _logger = Logger.getLogger(MessageStoreQuotaEventsTestBase.class);
-
- protected static final byte[] MESSAGE_DATA = new byte[32 * 1024];
-
- private MessageStore _store;
- private File _storeLocation;
-
- private List<Event> _events;
- private UUID _transactionResource;
-
- protected abstract MessageStore createStore() throws Exception;
-
- protected abstract void applyStoreSpecificConfiguration(VirtualHost virtualHost);
-
- protected abstract int getNumberOfMessagesToFillStore();
-
- @Override
- public void setUp() throws Exception
- {
- super.setUp();
-
- _storeLocation = new File(new File(TMP_FOLDER), getTestName());
- FileUtils.delete(_storeLocation, true);
-
-
- VirtualHost vhost = mock(VirtualHost.class);
- when(vhost.getAttribute(eq(VirtualHost.STORE_PATH))).thenReturn(_storeLocation.getAbsolutePath());
- when(vhost.getName()).thenReturn("test");
-
- applyStoreSpecificConfiguration(vhost);
-
- _store = createStore();
- ((DurableConfigurationStore)_store).configureConfigStore(vhost, null);
- _store.configureMessageStore(vhost, null, null);
-
- _transactionResource = UUID.randomUUID();
- _events = new ArrayList<Event>();
- _store.addEventListener(this, Event.PERSISTENT_MESSAGE_SIZE_OVERFULL, Event.PERSISTENT_MESSAGE_SIZE_UNDERFULL);
- }
-
- @Override
- public void tearDown() throws Exception
- {
- try
- {
- super.tearDown();
- }
- finally
- {
- if (_store != null)
- {
- _store.close();
- }
- FileUtils.delete(_storeLocation, true);
- }
- }
-
- public void testOverflow() throws Exception
- {
- Transaction transaction = _store.newTransaction();
-
- List<EnqueableMessage> messages = new ArrayList<EnqueableMessage>();
- for (int i = 0; i < getNumberOfMessagesToFillStore(); i++)
- {
- EnqueableMessage m = addMessage(i);
- messages.add(m);
- transaction.enqueueMessage(this, m);
- }
- transaction.commitTran();
-
- assertEvent(1, Event.PERSISTENT_MESSAGE_SIZE_OVERFULL);
-
- for (EnqueableMessage m : messages)
- {
- m.getStoredMessage().remove();
- }
-
- assertEvent(2, Event.PERSISTENT_MESSAGE_SIZE_UNDERFULL);
- }
-
- protected EnqueableMessage addMessage(long id)
- {
- StorableMessageMetaData metaData = createMetaData(id, MESSAGE_DATA.length);
- StoredMessage handle = _store.addMessage(metaData);
- handle.addContent(0, ByteBuffer.wrap(MESSAGE_DATA));
- TestMessage message = new TestMessage(id, handle);
- return message;
- }
-
- private StorableMessageMetaData createMetaData(long id, int length)
- {
- StorableMessageMetaData metaData = mock(StorableMessageMetaData.class);
- when(metaData.isPersistent()).thenReturn(true);
- when(metaData.getContentSize()).thenReturn(length);
- when(metaData.getStorableSize()).thenReturn(0);
- MessageMetaDataType type = mock(MessageMetaDataType.class);
- when(type.ordinal()).thenReturn(-1);
- when(metaData.getType()).thenReturn(type);
- return metaData;
- }
-
- @Override
- public void event(Event event)
- {
- _logger.debug("Test event listener received event " + event);
- _events.add(event);
- }
-
- private void assertEvent(int expectedNumberOfEvents, Event... expectedEvents)
- {
- assertEquals("Unexpected number of events received ", expectedNumberOfEvents, _events.size());
- for (Event event : expectedEvents)
- {
- assertTrue("Expected event is not found:" + event, _events.contains(event));
- }
- }
-
- @Override
- public UUID getId()
- {
- return _transactionResource;
- }
-
- private static class TestMessage implements EnqueableMessage
- {
- private final StoredMessage<?> _handle;
- private final long _messageId;
-
- public TestMessage(long messageId, StoredMessage<?> handle)
- {
- _messageId = messageId;
- _handle = handle;
- }
-
- public long getMessageNumber()
- {
- return _messageId;
- }
-
- public boolean isPersistent()
- {
- return true;
- }
-
- public StoredMessage<?> getStoredMessage()
- {
- return _handle;
- }
- }
-}