mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-10-29 15:21:53 +00:00
ab4ca7177f
51903 to 54309 git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@54310 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
387 lines
14 KiB
Java
387 lines
14 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.cmis.renditions;
|
|
|
|
import static org.junit.Assert.fail;
|
|
|
|
import java.io.File;
|
|
import java.io.Serializable;
|
|
import java.net.URL;
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.Comparator;
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
|
|
import org.alfresco.cmis.CMISRendition;
|
|
import org.alfresco.cmis.mapping.BaseCMISTest;
|
|
import org.alfresco.model.ContentModel;
|
|
import org.alfresco.repo.content.MimetypeMap;
|
|
import org.alfresco.repo.content.transform.AbstractContentTransformerTest;
|
|
import org.alfresco.repo.content.transform.magick.ImageTransformationOptions;
|
|
import org.alfresco.repo.thumbnail.ThumbnailDefinition;
|
|
import org.alfresco.service.cmr.repository.ContentReader;
|
|
import org.alfresco.service.cmr.repository.ContentWriter;
|
|
import org.alfresco.service.cmr.repository.NoTransformerException;
|
|
import org.alfresco.service.cmr.repository.NodeRef;
|
|
import org.alfresco.service.cmr.repository.TransformationOptions;
|
|
import org.alfresco.service.namespace.NamespaceService;
|
|
import org.alfresco.service.namespace.QName;
|
|
|
|
/**
|
|
* @author Stas Sokolovsky
|
|
*/
|
|
public class CMISRenditionServiceTest extends BaseCMISTest
|
|
{
|
|
private static final String[] THUMBNAIL_NAMES = new String[] { "doclib", "webpreview", "imgpreview" };
|
|
private static final String[] THUMBNAIL_KINDS = new String[] { "cmis:thumbnail", "alf:webpreview", "alf:webpreview" };
|
|
|
|
private NodeRef document;
|
|
private List<CMISRendition> documentRenditions = new ArrayList<CMISRendition>();
|
|
private CMISRendition icon16Rendition = new CMISRenditionImpl(null, "alf:icon16", "image/gif", "alf:icon16", 16, 16, null, null, null);
|
|
private CMISRendition icon32Rendition = new CMISRenditionImpl(null, "alf:icon32", "image/gif", "alf:icon32", 32, 32, null, null, null);
|
|
|
|
public void setUp() throws Exception
|
|
{
|
|
super.setUp();
|
|
|
|
String documentName = "TestDocument" + System.currentTimeMillis();
|
|
document = createPdfDocument(documentName);
|
|
|
|
documentRenditions = new ArrayList<CMISRendition>();
|
|
for (int i = 0; i < THUMBNAIL_NAMES.length; ++i)
|
|
{
|
|
documentRenditions.add(createRendition(document, THUMBNAIL_NAMES[i], THUMBNAIL_KINDS[i]));
|
|
}
|
|
}
|
|
|
|
public void testGetAllRenditions() throws Exception
|
|
{
|
|
List<CMISRendition> receivedRenditions = cmisRenditionService.getRenditions(document, "*");
|
|
|
|
List<CMISRendition> expectedRenditions = new ArrayList<CMISRendition>();
|
|
expectedRenditions.addAll(documentRenditions);
|
|
expectedRenditions.add(icon16Rendition);
|
|
expectedRenditions.add(icon32Rendition);
|
|
|
|
assertRendiions(receivedRenditions, expectedRenditions);
|
|
}
|
|
|
|
public void testGetRenditionsByKind() throws Exception
|
|
{
|
|
testGetRenditionsByKind("cmis:thumbnail");
|
|
testGetRenditionsByKind("alf:webpreview");
|
|
testGetRenditionsByKind("alf:icon16");
|
|
testGetRenditionsByKind("alf:icon32");
|
|
|
|
testGetRenditionsByKind("alf:webpreview", "alf:icon32");
|
|
testGetRenditionsByKind("cmis:thumbnail", "alf:webpreview", "alf:icon16", "alf:icon32");
|
|
}
|
|
|
|
public void testGetRenditionsByMimetype() throws Exception
|
|
{
|
|
for (CMISRendition rendition : documentRenditions)
|
|
{
|
|
testGetRenditionsByMimetype(rendition.getMimeType());
|
|
}
|
|
testGetRenditionsByMimetype(icon16Rendition.getMimeType());
|
|
testGetRenditionsByMimetype(icon32Rendition.getMimeType());
|
|
|
|
String[] mimetypes = getDifferentMimetypes();
|
|
testGetRenditionsByMimetype(mimetypes);
|
|
}
|
|
|
|
public void testEmptyFilter() throws Exception
|
|
{
|
|
List<CMISRendition> receivedRenditions = cmisRenditionService.getRenditions(document, null);
|
|
assertTrue(receivedRenditions == null || receivedRenditions.size() == 0);
|
|
|
|
receivedRenditions = cmisRenditionService.getRenditions(document, "cmis:none");
|
|
assertTrue(receivedRenditions == null || receivedRenditions.size() == 0);
|
|
}
|
|
|
|
public void testRenditionsByBaseMimetype() throws Exception
|
|
{
|
|
String[] mimetypes = getDifferentMimetypes();
|
|
List<String> baseMimetypeFilters = new ArrayList<String>();
|
|
for (String mimetype : mimetypes)
|
|
{
|
|
String baseMimetype = getBaseType(mimetype);
|
|
String filter = baseMimetype + "/*";
|
|
baseMimetypeFilters.add(filter);
|
|
|
|
testGetRenditionsByMimetype(filter);
|
|
}
|
|
|
|
testGetRenditionsByMimetype(baseMimetypeFilters.toArray(new String[0]));
|
|
}
|
|
|
|
public void testMixedFilter() throws Exception
|
|
{
|
|
String[] mimetypes = getDifferentMimetypes();
|
|
testGetRenditions(THUMBNAIL_KINDS, mimetypes);
|
|
}
|
|
|
|
private void testGetRenditionsByMimetype(String... mimetypes) throws Exception
|
|
{
|
|
testGetRenditions(null, mimetypes);
|
|
}
|
|
|
|
private void testGetRenditionsByKind(String... kinds) throws Exception
|
|
{
|
|
testGetRenditions(kinds, null);
|
|
}
|
|
|
|
private void testGetRenditions(String[] kinds, String[] mimetypes) throws Exception
|
|
{
|
|
String filter = createFilter(kinds, mimetypes);
|
|
List<CMISRendition> receivedRenditions = cmisRenditionService.getRenditions(document, filter);
|
|
|
|
List<CMISRendition> expectedRenditions = new ArrayList<CMISRendition>();
|
|
if (kinds != null)
|
|
{
|
|
for (String kind : kinds)
|
|
{
|
|
expectedRenditions.addAll(getRenditionsByKind(kind));
|
|
}
|
|
}
|
|
if (mimetypes != null)
|
|
{
|
|
for (String mimetype : mimetypes)
|
|
{
|
|
expectedRenditions.addAll(getRenditionsByMimetype(mimetype));
|
|
}
|
|
}
|
|
|
|
assertRendiions(receivedRenditions, expectedRenditions);
|
|
}
|
|
|
|
protected void tearDown() throws Exception
|
|
{
|
|
fileFolderService.delete(document);
|
|
super.tearDown();
|
|
}
|
|
|
|
private CMISRendition createRendition(NodeRef nodeRef, String thumbnailName, String kind)
|
|
{
|
|
ThumbnailDefinition details = thumbnailService.getThumbnailRegistry().getThumbnailDefinition(thumbnailName);
|
|
NodeRef thumbnailNodeRef = thumbnailService.createThumbnail(nodeRef, ContentModel.PROP_CONTENT, details.getMimetype(), details.getTransformationOptions(), details
|
|
.getName());
|
|
|
|
CMISRenditionImpl rendition = new CMISRenditionImpl();
|
|
rendition.setStreamId(thumbnailNodeRef.toString());
|
|
rendition.setKind(kind);
|
|
rendition.setMimeType(details.getMimetype());
|
|
if (details.getTransformationOptions() instanceof ImageTransformationOptions)
|
|
{
|
|
ImageTransformationOptions imageOptions = (ImageTransformationOptions) details.getTransformationOptions();
|
|
rendition.setWidth(imageOptions.getResizeOptions().getWidth());
|
|
rendition.setHeight(imageOptions.getResizeOptions().getHeight());
|
|
}
|
|
|
|
return rendition;
|
|
}
|
|
|
|
private NodeRef createPdfDocument(final String nodeName)
|
|
{
|
|
Map<QName, Serializable> props = new HashMap<QName, Serializable>();
|
|
props.put(ContentModel.PROP_NAME, nodeName);
|
|
NodeRef result = nodeService.createNode(rootNodeRef,
|
|
ContentModel.ASSOC_CONTAINS,
|
|
QName.createQName(NamespaceService.CONTENT_MODEL_1_0_URI, nodeName),
|
|
ContentModel.TYPE_CONTENT,
|
|
props).getChildRef();
|
|
|
|
File file = loadQuickPdfFile();
|
|
|
|
// Set the content
|
|
ContentWriter writer = contentService.getWriter(result, ContentModel.PROP_CONTENT, true);
|
|
writer.setMimetype(MimetypeMap.MIMETYPE_PDF);
|
|
writer.setEncoding("UTF-8");
|
|
|
|
writer.putContent(file);
|
|
|
|
return result;
|
|
}
|
|
|
|
private File loadQuickPdfFile()
|
|
{
|
|
URL url = AbstractContentTransformerTest.class.getClassLoader().getResource("quick/quick.pdf");
|
|
if (url == null)
|
|
{
|
|
fail("Could not load pdf file");
|
|
}
|
|
File file = new File(url.getFile());
|
|
if (!file.exists())
|
|
{
|
|
fail("Could not load pdf file");
|
|
}
|
|
return file;
|
|
}
|
|
|
|
private void assertRendiions(List<CMISRendition> receivedRenditions, List<CMISRendition> expectedRenditions)
|
|
{
|
|
assertNotNull(receivedRenditions);
|
|
expectedRenditions = removeReplication(expectedRenditions);
|
|
assertTrue(receivedRenditions.size() == expectedRenditions.size());
|
|
|
|
for (CMISRendition rendition : receivedRenditions)
|
|
{
|
|
assertNotNull(rendition);
|
|
assertNotNull(rendition.getStreamId());
|
|
}
|
|
|
|
Collections.sort(receivedRenditions, renditionsComparator);
|
|
Collections.sort(expectedRenditions, renditionsComparator);
|
|
|
|
for (int i = 0; i < expectedRenditions.size(); ++i)
|
|
{
|
|
assertRendition(receivedRenditions.get(i), expectedRenditions.get(i));
|
|
}
|
|
}
|
|
|
|
private void assertRendition(CMISRendition receivedRendition, CMISRendition expectedRendition)
|
|
{
|
|
assertEquals(expectedRendition.getStreamId(), receivedRendition.getStreamId());
|
|
assertEquals(expectedRendition.getKind(), receivedRendition.getKind());
|
|
assertEquals(expectedRendition.getMimeType(), receivedRendition.getMimeType());
|
|
assertEquals(expectedRendition.getWidth(), receivedRendition.getWidth());
|
|
assertEquals(expectedRendition.getHeight(), receivedRendition.getHeight());
|
|
}
|
|
|
|
private List<CMISRendition> removeReplication(List<CMISRendition> renditions)
|
|
{
|
|
return new ArrayList<CMISRendition>(new HashSet<CMISRendition>(renditions));
|
|
}
|
|
|
|
private Comparator<CMISRendition> renditionsComparator = new Comparator<CMISRendition>()
|
|
{
|
|
public int compare(CMISRendition rendition1, CMISRendition rendition2)
|
|
{
|
|
return rendition1.getStreamId().compareTo(rendition2.getStreamId());
|
|
}
|
|
};
|
|
|
|
private List<CMISRendition> getRenditionsByKind(String kind)
|
|
{
|
|
return getRenditions(kind, null);
|
|
}
|
|
|
|
private List<CMISRendition> getRenditionsByMimetype(String mimetype)
|
|
{
|
|
return getRenditions(null, mimetype);
|
|
}
|
|
|
|
private List<CMISRendition> getRenditions(String kind, String mimetype)
|
|
{
|
|
List<CMISRendition> result = new ArrayList<CMISRendition>();
|
|
|
|
List<CMISRendition> allRenditions = new ArrayList<CMISRendition>(documentRenditions);
|
|
allRenditions.add(icon16Rendition);
|
|
allRenditions.add(icon32Rendition);
|
|
for (CMISRendition rendition : allRenditions)
|
|
{
|
|
if (isRenditionSatisfyConditions(rendition, kind, mimetype))
|
|
{
|
|
result.add(rendition);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
private boolean isRenditionSatisfyConditions(CMISRendition rendition, String kind, String mimetype)
|
|
{
|
|
if (kind != null)
|
|
{
|
|
if (!kind.equals(rendition.getKind()))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
if (mimetype != null)
|
|
{
|
|
if (mimetype.endsWith("/*"))
|
|
{
|
|
String baseMimetype = getBaseType(mimetype);
|
|
if (!rendition.getMimeType().startsWith(baseMimetype))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
else if (!mimetype.equals(rendition.getMimeType()))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
private String[] getDifferentMimetypes()
|
|
{
|
|
List<CMISRendition> allRenditions = new ArrayList<CMISRendition>(documentRenditions);
|
|
allRenditions.add(icon16Rendition);
|
|
allRenditions.add(icon32Rendition);
|
|
Set<String> mimetypes = new HashSet<String>();
|
|
for (CMISRendition rendition : allRenditions)
|
|
{
|
|
mimetypes.add(rendition.getMimeType());
|
|
}
|
|
return mimetypes.toArray(new String[0]);
|
|
}
|
|
|
|
private String getBaseType(String mimetype)
|
|
{
|
|
String baseMymetype = mimetype;
|
|
int subTypeIndex = mimetype.indexOf("/");
|
|
if (subTypeIndex > 0 || subTypeIndex < mimetype.length())
|
|
{
|
|
baseMymetype = mimetype.substring(0, subTypeIndex);
|
|
}
|
|
return baseMymetype;
|
|
}
|
|
|
|
private String createFilter(String[] kinds, String[] mimetypes)
|
|
{
|
|
StringBuilder filter = new StringBuilder();
|
|
if (kinds != null)
|
|
{
|
|
for (String kind : kinds)
|
|
{
|
|
filter.append(kind);
|
|
filter.append(",");
|
|
}
|
|
}
|
|
if (mimetypes != null)
|
|
{
|
|
for (String mimetype : mimetypes)
|
|
{
|
|
filter.append(mimetype);
|
|
filter.append(",");
|
|
}
|
|
}
|
|
filter.delete(filter.length() - 1, filter.length());
|
|
return filter.toString();
|
|
}
|
|
|
|
}
|