SimpleOutputHandlerTest.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 2008 Pentaho Corporation.  All rights reserved. 
 * 
 */
package org.pentaho.test.platform.engine.core;

import java.io.ByteArrayOutputStream;
import java.io.OutputStream;
import java.math.BigDecimal;
import java.text.DateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import junit.framework.TestCase;

import org.pentaho.commons.connection.memory.MemoryResultSet;
import org.pentaho.platform.api.engine.IContentOutputHandler;
import org.pentaho.platform.api.engine.IOutputHandler;
import org.pentaho.platform.api.repository.IContentItem;
import org.pentaho.platform.engine.core.output.BufferedContentItem;
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.PentahoSystem;
import org.pentaho.platform.engine.core.system.StandaloneSession;
import org.pentaho.platform.engine.core.system.objfac.StandaloneObjectFactory;

@SuppressWarnings({"all"})
public class SimpleOutputHandlerTest extends TestCase {

  public void test1() throws Exception {

    StandaloneObjectFactory factory = new StandaloneObjectFactory();
    
    TestOutputHandler.contentItem = new SimpleContentItem();
    factory.defineObject("testoutut", TestOutputHandler.class.getName(), StandaloneObjectFactory.Scope.LOCAL);
    PentahoSystem.setObjectFactory( factory );
    
    StandaloneSession session = new StandaloneSession();
    
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    
    SimpleOutputHandler handler = new SimpleOutputHandler( out, false );
    
    MimeTypeListener listener = new MimeTypeListener();
    handler.setMimeTypeListener(listener);
    assertEquals( listener, handler.getMimeTypeListener() );
    
    assertFalse( handler.allowFeedback() );
    handler.setSession( session );
    assertEquals( session, handler.getSession() );
    handler.setRuntimeContext( null );

    handler.setContentItem(new SimpleContentItem(), null, null);
    
    IContentItem content2 = handler.getOutputContentItem("bogus", "testoutut:bogus", null, null, null, null, null);
    assertEquals( TestOutputHandler.contentItem, content2 );

    content2 = handler.getOutputContentItem("bogus", "bogus", null, null, null, null, null);
    assertNull( content2 );
    
    content2 = handler.getOutputContentItem(IOutputHandler.RESPONSE, IOutputHandler.CONTENT, null, null, null, null, null);
    
    assertEquals( out, content2.getOutputStream(null) );
    assertFalse( handler.contentDone() );
    assertNull( handler.getFeedbackContentItem() );
    assertFalse( handler.contentDone() );
    
    assertNull( handler.getOutputDefs() );
    assertNull( handler.getOutputDef(null) );
    
    handler.setOutputPreference( -1 );
    assertEquals( -1, handler.getOutputPreference() );
    
    handler.setMimeType("test/test");
    assertEquals( "test/test", handler.getMimeType() );

    handler.setOutput("bogus", null);
    
    handler.setOutput("file:bogus", null);
    
    IContentItem content3 = new BufferedContentItem( null );
    content3.getOutputStream(null).write( "test data".getBytes() );
    content3.closeOutputStream();
    handler.setOutput( IOutputHandler.CONTENT, content3);
    
    assertEquals( "test data", new String( out.toByteArray() ) );

    handler.setOutput( "test", "testoutput");

    assertEquals( "testoutput", handler.getResponseAttributes().get("test") );
    
    handler.setOutput( IOutputHandler.CONTENT, "+test data2");
    assertEquals( "test data+test data2", new String( out.toByteArray() ) );

  }
/*
  public void test2() throws Exception {

    StandaloneSession session = new StandaloneSession();
    
    OutputStream out = new ExceptionOutputStream();
    IContentItem content = new SimpleContentItem(out);
    
    SimpleOutputHandler handler = new SimpleOutputHandler( content, true );
    
    assertTrue( handler.allowFeedback() );

    assertFalse( handler.contentDone() );
    assertNotNull( handler.getFeedbackContentItem() );
    assertTrue( handler.contentDone() );
    IContentItem content2 = handler.getFeedbackContentItem();
    
    assertEquals( content.getOutputStream(null), content2.getOutputStream(null) );
    
    IContentItem content3 = new BufferedContentItem( null );
    content3.getOutputStream(null).write( "test data".getBytes() );
    content3.closeOutputStream();
    handler.setOutput( IOutputHandler.CONTENT, content3);
    
  }
*/
  public void test3() throws Exception {

    StandaloneSession session = new StandaloneSession();
    
    OutputStream out = new ExceptionOutputStream();
    
    SimpleOutputHandler handler = new SimpleOutputHandler( out, false );
    
    assertFalse( handler.allowFeedback() );
    handler.setSession( session );
    assertEquals( session, handler.getSession() );
    handler.setRuntimeContext( null );

    IContentItem content2 = handler.getOutputContentItem(IOutputHandler.RESPONSE, IOutputHandler.CONTENT, null, null, null, null, null);
    
    assertEquals( out, content2.getOutputStream(null) );
    assertNull( handler.getFeedbackContentItem() );
    
    handler.setOutput( IOutputHandler.CONTENT, "test data");

  }

  public void test4() throws Exception {

    StandaloneSession session = new StandaloneSession();
    
    OutputStream out = new ByteArrayOutputStream();
    
    SimpleOutputHandler handler = new SimpleOutputHandler( out, true );
    
    assertTrue( handler.allowFeedback() );

    assertNotNull( handler.getFeedbackContentItem() );
    IContentItem content2 = handler.getFeedbackContentItem();
    
    assertEquals( out, content2.getOutputStream(null) );
    
    OutputStream out3 = new ByteArrayOutputStream();

    handler.setOutputStream(out3, IOutputHandler.RESPONSE, IOutputHandler.CONTENT);
    IContentItem content3 = handler.getOutputContentItem(IOutputHandler.RESPONSE, IOutputHandler.CONTENT, null, null, null, null, null);

    assertEquals( out3, content3.getOutputStream(null) );
    
  }

}