Package org.pentaho.reporting.engine.classic.core.modules.misc.datafactory

Source Code of org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.DataFactoryScriptSupportTest

/*!
* This program is free software; you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License, version 2.1 as published by the Free Software
* Foundation.
*
* You should have received a copy of the GNU Lesser General Public License along with this
* program; if not, you can obtain a copy at http://www.gnu.org/licenses/old-licenses/lgpl-2.1.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 Lesser General Public License for more details.
*
* Copyright (c) 2002-2013 Pentaho Corporation..  All rights reserved.
*/

package org.pentaho.reporting.engine.classic.core.modules.misc.datafactory;

import java.util.Arrays;
import java.util.HashMap;
import javax.swing.table.TableModel;

import junit.framework.Assert;
import junit.framework.TestCase;
import org.pentaho.reporting.engine.classic.core.ClassicEngineBoot;
import org.pentaho.reporting.engine.classic.core.DefaultResourceBundleFactory;
import org.pentaho.reporting.engine.classic.core.ParameterDataRow;
import org.pentaho.reporting.engine.classic.core.ReportDataFactoryException;
import org.pentaho.reporting.engine.classic.core.StaticDataRow;
import org.pentaho.reporting.engine.classic.core.TableDataFactory;
import org.pentaho.reporting.engine.classic.core.designtime.datafactory.DesignTimeDataFactoryContext;
import org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.sql.DriverConnectionProvider;
import org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.sql.SQLReportDataFactory;
import org.pentaho.reporting.libraries.resourceloader.ResourceKey;
import org.pentaho.reporting.libraries.resourceloader.ResourceManager;

public class DataFactoryScriptSupportTest extends TestCase
{
  private String globalScript = "var globalScopeVariable = 'global'; \n" +
      "function init (dataRow)  \n" +
      "{ \n" +
      "  if (globalScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error';  \n" +
      "} \n" +
      "\n" +
      "function shutdown ()  \n" +
      "{ \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error'; \n" +
      "} \n" +
      "\n";

  private String queryScript = "var localScopeVariable = 'local'; \n" +
      "function initQuery ()  \n" +
      "{ \n" +
      "  println ('Init ');\n" +
      "  if (localScopeVariable == null) throw 'error';  \n" +
      "  if (globalScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error';  \n" +
      "} \n" +
      "\n" +
      "function computeQueryFields (query, queryName)  \n" +
      "{ \n" +
      "  println ('computeQueryFields ' + query);\n" +
      "  if (localScopeVariable == null) throw 'error';  \n" +
      "  if (globalScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error'; \n" +
      "  return ['one', 'two'];\n" +
      "} \n" +
      "\n" +
      "function computeQuery (query, queryName, dataRow)  \n" +
      "{ \n" +
      "  println ('computeQuery ' + query);" +
      "  if (localScopeVariable == null) throw 'error';  \n" +
      "  if (globalScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error'; \n" +
      "  return 'result'; \n" +
      "} \n" +
      "\n" +
      "function shutdown ()  \n" +
      "{ \n" +
      "  println ('shutdown ');" +
      "  if (localScopeVariable == null) throw 'error';  \n" +
      "  if (globalScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error'; \n" +
      "} \n" +
      "\n";

  private String queryScript2 = "var localScopeVariable = 'local'; \n" +
      "function initQuery ()  \n" +
      "{ \n" +
      "  println ('Init ');\n" +
      "  if (localScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error';  \n" +
      "} \n" +
      "\n" +
      "function computeQueryFields (query, queryName)  \n" +
      "{ \n" +
      "  println ('computeQueryFields ' + query);\n" +
      "  if (localScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error'; \n" +
      "  return ['one', 'two'];\n" +
      "} \n" +
      "\n" +
      "function computeQuery (query, queryName, dataRow)  \n" +
      "{ \n" +
      "  println ('computeQuery ' + query);" +
      "  if (localScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error'; \n" +
      "  return 'result'; \n" +
      "} \n" +
      "\n" +
      "function shutdown ()  \n" +
      "{ \n" +
      "  println ('shutdown ');" +
      "  if (localScopeVariable == null) throw 'error';  \n" +
      "  if (resourceManager == null) throw 'error';  \n" +
      "  if (dataFactory == null) throw 'error';  \n" +
      "  if (configuration == null) throw 'error'; \n" +
      "  if (contextKey == null) throw 'error';  \n" +
      "  if (resourceBundleFactory == null) throw 'error'; \n" +
      "} \n" +
      "\n";


  public DataFactoryScriptSupportTest()
  {
  }

  public DataFactoryScriptSupportTest(final String name)
  {
    super(name);
  }

  public void setUp()
  {
    ClassicEngineBoot.getInstance().start();
  }

