/**
* 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.trash.test;
import com.liferay.arquillian.extension.junit.bridge.junit.Arquillian;
import com.liferay.asset.kernel.model.AssetEntry;
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.model.MBThread;
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.message.boards.kernel.service.MBThreadLocalServiceUtil;
import com.liferay.message.boards.kernel.service.MBThreadServiceUtil;
import com.liferay.portal.kernel.model.BaseModel;
import com.liferay.portal.kernel.model.ClassedModel;
import com.liferay.portal.kernel.model.Group;
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.service.test.ServiceTestUtil;
import com.liferay.portal.test.rule.LiferayIntegrationTestRule;
import com.liferay.portlet.messageboards.util.test.MBTestUtil;
import com.liferay.trash.test.util.BaseTrashHandlerTestCase;
import com.liferay.trash.test.util.DefaultWhenIsAssetable;
import com.liferay.trash.test.util.DefaultWhenIsIndexableBaseModel;
import com.liferay.trash.test.util.WhenHasGrandParent;
import com.liferay.trash.test.util.WhenHasMyBaseModel;
import com.liferay.trash.test.util.WhenHasRecentBaseModelCount;
import com.liferay.trash.test.util.WhenIsAssetable;
import com.liferay.trash.test.util.WhenIsAssetableBaseModel;
import com.liferay.trash.test.util.WhenIsIndexableBaseModel;
import com.liferay.trash.test.util.WhenIsMoveableFromTrashBaseModel;
import com.liferay.trash.test.util.WhenIsRestorableBaseModel;
import com.liferay.trash.test.util.WhenIsUpdatableBaseModel;
import java.io.InputStream;
import java.util.Calendar;
import java.util.Collections;
import org.junit.Assert;
import org.junit.Before;
import org.junit.ClassRule;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
/**
* @author Zsolt Berentey
* @author Eduardo Garcia
*/
@RunWith(Arquillian.class)
@Sync
public class MBThreadTrashHandlerTest
extends BaseTrashHandlerTestCase
implements WhenHasGrandParent, WhenHasMyBaseModel,
WhenHasRecentBaseModelCount, WhenIsAssetableBaseModel,
WhenIsIndexableBaseModel, WhenIsMoveableFromTrashBaseModel,
WhenIsRestorableBaseModel, WhenIsUpdatableBaseModel {
@ClassRule
@Rule
public static final AggregateTestRule aggregateTestRule =
new AggregateTestRule(
new LiferayIntegrationTestRule(),
SynchronousDestinationTestRule.INSTANCE);
@Override
public AssetEntry fetchAssetEntry(ClassedModel classedModel)
throws Exception {
return _whenIsAssetable.fetchAssetEntry(classedModel);
}
@Override
public int getMineBaseModelsCount(long groupId, long userId)
throws Exception {
return MBThreadServiceUtil.getGroupThreadsCount(
groupId, userId, WorkflowConstants.STATUS_APPROVED);
}
@Override
public String getParentBaseModelClassName() {
Class<MBCategory> mbCategoryClass = MBCategory.class;
return mbCategoryClass.getName();
}
@Override
public int getRecentBaseModelsCount(long groupId) throws Exception {
Calendar calendar = Calendar.getInstance();
calendar.add(Calendar.HOUR, -1);
return MBThreadServiceUtil.getGroupThreadsCount(
groupId, 0, calendar.getTime(), WorkflowConstants.STATUS_APPROVED);
}
@Override
public String getSearchKeywords() {
return _SUBJECT;
}
@Override
public boolean isAssetEntryVisible(ClassedModel classedModel, long classPK)
throws Exception {
MBMessage rootMessage = MBMessageLocalServiceUtil.getMBMessage(
((MBThread)classedModel).getRootMessageId());
return _whenIsAssetable.isAssetEntryVisible(
rootMessage, getAssetClassPK(rootMessage));
}
@Override
public BaseModel<?> moveBaseModelFromTrash(
ClassedModel classedModel, Group group,
ServiceContext serviceContext)
throws Exception {
BaseModel<?> parentBaseModel = getParentBaseModel(
group, serviceContext);
MBThreadServiceUtil.moveThreadFromTrash(
(Long)parentBaseModel.getPrimaryKeyObj(),
(Long)classedModel.getPrimaryKeyObj());
return parentBaseModel;
}
@Override
public void moveParentBaseModelToTrash(long primaryKey) throws Exception {
MBCategoryServiceUtil.moveCategoryToTrash(primaryKey);
}
@Override
public int searchBaseModelsCount(Class<?> clazz, long groupId)
throws Exception {
return _whenIsIndexableBaseModel.searchBaseModelsCount(
MBMessage.class, groupId);
}
@Override
public int searchTrashEntriesCount(
String keywords, ServiceContext serviceContext)
throws Exception {
return _whenIsIndexableBaseModel.searchTrashEntriesCount(
keywords, serviceContext);
}
@Before
@Override
public void setUp() throws Exception {
super.setUp();
ServiceTestUtil.setUser(TestPropsValues.getUser());
}
@Test
public void testCategoryMessageCount() throws Exception {
ServiceContext serviceContext =
ServiceContextTestUtil.getServiceContext(group.getGroupId());
BaseModel<?> parentBaseModel = getParentBaseModel(
group, serviceContext);
int initialBaseModelsCount = getMessageCount(
(Long)parentBaseModel.getPrimaryKeyObj());
baseModel = addBaseModel(parentBaseModel, serviceContext);
Assert.assertEquals(
initialBaseModelsCount + 1,
getMessageCount((Long)parentBaseModel.getPrimaryKeyObj()));
moveBaseModelToTrash((Long)baseModel.getPrimaryKeyObj());
Assert.assertEquals(
initialBaseModelsCount,
getMessageCount((Long)parentBaseModel.getPrimaryKeyObj()));
baseModel = addBaseModel(parentBaseModel, serviceContext);
Assert.assertEquals(
initialBaseModelsCount + 1,
getMessageCount((Long)parentBaseModel.getPrimaryKeyObj()));
replyMessage(baseModel);
Assert.assertEquals(
initialBaseModelsCount + 2,
getMessageCount((Long)parentBaseModel.getPrimaryKeyObj()));
}
@Override
public BaseModel<?> updateBaseModel(
long primaryKey, ServiceContext serviceContext)
throws Exception {
MBThread thread = MBThreadLocalServiceUtil.getThread(primaryKey);
if (serviceContext.getWorkflowAction() ==
WorkflowConstants.ACTION_SAVE_DRAFT) {
thread = MBThreadLocalServiceUtil.updateStatus(
TestPropsValues.getUserId(), primaryKey,
WorkflowConstants.STATUS_DRAFT);
}
return thread;
}
@Override
protected BaseModel<?> addBaseModelWithWorkflow(
BaseModel<?> parentBaseModel, ServiceContext serviceContext)
throws Exception {
MBCategory category = (MBCategory)parentBaseModel;
MBMessage message = MBTestUtil.addMessageWithWorkflow(
serviceContext.getUserId(), category.getGroupId(),
category.getCategoryId(), getSearchKeywords(), getSearchKeywords(),
true, serviceContext);
return message.getThread();
}
@Override
protected BaseModel<?> addBaseModelWithWorkflow(
ServiceContext serviceContext)
throws Exception {
MBMessage message = MBMessageLocalServiceUtil.addMessage(
TestPropsValues.getUserId(), RandomTestUtil.randomString(),
serviceContext.getScopeGroupId(),
MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
RandomTestUtil.randomString(), RandomTestUtil.randomString(),
serviceContext);
return message.getThread();
}
@Override
protected void deleteParentBaseModel(
BaseModel<?> parentBaseModel, boolean includeTrashedEntries)
throws Exception {
MBCategory parentCategory = (MBCategory)parentBaseModel;
MBCategoryLocalServiceUtil.deleteCategory(parentCategory, false);
}
@Override
protected BaseModel<?> getBaseModel(long primaryKey) throws Exception {
return MBThreadLocalServiceUtil.getThread(primaryKey);
}
@Override
protected Class<?> getBaseModelClass() {
return MBThread.class;
}
protected int getMessageCount(long categoryId) throws Exception {
MBCategory category = MBCategoryLocalServiceUtil.getCategory(
categoryId);
return category.getMessageCount();
}
@Override
protected int getNotInTrashBaseModelsCount(BaseModel<?> parentBaseModel)
throws Exception {
MBCategory category = (MBCategory)parentBaseModel;
return MBThreadLocalServiceUtil.getCategoryThreadsCount(
category.getGroupId(), category.getCategoryId(),
WorkflowConstants.STATUS_APPROVED);
}
@Override
protected BaseModel<?> getParentBaseModel(
Group group, long parentBaseModelId, ServiceContext serviceContext)
throws Exception {
return MBCategoryLocalServiceUtil.addCategory(
TestPropsValues.getUserId(), parentBaseModelId,
RandomTestUtil.randomString(), StringPool.BLANK, serviceContext);
}
@Override
protected BaseModel<?> getParentBaseModel(
Group group, ServiceContext serviceContext)
throws Exception {
return MBCategoryLocalServiceUtil.addCategory(
TestPropsValues.getUserId(),
MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
RandomTestUtil.randomString(), StringPool.BLANK, serviceContext);
}
@Override
protected String getUniqueTitle(BaseModel<?> baseModel) {
return null;
}
@Override
protected boolean isBaseModelContainerModel() {
return false;
}
@Override
protected void moveBaseModelToTrash(long primaryKey) throws Exception {
MBThreadServiceUtil.moveThreadToTrash(primaryKey);
}
protected void replyMessage(BaseModel<?> baseModel) throws Exception {
MBThread thread = (MBThread)baseModel;
ServiceContext serviceContext =
ServiceContextTestUtil.getServiceContext(
thread.getGroupId(), TestPropsValues.getUserId());
MBMessageLocalServiceUtil.addMessage(
TestPropsValues.getUserId(), RandomTestUtil.randomString(),
thread.getGroupId(), thread.getCategoryId(), thread.getThreadId(),
thread.getRootMessageId(), RandomTestUtil.randomString(),
RandomTestUtil.randomString(), MBMessageConstants.DEFAULT_FORMAT,
Collections.<ObjectValuePair<String, InputStream>>emptyList(),
false, 0.0, false, serviceContext);
}
private static final String _SUBJECT = "Subject";
private final WhenIsAssetable _whenIsAssetable =
new DefaultWhenIsAssetable();
private final WhenIsIndexableBaseModel _whenIsIndexableBaseModel =
new DefaultWhenIsIndexableBaseModel();
}