mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-06-30 18:15:39 +00:00
79508da9fc
* Fixed spreadsheet download * Refactored use of DocService to improve reliability * Reuse security tokens * Consolidate code as a result of above changes git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@30832 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
366 lines
17 KiB
Java
366 lines
17 KiB
Java
/*
|
|
* 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.googledocs;
|
|
|
|
import java.io.File;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.net.MalformedURLException;
|
|
|
|
import javax.transaction.UserTransaction;
|
|
|
|
import junit.framework.TestCase;
|
|
|
|
import org.alfresco.model.ContentModel;
|
|
import org.alfresco.repo.content.MimetypeMap;
|
|
import org.alfresco.repo.content.transform.AbstractContentTransformerTest;
|
|
import org.alfresco.repo.management.subsystems.ApplicationContextFactory;
|
|
import org.alfresco.repo.rendition.executer.AbstractRenderingEngine;
|
|
import org.alfresco.repo.rendition.executer.ReformatRenderingEngine;
|
|
import org.alfresco.repo.security.authentication.AuthenticationUtil;
|
|
import org.alfresco.repo.site.SiteServiceImpl;
|
|
import org.alfresco.service.cmr.coci.CheckOutCheckInService;
|
|
import org.alfresco.service.cmr.model.FileFolderService;
|
|
import org.alfresco.service.cmr.rendition.RenditionDefinition;
|
|
import org.alfresco.service.cmr.rendition.RenditionService;
|
|
import org.alfresco.service.cmr.repository.ContentReader;
|
|
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.security.MutableAuthenticationService;
|
|
import org.alfresco.service.cmr.security.PersonService;
|
|
import org.alfresco.service.cmr.site.SiteService;
|
|
import org.alfresco.service.cmr.site.SiteVisibility;
|
|
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.alfresco.util.GUID;
|
|
import org.alfresco.util.PropertyMap;
|
|
import org.springframework.context.ApplicationContext;
|
|
import org.springframework.context.ConfigurableApplicationContext;
|
|
|
|
import com.google.gdata.util.ServiceException;
|
|
|
|
public class GoogleDocumentServiceSystemTest extends TestCase implements GoogleDocsModel
|
|
{
|
|
private NodeService nodeService;
|
|
private GoogleDocsService googleDocsService;
|
|
private SiteService siteService;
|
|
private TransactionService transactionService;
|
|
private FileFolderService fileFolderService;
|
|
private ContentService contentService;
|
|
private CheckOutCheckInService checkOutCheckInService;
|
|
private MutableAuthenticationService authenticationService;
|
|
private PersonService personService;
|
|
private ApplicationContextFactory subsystem;
|
|
private RenditionService renditionService;
|
|
|
|
private static final String USER_ONE = "GoogleDocUserOne";
|
|
private static final String USER_TWO = "GoogleDocUserTwo";
|
|
private static final String USER_THREE = "GoogleDocUserThree";
|
|
private static final String USER_FOUR = "GoogleDocUserFour";
|
|
private static final String USER_FIVE = "GoogleDocUserFive";
|
|
private static final String USER_SIX = "GoogleDocUserSix";
|
|
private static final String USER_SEVEN = "GoogleDocUserSeven";
|
|
|
|
private NodeRef folder;
|
|
private NodeRef nodeRefDoc;
|
|
private NodeRef nodeRefSpread;
|
|
private NodeRef nodeRefPres;
|
|
private NodeRef nodeRefPdf;
|
|
private NodeRef nodeRef2;
|
|
private UserTransaction userTransaction;
|
|
private String siteId;
|
|
|
|
@Override
|
|
protected void setUp() throws Exception
|
|
{
|
|
ApplicationContext appContext = ApplicationContextHelper.getApplicationContext();
|
|
|
|
nodeService = (NodeService)appContext.getBean("nodeService");
|
|
siteService = (SiteService)appContext.getBean("siteService");
|
|
transactionService = (TransactionService)appContext.getBean("transactionService");
|
|
fileFolderService = (FileFolderService)appContext.getBean("fileFolderService");
|
|
contentService = (ContentService)appContext.getBean("contentService");
|
|
checkOutCheckInService = (CheckOutCheckInService)appContext.getBean("checkOutCheckInService");
|
|
authenticationService = (MutableAuthenticationService)appContext.getBean("authenticationService");
|
|
personService = (PersonService)appContext.getBean("personService");
|
|
renditionService = (RenditionService)appContext.getBean("renditionService");
|
|
|
|
// Start the user transaction
|
|
userTransaction = transactionService.getUserTransaction();
|
|
userTransaction.begin();
|
|
|
|
// Get the sub-system and make sure the googleeditable feature is turned on
|
|
subsystem = (ApplicationContextFactory)appContext.getBean("googledocs");
|
|
if (subsystem.getProperty("googledocs.googleeditable.enabled").equals("false") == true)
|
|
{
|
|
subsystem.stop();
|
|
subsystem.setProperty("googledocs.googleeditable.enabled", "true");
|
|
subsystem.start();
|
|
}
|
|
|
|
// Get the google docs service
|
|
ConfigurableApplicationContext childContext = (ConfigurableApplicationContext)subsystem.getApplicationContext();
|
|
googleDocsService = (GoogleDocsService)childContext.getBean("googleDocsService");
|
|
|
|
// Create the test users
|
|
createUser(USER_ONE, "rwetherall@alfresco.com", null);
|
|
createUser(USER_TWO, "admin@alfresco.com", "rwetherall@activiti.com");
|
|
createUser(USER_THREE, "roy.wetherall@alfresco.com", null);
|
|
createUser(USER_FOUR, "roy.wetherall@activiti.com", null);
|
|
createUser(USER_FIVE, "admin@alfresco.com", "admin@alfresco.com");
|
|
createUser(USER_SIX, "admin@alfresco.com", null);
|
|
createUser(USER_SEVEN, null, null);
|
|
|
|
// Authenticate as user one
|
|
AuthenticationUtil.setFullyAuthenticatedUser(USER_ONE);
|
|
|
|
siteId = GUID.generate();
|
|
|
|
// Create a site to use as holder for our test google documents
|
|
siteService.createSite("sitePreset", siteId, "My Title", "My Description", SiteVisibility.PUBLIC);
|
|
NodeRef container = siteService.createContainer(siteId, "testComponent", null, null);
|
|
|
|
// Add some memberships to the site
|
|
siteService.setMembership(siteId, USER_TWO, SiteServiceImpl.SITE_COLLABORATOR);
|
|
siteService.setMembership(siteId, USER_THREE, SiteServiceImpl.SITE_CONTRIBUTOR);
|
|
siteService.setMembership(siteId, USER_FOUR, SiteServiceImpl.SITE_CONSUMER);
|
|
siteService.setMembership(siteId, USER_FIVE, SiteServiceImpl.SITE_COLLABORATOR);
|
|
siteService.setMembership(siteId, USER_SIX, SiteServiceImpl.SITE_COLLABORATOR);
|
|
siteService.setMembership(siteId, USER_SEVEN, SiteServiceImpl.SITE_COLLABORATOR);
|
|
|
|
// Create a folder in our site container
|
|
folder = fileFolderService.create(container, "myfolder" + GUID.generate(), ContentModel.TYPE_FOLDER).getNodeRef();
|
|
|
|
// Create test documents
|
|
nodeRefDoc = createTestDocument("mydoc.docx", "alfresco/subsystems/googledocs/default/test.docx", MimetypeMap.MIMETYPE_WORD);
|
|
nodeRefSpread = createTestDocument("mydoc.xls", "alfresco/subsystems/googledocs/default/testBook.xls", MimetypeMap.MIMETYPE_EXCEL);
|
|
//nodeRefSpread = createTestDocument("mydoc2.xlsx", "alfresco/subsystems/googledocs/default/test.xlsx", MimetypeMap.MIMETYPE_EXCEL);
|
|
|
|
// Create an empty content node (simulate creation of a new google doc in UI)
|
|
nodeRef2 = fileFolderService.create(folder, "mygoogledoc.xls", ContentModel.TYPE_CONTENT).getNodeRef();
|
|
nodeService.addAspect(nodeRef2, ASPECT_GOOGLEEDITABLE, null);
|
|
ContentWriter contentWriter = contentService.getWriter(nodeRef2, ContentModel.PROP_CONTENT, true);
|
|
contentWriter.setEncoding("UTF-8");
|
|
contentWriter.setMimetype(MimetypeMap.MIMETYPE_EXCEL);
|
|
contentWriter.putContent("");
|
|
}
|
|
|
|
private NodeRef createTestDocument(String name, String contentPath, String mimetype)
|
|
{
|
|
NodeRef nodeRef = fileFolderService.create(folder, name, ContentModel.TYPE_CONTENT).getNodeRef();
|
|
ContentWriter writer = contentService.getWriter(nodeRef, ContentModel.PROP_CONTENT, true);
|
|
writer.setEncoding("UTF-8");
|
|
writer.setMimetype(mimetype);
|
|
InputStream is = getClass().getClassLoader().getResourceAsStream(contentPath);
|
|
writer.putContent(is);
|
|
return nodeRef;
|
|
}
|
|
|
|
private void createUser(String userName, String email, String googleEmail)
|
|
{
|
|
if (authenticationService.authenticationExists(userName) == false)
|
|
{
|
|
authenticationService.createAuthentication(userName, "PWD".toCharArray());
|
|
|
|
PropertyMap ppOne = new PropertyMap(4);
|
|
ppOne.put(ContentModel.PROP_USERNAME, userName);
|
|
ppOne.put(ContentModel.PROP_FIRSTNAME, "firstName");
|
|
ppOne.put(ContentModel.PROP_LASTNAME, "lastName");
|
|
ppOne.put(ContentModel.PROP_JOBTITLE, "jobTitle");
|
|
|
|
if (email != null)
|
|
{
|
|
ppOne.put(ContentModel.PROP_EMAIL, email);
|
|
}
|
|
|
|
if (googleEmail != null)
|
|
{
|
|
ppOne.put(ContentModel.PROP_GOOGLEUSERNAME, googleEmail);
|
|
}
|
|
|
|
personService.createPerson(ppOne);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
protected void tearDown() throws Exception
|
|
{
|
|
siteService.deleteSite(siteId);
|
|
|
|
if (userTransaction != null)
|
|
{
|
|
userTransaction.commit();
|
|
}
|
|
}
|
|
|
|
private boolean isGoogleServiceAvailable()
|
|
{
|
|
return googleDocsService.isEnabled();
|
|
}
|
|
|
|
public void testGoogleDocUploadDownload() throws Exception
|
|
{
|
|
if (isGoogleServiceAvailable() == true)
|
|
{
|
|
googleDocsService.createGoogleDoc(nodeRefDoc, GoogleDocsPermissionContext.SHARE_WRITE);
|
|
|
|
assertTrue(nodeService.hasAspect(nodeRefDoc, ASPECT_GOOGLERESOURCE));
|
|
assertNotNull(nodeService.getProperty(nodeRefDoc, PROP_URL));
|
|
assertNotNull(nodeService.getProperty(nodeRefDoc, PROP_RESOURCE_ID));
|
|
assertNotNull(nodeService.getProperty(nodeRefDoc, PROP_RESOURCE_TYPE));
|
|
|
|
System.out.println("For node ref " + nodeRefDoc.toString());
|
|
System.out.println("Google doc URL: " + nodeService.getProperty(nodeRefDoc, PROP_URL));
|
|
System.out.println("Google doc type: " + nodeService.getProperty(nodeRefDoc, PROP_RESOURCE_TYPE));
|
|
System.out.println("Google doc id: " + nodeService.getProperty(nodeRefDoc, PROP_RESOURCE_ID));
|
|
String downloadFile = downloadFile(googleDocsService.getGoogleDocContent(nodeRefDoc), ".doc");
|
|
System.out.println("Download file: " + downloadFile);
|
|
|
|
googleDocsService.createGoogleDoc(nodeRefSpread, GoogleDocsPermissionContext.SHARE_WRITE);
|
|
|
|
assertTrue(nodeService.hasAspect(nodeRefSpread, ASPECT_GOOGLERESOURCE));
|
|
assertNotNull(nodeService.getProperty(nodeRefSpread, PROP_URL));
|
|
assertNotNull(nodeService.getProperty(nodeRefSpread, PROP_RESOURCE_ID));
|
|
assertNotNull(nodeService.getProperty(nodeRefSpread, PROP_RESOURCE_TYPE));
|
|
|
|
System.out.println("Google doc URL: " + nodeService.getProperty(nodeRefSpread, PROP_URL));
|
|
System.out.println("Google doc type: " + nodeService.getProperty(nodeRefSpread, PROP_RESOURCE_TYPE));
|
|
System.out.println("Google doc id: " + nodeService.getProperty(nodeRefSpread, PROP_RESOURCE_ID));
|
|
downloadFile = downloadFile(googleDocsService.getGoogleDocContent(nodeRefSpread), ".xls");
|
|
System.out.println("Download file: " + downloadFile);
|
|
|
|
googleDocsService.createGoogleDoc(nodeRef2, GoogleDocsPermissionContext.SHARE_WRITE);
|
|
}
|
|
|
|
}
|
|
|
|
public void testCheckOutCheckIn() throws Exception
|
|
{
|
|
if (isGoogleServiceAvailable() == true)
|
|
{
|
|
// Check out the empty google document
|
|
NodeRef workingCopy = checkOutCheckInService.checkout(nodeRef2);
|
|
|
|
assertTrue(nodeService.hasAspect(workingCopy, ASPECT_GOOGLERESOURCE));
|
|
assertNotNull(nodeService.getProperty(workingCopy, PROP_URL));
|
|
assertNotNull(nodeService.getProperty(workingCopy, PROP_RESOURCE_ID));
|
|
assertNotNull(nodeService.getProperty(workingCopy, PROP_RESOURCE_TYPE));
|
|
|
|
System.out.println("Google doc URL: " + nodeService.getProperty(workingCopy, PROP_URL));
|
|
System.out.println("Google doc type: " + nodeService.getProperty(workingCopy, PROP_RESOURCE_TYPE));
|
|
System.out.println("Google doc id: " + nodeService.getProperty(workingCopy, PROP_RESOURCE_ID));
|
|
|
|
checkOutCheckInService.checkin(workingCopy, null);
|
|
|
|
assertFalse(nodeService.hasAspect(nodeRef2, ASPECT_GOOGLERESOURCE));
|
|
ContentReader contentReader = contentService.getReader(nodeRef2, ContentModel.PROP_CONTENT);
|
|
assertNotNull(contentReader);
|
|
|
|
// Lets try and delete the checked in node reference
|
|
nodeService.deleteNode(nodeRef2);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* http://issues.alfresco.com/jira/browse/ALF-5060
|
|
*/
|
|
public void testALF5060() throws Exception
|
|
{
|
|
if (isGoogleServiceAvailable() == true)
|
|
{
|
|
// Create a rendition definition (doc -> pdf)
|
|
RenditionDefinition def = renditionService.createRenditionDefinition(
|
|
QName.createQName(NamespaceService.CONTENT_MODEL_1_0_URI, "myDef"),
|
|
ReformatRenderingEngine.NAME);
|
|
def.setExecuteAsynchronously(false);
|
|
def.setParameterValue(AbstractRenderingEngine.PARAM_MIME_TYPE, MimetypeMap.MIMETYPE_FLASH);
|
|
|
|
// Create a word document
|
|
File fileDoc = AbstractContentTransformerTest.loadQuickTestFile("doc");
|
|
NodeRef nodeRef = fileFolderService.create(folder, "testing.doc", ContentModel.TYPE_CONTENT).getNodeRef();
|
|
nodeService.addAspect(nodeRef, ASPECT_GOOGLEEDITABLE, null);
|
|
ContentWriter contentWriter = contentService.getWriter(nodeRef, ContentModel.PROP_CONTENT, true);
|
|
contentWriter.setEncoding("UTF-8");
|
|
contentWriter.setMimetype(MimetypeMap.MIMETYPE_WORD);
|
|
contentWriter.putContent(fileDoc);
|
|
|
|
renditionService.render(nodeRef, def);
|
|
|
|
NodeRef workingCopy = checkOutCheckInService.checkout(nodeRef);
|
|
assertTrue(nodeService.hasAspect(workingCopy, ASPECT_GOOGLERESOURCE));
|
|
assertNotNull(nodeService.getProperty(workingCopy, PROP_URL));
|
|
assertNotNull(nodeService.getProperty(workingCopy, PROP_RESOURCE_ID));
|
|
assertNotNull(nodeService.getProperty(workingCopy, PROP_RESOURCE_TYPE));
|
|
System.out.println("Google doc URL: " + nodeService.getProperty(workingCopy, PROP_URL));
|
|
System.out.println("Google doc type: " + nodeService.getProperty(workingCopy, PROP_RESOURCE_TYPE));
|
|
System.out.println("Google doc id: " + nodeService.getProperty(workingCopy, PROP_RESOURCE_ID));
|
|
checkOutCheckInService.checkin(workingCopy, null);
|
|
|
|
renditionService.render(nodeRef, def);
|
|
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Utility method to download input stream to a file for inspection
|
|
*
|
|
* @param inStream
|
|
* @param ext
|
|
* @return
|
|
* @throws IOException
|
|
* @throws MalformedURLException
|
|
* @throws ServiceException
|
|
*/
|
|
private String downloadFile(InputStream inStream, String ext) throws IOException, MalformedURLException, ServiceException
|
|
{
|
|
File file = File.createTempFile("googleDocTest", ext);
|
|
String filePath = file.getAbsolutePath();
|
|
FileOutputStream outStream = null;
|
|
try
|
|
{
|
|
outStream = new FileOutputStream(filePath);
|
|
|
|
int c;
|
|
while ((c = inStream.read()) != -1)
|
|
{
|
|
outStream.write(c);
|
|
}
|
|
}
|
|
finally
|
|
{
|
|
if (inStream != null)
|
|
{
|
|
inStream.close();
|
|
}
|
|
if (outStream != null)
|
|
{
|
|
outStream.flush();
|
|
outStream.close();
|
|
}
|
|
}
|
|
|
|
return filePath;
|
|
}
|
|
}
|