AssetLinkExportImportTest.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.asset.lar.test;

import com.liferay.arquillian.extension.junit.bridge.junit.Arquillian;
import com.liferay.asset.kernel.model.AssetEntry;
import com.liferay.asset.kernel.model.AssetLink;
import com.liferay.asset.kernel.model.adapter.StagedAssetLink;
import com.liferay.asset.kernel.service.AssetEntryLocalServiceUtil;
import com.liferay.asset.kernel.service.AssetLinkLocalServiceUtil;
import com.liferay.bookmarks.model.BookmarksEntry;
import com.liferay.bookmarks.util.test.BookmarksTestUtil;
import com.liferay.exportimport.kernel.configuration.ExportImportConfigurationParameterMapFactory;
import com.liferay.exportimport.kernel.lar.ExportImportHelperUtil;
import com.liferay.exportimport.kernel.lar.ExportImportPathUtil;
import com.liferay.exportimport.kernel.lar.PortletDataContext;
import com.liferay.exportimport.kernel.lar.PortletDataContextFactoryUtil;
import com.liferay.exportimport.kernel.lar.PortletDataHandlerKeys;
import com.liferay.exportimport.kernel.lar.UserIdStrategy;
import com.liferay.exportimport.kernel.service.StagingLocalServiceUtil;
import com.liferay.exportimport.kernel.staging.StagingUtil;
import com.liferay.journal.constants.JournalPortletKeys;
import com.liferay.journal.model.JournalArticle;
import com.liferay.journal.model.JournalFolderConstants;
import com.liferay.journal.service.JournalArticleLocalServiceUtil;
import com.liferay.journal.test.util.JournalTestUtil;
import com.liferay.portal.kernel.model.Group;
import com.liferay.portal.kernel.model.adapter.ModelAdapterUtil;
import com.liferay.portal.kernel.service.ServiceContext;
import com.liferay.portal.kernel.test.rule.AggregateTestRule;
import com.liferay.portal.kernel.test.rule.Sync;
import com.liferay.portal.kernel.test.rule.SynchronousDestinationTestRule;
import com.liferay.portal.kernel.test.util.TestPropsValues;
import com.liferay.portal.kernel.util.MapUtil;
import com.liferay.portal.kernel.util.StringPool;
import com.liferay.portal.kernel.zip.ZipReaderFactoryUtil;
import com.liferay.portal.lar.test.BaseExportImportTestCase;
import com.liferay.portal.service.test.ServiceTestUtil;
import com.liferay.portal.test.rule.LiferayIntegrationTestRule;

import java.util.List;
import java.util.Map;

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 Akos Thurzo
 */
@RunWith(Arquillian.class)
@Sync
public class AssetLinkExportImportTest extends BaseExportImportTestCase {

	@ClassRule
	@Rule
	public static final AggregateTestRule aggregateTestRule =
		new AggregateTestRule(
			new LiferayIntegrationTestRule(),
			SynchronousDestinationTestRule.INSTANCE);

	@Before
	@Override
	public void setUp() throws Exception {
		super.setUp();

		ServiceTestUtil.setUser(TestPropsValues.getUser());

		_journalArticle = JournalTestUtil.addArticle(
			group.getGroupId(),
			JournalFolderConstants.DEFAULT_PARENT_FOLDER_ID);

		_bookmarksEntry = BookmarksTestUtil.addEntry(group.getGroupId(), true);

		addAssetLink(_journalArticle, _bookmarksEntry, 1);
	}

	@Test
	public void testBothAssetEntriesExported() throws Exception {
		long[] layoutIds = new long[] {layout.getLayoutId()};

		exportImportLayouts(layoutIds, getImportParameterMap());

		AssetEntry assetEntry = AssetEntryLocalServiceUtil.getEntry(
			importedGroup.getGroupId(),
			_journalArticle.getArticleResourceUuid());

		List<AssetLink> assetLinks = AssetLinkLocalServiceUtil.getLinks(
			assetEntry.getEntryId());

		Assert.assertNotNull(assetLinks);
		Assert.assertTrue(!assetLinks.isEmpty());
	}

