mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-08-07 17:49:17 +00:00
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
This commit is contained in:
Samuel Langlois
@@ -0,0 +1,436 @@
|
||||
/*
|
||||
* Copyright (C) 2005-2012 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.usage;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
import javax.transaction.UserTransaction;
|
||||
|
||||
import junit.framework.TestCase;
|
||||
|
||||
import org.alfresco.model.ContentModel;
|
||||
import org.alfresco.repo.content.MimetypeMap;
|
||||
import org.alfresco.repo.security.authentication.AuthenticationUtil;
|
||||
import org.alfresco.service.cmr.repository.ChildAssociationRef;
|
||||
import org.alfresco.service.cmr.repository.ContentService;
|
||||
import org.alfresco.service.cmr.repository.ContentWriter;
|
||||
import org.alfresco.service.cmr.repository.NodeRef;
|
||||
import org.alfresco.service.cmr.repository.NodeService;
|
||||
import org.alfresco.service.cmr.repository.StoreRef;
|
||||
import org.alfresco.service.cmr.security.MutableAuthenticationService;
|
||||
import org.alfresco.service.cmr.security.PersonService;
|
||||
import org.alfresco.service.cmr.usage.ContentUsageService;
|
||||
import org.alfresco.service.namespace.NamespaceService;
|
||||
import org.alfresco.service.namespace.QName;
|
||||
import org.alfresco.service.transaction.TransactionService;
|
||||
import org.alfresco.util.ApplicationContextHelper;
|
||||
import org.apache.commons.logging.Log;
|
||||
import org.apache.commons.logging.LogFactory;
|
||||
import org.springframework.context.ApplicationContext;
|
||||
|
||||
/**
|
||||
* Test enabling (recalculating) and disabling (clearing) and user usages, and also collapsing user usage deltas.
|
||||
*/
|
||||
public class UserUsageTrackingComponentTest extends TestCase
|
||||
{
|
||||
private static ApplicationContext applicationContext = ApplicationContextHelper.getApplicationContext();
|
||||
|
||||
private static Log logger = LogFactory.getLog(UserUsageTrackingComponentTest.class);
|
||||
|
||||
private boolean clean = true;
|
||||
|
||||
private MutableAuthenticationService authenticationService;
|
||||
private ContentService contentService;
|
||||
private TransactionService transactionService;
|
||||
private PersonService personService;
|
||||
private NodeService nodeService;
|
||||
private UserUsageTrackingComponent userUsageTrackingComponent;
|
||||
private ContentUsageService contentUsageService;
|
||||
|
||||
private UserTransaction testTX;
|
||||
|
||||
public static StoreRef SPACES_STORE = new StoreRef(StoreRef.PROTOCOL_WORKSPACE, "SpacesStore");
|
||||
|
||||
private static final String TEST_RUN = System.currentTimeMillis()+"";
|
||||
private static final String TEST_USER_PREFIX = "user-"+TEST_RUN+"-";
|
||||
|
||||
private static final int MAX_USERS = 5;
|
||||
private static final int BATCH_SIZE = 5;
|
||||
private static final int PROGRESS_SIZE = 100;
|
||||
|
||||
protected void setUp() throws Exception
|
||||
{
|
||||
nodeService = (NodeService)applicationContext.getBean("NodeService");
|
||||
authenticationService = (MutableAuthenticationService)applicationContext.getBean("authenticationService");
|
||||
transactionService = (TransactionService)applicationContext.getBean("transactionComponent");
|
||||
personService = (PersonService)applicationContext.getBean("PersonService");
|
||||
contentService = (ContentService)applicationContext.getBean("ContentService");
|
||||
contentUsageService = (ContentUsageService)applicationContext.getBean("ContentUsageService");
|
||||
|
||||
userUsageTrackingComponent = (UserUsageTrackingComponent)applicationContext.getBean("userUsageTrackingComponent");
|
||||
|
||||
// Enable
|
||||
ContentUsageImpl contentUsage = (ContentUsageImpl)applicationContext.getBean("contentUsageImpl");
|
||||
contentUsage.setEnabled(true);
|
||||
contentUsage.init();
|
||||
userUsageTrackingComponent.setEnabled(true);
|
||||
userUsageTrackingComponent.init();
|
||||
userUsageTrackingComponent.bootstrapInternal();
|
||||
|
||||
AuthenticationUtil.setRunAsUserSystem();
|
||||
}
|
||||
|
||||
protected void tearDown() throws Exception
|
||||
{
|
||||
// Disable
|
||||
ContentUsageImpl contentUsage = (ContentUsageImpl)applicationContext.getBean("contentUsageImpl");
|
||||
contentUsage.setEnabled(false);
|
||||
contentUsage.init();
|
||||
userUsageTrackingComponent.setEnabled(false);
|
||||
userUsageTrackingComponent.bootstrapInternal();
|
||||
|
||||
if (clean)
|
||||
{
|
||||
deleteUsersAndContent(true);
|
||||
}
|
||||
|
||||
super.tearDown();
|
||||
}
|
||||
|
||||
private Map<String, Long> createUsersAndContent(String namePrefix)
|
||||
{
|
||||
long start = System.currentTimeMillis();
|
||||
long progressStart = System.currentTimeMillis();
|
||||
|
||||
Map<String, Long> personContent = new HashMap<String, Long>(MAX_USERS);
|
||||
|
||||
try
|
||||
{
|
||||
int count = 0;
|
||||
|
||||
for (int i = 1; i <= MAX_USERS; i++)
|
||||
{
|
||||
if (count == 0)
|
||||
{
|
||||
testTX = transactionService.getUserTransaction();
|
||||
testTX.begin();
|
||||
}
|
||||
|
||||
count++;
|
||||
String userName = TEST_USER_PREFIX+i;
|
||||
|
||||
if (! authenticationService.authenticationExists(userName))
|
||||
{
|
||||
// Note: this will auto-create the user home
|
||||
HashMap<QName, Serializable> props = new HashMap<QName, Serializable>();
|
||||
props.put(ContentModel.PROP_USERNAME, userName);
|
||||
|
||||
personService.createPerson(props); // ignore result
|
||||
|
||||
authenticationService.createAuthentication(userName, userName.toCharArray());
|
||||
authenticationService.setAuthenticationEnabled(userName, false);
|
||||
|
||||
Long contentSize = null;
|
||||
|
||||
NodeRef homeFolder = getHomeSpaceFolderNode(userName);
|
||||
|
||||
StringBuilder sb = new StringBuilder();
|
||||
for (int j = 1; j <= i; j++)
|
||||
{
|
||||
int k = j % 10;
|
||||
sb.append(k);
|
||||
}
|
||||
|
||||
AuthenticationUtil.setFullyAuthenticatedUser(userName);
|
||||
|
||||
NodeRef content = addTextContent(homeFolder, namePrefix + "-" + userName + "-1.txt", sb.toString());
|
||||
contentSize = contentService.getReader(content, ContentModel.PROP_CONTENT).getSize();
|
||||
|
||||
content = addTextContent(homeFolder, namePrefix + "-" + userName + "-2.txt", sb.toString());
|
||||
contentSize += contentService.getReader(content, ContentModel.PROP_CONTENT).getSize();
|
||||
|
||||
personContent.put(userName, contentSize);
|
||||
}
|
||||
|
||||
AuthenticationUtil.setRunAsUserSystem();
|
||||
|
||||
if ((count == BATCH_SIZE) || (i == MAX_USERS))
|
||||
{
|
||||
testTX.commit();
|
||||
count = 0;
|
||||
}
|
||||
|
||||
if (((i % PROGRESS_SIZE) == 0) && (i != MAX_USERS))
|
||||
{
|
||||
logger.debug("Progress: "+PROGRESS_SIZE+" users created in "+((System.currentTimeMillis()-progressStart)/1000)+" secs");
|
||||
progressStart = System.currentTimeMillis();
|
||||
}
|
||||
}
|
||||
|
||||
logger.debug("Total: "+MAX_USERS+" users created in "+((System.currentTimeMillis()-start)/1000)+" secs");
|
||||
}
|
||||
catch (Throwable t)
|
||||
{
|
||||
t.printStackTrace();
|
||||
|
||||
try { testTX.rollback(); } catch (Exception e) { e.printStackTrace(); }
|
||||
}
|
||||
return personContent;
|
||||
}
|
||||
|
||||
public void test1EnableDisableCollapse()
|
||||
{
|
||||
String fileNamePrefix = "TEST";
|
||||
|
||||
createUsersAndContent(fileNamePrefix);
|
||||
|
||||
logger.debug("Test: " + getName());
|
||||
|
||||
userUsageTrackingComponent.setEnabled(false);
|
||||
userUsageTrackingComponent.bootstrapInternal(); // false => clear
|
||||
|
||||
logger.debug("Cleared usages");
|
||||
|
||||
checkCleared(true);
|
||||
|
||||
userUsageTrackingComponent.setEnabled(true);
|
||||
userUsageTrackingComponent.bootstrapInternal(); // true => recalculate
|
||||
|
||||
logger.debug("Recalculated usages");
|
||||
|
||||
userUsageTrackingComponent.execute(); // collapse usages
|
||||
|
||||
checkCalculated(2L);
|
||||
|
||||
checkUsage(2L);
|
||||
|
||||
// add content
|
||||
for (int i = 1; i <= MAX_USERS; i++)
|
||||
{
|
||||
String userName = TEST_USER_PREFIX+i;
|
||||
|
||||
NodeRef homeFolder = getHomeSpaceFolderNode(userName);
|
||||
|
||||
AuthenticationUtil.setFullyAuthenticatedUser(userName);
|
||||
|
||||
StringBuilder sb = new StringBuilder();
|
||||
for (int j = 1; j <= i; j++)
|
||||
{
|
||||
int k = j % 10;
|
||||
sb.append(k);
|
||||
}
|
||||
|
||||
addTextContent(homeFolder, "c-"+userName+".txt", sb.toString());
|
||||
addTextContent(homeFolder, "d-"+userName+".txt", sb.toString());
|
||||
|
||||
AuthenticationUtil.setRunAsUserSystem();
|
||||
}
|
||||
|
||||
logger.debug("Added content");
|
||||
|
||||
checkUsage(4L);
|
||||
|
||||
userUsageTrackingComponent.execute(); // collapse usages
|
||||
|
||||
logger.debug("Collapsed usages");
|
||||
|
||||
checkUsage(4L);
|
||||
|
||||
// delete content
|
||||
for (int i = 1; i <= MAX_USERS; i++)
|
||||
{
|
||||
String userName = TEST_USER_PREFIX+i;
|
||||
|
||||
NodeRef homeFolder = getHomeSpaceFolderNode(userName);
|
||||
|
||||
AuthenticationUtil.setFullyAuthenticatedUser(userName);
|
||||
|
||||
NodeRef childNodeRef = nodeService.getChildByName(homeFolder, ContentModel.ASSOC_CONTAINS, fileNamePrefix+"-"+userName+"-1.txt");
|
||||
nodeService.deleteNode(childNodeRef);
|
||||
|
||||
childNodeRef = nodeService.getChildByName(homeFolder, ContentModel.ASSOC_CONTAINS, fileNamePrefix+"-"+userName+"-2.txt");
|
||||
nodeService.deleteNode(childNodeRef);
|
||||
|
||||
AuthenticationUtil.setRunAsUserSystem();
|
||||
}
|
||||
|
||||
logger.debug("Deleted content");
|
||||
|
||||
checkUsage(2L);
|
||||
|
||||
userUsageTrackingComponent.execute(); // collapse usages
|
||||
|
||||
logger.debug("Collapsed usages");
|
||||
|
||||
checkUsage(2L);
|
||||
|
||||
userUsageTrackingComponent.setEnabled(false);
|
||||
userUsageTrackingComponent.bootstrapInternal(); // false => clear
|
||||
|
||||
logger.debug("Cleared usages");
|
||||
checkCleared(false);
|
||||
}
|
||||
|
||||
public void test2RecalculateUserUsage() throws Exception
|
||||
{
|
||||
Map<String, Long> expectedResultsA = createUsersAndContent("A");
|
||||
|
||||
deleteUsersAndContent(false);
|
||||
|
||||
Map<String, Long> expectedResultsB = createUsersAndContent("B");
|
||||
|
||||
for (int i = 1; i <= MAX_USERS; i++)
|
||||
{
|
||||
String userName = TEST_USER_PREFIX + i;
|
||||
Long totalExpected = expectedResultsA.get(userName)+expectedResultsB.get(userName);
|
||||
assertEquals(totalExpected, Long.valueOf(contentUsageService.getUserUsage(userName)));
|
||||
}
|
||||
}
|
||||
|
||||
private void checkCalculated(long factor)
|
||||
{
|
||||
for (int i = 1; i <= MAX_USERS; i++)
|
||||
{
|
||||
String userName = TEST_USER_PREFIX+i;
|
||||
NodeRef personNodeRef = personService.getPerson(userName);
|
||||
|
||||
// get user stored usage (not including deltas, if any)
|
||||
Long sizeProp = (Long)nodeService.getProperty(personNodeRef, ContentModel.PROP_SIZE_CURRENT);
|
||||
assertNotNull("Property " + ContentModel.PROP_SIZE_CURRENT + "does not exist, yet", sizeProp);
|
||||
assertEquals(userName, i*factor, sizeProp.longValue());
|
||||
}
|
||||
}
|
||||
|
||||
private void checkUsage(long factor)
|
||||
{
|
||||
for (int i = 1; i <= MAX_USERS; i++)
|
||||
{
|
||||
String userName = TEST_USER_PREFIX+i;
|
||||
|
||||
// get user usage (including deltas, if any)
|
||||
assertEquals(userName, i*factor, contentUsageService.getUserUsage(userName));
|
||||
}
|
||||
}
|
||||
|
||||
private void checkCleared(boolean isCleared)
|
||||
{
|
||||
for (int i = 1; i <= MAX_USERS; i++)
|
||||
{
|
||||
String userName = TEST_USER_PREFIX+i;
|
||||
NodeRef personNodeRef = personService.getPerson(userName);
|
||||
Long sizeCurrent = (Long) nodeService.getProperty(personNodeRef, ContentModel.PROP_SIZE_CURRENT);
|
||||
|
||||
if (isCleared)
|
||||
{
|
||||
assertTrue(sizeCurrent == null || (new Long(0L)).equals(sizeCurrent));
|
||||
}
|
||||
else
|
||||
{
|
||||
assertFalse(sizeCurrent == null || (new Long(0L)).equals(sizeCurrent));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void deleteUsersAndContent(boolean deleteContent)
|
||||
{
|
||||
long start = System.currentTimeMillis();
|
||||
|
||||
try
|
||||
{
|
||||
int count = 0;
|
||||
|
||||
int deleteCount = 0;
|
||||
|
||||
for (int i = 1; i <= MAX_USERS; i++)
|
||||
{
|
||||
if (count == 0)
|
||||
{
|
||||
testTX = transactionService.getUserTransaction();
|
||||
testTX.begin();
|
||||
}
|
||||
|
||||
count++;
|
||||
String userName = TEST_USER_PREFIX+i;
|
||||
|
||||
if (authenticationService.authenticationExists(userName))
|
||||
{
|
||||
if (deleteContent)
|
||||
{
|
||||
NodeRef homeFolder = getHomeSpaceFolderNode(userName);
|
||||
nodeService.deleteNode(homeFolder);
|
||||
}
|
||||
|
||||
personService.deletePerson(userName);
|
||||
deleteCount++;
|
||||
}
|
||||
|
||||
if ((count == BATCH_SIZE) || (i == MAX_USERS))
|
||||
{
|
||||
testTX.commit();
|
||||
count = 0;
|
||||
}
|
||||
}
|
||||
|
||||
logger.debug("Total: "+deleteCount+" users deleted in "+((System.currentTimeMillis()-start)/1000)+" secs");
|
||||
}
|
||||
catch (Throwable t)
|
||||
{
|
||||
t.printStackTrace();
|
||||
|
||||
try { testTX.rollback(); } catch (Exception e) { e.printStackTrace(); }
|
||||
}
|
||||
}
|
||||
|
||||
private NodeRef getHomeSpaceFolderNode(String userName)
|
||||
{
|
||||
return (NodeRef)this.nodeService.getProperty(personService.getPerson(userName), ContentModel.PROP_HOMEFOLDER);
|
||||
}
|
||||
|
||||
private NodeRef addTextContent(NodeRef spaceRef, String fileName, String textData)
|
||||
{
|
||||
Map<QName, Serializable> contentProps = new HashMap<QName, Serializable>();
|
||||
contentProps.put(ContentModel.PROP_NAME, fileName);
|
||||
|
||||
ChildAssociationRef association = nodeService.createNode(spaceRef,
|
||||
ContentModel.ASSOC_CONTAINS,
|
||||
QName.createQName(NamespaceService.CONTENT_MODEL_1_0_URI, fileName),
|
||||
ContentModel.TYPE_CONTENT,
|
||||
contentProps);
|
||||
|
||||
NodeRef content = association.getChildRef();
|
||||
|
||||
// add titled aspect (for Web Client display)
|
||||
Map<QName, Serializable> titledProps = new HashMap<QName, Serializable>();
|
||||
titledProps.put(ContentModel.PROP_TITLE, fileName);
|
||||
titledProps.put(ContentModel.PROP_DESCRIPTION, fileName);
|
||||
this.nodeService.addAspect(content, ContentModel.ASPECT_TITLED, titledProps);
|
||||
|
||||
ContentWriter writer = contentService.getWriter(content, ContentModel.PROP_CONTENT, true);
|
||||
|
||||
writer.setMimetype(MimetypeMap.MIMETYPE_TEXT_PLAIN);
|
||||
writer.setEncoding("UTF-8");
|
||||
|
||||
writer.putContent(textData);
|
||||
|
||||
return content;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user