/*
* This program is free software; you can redistribute it and/or modify it under the
* terms of the GNU General Public License, version 2 as published by the Free Software
* Foundation.
*
* You should have received a copy of the GNU General Public License along with this
* program; if not, you can obtain a copy at http://www.gnu.org/licenses/gpl-2.0.html
* or from the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
* This program 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 General Public License for more details.
*
*
* Copyright 2006 - 2013 Pentaho Corporation. All rights reserved.
*/
package org.pentaho.test.platform.engine.services;
import org.pentaho.platform.api.engine.IContentGenerator;
import org.pentaho.platform.api.engine.IMimeTypeListener;
import org.pentaho.platform.api.engine.IOutputHandler;
import org.pentaho.platform.api.engine.IParameterProvider;
import org.pentaho.platform.engine.core.output.SimpleOutputHandler;
import org.pentaho.platform.engine.core.solution.SimpleParameterProvider;
import org.pentaho.platform.engine.core.system.PentahoSessionHolder;
import org.pentaho.platform.util.web.SimpleUrlFactory;
import java.io.ByteArrayOutputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class ContentGeneratorUtil {
/**
* Convenience method for executing a content generator and getting back it's output as a string. Useful for
* testing.
*
* @param cg
* the content generator to execute
* @return the output of the content generator
* @throws Exception
* if there was a problem creating the content
*/
public static String getContentAsString( IContentGenerator cg ) throws Exception {
ByteArrayOutputStream out = new ByteArrayOutputStream();
IOutputHandler outputHandler = new SimpleOutputHandler( out, false );
Map<String, IParameterProvider> parameterProviders = new HashMap<String, IParameterProvider>();
SimpleParameterProvider requestParams = new SimpleParameterProvider();
parameterProviders.put( IParameterProvider.SCOPE_REQUEST, requestParams );
SimpleUrlFactory urlFactory = new SimpleUrlFactory( "?" ); //$NON-NLS-1$
List<String> messages = new ArrayList<String>();
cg.setOutputHandler( outputHandler );
IMimeTypeListener mimeTypeListener = new IMimeTypeListener() {
@SuppressWarnings( "unused" )
public String mimeType = null;
@SuppressWarnings( "unused" )
public String name = null;
public void setMimeType( String mimeType ) {
this.mimeType = mimeType;
}
public void setName( String name ) {
this.name = name;
}
};
outputHandler.setMimeTypeListener( mimeTypeListener );
cg.setMessagesList( messages );
cg.setParameterProviders( parameterProviders );
cg.setSession( PentahoSessionHolder.getSession() );
cg.setUrlFactory( urlFactory );
cg.createContent();
String content = new String( out.toByteArray() );
return content;
}
}