diff --git a/source/java/org/alfresco/repo/importer/FileImporterTest.java b/source/java/org/alfresco/repo/importer/FileImporterTest.java
index 40939ebade..746e8a0ee4 100644
--- a/source/java/org/alfresco/repo/importer/FileImporterTest.java
+++ b/source/java/org/alfresco/repo/importer/FileImporterTest.java
@@ -45,10 +45,12 @@ import org.alfresco.service.cmr.search.ResultSet;
import org.alfresco.service.cmr.search.SearchParameters;
import org.alfresco.service.cmr.search.SearchService;
import org.alfresco.service.cmr.security.AuthenticationService;
+import org.alfresco.service.cmr.security.PermissionService;
import org.alfresco.service.namespace.NamespaceService;
import org.alfresco.service.transaction.TransactionService;
import org.alfresco.util.ApplicationContextHelper;
import org.alfresco.util.TempFileProvider;
+import org.alfresco.util.TestWithUserUtils;
import org.springframework.context.ApplicationContext;
public class FileImporterTest extends TestCase
@@ -60,6 +62,7 @@ public class FileImporterTest extends TestCase
private ContentService contentService;
private AuthenticationService authenticationService;
private AuthenticationComponent authenticationComponent;
+ private PermissionService permissionService;
private MimetypeService mimetypeService;
private NamespaceService namespaceService;
@@ -86,6 +89,7 @@ public class FileImporterTest extends TestCase
contentService = serviceRegistry.getContentService();
authenticationService = (AuthenticationService) ctx.getBean("authenticationService");
authenticationComponent = (AuthenticationComponent) ctx.getBean("authenticationComponent");
+ permissionService = serviceRegistry.getPermissionService();
mimetypeService = serviceRegistry.getMimetypeService();
namespaceService = serviceRegistry.getNamespaceService();
@@ -170,6 +174,8 @@ public class FileImporterTest extends TestCase
*
String: Directory to use as source (e.g. c:/temp)
* String: New name to give the source. It may have a suffix added (e.g. upload_xxx)
* Integer: Number of times to repeat the load.
+ * String: (optional) user to authenticate as
+ * String: (optional) password for authentication
*
* @throws SystemException
* @throws NotSupportedException
@@ -186,6 +192,8 @@ public class FileImporterTest extends TestCase
File sourceFile = new File(args[2]);
String baseName = args[3];
int target = Integer.parseInt(args[4]);
+ String userName = args.length > 5 ? args[5] : null;
+ String userPwd = args.length > 6 ? args[6] : "";
while (count < target)
{
File directory = TempFileProvider.getTempDir();
@@ -210,13 +218,13 @@ public class FileImporterTest extends TestCase
}
NodeRef storeRoot = test.nodeService.getRootNode(spacesStore);
- List location = test.searchService.selectNodes(
+ List importLocations = test.searchService.selectNodes(
storeRoot,
args[1],
null,
test.namespaceService,
false);
- if (location.size() == 0)
+ if (importLocations.size() == 0)
{
throw new AlfrescoRuntimeException(
"Root node not found, " +
@@ -224,13 +232,30 @@ public class FileImporterTest extends TestCase
" not found in store, " +
storeRoot);
}
+ NodeRef importLocation = importLocations.get(0);
+
+ // optionally authenticate as a specific user
+ if (userName != null)
+ {
+ // give the user all necessary permissions on the root
+ test.permissionService.setPermission(importLocation, userName, PermissionService.ALL_PERMISSIONS, true);
+ // authenticate as the designated user
+ TestWithUserUtils.authenticateUser(
+ userName,
+ userPwd,
+ test.authenticationService,
+ test.authenticationComponent);
+ tx.commit();
+ tx = transactionService.getUserTransaction();
+ tx.begin();
+ }
long start = System.nanoTime();
int importCount = test.createFileImporter().loadNamedFile(
- location.get(0),
+ importLocation,
sourceFile,
true,
- String.format("%s-%05d", baseName, count));
+ String.format("%s-%05d-%s", baseName, count, System.currentTimeMillis()));
grandTotal += importCount;
long end = System.nanoTime();
long first = end-start;
diff --git a/source/java/org/alfresco/util/TestWithUserUtils.java b/source/java/org/alfresco/util/TestWithUserUtils.java
index e8b5bd8eb2..6d2d37b811 100644
--- a/source/java/org/alfresco/util/TestWithUserUtils.java
+++ b/source/java/org/alfresco/util/TestWithUserUtils.java
@@ -20,6 +20,7 @@ import java.io.Serializable;
import java.util.HashMap;
import org.alfresco.model.ContentModel;
+import org.alfresco.repo.security.authentication.AuthenticationComponent;
import org.alfresco.service.cmr.repository.NodeRef;
import org.alfresco.service.cmr.repository.NodeService;
import org.alfresco.service.cmr.security.AuthenticationService;
@@ -89,6 +90,32 @@ public abstract class TestWithUserUtils extends BaseSpringTest
authenticationService.authenticate(userName, password.toCharArray());
}
+ /**
+ * Authenticate as the given user. If the user does not exist, then authenticate as the system user
+ * and create the authentication first.
+ */
+ public static void authenticateUser(
+ String userName,
+ String password,
+ AuthenticationService authenticationService,
+ AuthenticationComponent authenticationComponent)
+ {
+ // go system
+ try
+ {
+ authenticationComponent.setSystemUserAsCurrentUser();
+ if (!authenticationService.authenticationExists(userName))
+ {
+ authenticationService.createAuthentication(userName, password.toCharArray());
+ }
+ }
+ finally
+ {
+ authenticationComponent.clearCurrentSecurityContext();
+ }
+ authenticationService.authenticate(userName, password.toCharArray());
+ }
+
/**
* Get the current user node reference
*