	@Test
	public void testOnlyAssetLinkExported() throws Exception {
		long[] layoutIds = new long[] {layout.getLayoutId()};

		Map<String, String[]> exportParameterMap = getExportParameterMap();

		exportParameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA,
			new String[] {Boolean.FALSE.toString()});
		exportParameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA_ALL,
			new String[] {Boolean.FALSE.toString()});

		exportLayouts(layoutIds, exportParameterMap);

		checkAssetLinksInLar(_journalArticle.getArticleResourceUuid());

		// Journal article should not be in the LAR

		PortletDataContext portletDataContext = getPortletDataContext();

		_journalArticle =
			(JournalArticle)portletDataContext.getZipEntryAsObject(
				ExportImportPathUtil.getModelPath(_journalArticle));

		Assert.assertNull(_journalArticle);

		// BookmarksE etry should not be in the LAR

		_bookmarksEntry =
			(BookmarksEntry)portletDataContext.getZipEntryAsObject(
				ExportImportPathUtil.getModelPath(_bookmarksEntry));

		Assert.assertNull(_bookmarksEntry);

		importLayouts(getImportParameterMap());
	}

	@Test
	public void testOnlyOneAssetEntryExported() throws Exception {
		long[] layoutIds = new long[] {layout.getLayoutId()};

		Map<String, String[]> exportParameterMap = getExportParameterMap();

		exportParameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA,
			new String[] {Boolean.TRUE.toString()});
		exportParameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA + StringPool.UNDERLINE +
				JournalPortletKeys.JOURNAL,
			new String[] {Boolean.TRUE.toString()});
		exportParameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA_ALL,
			new String[] {Boolean.FALSE.toString()});

		exportLayouts(layoutIds, exportParameterMap);

		checkAssetLinksInLar(_journalArticle.getArticleResourceUuid());

		// Journal article should be in the LAR

		PortletDataContext portletDataContext = getPortletDataContext();

		_journalArticle =
			(JournalArticle)portletDataContext.getZipEntryAsObject(
				ExportImportPathUtil.getModelPath(_journalArticle));

		Assert.assertNotNull(_journalArticle);

		// Bookmarks entry should not be in the LAR

		_bookmarksEntry =
			(BookmarksEntry)portletDataContext.getZipEntryAsObject(
				ExportImportPathUtil.getModelPath(_bookmarksEntry));

		Assert.assertNull(_bookmarksEntry);

		importLayouts(getImportParameterMap());
	}

	@Test
	public void testStaging() throws Exception {
		StagingLocalServiceUtil.enableLocalStaging(
			TestPropsValues.getUserId(), group, false, false,
			new ServiceContext());

		Group stagingGroup = group.getStagingGroup();

		JournalArticle article = JournalTestUtil.addArticle(
			stagingGroup.getGroupId(),
			JournalFolderConstants.DEFAULT_PARENT_FOLDER_ID);

		BookmarksEntry bookmarksEntry = BookmarksTestUtil.addEntry(
			stagingGroup.getGroupId(), true);

		addAssetLink(article, bookmarksEntry, 1);

		Map<String, String[]> parameterMap =
			ExportImportConfigurationParameterMapFactory.buildParameterMap();

		parameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA,
			new String[] {Boolean.FALSE.toString()});
		parameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA_ALL,
			new String[] {Boolean.FALSE.toString()});

		StagingUtil.publishLayouts(
			TestPropsValues.getUserId(), stagingGroup.getGroupId(),
			group.getGroupId(), false, parameterMap);

		JournalArticle liveJournalArticle =
			JournalArticleLocalServiceUtil.fetchJournalArticleByUuidAndGroupId(
				article.getUuid(), group.getGroupId());

		Assert.assertNull(liveJournalArticle);

		parameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA,
			new String[] {Boolean.TRUE.toString()});
		parameterMap.put(
			PortletDataHandlerKeys.PORTLET_DATA_ALL,
			new String[] {Boolean.TRUE.toString()});

		StagingUtil.publishLayouts(
			TestPropsValues.getUserId(), stagingGroup.getGroupId(),
			group.getGroupId(), false, parameterMap);

		AssetEntry assetEntry = AssetEntryLocalServiceUtil.fetchEntry(
			group.getGroupId(), article.getArticleResourceUuid());

		Assert.assertNotNull(assetEntry);

		List<AssetLink> assetLinks = AssetLinkLocalServiceUtil.getLinks(
			assetEntry.getEntryId());

		Assert.assertNotNull(assetLinks);
		Assert.assertTrue(assetLinks.size() == 2);
	}

	protected void checkAssetLinksInLar(String assetEntryClassUuid)
		throws Exception {

		AssetEntry assetEntry = AssetEntryLocalServiceUtil.fetchEntry(
			group.getGroupId(), assetEntryClassUuid);

		Assert.assertNotNull(assetEntry);

		List<AssetLink> assetLinks = AssetLinkLocalServiceUtil.getLinks(
			assetEntry.getEntryId());

		Assert.assertNotNull(assetLinks);
		Assert.assertTrue(assetLinks.size() == 2);

		PortletDataContext portletDataContext = getPortletDataContext();

		for (AssetLink assetLink : assetLinks) {
			StagedAssetLink stagedAssetLink = ModelAdapterUtil.adapt(
				assetLink, AssetLink.class, StagedAssetLink.class);

			stagedAssetLink =
				(StagedAssetLink)portletDataContext.getZipEntryAsObject(
					ExportImportPathUtil.getModelPath(stagedAssetLink));

			Assert.assertNotNull(stagedAssetLink);
		}
	}

	protected PortletDataContext getPortletDataContext() throws Exception {
		Map<String, String[]> parameterMap =
			ExportImportConfigurationParameterMapFactory.buildParameterMap();

		String userIdStrategyString = MapUtil.getString(
			parameterMap, PortletDataHandlerKeys.USER_ID_STRATEGY);

		UserIdStrategy userIdStrategy =
			ExportImportHelperUtil.getUserIdStrategy(
				TestPropsValues.getUserId(), userIdStrategyString);

		return PortletDataContextFactoryUtil.createImportPortletDataContext(
			group.getCompanyId(), importedGroup.getGroupId(), parameterMap,
			userIdStrategy, ZipReaderFactoryUtil.getZipReader(larFile));
	}

	private BookmarksEntry _bookmarksEntry;
	private JournalArticle _journalArticle;

}