mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-06-30 18:15:39 +00:00
4d39879ab6
git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@42029 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
427 lines
16 KiB
Java
427 lines
16 KiB
Java
/*
|
|
* 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();
|
|
|
|
userUsageTrackingComponent.setEnabled(true);
|
|
userUsageTrackingComponent.bootstrapInternal(); // true => recalculate
|
|
|
|
logger.debug("Recalculated 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();
|
|
}
|
|
|
|
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()
|
|
{
|
|
for (int i = 1; i <= MAX_USERS; i++)
|
|
{
|
|
String userName = TEST_USER_PREFIX+i;
|
|
NodeRef personNodeRef = personService.getPerson(userName);
|
|
|
|
assertNull(nodeService.getProperty(personNodeRef, ContentModel.PROP_SIZE_CURRENT));
|
|
}
|
|
}
|
|
|
|
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;
|
|
}
|
|
}
|