This checks in the current state of WCM development.

git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/BRANCHES/WCM-DEV2/root@3128 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261

source/java/org/alfresco/repo/avm/AVMServiceTestBase.java

@@ -0,0 +1,168 @@
+/*
+ * Copyright (C) 2006 Alfresco, Inc.
+ *
+ * Licensed under the Mozilla Public License version 1.1 
+ * with a permitted attribution clause. You may obtain a
+ * copy of the License at
+ *
+ *   http://www.alfresco.org/legal/license.txt
+ *
+ * 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.alfresco.repo.avm;
+
+import java.io.IOException;
+import java.io.PrintStream;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.TreeMap;
+
+import org.alfresco.repo.avm.hibernate.HibernateHelper;
+import org.hibernate.stat.Statistics;
+
+import junit.framework.TestCase;
+
+/**
+ * Base class for AVMService tests.
+ * @author britt
+ */
+public class AVMServiceTestBase extends TestCase
+{
+    /**
+     * The AVMService we are testing.
+     */
+    protected AVMService fService;
+    
+    /**
+     * The start time of actual work for a test.
+     */
+    private long fStartTime;
+
+    /* (non-Javadoc)
+     * @see junit.framework.TestCase#setUp()
+     */
+    @Override
+    protected void setUp() throws Exception
+    {
+        HibernateHelper.GetSessionFactory().getStatistics().setStatisticsEnabled(true);
+        AVMServiceImpl service = new AVMServiceImpl();
+        service.setStorage("build/test-results/storage");
+        service.init(true);
+        fStartTime = System.currentTimeMillis();
+        fService = service;
+    }
+
+    /* (non-Javadoc)
+     * @see junit.framework.TestCase#tearDown()
+     */
+    @Override
+    protected void tearDown() throws Exception
+    {
+        long now = System.currentTimeMillis();
+        System.out.println("Timing: " + (now - fStartTime) + "ms");
+        Statistics stats = HibernateHelper.GetSessionFactory().getStatistics();
+        stats.logSummary();
+        stats.clear();
+        HibernateHelper.Reset();
+    }
+    
+    /**
+     * Get the recursive contents of the given path and version.
+     * @param path 
+     * @param version
+     * @return
+     */
+    protected String recursiveContents(String path, int version, boolean followLinks)
+    {
+        String val = recursiveList(path, version, 0, followLinks);
+        return val.substring(val.indexOf('\n'));
+    }
+
+    
+    /**
+     * Helper to write a recursive listing of a repository at a given version.
+     * @param repoName The name of the repository.
+     * @param version The version to look under.
+     */
+    protected String recursiveList(String repoName, int version, boolean followLinks)
+    {
+        return recursiveList(repoName + ":/", version, 0, followLinks);
+    }
+    
+    /**
+     * Recursive list the given path.
+     * @param path The path.
+     * @param version The version.
+     * @param indent The current indent level.
+     */
+    protected String recursiveList(String path, int version, int indent, boolean followLinks)
+    {
+        StringBuilder builder = new StringBuilder();
+        for (int i = 0; i < indent; i++)
+        {
+            builder.append(' ');
+        }
+        builder.append(path.substring(path.lastIndexOf('/') + 1));
+        builder.append(' ');
+        AVMNodeDescriptor desc = fService.lookup(version, path);
+        builder.append(desc.toString());
+        builder.append('\n');
+        if (desc.getType() == AVMNodeType.PLAIN_DIRECTORY ||
+            (desc.getType() == AVMNodeType.LAYERED_DIRECTORY && followLinks))
+        {
+            String basename = path.endsWith("/") ? path : path + "/";
+            List<FolderEntry> listing = fService.getDirectoryListing(version, path);
+            for (FolderEntry entry : listing)
+            {
+                builder.append(recursiveList(basename + entry.getName(), version, indent + 2, followLinks));
+            }
+        }
+        return builder.toString();
+    }
+    
+    /**
+     * Setup a basic tree.
+     */
+    protected void setupBasicTree()
+        throws IOException
+    {
+        fService.createDirectory("main:/", "a");
+        fService.createDirectory("main:/a", "b");
+        fService.createDirectory("main:/a/b", "c");
+        fService.createDirectory("main:/", "d");
+        fService.createDirectory("main:/d", "e");
+        fService.createDirectory("main:/d/e", "f");
+        fService.createFile("main:/a/b/c", "foo").close();
+        PrintStream out = new PrintStream(fService.getFileOutputStream("main:/a/b/c/foo"));
+        out.println("I am main:/a/b/c/foo");
+        out.flush();
+        out.close();
+        fService.createFile("main:/a/b/c", "bar").close();
+        out = new PrintStream(fService.getFileOutputStream("main:/a/b/c/bar"));
+        out.println("I am main:/a/b/c/bar");
+        out.flush();
+        out.close();
+        ArrayList<String> toSnapshot = new ArrayList<String>();
+        toSnapshot.add("main");
+        fService.createSnapshot(toSnapshot);
+    }
+    
+    /**
+     * Check that history has not been screwed up.
+     */
+    protected void checkHistory(TreeMap<Integer, String> history, String repName)
+    {
+        for (Integer i : history.keySet())
+        {
+            assertEquals(history.get(i), recursiveList(repName, i, false));
+        }
+        int latest = fService.getLatestVersionID(repName);
+        history.put(latest - 1, recursiveList(repName, -1, false));
+    }
+}