1 files changed, 153 insertions, 0 deletions
diff --git a/java/bdbstore/src/test/java/org/apache/qpid/server/store/berkeleydb/upgrade/AbstractUpgradeTestCase.java b/java/bdbstore/src/test/java/org/apache/qpid/server/store/berkeleydb/upgrade/AbstractUpgradeTestCase.java
new file mode 100644
index 0000000000..36991b90d0
--- /dev/null
+++ b/java/bdbstore/src/test/java/org/apache/qpid/server/store/berkeleydb/upgrade/AbstractUpgradeTestCase.java
@@ -0,0 +1,153 @@
+/*
+ *
+ * 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.berkeleydb.upgrade;
+
+import java.io.File;
+
+import org.apache.qpid.server.logging.LogSubject;
+import org.apache.qpid.server.logging.subjects.TestBlankSubject;
+import org.apache.qpid.test.utils.QpidTestCase;
+import org.apache.qpid.util.FileUtils;
+
+import com.sleepycat.je.Database;
+import com.sleepycat.je.Environment;
+import com.sleepycat.je.EnvironmentConfig;
+import com.sleepycat.je.Transaction;
+
+public abstract class AbstractUpgradeTestCase extends QpidTestCase
+{
+    protected static final class StaticAnswerHandler implements UpgradeInteractionHandler
+    {
+        private UpgradeInteractionResponse _response;
+
+        public StaticAnswerHandler(UpgradeInteractionResponse response)
+        {
+            _response = response;
+        }
+
+        @Override
+        public UpgradeInteractionResponse requireResponse(String question, UpgradeInteractionResponse defaultResponse,
+                UpgradeInteractionResponse... possibleResponses)
+        {
+            return _response;
+        }
+    }
+
+    public static final String[] QUEUE_NAMES = { "clientid:myDurSubName", "clientid:mySelectorDurSubName", "myUpgradeQueue",
+            "queue-non-durable" };
+    public static int[] QUEUE_SIZES = { 1, 1, 10, 3 };
+    public static int TOTAL_MESSAGE_NUMBER = 15;
+    protected static final LogSubject LOG_SUBJECT = new TestBlankSubject();
+
+    // one binding per exchange
+    protected static final int TOTAL_BINDINGS = QUEUE_NAMES.length * 2;
+    protected static final int TOTAL_EXCHANGES = 5;
+
+    private File _storeLocation;
+    protected Environment _environment;
+
+    @Override
+    public void setUp() throws Exception
+    {
+        super.setUp();
+        _storeLocation = copyStore(getStoreDirectoryName());
+
+        _environment = createEnvironment(_storeLocation);
+    }
+
+    /** @return eg "bdbstore-v4" - used for copying store */
+    protected abstract String getStoreDirectoryName();
+
+    protected Environment createEnvironment(File storeLocation)
+    {
+        EnvironmentConfig envConfig = new EnvironmentConfig();
+        envConfig.setAllowCreate(true);
+        envConfig.setTransactional(true);
+        envConfig.setConfigParam("je.lock.nLockTables", "7");
+        envConfig.setReadOnly(false);
+        envConfig.setSharedCache(false);
+        envConfig.setCacheSize(0);
+        return new Environment(storeLocation, envConfig);
+    }
+
+    @Override
+    public void tearDown() throws Exception
+    {
+        try
+        {
+            _environment.close();
+        }
+        finally
+        {
+            _environment = null;
+            deleteDirectoryIfExists(_storeLocation);
+        }
+        super.tearDown();
+    }
+
+    private File copyStore(String storeDirectoryName) throws Exception
+    {
+        String src = getClass().getClassLoader().getResource("upgrade/" + storeDirectoryName).toURI().getPath();
+        File storeLocation = new File(new File(TMP_FOLDER), "test-store");
+        deleteDirectoryIfExists(storeLocation);
+        FileUtils.copyRecursive(new File(src), new File(TMP_FOLDER));
+        return storeLocation;
+    }
+
+    protected void deleteDirectoryIfExists(File dir)
+    {
+        if (dir.exists())
+        {
+            assertTrue("The provided file " + dir + " is not a directory", dir.isDirectory());
+
+            boolean deletedSuccessfully = FileUtils.delete(dir, true);
+
+            assertTrue("Files at '" + dir + "' should have been deleted", deletedSuccessfully);
+        }
+    }
+
+    protected void assertDatabaseRecordCount(String databaseName, final long expectedCountNumber)
+    {
+        long count = getDatabaseCount(databaseName);
+        assertEquals("Unexpected database '" + databaseName + "' entry number", expectedCountNumber, count);
+    }
+
+    protected long getDatabaseCount(String databaseName)
+    {
+        DatabaseCallable<Long> operation = new DatabaseCallable<Long>()
+        {
+
+            @Override
+            public Long call(Database sourceDatabase, Database targetDatabase, Transaction transaction)
+            {
+                return new Long(sourceDatabase.count());
+
+            }
+        };
+        Long count = new DatabaseTemplate(_environment, databaseName, null).call(operation);
+        return count.longValue();
+    }
+
+    public String getVirtualHostName()
+    {
+        return getName();
+    }
+}