  public void testSimpleSetup() throws ReportDataFactoryException
  {
    final ResourceManager mgr = new ResourceManager();
    mgr.registerDefaults();

    final DataFactoryScriptingSupport support = new DataFactoryScriptingSupport();
    support.setGlobalScriptLanguage("JavaScript");
    support.setGlobalScript(globalScript);
    support.initialize(new TableDataFactory(), new DesignTimeDataFactoryContext
        (ClassicEngineBoot.getInstance().getGlobalConfig(), mgr,
        new ResourceKey("dummy", "dummy", new HashMap()), new DefaultResourceBundleFactory()));
    support.shutdown();
  }

  public void testQuerySetup() throws ReportDataFactoryException
  {
    final ResourceManager mgr = new ResourceManager();
    mgr.registerDefaults();

    final DataFactoryScriptingSupport support = new DataFactoryScriptingSupport();
    support.setGlobalScriptLanguage("JavaScript");
    support.setGlobalScript(globalScript);
    support.initialize(new TableDataFactory(), new DesignTimeDataFactoryContext
        (ClassicEngineBoot.getInstance().getGlobalConfig(), mgr,
        new ResourceKey("dummy", "dummy", new HashMap()), new DefaultResourceBundleFactory()));
    support.setQuery("test", "test-query", null, null);
    support.setQuery("test-script", "test-query-2", "JavaScript", queryScript);

    assertEquals("test-query", support.computeQuery("test", new ParameterDataRow()));
    assertEqualsArray(null, support.computeAdditionalQueryFields("test", new ParameterDataRow()));
    assertEquals("result", support.computeQuery("test-script", new ParameterDataRow()));
    assertEqualsArray(null, support.computeAdditionalQueryFields("test-script", new ParameterDataRow()));
    support.shutdown();
  }

  public void testQueryOnly() throws ReportDataFactoryException
  {
    final ResourceManager mgr = new ResourceManager();
    mgr.registerDefaults();

    final DataFactoryScriptingSupport support = new DataFactoryScriptingSupport();
    support.initialize(new TableDataFactory(), new DesignTimeDataFactoryContext
        (ClassicEngineBoot.getInstance().getGlobalConfig(), mgr,
        new ResourceKey("dummy", "dummy", new HashMap()), new DefaultResourceBundleFactory()));
    support.setQuery("test", "test-query", null, null);
    support.setQuery("test-script", "test-query-2", "JavaScript", queryScript2);

    assertEquals("test-query", support.computeQuery("test", new ParameterDataRow()));
    assertEqualsArray(null, support.computeAdditionalQueryFields("test", new ParameterDataRow()));
    assertEquals("result", support.computeQuery("test-script", new ParameterDataRow()));
    assertEqualsArray(null, support.computeAdditionalQueryFields("test-script", new ParameterDataRow()));
    support.shutdown();
  }
 
  public void testPostProcessResult() throws ReportDataFactoryException {
    final DriverConnectionProvider driverConnectionProvider = new DriverConnectionProvider();

    driverConnectionProvider.setDriver( "org.hsqldb.jdbcDriver" );
    driverConnectionProvider.setUrl( "jdbc:hsqldb:mem:SampleData" );
    driverConnectionProvider.setProperty( "user", "sa" );
    driverConnectionProvider.setProperty( "password", "" );

    final SQLReportDataFactory sqlReportDataFactory = new SQLReportDataFactory( driverConnectionProvider );

    sqlReportDataFactory.setQuery( "default", "SELECT Count(*) FROM CUSTOMERS", "Groovy",
        "import org.pentaho.reporting.engine.classic.core.util.TypedTableModel\n"
            + "def postProcessResult(query, queryName, dataRow, tableModel){\n"
            + "TypedTableModel model = new TypedTableModel([\"column1\"] as String[], [String.class] as Class[]);"
            + "model.addRow(\"row1\");" + "return model;" + "}" );

    try {
      sqlReportDataFactory.initialize( new DesignTimeDataFactoryContext() );
      TableModel data = sqlReportDataFactory.queryData( "default", new StaticDataRow() );
      Assert.assertEquals( 1, data.getColumnCount() );
      Assert.assertEquals( 1, data.getRowCount() );
      Assert.assertEquals( "row1", data.getValueAt( 0, 0 ) );
    } finally {
      sqlReportDataFactory.close();
    }
  }

  private void assertEqualsArray(final String[] o, final String[] strings)
  {
    if (o == null)
    {
      System.out.println(Arrays.asList(strings));
      return;
    }
    assertNotNull(strings);
    assertEquals(Arrays.asList(o), Arrays.asList(strings));
  }
}
TOP

Related Classes of org.pentaho.reporting.engine.classic.core.modules.misc.datafactory.DataFactoryScriptSupportTest

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.