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

Java tutorial

Introduction

Here is the source code for com.liferay.bookmarks.trash.test.BookmarksEntryTrashHandlerTest.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.BookmarksEntry;
import com.liferay.bookmarks.model.BookmarksFolder;
import com.liferay.bookmarks.model.BookmarksFolderConstants;
import com.liferay.bookmarks.service.BookmarksEntryLocalServiceUtil;
import com.liferay.bookmarks.service.BookmarksEntryServiceUtil;
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.model.WorkflowedModel;
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.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.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 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 BookmarksEntryTrashHandlerTest extends BaseTrashHandlerTestCase
        implements WhenHasGrandParent, WhenHasMyBaseModel, WhenHasRecentBaseModelCount, WhenIsAssetableBaseModel,
        WhenIsAssetableParentModel, 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 BookmarksEntryServiceUtil.getGroupEntriesCount(groupId, userId);
    }

    @Override
    public String getParentBaseModelClassName() {
        Class<BookmarksFolder> bookmarksFolderClass = BookmarksFolder.class;

        return bookmarksFolderClass.getName();
    }

    @Override
    public int getRecentBaseModelsCount(long groupId) throws Exception {
        return BookmarksEntryServiceUtil.getGroupEntriesCount(groupId, 0);
    }

    @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);

        BookmarksEntryServiceUtil.moveEntryFromTrash((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 {

        BookmarksEntry entry = BookmarksEntryLocalServiceUtil.getEntry(primaryKey);

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

            entry = BookmarksEntryLocalServiceUtil.updateStatus(TestPropsValues.getUserId(), entry,
                    WorkflowConstants.STATUS_DRAFT);
        }

        return entry;
    }

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

        BookmarksFolder folder = (BookmarksFolder) parentBaseModel;

        return addBaseModelWithWorkflow(folder.getUserId(), folder.getGroupId(), folder.getFolderId(),
                serviceContext);
    }

    protected BaseModel<?> addBaseModelWithWorkflow(long userId, long groupId, long folderId,
            ServiceContext serviceContext) throws Exception {

        String name = getSearchKeywords();
        String url = "http://www.liferay.com";
        String description = "Content: Enterprise. Open Source.";

        return BookmarksEntryLocalServiceUtil.addEntry(userId, groupId, folderId, name, url, description,
                serviceContext);
    }

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

        return addBaseModelWithWorkflow(TestPropsValues.getUserId(), serviceContext.getScopeGroupId(),
                BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID, 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 BookmarksEntryLocalServiceUtil.getEntry(primaryKey);
    }

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

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

        BookmarksFolder folder = (BookmarksFolder) parentBaseModel;

        return BookmarksEntryServiceUtil.getEntriesCount(folder.getGroupId(), folder.getFolderId());
    }

    @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 WorkflowedModel getWorkflowedModel(ClassedModel baseModel) throws Exception {

        BookmarksEntry entry = (BookmarksEntry) baseModel;

        return entry;
    }

    @Override
    protected void moveBaseModelToTrash(long primaryKey) throws Exception {
        BookmarksEntryServiceUtil.moveEntryToTrash(primaryKey);
    }

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

}