com.liferay.portlet.bookmarks.service.BookmarksFolderServiceTest.java Source code

Introduction

Here is the source code for com.liferay.portlet.bookmarks.service.BookmarksFolderServiceTest.java

Source

/**
 * Copyright (c) 2000-2012 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.portlet.bookmarks.service;

import com.liferay.portal.kernel.search.Document;
import com.liferay.portal.kernel.search.Field;
import com.liferay.portal.kernel.search.Hits;
import com.liferay.portal.kernel.search.Indexer;
import com.liferay.portal.kernel.search.IndexerRegistryUtil;
import com.liferay.portal.kernel.search.Query;
import com.liferay.portal.kernel.search.QueryConfig;
import com.liferay.portal.kernel.search.SearchContext;
import com.liferay.portal.kernel.util.GetterUtil;
import com.liferay.portlet.bookmarks.model.BookmarksEntry;
import com.liferay.portlet.bookmarks.model.BookmarksFolder;

import java.util.List;

/**
 * @author Brian Wing Shun Chan
 */
public class BookmarksFolderServiceTest extends BaseBookmarksServiceTestCase {

    public void testAddFolder() throws Exception {
        addFolder();
    }

    public void testAddSubfolder() throws Exception {
        BookmarksFolder folder = addFolder();

        addFolder(folder.getFolderId());
    }

    public void testDeleteFolder() throws Exception {
        BookmarksFolder folder = addFolder();

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

    public void testGetFolder() throws Exception {
        BookmarksFolder folder = addFolder();

        BookmarksFolderServiceUtil.getFolder(folder.getFolderId());
    }

    public void testSearch() throws Exception {
        BookmarksEntry entry = addEntry();

        Thread.sleep(1000);

        long companyId = entry.getCompanyId();
        long groupId = entry.getFolder().getGroupId();
        long folderId = entry.getFolderId();
        String keywords = "test";

        SearchContext searchContext = new SearchContext();

        searchContext.setCompanyId(companyId);
        searchContext.setFolderIds(new long[] { folderId });
        searchContext.setGroupIds(new long[] { groupId });
        searchContext.setKeywords(keywords);

        QueryConfig queryConfig = new QueryConfig();

        queryConfig.setHighlightEnabled(false);
        queryConfig.setScoreEnabled(false);

        searchContext.setQueryConfig(queryConfig);

        Indexer indexer = IndexerRegistryUtil.getIndexer(BookmarksEntry.class);

        Hits hits = indexer.search(searchContext);

        assertEquals(1, hits.getLength());

        List<Document> results = hits.toList();

        for (Document doc : results) {
            assertEquals(companyId, GetterUtil.getLong(doc.get(Field.COMPANY_ID)));

            assertEquals(groupId, GetterUtil.getLong(doc.get(Field.GROUP_ID)));

            assertEqualsIgnoreCase(entry.getName(), doc.get(Field.TITLE));
            assertEquals(entry.getUrl(), doc.get(Field.URL));
            assertEqualsIgnoreCase(entry.getDescription(), doc.get(Field.DESCRIPTION));

            assertEquals(folderId, GetterUtil.getLong(doc.get("folderId")));
            assertEquals(entry.getEntryId(), GetterUtil.getLong(doc.get(Field.ENTRY_CLASS_PK)));
        }

        BookmarksFolderLocalServiceUtil.deleteFolder(folderId);

        Thread.sleep(1000);

        hits = indexer.search(searchContext);

        Query query = hits.getQuery();

        assertEquals(query.toString(), 0, hits.getLength());

        addEntry();
        addEntry();
        addEntry();
        addEntry();

        Thread.sleep(1000);

        searchContext.setEnd(3);
        searchContext.setFolderIds(null);
        searchContext.setStart(1);

        hits = indexer.search(searchContext);

        assertEquals(4, hits.getLength());
        assertEquals(2, hits.getDocs().length);
    }

}