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.dynamic.data.mapping.service.test; import com.liferay.arquillian.extension.junit.bridge.junit.Arquillian; import com.liferay.dynamic.data.mapping.model.DDMStructure; import com.liferay.dynamic.data.mapping.model.DDMStructureConstants; import com.liferay.dynamic.data.mapping.service.DDMStructureServiceUtil; import com.liferay.dynamic.data.mapping.storage.StorageType; import com.liferay.petra.string.StringPool; import com.liferay.portal.kernel.dao.orm.QueryUtil; import com.liferay.portal.kernel.model.Group; import com.liferay.portal.kernel.model.ResourceConstants; import com.liferay.portal.kernel.model.Role; import com.liferay.portal.kernel.model.RoleConstants; import com.liferay.portal.kernel.model.User; import com.liferay.portal.kernel.security.auth.PrincipalThreadLocal; import com.liferay.portal.kernel.security.permission.ActionKeys; import com.liferay.portal.kernel.security.permission.PermissionChecker; import com.liferay.portal.kernel.security.permission.PermissionCheckerFactoryUtil; import com.liferay.portal.kernel.security.permission.PermissionThreadLocal; import com.liferay.portal.kernel.security.permission.ResourceActionsUtil; import com.liferay.portal.kernel.service.ResourcePermissionServiceUtil; import com.liferay.portal.kernel.service.RoleLocalServiceUtil; 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.TestPropsValues; import com.liferay.portal.kernel.test.util.UserTestUtil; import com.liferay.portal.kernel.util.PortalUtil; import com.liferay.portal.kernel.util.StringUtil; import com.liferay.portal.kernel.workflow.WorkflowConstants; import com.liferay.portal.test.rule.LiferayIntegrationTestRule; import java.util.List; 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 Rafael Praxedes */ @RunWith(Arquillian.class) public class DDMStructureServiceTest extends BaseDDMServiceTestCase { @ClassRule @Rule public static final AggregateTestRule aggregateTestRule = new LiferayIntegrationTestRule(); @Before @Override public void setUp() throws Exception { super.setUp(); _classNameId = PortalUtil.getClassNameId(DDL_RECORD_SET_CLASS_NAME); _group = GroupTestUtil.addGroup(); _siteAdminUser = UserTestUtil.addGroupAdminUser(group); setUpPermissionThreadLocal(); setUpPrincipalThreadLocal(); } @After public void tearDown() throws Exception { PermissionThreadLocal.setPermissionChecker(_originalPermissionChecker); PrincipalThreadLocal.setName(_originalName); } @Test public void testGetStructures() throws Exception { addStructure(_classNameId, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString()); long[] groupIds = { group.getGroupId(), _group.getGroupId() }; List<DDMStructure> structures = DDMStructureServiceUtil.getStructures(TestPropsValues.getCompanyId(), groupIds, _classNameId, WorkflowConstants.STATUS_ANY); Assert.assertEquals(structures.toString(), 3, structures.size()); } @Test public void testSearch() throws Exception { addStructure(_classNameId, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString()); long[] groupIds = { group.getGroupId(), _group.getGroupId() }; List<DDMStructure> structures = DDMStructureServiceUtil.search(TestPropsValues.getCompanyId(), groupIds, _classNameId, StringPool.BLANK, WorkflowConstants.STATUS_ANY, QueryUtil.ALL_POS, QueryUtil.ALL_POS, null); Assert.assertEquals(structures.toString(), 3, structures.size()); } @Test public void testSearchByNameAndDescription() throws Exception { String name = StringUtil.randomString(); String description = StringUtil.randomString(); DDMStructure structure = addStructure(_classNameId, name, description); addStructure(_classNameId, name, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString(), description); long[] groupIds = { group.getGroupId(), _group.getGroupId() }; List<DDMStructure> structures = DDMStructureServiceUtil.search(TestPropsValues.getCompanyId(), groupIds, _classNameId, name, description, StorageType.JSON.getValue(), DDMStructureConstants.TYPE_DEFAULT, WorkflowConstants.STATUS_ANY, true, QueryUtil.ALL_POS, QueryUtil.ALL_POS, null); Assert.assertEquals(structures.toString(), 1, structures.size()); Assert.assertEquals(structure, structures.get(0)); } @Test public void testSearchByNameOrDescription() throws Exception { String name = StringUtil.randomString(); String description = StringUtil.randomString(); addStructure(_classNameId, name, description); addStructure(_classNameId, name, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString(), description); long[] groupIds = { group.getGroupId(), _group.getGroupId() }; List<DDMStructure> structures = DDMStructureServiceUtil.search(TestPropsValues.getCompanyId(), groupIds, _classNameId, name, description, StorageType.JSON.getValue(), DDMStructureConstants.TYPE_DEFAULT, WorkflowConstants.STATUS_ANY, false, QueryUtil.ALL_POS, QueryUtil.ALL_POS, null); Assert.assertEquals(structures.toString(), 3, structures.size()); } @Test public void testSearchByType() throws Exception { addStructure(0, _classNameId, null, StringUtil.randomString(), StringPool.BLANK, read("test-structure.xsd"), StorageType.JSON.getValue(), DDMStructureConstants.TYPE_DEFAULT, WorkflowConstants.STATUS_APPROVED); addStructure(0, _classNameId, null, StringUtil.randomString(), StringPool.BLANK, read("test-structure.xsd"), StorageType.JSON.getValue(), DDMStructureConstants.TYPE_FRAGMENT, WorkflowConstants.STATUS_APPROVED); List<DDMStructure> structures = DDMStructureServiceUtil.search(TestPropsValues.getCompanyId(), new long[] { group.getGroupId() }, _classNameId, null, null, null, DDMStructureConstants.TYPE_DEFAULT, WorkflowConstants.STATUS_APPROVED, true, QueryUtil.ALL_POS, QueryUtil.ALL_POS, null); Assert.assertEquals(structures.toString(), 1, structures.size()); structures = DDMStructureServiceUtil.search(TestPropsValues.getCompanyId(), new long[] { group.getGroupId() }, _classNameId, null, null, null, DDMStructureConstants.TYPE_FRAGMENT, WorkflowConstants.STATUS_APPROVED, true, QueryUtil.ALL_POS, QueryUtil.ALL_POS, null); Assert.assertEquals(structures.toString(), 1, structures.size()); } @Test public void testSearchCount() throws Exception { addStructure(_classNameId, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString()); long[] groupIds = { group.getGroupId(), _group.getGroupId() }; int count = DDMStructureServiceUtil.searchCount(TestPropsValues.getCompanyId(), groupIds, _classNameId, StringPool.BLANK, WorkflowConstants.STATUS_ANY); Assert.assertEquals(3, count); } @Test public void testSearchCountByNameAndDescription() throws Exception { String name = StringUtil.randomString(); String description = StringUtil.randomString(); addStructure(_classNameId, name, description); addStructure(_classNameId, name, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString(), description); long[] groupIds = { group.getGroupId(), _group.getGroupId() }; int count = DDMStructureServiceUtil.searchCount(TestPropsValues.getCompanyId(), groupIds, _classNameId, name, description, StorageType.JSON.getValue(), DDMStructureConstants.TYPE_DEFAULT, WorkflowConstants.STATUS_ANY, true); Assert.assertEquals(1, count); } @Test public void testSearchCountByNameOrDescription() throws Exception { String name = StringUtil.randomString(); String description = StringUtil.randomString(); addStructure(_classNameId, name, description); addStructure(_classNameId, name, StringUtil.randomString()); addStructure(_classNameId, StringUtil.randomString(), description); long[] groupIds = { group.getGroupId(), _group.getGroupId() }; int count = DDMStructureServiceUtil.searchCount(TestPropsValues.getCompanyId(), groupIds, _classNameId, name, description, StorageType.JSON.getValue(), DDMStructureConstants.TYPE_DEFAULT, WorkflowConstants.STATUS_ANY, false); Assert.assertEquals(3, count); } @Test public void testSearchCountByType() throws Exception { int initialCount = DDMStructureServiceUtil.searchCount(TestPropsValues.getCompanyId(), new long[] { group.getGroupId() }, _classNameId, null, null, null, DDMStructureConstants.TYPE_FRAGMENT, WorkflowConstants.STATUS_ANY, true); addStructure(0, _classNameId, null, StringUtil.randomString(), StringPool.BLANK, read("test-structure.xsd"), StorageType.JSON.getValue(), DDMStructureConstants.TYPE_FRAGMENT, WorkflowConstants.STATUS_APPROVED); int count = DDMStructureServiceUtil.searchCount(TestPropsValues.getCompanyId(), new long[] { group.getGroupId() }, _classNameId, null, null, null, DDMStructureConstants.TYPE_FRAGMENT, WorkflowConstants.STATUS_ANY, true); Assert.assertEquals(initialCount + 1, count); } @Test public void testSearchWithSiteAdminPermission() throws Exception { DDMStructure structure = addStructure(_classNameId, StringUtil.randomString()); structure = addStructure(_classNameId, StringUtil.randomString()); String modelName = ResourceActionsUtil.getCompositeModelName(PortalUtil.getClassName(_classNameId), DDMStructure.class.getName()); List<Role> roles = RoleLocalServiceUtil.getRoles(TestPropsValues.getCompanyId()); for (Role role : roles) { ResourcePermissionServiceUtil.removeResourcePermission(structure.getGroupId(), structure.getCompanyId(), modelName, ResourceConstants.SCOPE_INDIVIDUAL, String.valueOf(structure.getPrimaryKey()), role.getRoleId(), ActionKeys.VIEW); } long[] groupIds = { group.getGroupId(), group.getGroupId() }; List<DDMStructure> structures = DDMStructureServiceUtil.search(TestPropsValues.getCompanyId(), groupIds, _classNameId, StringPool.BLANK, WorkflowConstants.STATUS_ANY, QueryUtil.ALL_POS, QueryUtil.ALL_POS, null); Assert.assertEquals(structures.toString(), 2, structures.size()); } @Test public void testSearchWithSiteMemberPermission() throws Exception { DDMStructure structure = addStructure(_classNameId, StringUtil.randomString()); structure = addStructure(_classNameId, StringUtil.randomString()); String modelName = ResourceActionsUtil.getCompositeModelName(PortalUtil.getClassName(_classNameId), DDMStructure.class.getName()); List<Role> roles = RoleLocalServiceUtil.getRoles(TestPropsValues.getCompanyId()); for (Role role : roles) { ResourcePermissionServiceUtil.removeResourcePermission(structure.getGroupId(), structure.getCompanyId(), modelName, ResourceConstants.SCOPE_INDIVIDUAL, String.valueOf(structure.getPrimaryKey()), role.getRoleId(), ActionKeys.VIEW); } long[] groupIds = { group.getGroupId(), group.getGroupId() }; User siteMemberUser = UserTestUtil.addGroupUser(group, RoleConstants.SITE_MEMBER); PermissionChecker permissionChecker = PermissionCheckerFactoryUtil.create(siteMemberUser); PermissionThreadLocal.setPermissionChecker(permissionChecker); List<DDMStructure> structures = DDMStructureServiceUtil.search(TestPropsValues.getCompanyId(), groupIds, _classNameId, StringPool.BLANK, WorkflowConstants.STATUS_ANY, QueryUtil.ALL_POS, QueryUtil.ALL_POS, null); Assert.assertEquals(structures.toString(), 1, structures.size()); } protected void setUpPermissionThreadLocal() throws Exception { _originalPermissionChecker = PermissionThreadLocal.getPermissionChecker(); PermissionThreadLocal.setPermissionChecker(PermissionCheckerFactoryUtil.create(_siteAdminUser)); } protected void setUpPrincipalThreadLocal() throws Exception { _originalName = PrincipalThreadLocal.getName(); PrincipalThreadLocal.setName(_siteAdminUser.getUserId()); } private static long _classNameId; @DeleteAfterTestRun private Group _group; private String _originalName; private PermissionChecker _originalPermissionChecker; @DeleteAfterTestRun private User _siteAdminUser; }