Merged HEAD-QA to HEAD (4.2) (including moving test classes into separate folders)

51903 to 54309 


git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@54310 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261

source/test-java/org/alfresco/repo/web/scripts/TestWebScriptRepoServer.java

@@ -0,0 +1,225 @@
+/*
+ * Copyright (C) 2005-2010 Alfresco Software Limited.
+ *
+ * This file is part of Alfresco
+ *
+ * Alfresco is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Alfresco is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.alfresco.repo.web.scripts;
+
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.io.StringWriter;
+
+import org.alfresco.repo.security.authentication.AuthenticationException;
+import org.alfresco.repo.security.authentication.AuthenticationUtil;
+import org.alfresco.repo.security.authentication.AuthenticationUtil.RunAsWork;
+import org.alfresco.repo.transaction.RetryingTransactionHelper;
+import org.alfresco.repo.transaction.RetryingTransactionHelper.RetryingTransactionCallback;
+import org.alfresco.service.cmr.security.AuthenticationService;
+import org.alfresco.util.ApplicationContextHelper;
+import org.alfresco.util.EqualsHelper;
+import org.springframework.context.support.ClassPathXmlApplicationContext;
+import org.springframework.extensions.webscripts.TestWebScriptServer;
+
+
+/**
+ * Stand-alone Web Script Test Server
+ * 
+ * @author davidc
+ */
+public class TestWebScriptRepoServer extends TestWebScriptServer
+{
+    /**
+     * Main entry point.
+     */
+    public static void main(String[] args)
+    {
+        try
+        {
+            TestWebScriptServer testServer = getTestServer();
+            AuthenticationUtil.setRunAsUserSystem();
+            testServer.rep();
+        }
+        catch(Throwable e)
+        {
+            StringWriter strWriter = new StringWriter();
+            PrintWriter printWriter = new PrintWriter(strWriter);
+            e.printStackTrace(printWriter);
+            System.out.println(strWriter.toString());
+        }
+        finally
+        {
+            System.exit(0);
+        }
+    }
+
+    private final static String[] CONFIG_LOCATIONS = new String[]
+    {
+        "classpath:alfresco/application-context.xml",
+        "classpath:alfresco/web-scripts-application-context.xml",
+        "classpath:alfresco/web-scripts-application-context-test.xml"
+    };
+    
+    /** A static reference to the application context being used */
+    private static ClassPathXmlApplicationContext ctx;
+    private static String appendedTestConfiguration;
+    
+    private RetryingTransactionHelper retryingTransactionHelper;
+    private AuthenticationService authenticationService;
+    
+    
+    /**
+     * Sets helper that provides transaction callbacks
+     */
+    public void setTransactionHelper(RetryingTransactionHelper retryingTransactionHelper)
+    {
+        this.retryingTransactionHelper = retryingTransactionHelper;
+    }
+    
+    /**
+     * @param authenticationService
+     */
+    public void setAuthenticationService(AuthenticationService authenticationService)
+    {
+        this.authenticationService = authenticationService;
+    }
+
+    /**
+     * Get default user name
+     */
+    protected String getDefaultUserName()
+    {
+        return AuthenticationUtil.getAdminUserName();
+    }
+
+    /**
+     * {@inheritDoc #getTestServer(String)}
+     */
+    public static TestWebScriptServer getTestServer()
+    {
+        return getTestServer(null);
+    }
+    
+    /**
+     * Start up a context and get the server bean.
+     * <p>
+     * This method will close and restart the application context only if the configuration has
+     * changed.
+     * 
+     * @param appendTestConfigLocation      additional context file to include in the application context
+     * @return  Test Server
+     */
+    public static synchronized TestWebScriptServer getTestServer(String appendTestConfigLocation)
+    {
+        if (TestWebScriptRepoServer.ctx != null)
+        {
+            boolean configChanged = !EqualsHelper.nullSafeEquals(
+                    appendTestConfigLocation,
+                    TestWebScriptRepoServer.appendedTestConfiguration);
+            if (configChanged)
+            {
+                // The config changed, so close the context (it'll be restarted later)
+                try
+                {
+                    ctx.close();
+                    ctx = null;
+                }
+                catch (Throwable e)
+                {
+                    throw new RuntimeException("Failed to shut down existing application context", e);
+                }
+            }
+            else
+            {
+                // There is already a context with the required configuration
+            }
+        }
+        
+        // Check if we need to start/restart the context
+        if (TestWebScriptRepoServer.ctx == null || (TestWebScriptRepoServer.ctx != null && !TestWebScriptRepoServer.ctx.isActive()))
+        {
+            // Restart it
+            final String[] configLocations;
+            if (appendTestConfigLocation == null)
+            {
+                configLocations = CONFIG_LOCATIONS;
+            }
+            else
+            {
+                configLocations = new String[CONFIG_LOCATIONS.length+1];
+                System.arraycopy(CONFIG_LOCATIONS, 0, configLocations, 0, CONFIG_LOCATIONS.length);
+                configLocations[CONFIG_LOCATIONS.length] = appendTestConfigLocation;
+            }
+            
+            TestWebScriptRepoServer.ctx = (ClassPathXmlApplicationContext)ApplicationContextHelper.getApplicationContext(configLocations);
+            
+//            TestWebScriptRepoServer.ctx = new ClassPathXmlApplicationContext(configLocations);
+            TestWebScriptRepoServer.appendedTestConfiguration = appendTestConfigLocation;
+        }
+        
+        // Get the bean
+        TestWebScriptServer testServer = (TestWebScriptRepoServer)TestWebScriptRepoServer.ctx.getBean("webscripts.test");
+        return testServer;
+    }
+    
+    /**
+     * Interpret a single command using the BufferedReader passed in for any data needed.
+     * 
+     * @param line The unparsed command
+     * @return The textual output of the command.
+     */
+    @Override
+    protected String interpretCommand(final String line)
+        throws IOException
+    {
+        try
+        {
+            if (username.startsWith("TICKET_"))
+            {
+                try
+                {
+                    retryingTransactionHelper.doInTransaction(new RetryingTransactionCallback<Object>()
+                    {
+                        public Object execute() throws Exception
+                        {
+                            authenticationService.validate(username);
+                            return null;
+                        }
+                    });
+                    return executeCommand(line);
+                }
+                finally
+                {
+                    authenticationService.clearCurrentSecurityContext();
+                }
+            }
+        }
+        catch(AuthenticationException e)
+        {
+            executeCommand("user " + getDefaultUserName());
+        }
+        
+        // execute command in context of currently selected user
+        return AuthenticationUtil.runAs(new RunAsWork<String>()
+        {
+            @SuppressWarnings("synthetic-access")
+            public String doWork() throws Exception
+            {
+                return executeCommand(line);
+            }
+        }, username);
+    }
+
+}