/**
* Copyright (c) 2000-present Liferay, Inc. All rights reserved.
*
* This library 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 2.1 of the License, or (at your option)
* any later version.
*
* This library 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.
*/
package com.liferay.message.boards.lar.test;
import com.liferay.arquillian.extension.junit.bridge.junit.Arquillian;
import com.liferay.document.library.kernel.exception.NoSuchFileEntryException;
import com.liferay.document.library.kernel.exception.NoSuchFolderException;
import com.liferay.document.library.kernel.model.DLFileEntry;
import com.liferay.document.library.kernel.model.DLFolder;
import com.liferay.exportimport.kernel.lar.ExportImportClassedModelUtil;
import com.liferay.exportimport.kernel.lar.StagedModelDataHandler;
import com.liferay.exportimport.kernel.lar.StagedModelDataHandlerRegistryUtil;
import com.liferay.message.boards.kernel.model.MBCategory;
import com.liferay.message.boards.kernel.model.MBCategoryConstants;
import com.liferay.message.boards.kernel.model.MBMessage;
import com.liferay.message.boards.kernel.model.MBMessageConstants;
import com.liferay.message.boards.kernel.service.MBCategoryLocalServiceUtil;
import com.liferay.message.boards.kernel.service.MBCategoryServiceUtil;
import com.liferay.message.boards.kernel.service.MBMessageLocalServiceUtil;
import com.liferay.portal.kernel.exception.NoSuchModelException;
import com.liferay.portal.kernel.model.Group;
import com.liferay.portal.kernel.model.Repository;
import com.liferay.portal.kernel.model.StagedModel;
import com.liferay.portal.kernel.repository.model.FileEntry;
import com.liferay.portal.kernel.repository.model.Folder;
import com.liferay.portal.kernel.service.RepositoryLocalServiceUtil;
import com.liferay.portal.kernel.service.ServiceContext;
import com.liferay.portal.kernel.test.rule.AggregateTestRule;
import com.liferay.portal.kernel.test.rule.Sync;
import com.liferay.portal.kernel.test.rule.SynchronousDestinationTestRule;
import com.liferay.portal.kernel.test.util.RandomTestUtil;
import com.liferay.portal.kernel.test.util.ServiceContextTestUtil;
import com.liferay.portal.kernel.test.util.TestPropsValues;
import com.liferay.portal.kernel.util.ObjectValuePair;
import com.liferay.portal.kernel.util.StringPool;
import com.liferay.portal.kernel.workflow.WorkflowConstants;
import com.liferay.portal.lar.test.BaseWorkflowedStagedModelDataHandlerTestCase;
import com.liferay.portal.test.rule.LiferayIntegrationTestRule;
import com.liferay.portlet.messageboards.util.test.MBTestUtil;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.junit.Assert;
import org.junit.ClassRule;
import org.junit.Rule;
import org.junit.runner.RunWith;
/**
* @author Daniel Kocsis
*/
@RunWith(Arquillian.class)
@Sync
public class MBMessageStagedModelDataHandlerTest
extends BaseWorkflowedStagedModelDataHandlerTestCase {
@ClassRule
@Rule
public static final AggregateTestRule aggregateTestRule =
new AggregateTestRule(
new LiferayIntegrationTestRule(),
SynchronousDestinationTestRule.INSTANCE);
@Override
protected Map<String, List<StagedModel>> addDependentStagedModelsMap(
Group group)
throws Exception {
Map<String, List<StagedModel>> dependentStagedModelsMap =
new HashMap<>();
ServiceContext serviceContext =
ServiceContextTestUtil.getServiceContext(
group.getGroupId(), TestPropsValues.getUserId());
MBCategory category = MBCategoryServiceUtil.addCategory(
TestPropsValues.getUserId(),
MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
RandomTestUtil.randomString(), StringPool.BLANK, serviceContext);
addDependentStagedModel(
dependentStagedModelsMap, MBCategory.class, category);
return dependentStagedModelsMap;
}
@Override
protected StagedModel addStagedModel(
Group group,
Map<String, List<StagedModel>> dependentStagedModelsMap)
throws Exception {
List<StagedModel> dependentStagedModels = dependentStagedModelsMap.get(
MBCategory.class.getSimpleName());
MBCategory category = (MBCategory)dependentStagedModels.get(0);
List<ObjectValuePair<String, InputStream>> objectValuePairs =
MBTestUtil.getInputStreamOVPs(
"attachment.txt", getClass(), StringPool.BLANK);
ServiceContext serviceContext =
ServiceContextTestUtil.getServiceContext(
group.getGroupId(), TestPropsValues.getUserId());
serviceContext.setWorkflowAction(WorkflowConstants.ACTION_PUBLISH);
MBMessage message = MBMessageLocalServiceUtil.addMessage(
TestPropsValues.getUserId(), RandomTestUtil.randomString(),
group.getGroupId(), category.getCategoryId(), 0, 0,
RandomTestUtil.randomString(), RandomTestUtil.randomString(),
MBMessageConstants.DEFAULT_FORMAT, objectValuePairs, false, 0.0,
false, serviceContext);
MBMessageLocalServiceUtil.updateAnswer(message, true, false);
List<FileEntry> attachmentsFileEntries =
message.getAttachmentsFileEntries();
FileEntry fileEntry = attachmentsFileEntries.get(0);
Folder folder = fileEntry.getFolder();
while (folder != null) {
addDependentStagedModel(
dependentStagedModelsMap, DLFolder.class, folder);
folder = folder.getParentFolder();
}
addDependentStagedModel(
dependentStagedModelsMap, DLFileEntry.class,
attachmentsFileEntries.get(0));
Repository repository = RepositoryLocalServiceUtil.getRepository(
fileEntry.getRepositoryId());
addDependentStagedModel(
dependentStagedModelsMap, Repository.class, repository);
return message;
}
@Override
protected List<StagedModel> addWorkflowedStagedModels(Group group)
throws Exception {
List<StagedModel> stagedModels = new ArrayList<>();
ServiceContext serviceContext =
ServiceContextTestUtil.getServiceContext(
group.getGroupId(), TestPropsValues.getUserId());
MBMessage approvedMessage = MBTestUtil.addMessageWithWorkflow(
TestPropsValues.getUserId(), group.getGroupId(),
MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
RandomTestUtil.randomString(), RandomTestUtil.randomString(), true,
serviceContext);
stagedModels.add(approvedMessage);
MBMessage pendingMessage = MBTestUtil.addMessageWithWorkflow(
TestPropsValues.getUserId(), group.getGroupId(),
MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
RandomTestUtil.randomString(), RandomTestUtil.randomString(), false,
serviceContext);
stagedModels.add(pendingMessage);
return stagedModels;
}
@Override
protected void deleteStagedModel(
StagedModel stagedModel,
Map<String, List<StagedModel>> dependentStagedModelsMap,
Group group)
throws Exception {
@SuppressWarnings("rawtypes")
StagedModelDataHandler stagedModelDataHandler =
StagedModelDataHandlerRegistryUtil.getStagedModelDataHandler(
ExportImportClassedModelUtil.getClassName(stagedModel));
stagedModelDataHandler.deleteStagedModel(stagedModel);
for (List<StagedModel> dependentStagedModels :
dependentStagedModelsMap.values()) {
for (StagedModel dependentStagedModel : dependentStagedModels) {
try {
stagedModelDataHandler =
StagedModelDataHandlerRegistryUtil.
getStagedModelDataHandler(
ExportImportClassedModelUtil.getClassName(
dependentStagedModel));
stagedModelDataHandler.deleteStagedModel(
dependentStagedModel);
}
catch (NoSuchModelException nsme) {
if (!(nsme instanceof NoSuchFileEntryException) &&
!(nsme instanceof NoSuchFolderException)) {
throw nsme;
}
}
}
}
}
@Override
protected StagedModel getStagedModel(String uuid, Group group) {
try {
return MBMessageLocalServiceUtil.getMBMessageByUuidAndGroupId(
uuid, group.getGroupId());
}
catch (Exception e) {
return null;
}
}
@Override
protected Class<? extends StagedModel> getStagedModelClass() {
return MBMessage.class;
}
@Override
protected void validateImport(
Map<String, List<StagedModel>> dependentStagedModelsMap,
Group group)
throws Exception {
List<StagedModel> dependentStagedModels = dependentStagedModelsMap.get(
MBCategory.class.getSimpleName());
Assert.assertEquals(
dependentStagedModels.toString(), 1, dependentStagedModels.size());
MBCategory category = (MBCategory)dependentStagedModels.get(0);
MBCategoryLocalServiceUtil.getMBCategoryByUuidAndGroupId(
category.getUuid(), group.getGroupId());
}
@Override
protected void validateImport(
StagedModel stagedModel, StagedModelAssets stagedModelAssets,
Map<String, List<StagedModel>> dependentStagedModelsMap,
Group group)
throws Exception {
super.validateImport(
stagedModel, stagedModelAssets, dependentStagedModelsMap, group);
MBMessage importedMessage = (MBMessage)getStagedModel(
stagedModel.getUuid(), group);
Assert.assertEquals(
1, importedMessage.getAttachmentsFileEntriesCount());
Assert.assertTrue(importedMessage.isAnswer());
}
@Override
protected void validateImportedStagedModel(
StagedModel stagedModel, StagedModel importedStagedModel)
throws Exception {
super.validateImportedStagedModel(stagedModel, importedStagedModel);
MBMessage message = (MBMessage)stagedModel;
MBMessage importedMessage = (MBMessage)importedStagedModel;
Assert.assertEquals(message.getSubject(), importedMessage.getSubject());
Assert.assertEquals(message.getFormat(), importedMessage.getFormat());
Assert.assertEquals(
message.isAnonymous(), importedMessage.isAnonymous());
Assert.assertEquals(
message.getPriority(), importedMessage.getPriority(), 0L);
Assert.assertEquals(
message.isAllowPingbacks(), importedMessage.isAllowPingbacks());
Assert.assertEquals(message.isAnswer(), importedMessage.isAnswer());
}
}