ContentItemHandlerTest.java

/*
 * 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 2005-2008 Pentaho Corporation.  All rights reserved. 
 * 
 * @created Aug 15, 2005 
 * @author James Dixon
 */

package org.pentaho.test.platform.repository;

import java.io.File;
import java.io.FileOutputStream;

import org.pentaho.platform.api.engine.IPentahoSession;
import org.pentaho.platform.api.engine.IRuntimeContext;
import org.pentaho.platform.engine.core.output.SimpleContentItem;
import org.pentaho.platform.engine.core.output.SimpleOutputHandler;
import org.pentaho.platform.engine.core.solution.SimpleParameterProvider;
import org.pentaho.platform.engine.core.system.StandaloneSession;
import org.pentaho.platform.repository.messages.Messages;
import org.pentaho.test.platform.engine.core.BaseTest;

@SuppressWarnings("nls")
public class ContentItemHandlerTest extends BaseTest {

	public static final String SOLUTION_PATH = "test-src/solution";
	  private static final String ALT_SOLUTION_PATH = "test-src/solution";
	  private static final String PENTAHO_XML_PATH = "/system/pentaho.xml";
	  final String SYSTEM_FOLDER = "/system";
//	  private static final String DEFAULT_SPRING_CONFIG_FILE_NAME = "pentahoObjects.spring.xml";

		  public String getSolutionPath() {
		      File file = new File(SOLUTION_PATH + PENTAHO_XML_PATH);
		      if(file.exists()) {
		        System.out.println("File exist returning " + SOLUTION_PATH);
		        return SOLUTION_PATH;  
		      } else {
		        System.out.println("File does not exist returning " + ALT_SOLUTION_PATH);      
		        return ALT_SOLUTION_PATH;
		      }
		  }
		  
  public void testSimpleOutputHandler() {
    startTest();
    IPentahoSession session = new StandaloneSession(Messages.getInstance().getString("BaseTest.DEBUG_JUNIT_SESSION")); //$NON-NLS-1$
    SimpleParameterProvider parameterProvider = new SimpleParameterProvider();
    try {
      FileOutputStream outputStream = new FileOutputStream("c:/test.txt"); //$NON-NLS-1$

      SimpleOutputHandler handler = new SimpleOutputHandler(outputStream, true);

      parameterProvider.setParameter("type", "html"); //$NON-NLS-1$ //$NON-NLS-2$
      IRuntimeContext context = run(
          "test", "platform", "ContentOutputTest_Bytearray.xaction", null, false, parameterProvider, handler, session); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$ 
      SimpleContentItem contentItem = (SimpleContentItem) context.getOutputContentItem("content", "text/html"); //$NON-NLS-1$ //$NON-NLS-2$
      handler.setContentItem(contentItem, "outputs", "content"); //$NON-NLS-1$ //$NON-NLS-2$
      contentItem.closeOutputStream();

      assertEquals(handler.getSession(), session);

    } catch (Exception e) {
      e.printStackTrace();
    }

    finishTest();
  }

  public void testSimpleOutputHandlerWithContentItem() {
    startTest();
    IPentahoSession session = new StandaloneSession(Messages.getInstance().getString("BaseTest.DEBUG_JUNIT_SESSION")); //$NON-NLS-1$
    SimpleParameterProvider parameterProvider = new SimpleParameterProvider();
    try {
      FileOutputStream outputStream = new FileOutputStream("c:/test.txt"); //$NON-NLS-1$
      SimpleOutputHandler handler = new SimpleOutputHandler(outputStream, true);

      parameterProvider.setParameter("type", "html"); //$NON-NLS-1$ //$NON-NLS-2$
      IRuntimeContext context = run(
          "test", "platform", "ContentOutputTest_Bytearray.xaction", null, false, parameterProvider, handler, session); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$ 
      SimpleContentItem contentItem = (SimpleContentItem) context.getOutputContentItem("content", "text/html"); //$NON-NLS-1$ //$NON-NLS-2$
      handler.setContentItem(contentItem, "outputs", "content"); //$NON-NLS-1$ //$NON-NLS-2$
      contentItem.closeOutputStream();

      assertEquals(handler.getMimeType(), "text/html"); //$NON-NLS-1$
      assertEquals(handler.getOutputContentItem("outputs", "content", null, null, "text/html"), contentItem); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$

    } catch (Exception e) {
      e.printStackTrace();
    }
    assertTrue(true);
    finishTest();
  }

  public static void main(String[] args) {
    ContentItemHandlerTest test = new ContentItemHandlerTest();
    test.setUp();
    test.testSimpleOutputHandler();
    test.testSimpleOutputHandlerWithContentItem();
    try {

    } finally {
      test.tearDown();
      BaseTest.shutdown();
    }
  }

}