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.calendar.service.test; import com.liferay.arquillian.extension.junit.bridge.junit.Arquillian; import com.liferay.calendar.model.Calendar; import com.liferay.calendar.model.CalendarBooking; import com.liferay.calendar.service.CalendarBookingLocalServiceUtil; import com.liferay.calendar.service.CalendarBookingServiceUtil; import com.liferay.calendar.test.util.CalendarBookingTestUtil; import com.liferay.calendar.test.util.CalendarStagingTestUtil; import com.liferay.calendar.test.util.CalendarTestUtil; 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.PrincipalException; 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.service.ResourcePermissionLocalServiceUtil; import com.liferay.portal.kernel.service.RoleLocalServiceUtil; import com.liferay.portal.kernel.service.ServiceContext; import com.liferay.portal.kernel.test.context.ContextUserReplace; 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.workflow.WorkflowConstants; import com.liferay.portal.test.rule.LiferayIntegrationTestRule; import com.liferay.portal.test.rule.PermissionCheckerTestRule; import com.liferay.portal.test.rule.SynchronousMailTestRule; import java.util.Collections; 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 Lino Alves */ @RunWith(Arquillian.class) public class CalendarBookingServiceTest { @ClassRule @Rule public static final AggregateTestRule aggregateTestRule = new AggregateTestRule( new LiferayIntegrationTestRule(), PermissionCheckerTestRule.INSTANCE, SynchronousMailTestRule.INSTANCE); @Before public void setUp() throws Exception { _omnidminUser = UserTestUtil.addOmniAdminUser(); _user1 = UserTestUtil.addUser(); _user2 = UserTestUtil.addUser(); _permissionChecker = PermissionThreadLocal.getPermissionChecker(); } @After public void tearDown() { CalendarStagingTestUtil.cleanUp(); PermissionThreadLocal.setPermissionChecker(_permissionChecker); } @Test public void testGetUnapprovedCalendarBookingsForOmniadmin() throws Exception { ServiceContext serviceContext = createServiceContext(); Calendar calendar = CalendarTestUtil.addCalendar(_user1, serviceContext); CalendarBooking calendarBooking = CalendarBookingTestUtil.addRegularCalendarBooking(_user1, calendar, serviceContext); calendarBooking.setStatus(WorkflowConstants.STATUS_PENDING); CalendarBookingLocalServiceUtil.updateCalendarBooking(calendarBooking); int[] statuses = { WorkflowConstants.STATUS_PENDING }; List<CalendarBooking> calendarBookings = Collections.emptyList(); try (ContextUserReplace contextUserReplacer = new ContextUserReplace(_omnidminUser)) { calendarBookings = CalendarBookingServiceUtil.getCalendarBookings(calendar.getCalendarId(), statuses); } Assert.assertTrue(!calendarBookings.isEmpty()); } @Test public void testGetUnapprovedCalendarBookingsForRegularUser() throws Exception { ServiceContext serviceContext = createServiceContext(); Calendar calendar = CalendarTestUtil.addCalendar(_user1, serviceContext); CalendarBooking calendarBooking = CalendarBookingTestUtil.addRegularCalendarBooking(_user1, calendar, serviceContext); calendarBooking.setStatus(WorkflowConstants.STATUS_PENDING); CalendarBookingLocalServiceUtil.updateCalendarBooking(calendarBooking); int[] statuses = { WorkflowConstants.STATUS_PENDING }; List<CalendarBooking> calendarBookings = Collections.emptyList(); try (ContextUserReplace contextUserReplacer = new ContextUserReplace(_user1)) { calendarBookings = CalendarBookingServiceUtil.getCalendarBookings(calendar.getCalendarId(), statuses); } Assert.assertTrue(!calendarBookings.isEmpty()); try (ContextUserReplace contextUserReplacer = new ContextUserReplace(_user2)) { calendarBookings = CalendarBookingServiceUtil.getCalendarBookings(calendar.getCalendarId(), statuses); } Assert.assertTrue(calendarBookings.toString(), calendarBookings.isEmpty()); } @Test public void testLiveSiteCalendarInvitesStagingSiteCalendarShouldNotAppearOnLiveResourceUntilPublish() throws Exception { _liveGroup = GroupTestUtil.addGroup(); Calendar invitingCalendar = CalendarTestUtil.addCalendar(_user1); Calendar liveCalendar = CalendarTestUtil.getDefaultCalendar(_liveGroup); CalendarStagingTestUtil.enableLocalStaging(_liveGroup, true); CalendarBooking childCalendarBooking = CalendarBookingTestUtil.addChildCalendarBooking(invitingCalendar, liveCalendar); List<CalendarBooking> childCalendarBookings = CalendarBookingServiceUtil .getChildCalendarBookings(childCalendarBooking.getParentCalendarBookingId(), true); Assert.assertEquals(childCalendarBookings.toString(), 2, childCalendarBookings.size()); childCalendarBookings = CalendarBookingServiceUtil .getChildCalendarBookings(childCalendarBooking.getParentCalendarBookingId(), false); Assert.assertEquals(childCalendarBookings.toString(), 1, childCalendarBookings.size()); } @Test(expected = PrincipalException.MustHavePermission.class) public void testUserWithoutPermissionInCalendarShouldNotViewCalendarBooking() throws Exception { ServiceContext serviceContext = createServiceContext(); Calendar calendar = CalendarTestUtil.addCalendar(_user1, serviceContext); deleteGuestAndUserPermission(calendar); CalendarBooking calendarBooking = CalendarBookingTestUtil.addRegularCalendarBooking(_user1, calendar, serviceContext); PermissionThreadLocal.setPermissionChecker(PermissionCheckerFactoryUtil.create(_user2)); CalendarBookingServiceUtil.getCalendarBooking(calendarBooking.getCalendarBookingId()); } protected ServiceContext createServiceContext() { ServiceContext serviceContext = new ServiceContext(); serviceContext.setCompanyId(_user1.getCompanyId()); serviceContext.setUserId(_user1.getUserId()); return serviceContext; } protected void deleteGuestAndUserPermission(Calendar calendar) throws Exception { Role role = RoleLocalServiceUtil.getRole(TestPropsValues.getCompanyId(), RoleConstants.GUEST); ResourcePermissionLocalServiceUtil.setResourcePermissions(TestPropsValues.getCompanyId(), Calendar.class.getName(), ResourceConstants.SCOPE_INDIVIDUAL, String.valueOf(calendar.getPrimaryKey()), role.getRoleId(), new String[0]); role = RoleLocalServiceUtil.getRole(TestPropsValues.getCompanyId(), RoleConstants.USER); ResourcePermissionLocalServiceUtil.setResourcePermissions(TestPropsValues.getCompanyId(), Calendar.class.getName(), ResourceConstants.SCOPE_INDIVIDUAL, String.valueOf(calendar.getPrimaryKey()), role.getRoleId(), new String[0]); } @DeleteAfterTestRun private Group _liveGroup; @DeleteAfterTestRun private User _omnidminUser; private PermissionChecker _permissionChecker; @DeleteAfterTestRun private User _user1; @DeleteAfterTestRun private User _user2; }