MockPortletContainer.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.portal.resiliency.spi.action;

import com.liferay.portal.kernel.model.Layout;
import com.liferay.portal.kernel.model.Portlet;
import com.liferay.portal.kernel.portlet.ActionResult;
import com.liferay.portal.kernel.portlet.PortletContainer;
import com.liferay.portal.kernel.util.WebKeys;

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

import javax.portlet.Event;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * @author Shuyang Zhou
 */
public class MockPortletContainer implements PortletContainer {

	public static final String MOCK_LAYOUT_TYPE_SETTINGS =
		"MOCK_LAYOUT_TYPE_SETTINGS";

	@Override
	public void preparePortlet(HttpServletRequest request, Portlet portlet) {
		prepared = true;
	}

	@Override
	public ActionResult processAction(
		HttpServletRequest request, HttpServletResponse response,
		Portlet portlet) {

		this.request = request;
		this.response = response;
		this.portlet = portlet;

		if (modifyLayoutTypeSettings) {
			Layout requestLayout = (Layout)request.getAttribute(WebKeys.LAYOUT);

			requestLayout.setTypeSettings(MOCK_LAYOUT_TYPE_SETTINGS);
		}

		actionResult = new ActionResult(null, null);

		return actionResult;
	}

	@Override
	public List<Event> processEvent(
		HttpServletRequest request, HttpServletResponse response,
		Portlet portlet, Layout layout, Event event) {

		this.request = request;
		this.response = response;
		this.portlet = portlet;
		this.layout = layout;
		this.event = event;

		if (modifyLayoutTypeSettings) {
			Layout requestLayout = (Layout)request.getAttribute(WebKeys.LAYOUT);

			requestLayout.setTypeSettings(MOCK_LAYOUT_TYPE_SETTINGS);
		}

		events = new ArrayList<>();

		return events;
	}

	@Override
	public void render(
		HttpServletRequest request, HttpServletResponse response,
		Portlet portlet) {

		this.request = request;
		this.response = response;
		this.portlet = portlet;
	}

	@Override
	public void serveResource(
		HttpServletRequest request, HttpServletResponse response,
		Portlet portlet) {

		this.request = request;
		this.response = response;
		this.portlet = portlet;
	}

	public ActionResult actionResult;
	public Event event;
	public List<Event> events;
	public Layout layout;
	public boolean modifyLayoutTypeSettings;
	public Portlet portlet;
	public boolean prepared;
	public HttpServletRequest request;
	public HttpServletResponse response;

}