Java tutorial
/** * 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.PortalRunMode; 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.portlet.trash.test.BaseTrashHandlerTestCase; import com.liferay.portlet.trash.test.DefaultWhenIsAssetable; import com.liferay.portlet.trash.test.DefaultWhenIsIndexableBaseModel; import com.liferay.portlet.trash.test.WhenHasGrandParent; import com.liferay.portlet.trash.test.WhenHasMyBaseModel; import com.liferay.portlet.trash.test.WhenHasRecentBaseModelCount; import com.liferay.portlet.trash.test.WhenIsAssetable; import com.liferay.portlet.trash.test.WhenIsAssetableBaseModel; import com.liferay.portlet.trash.test.WhenIsIndexableBaseModel; import com.liferay.portlet.trash.test.WhenIsMoveableFromTrashBaseModel; import com.liferay.portlet.trash.test.WhenIsRestorableBaseModel; import com.liferay.portlet.trash.test.WhenIsUpdatableBaseModel; import java.io.InputStream; import java.util.Calendar; import java.util.Collections; import org.junit.After; 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()); _testMode = PortalRunMode.isTestMode(); PortalRunMode.setTestMode(true); } @After public void tearDown() throws Exception { PortalRunMode.setTestMode(_testMode); } @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 boolean _testMode; private final WhenIsAssetable _whenIsAssetable = new DefaultWhenIsAssetable(); private final WhenIsIndexableBaseModel _whenIsIndexableBaseModel = new DefaultWhenIsIndexableBaseModel(); }