com.liferay.calendar.upgrade.v1_0_2.test.UpgradeCalendarTest.java Source code

Introduction

Here is the source code for com.liferay.calendar.upgrade.v1_0_2.test.UpgradeCalendarTest.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.calendar.upgrade.v1_0_2.test;

import com.liferay.arquillian.extension.junit.bridge.junit.Arquillian;
import com.liferay.calendar.model.Calendar;
import com.liferay.calendar.service.CalendarLocalServiceUtil;
import com.liferay.calendar.test.util.CalendarTestUtil;
import com.liferay.calendar.test.util.CalendarUpgradeTestUtil;
import com.liferay.calendar.test.util.UpgradeDatabaseTestHelper;
import com.liferay.portal.kernel.dao.orm.EntityCacheUtil;
import com.liferay.portal.kernel.exception.PortalException;
import com.liferay.portal.kernel.model.Group;
import com.liferay.portal.kernel.model.User;
import com.liferay.portal.kernel.service.UserLocalServiceUtil;
import com.liferay.portal.kernel.test.rule.DeleteAfterTestRun;
import com.liferay.portal.kernel.test.util.GroupTestUtil;
import com.liferay.portal.kernel.test.util.UserTestUtil;
import com.liferay.portal.kernel.upgrade.UpgradeProcess;
import com.liferay.portal.kernel.util.PropsKeys;
import com.liferay.portal.kernel.util.PropsUtil;
import com.liferay.portal.test.rule.LiferayIntegrationTestRule;

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 Adam Brandizzi
 */
@RunWith(Arquillian.class)
public class UpgradeCalendarTest {

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

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

        _user = UserTestUtil.addUser();

        _upgradeProcess = CalendarUpgradeTestUtil
                .getServiceUpgradeStep("com.liferay.calendar.internal.upgrade.v1_0_2.UpgradeCalendar");
        _upgradeDatabaseTestHelper = CalendarUpgradeTestUtil.getUpgradeDatabaseTestHelper();
    }

    @Test
    public void testUpgradeCreatesCalendarTimeZoneId() throws Exception {
        _upgradeProcess.upgrade();

        assertHasColumn("timeZoneId");
    }

    @Test
    public void testUpgradeSetsSiteCalendarTimeZoneId() throws Exception {
        Calendar calendar = CalendarTestUtil.addCalendar(_group);

        _upgradeProcess.upgrade();

        assertCalendarTimeZoneId(calendar, PropsUtil.get(PropsKeys.COMPANY_DEFAULT_TIME_ZONE));
    }

    @Test
    public void testUpgradeSetsUserCalendarTimeZoneId() throws Exception {
        setUserTimeZoneId("Asia/Shangai");

        Calendar calendar = CalendarTestUtil.addCalendar(_user);

        _upgradeProcess.upgrade();

        assertCalendarTimeZoneId(calendar, "Asia/Shangai");
    }

    protected void assertCalendarTimeZoneId(Calendar calendar, String timeZoneId) throws PortalException {

        EntityCacheUtil.clearCache();

        calendar = CalendarLocalServiceUtil.getCalendar(calendar.getCalendarId());

        Assert.assertEquals(timeZoneId, calendar.getTimeZoneId());
    }

    protected void assertHasColumn(String columnName) throws Exception {
        Assert.assertTrue(_upgradeDatabaseTestHelper.hasColumn("Calendar", columnName));
    }

    protected void setUserTimeZoneId(String timeZoneId) {
        _user.setTimeZoneId(timeZoneId);

        UserLocalServiceUtil.updateUser(_user);
    }

    @DeleteAfterTestRun
    private Group _group;

    private UpgradeDatabaseTestHelper _upgradeDatabaseTestHelper;
    private UpgradeProcess _upgradeProcess;

    @DeleteAfterTestRun
    private User _user;

}