com.liferay.bookmarks.trash.test.BookmarksFolderTrashHandlerTest.java Source code

Java tutorial

Introduction

Here is the source code for com.liferay.bookmarks.trash.test.BookmarksFolderTrashHandlerTest.java

Source

/**
 * 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.bookmarks.trash.test;

import com.liferay.arquillian.extension.junit.bridge.junit.Arquillian;
import com.liferay.asset.kernel.model.AssetEntry;
import com.liferay.bookmarks.model.BookmarksFolder;
import com.liferay.bookmarks.model.BookmarksFolderConstants;
import com.liferay.bookmarks.service.BookmarksFolderLocalServiceUtil;
import com.liferay.bookmarks.service.BookmarksFolderServiceUtil;
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.TestPropsValues;
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.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.WhenIsAssetable;
import com.liferay.portlet.trash.test.WhenIsAssetableBaseModel;
import com.liferay.portlet.trash.test.WhenIsAssetableParentModel;
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 com.liferay.portlet.trash.test.WhenParentModelIsSameType;

import org.junit.After;
import org.junit.Before;
import org.junit.ClassRule;
import org.junit.Rule;
import org.junit.runner.RunWith;

/**
 * @author Eudaldo Alonso
 */
@RunWith(Arquillian.class)
@Sync
public class BookmarksFolderTrashHandlerTest extends BaseTrashHandlerTestCase
        implements WhenHasGrandParent, WhenIsAssetableBaseModel, WhenIsAssetableParentModel,
        WhenIsIndexableBaseModel, WhenIsMoveableFromTrashBaseModel, WhenIsRestorableBaseModel,
        WhenIsUpdatableBaseModel, WhenParentModelIsSameType {

    @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 String getParentBaseModelClassName() {
        Class<BookmarksFolder> bookmarksFolderClass = BookmarksFolder.class;

        return bookmarksFolderClass.getName();
    }

    @Override
    public String getSearchKeywords() {
        return _whenIsIndexableBaseModel.getSearchKeywords();
    }

    @Override
    public boolean isAssetEntryVisible(ClassedModel classedModel, long classPK) throws Exception {

        return _whenIsAssetable.isAssetEntryVisible(classedModel, classPK);
    }

    @Override
    public BaseModel<?> moveBaseModelFromTrash(ClassedModel classedModel, Group group,
            ServiceContext serviceContext) throws Exception {

        BaseModel<?> parentBaseModel = getParentBaseModel(group, serviceContext);

        BookmarksFolderServiceUtil.moveFolderFromTrash((Long) classedModel.getPrimaryKeyObj(),
                (Long) parentBaseModel.getPrimaryKeyObj());

        return parentBaseModel;
    }

    @Override
    public void moveParentBaseModelToTrash(long primaryKey) throws Exception {
        BookmarksFolderServiceUtil.moveFolderToTrash(primaryKey);
    }

    @Override
    public int searchBaseModelsCount(Class<?> clazz, long groupId) throws Exception {

        return _whenIsIndexableBaseModel.searchBaseModelsCount(clazz, groupId);
    }

    @Override
    public int searchTrashEntriesCount(String keywords, ServiceContext serviceContext) throws Exception {

        return _whenIsIndexableBaseModel.searchTrashEntriesCount(keywords, serviceContext);
    }

    @Before
    @Override
    public void setUp() throws Exception {
        _testMode = PortalRunMode.isTestMode();

        PortalRunMode.setTestMode(true);

        ServiceTestUtil.setUser(TestPropsValues.getUser());

        super.setUp();
    }

    @After
    public void tearDown() throws Exception {
        PortalRunMode.setTestMode(_testMode);
    }

    @Override
    public BaseModel<?> updateBaseModel(long primaryKey, ServiceContext serviceContext) throws Exception {

        BookmarksFolder folder = BookmarksFolderLocalServiceUtil.getFolder(primaryKey);

        if (serviceContext.getWorkflowAction() == WorkflowConstants.ACTION_SAVE_DRAFT) {

            folder = BookmarksFolderLocalServiceUtil.updateStatus(TestPropsValues.getUserId(), folder,
                    WorkflowConstants.STATUS_DRAFT);
        }

        return folder;
    }

    @Override
    protected BaseModel<?> addBaseModelWithWorkflow(BaseModel<?> parentBaseModel, ServiceContext serviceContext)
            throws Exception {

        BookmarksFolder parentFolder = (BookmarksFolder) parentBaseModel;

        String name = getSearchKeywords();

        return BookmarksFolderLocalServiceUtil.addFolder(TestPropsValues.getUserId(), parentFolder.getFolderId(),
                name, StringPool.BLANK, serviceContext);
    }

    @Override
    protected BaseModel<?> addBaseModelWithWorkflow(ServiceContext serviceContext) throws Exception {

        String name = getSearchKeywords();

        return BookmarksFolderLocalServiceUtil.addFolder(TestPropsValues.getUserId(),
                BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID, name, StringPool.BLANK, serviceContext);
    }

    @Override
    protected void deleteParentBaseModel(BaseModel<?> parentBaseModel, boolean includeTrashedEntries)
            throws Exception {

        BookmarksFolder folder = (BookmarksFolder) parentBaseModel;

        BookmarksFolderServiceUtil.deleteFolder(folder.getFolderId(), false);
    }

    @Override
    protected BaseModel<?> getBaseModel(long primaryKey) throws Exception {
        return BookmarksFolderLocalServiceUtil.getFolder(primaryKey);
    }

    @Override
    protected Class<?> getBaseModelClass() {
        return BookmarksFolder.class;
    }

    @Override
    protected int getNotInTrashBaseModelsCount(BaseModel<?> parentBaseModel) throws Exception {

        BookmarksFolder parentFolder = (BookmarksFolder) parentBaseModel;

        return BookmarksFolderLocalServiceUtil.getFoldersCount(parentFolder.getGroupId(),
                parentFolder.getFolderId(), WorkflowConstants.STATUS_APPROVED);
    }

    @Override
    protected BaseModel<?> getParentBaseModel(Group group, long parentBaseModelId, ServiceContext serviceContext)
            throws Exception {

        return BookmarksFolderLocalServiceUtil.addFolder(TestPropsValues.getUserId(), parentBaseModelId,
                RandomTestUtil.randomString(), StringPool.BLANK, serviceContext);
    }

    @Override
    protected BaseModel<?> getParentBaseModel(Group group, ServiceContext serviceContext) throws Exception {

        return getParentBaseModel(group, BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID, serviceContext);
    }

    @Override
    protected String getUniqueTitle(BaseModel<?> baseModel) {
        return null;
    }

    @Override
    protected void moveBaseModelToTrash(long primaryKey) throws Exception {
        BookmarksFolderServiceUtil.moveFolderToTrash(primaryKey);
    }

    private boolean _testMode;
    private final WhenIsAssetable _whenIsAssetable = new DefaultWhenIsAssetable();
    private final WhenIsIndexableBaseModel _whenIsIndexableBaseModel = new DefaultWhenIsIndexableBaseModel();

}