MBCategoryLocalServiceTest.java

/**
 * 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.portlet.messageboards.service;

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.MBThreadConstants;
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.portal.kernel.dao.orm.QueryUtil;
import com.liferay.portal.kernel.model.Group;
import com.liferay.portal.kernel.service.ServiceContext;
import com.liferay.portal.kernel.test.AssertUtils;
import com.liferay.portal.kernel.test.rule.AggregateTestRule;
import com.liferay.portal.kernel.test.rule.DeleteAfterTestRun;
import com.liferay.portal.kernel.test.util.GroupTestUtil;
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.test.rule.LiferayIntegrationTestRule;

import java.io.InputStream;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.junit.Assert;
import org.junit.Before;
import org.junit.ClassRule;
import org.junit.Rule;
import org.junit.Test;

/**
 * @author Igor Beslic
 * @author Roberto Díaz
 * @author Sergio González
 */
public class MBCategoryLocalServiceTest {

	@ClassRule
	@Rule
	public static final AggregateTestRule aggregateTestRule =
		new LiferayIntegrationTestRule();

	@Before
	public void setUp() throws Exception {
		_group = GroupTestUtil.addGroup();
	}

	@Test
	public void testGetCategoriesAndThreadsCountInRootCategory()
		throws Exception {

		addMessage(MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		MBCategory category1 = addCategory();

		addMessage(category1.getCategoryId());

		MBCategory category2 = addCategory();

		addMessage(category2.getCategoryId());

		addMessage(MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		int categoriesAndThreadsCount =
			MBCategoryLocalServiceUtil.getCategoriesAndThreadsCount(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(4, categoriesAndThreadsCount);
	}

	@Test
	public void testGetCategoriesAndThreadsCountInRootCategoryWithOnlyCategories()
		throws Exception {

		MBCategory category1 = addCategory();

		addCategory(category1.getCategoryId());

		addCategory();

		int categoriesAndThreadsCount =
			MBCategoryLocalServiceUtil.getCategoriesAndThreadsCount(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(2, categoriesAndThreadsCount);
	}

	@Test
	public void testGetCategoriesAndThreadsCountInRootCategoryWithOnlyThreads()
		throws Exception {

		addMessage(MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		addMessage(MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		int categoriesAndThreadsCount =
			MBCategoryLocalServiceUtil.getCategoriesAndThreadsCount(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(2, categoriesAndThreadsCount);
	}

	@Test
	public void testGetCategoriesAndThreadsCountWithOnlyCategories()
		throws Exception {

		MBCategory category1 = addCategory();

		addCategory(category1.getCategoryId());

		addCategory();

		int categoriesAndThreadsCount =
			MBCategoryLocalServiceUtil.getCategoriesAndThreadsCount(
				_group.getGroupId(), category1.getCategoryId(),
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(1, categoriesAndThreadsCount);
	}

	@Test
	public void testGetCategoriesAndThreadsInRootCategory() throws Exception {
		MBMessage message1 = addMessage(
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		MBCategory category1 = addCategory();

		addMessage(category1.getCategoryId());

		MBCategory category2 = addCategory();

		addMessage(category2.getCategoryId());

		MBMessage message2 = addMessage(
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		List<Object> categoriesAndThreads =
			MBCategoryLocalServiceUtil.getCategoriesAndThreads(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(
			categoriesAndThreads.toString(), 4, categoriesAndThreads.size());
		Assert.assertEquals(category1, categoriesAndThreads.get(0));
		Assert.assertEquals(category2, categoriesAndThreads.get(1));
		Assert.assertEquals(message1.getThread(), categoriesAndThreads.get(2));
		Assert.assertEquals(message2.getThread(), categoriesAndThreads.get(3));
	}

	@Test
	public void testGetCategoriesAndThreadsInRootCategoryWithOnlyCategories()
		throws Exception {

		MBCategory category1 = addCategory();

		addCategory(category1.getCategoryId());

		MBCategory category2 = addCategory();

		List<Object> categoriesAndThreads =
			MBCategoryLocalServiceUtil.getCategoriesAndThreads(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(
			categoriesAndThreads.toString(), 2, categoriesAndThreads.size());
		Assert.assertEquals(category1, categoriesAndThreads.get(0));
		Assert.assertEquals(category2, categoriesAndThreads.get(1));
	}

	@Test
	public void testGetCategoriesAndThreadsInRootCategoryWithOnlyThreads()
		throws Exception {

		MBMessage message1 = addMessage(
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		MBMessage message2 = addMessage(
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		List<Object> categoriesAndThreads =
			MBCategoryLocalServiceUtil.getCategoriesAndThreads(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(
			categoriesAndThreads.toString(), 2, categoriesAndThreads.size());
		Assert.assertEquals(message1.getThread(), categoriesAndThreads.get(0));
		Assert.assertEquals(message2.getThread(), categoriesAndThreads.get(1));
	}

	@Test
	public void testGetCategoriesAndThreadsWithOnlyCategories()
		throws Exception {

		MBCategory category1 = addCategory();

		MBCategory subcategory1 = addCategory(category1.getCategoryId());

		addCategory();

		List<Object> categoriesAndThreads =
			MBCategoryLocalServiceUtil.getCategoriesAndThreads(
				_group.getGroupId(), category1.getCategoryId(),
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(
			categoriesAndThreads.toString(), 1, categoriesAndThreads.size());
		Assert.assertEquals(subcategory1, categoriesAndThreads.get(0));
	}

	@Test
	public void testGetCategoriesAndThreadsWithPriorityThread()
		throws Exception {

		MBMessage message1 = addMessage(
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		MBCategory category1 = addCategory();

		addMessage(category1.getCategoryId());

		MBMessage priorityMessage = addMessage(
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID, 2.0);

		MBMessage message2 = addMessage(
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);

		List<Object> categoriesAndThreads =
			MBCategoryLocalServiceUtil.getCategoriesAndThreads(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED);

		Assert.assertEquals(
			categoriesAndThreads.toString(), 4, categoriesAndThreads.size());
		Assert.assertEquals(category1, categoriesAndThreads.get(0));
		Assert.assertEquals(
			priorityMessage.getThread(), categoriesAndThreads.get(1));
		Assert.assertEquals(message1.getThread(), categoriesAndThreads.get(2));
		Assert.assertEquals(message2.getThread(), categoriesAndThreads.get(3));
	}

	@Test
	public void testGetCategoriesCountWithExcludedCategories()
		throws Exception {

		addCategory();

		MBCategory excludedCategory1 = addCategory();
		MBCategory excludedCategory2 = addCategory();

		Assert.assertEquals(
			3,
			MBCategoryLocalServiceUtil.getCategoriesCount(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_ANY));
		Assert.assertEquals(
			1,
			MBCategoryLocalServiceUtil.getCategoriesCount(
				_group.getGroupId(),
				new long[] {
					excludedCategory1.getCategoryId(),
					excludedCategory2.getCategoryId()
				},
				new long[] {MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID},
				WorkflowConstants.STATUS_ANY));
	}

	@Test
	public void testGetCategoriesCountWithExcludedCategory() throws Exception {
		addCategory();
		addCategory();

		MBCategory excludedCategory = addCategory();

		Assert.assertEquals(
			3,
			MBCategoryLocalServiceUtil.getCategoriesCount(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_ANY));
		Assert.assertEquals(
			2,
			MBCategoryLocalServiceUtil.getCategoriesCount(
				_group.getGroupId(), excludedCategory.getCategoryId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_ANY));
	}

	@Test
	public void testGetCategoriesCountWithStatusApprovedAndExcludedCategories()
		throws Exception {

		addCategory();

		MBCategory excludedCategory1 = addCategory();
		MBCategory excludedCategory2 = addCategory();

		MBCategory draftCategory = addCategory();

		MBCategoryLocalServiceUtil.updateStatus(
			draftCategory.getUserId(), draftCategory.getCategoryId(),
			WorkflowConstants.STATUS_DRAFT);

		Assert.assertEquals(
			3,
			MBCategoryLocalServiceUtil.getCategoriesCount(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED));
		Assert.assertEquals(
			1,
			MBCategoryLocalServiceUtil.getCategoriesCount(
				_group.getGroupId(),
				new long[] {
					excludedCategory1.getCategoryId(),
					excludedCategory2.getCategoryId()
				},
				new long[] {MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID},
				WorkflowConstants.STATUS_APPROVED));
	}

	@Test
	public void testGetCategoriesCountWithStatusApprovedAndExcludedCategory()
		throws Exception {

		addCategory();

		MBCategory excludedCategory = addCategory();
		MBCategory draftCategory = addCategory();

		MBCategoryLocalServiceUtil.updateStatus(
			draftCategory.getUserId(), draftCategory.getCategoryId(),
			WorkflowConstants.STATUS_DRAFT);

		Assert.assertEquals(
			2,
			MBCategoryLocalServiceUtil.getCategoriesCount(
				_group.getGroupId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED));
		Assert.assertEquals(
			1,
			MBCategoryLocalServiceUtil.getCategoriesCount(
				_group.getGroupId(), excludedCategory.getCategoryId(),
				MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
				WorkflowConstants.STATUS_APPROVED));
	}

	@Test
	public void testGetCategoriesWithExcludedCategories() throws Exception {
		List<MBCategory> expectedCategories = new ArrayList<>();

		expectedCategories.add(addCategory());

		MBCategory excludedCategory1 = addCategory();

		expectedCategories.add(excludedCategory1);

		MBCategory excludedCategory2 = addCategory();

		expectedCategories.add(excludedCategory2);

		List<MBCategory> categories = MBCategoryLocalServiceUtil.getCategories(
			_group.getGroupId());

		AssertUtils.assertEquals(expectedCategories, categories);

		expectedCategories.remove(excludedCategory1);
		expectedCategories.remove(excludedCategory2);

		categories = MBCategoryLocalServiceUtil.getCategories(
			_group.getGroupId(),
			new long[] {
				excludedCategory1.getCategoryId(),
				excludedCategory2.getCategoryId()
			},
			new long[] {MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID},
			WorkflowConstants.STATUS_ANY, QueryUtil.ALL_POS, QueryUtil.ALL_POS);

		AssertUtils.assertEquals(expectedCategories, categories);
	}

	@Test
	public void testGetCategoriesWithExcludedCategory() throws Exception {
		List<MBCategory> expectedCategories = new ArrayList<>();

		expectedCategories.add(addCategory());
		expectedCategories.add(addCategory());

		MBCategory excludedCategory = addCategory();

		expectedCategories.add(excludedCategory);

		List<MBCategory> categories = MBCategoryLocalServiceUtil.getCategories(
			_group.getGroupId());

		AssertUtils.assertEquals(expectedCategories, categories);

		expectedCategories.remove(excludedCategory);

		categories = MBCategoryLocalServiceUtil.getCategories(
			_group.getGroupId(), excludedCategory.getCategoryId(),
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
			WorkflowConstants.STATUS_ANY, QueryUtil.ALL_POS, QueryUtil.ALL_POS);

		AssertUtils.assertEquals(expectedCategories, categories);
	}

	@Test
	public void testGetCategoriesWithStatusApprovedAndExcludedCategories()
		throws Exception {

		List<MBCategory> expectedCategories = new ArrayList<>();

		expectedCategories.add(addCategory());

		MBCategory excludedCategory1 = addCategory();

		expectedCategories.add(excludedCategory1);

		MBCategory excludedCategory2 = addCategory();

		expectedCategories.add(excludedCategory2);

		MBCategory draftCategory = addCategory();

		MBCategoryLocalServiceUtil.updateStatus(
			draftCategory.getUserId(), draftCategory.getCategoryId(),
			WorkflowConstants.STATUS_DRAFT);

		List<MBCategory> categories = MBCategoryLocalServiceUtil.getCategories(
			_group.getGroupId(), WorkflowConstants.STATUS_APPROVED);

		AssertUtils.assertEquals(expectedCategories, categories);

		expectedCategories.remove(excludedCategory1);
		expectedCategories.remove(excludedCategory2);

		categories = MBCategoryLocalServiceUtil.getCategories(
			_group.getGroupId(),
			new long[] {
				excludedCategory1.getCategoryId(),
				excludedCategory2.getCategoryId()
			},
			new long[] {MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID},
			WorkflowConstants.STATUS_APPROVED, QueryUtil.ALL_POS,
			QueryUtil.ALL_POS);

		AssertUtils.assertEquals(expectedCategories, categories);
	}

	@Test
	public void testGetCategoriesWithStatusApprovedAndExcludedCategory()
		throws Exception {

		List<MBCategory> expectedCategories = new ArrayList<>();

		expectedCategories.add(addCategory());

		MBCategory excludedCategory = addCategory();

		expectedCategories.add(excludedCategory);

		MBCategory draftCategory = addCategory();

		MBCategoryLocalServiceUtil.updateStatus(
			draftCategory.getUserId(), draftCategory.getCategoryId(),
			WorkflowConstants.STATUS_DRAFT);

		List<MBCategory> categories = MBCategoryLocalServiceUtil.getCategories(
			_group.getGroupId(), WorkflowConstants.STATUS_APPROVED);

		AssertUtils.assertEquals(expectedCategories, categories);

		expectedCategories.remove(excludedCategory);

		categories = MBCategoryLocalServiceUtil.getCategories(
			_group.getGroupId(), excludedCategory.getCategoryId(),
			MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID,
			WorkflowConstants.STATUS_APPROVED, QueryUtil.ALL_POS,
			QueryUtil.ALL_POS);

		AssertUtils.assertEquals(expectedCategories, categories);
	}

	@Test
	public void testGetParentCategory() throws Exception {
		MBCategory parentCategory = addCategory();

		MBCategory category = addCategory(parentCategory.getCategoryId());

		Assert.assertNotNull(category.getParentCategory());

		Assert.assertNull(parentCategory.getParentCategory());
	}

	@Test
	public void testGetParentDiscussionCategory() throws Exception {
		MBCategory discussionCategory = MBCategoryLocalServiceUtil.getCategory(
			MBCategoryConstants.DISCUSSION_CATEGORY_ID);

		Assert.assertNotNull(discussionCategory);
		Assert.assertNull(discussionCategory.getParentCategory());
	}

	protected MBCategory addCategory() throws Exception {
		return addCategory(MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);
	}

	protected MBCategory addCategory(long parentCategoryId) throws Exception {
		ServiceContext serviceContext =
			ServiceContextTestUtil.getServiceContext(
				_group.getGroupId(), TestPropsValues.getUserId());

		return MBCategoryServiceUtil.addCategory(
			TestPropsValues.getUserId(), parentCategoryId,
			RandomTestUtil.randomString(), StringPool.BLANK, serviceContext);
	}

	protected MBMessage addMessage(long categoryId) throws Exception {
		return addMessage(categoryId, MBThreadConstants.PRIORITY_NOT_GIVEN);
	}

	protected MBMessage addMessage(long categoryId, double priority)
		throws Exception {

		ServiceContext serviceContext =
			ServiceContextTestUtil.getServiceContext(
				_group.getGroupId(), TestPropsValues.getUserId());

		List<ObjectValuePair<String, InputStream>> inputStreamOVPs =
			Collections.emptyList();

		return MBMessageLocalServiceUtil.addMessage(
			TestPropsValues.getUserId(), RandomTestUtil.randomString(),
			_group.getGroupId(), categoryId, RandomTestUtil.randomString(),
			RandomTestUtil.randomString(), MBMessageConstants.DEFAULT_FORMAT,
			inputStreamOVPs, false, priority, false, serviceContext);
	}

	@DeleteAfterTestRun
	private Group _group;

}