/*
 * 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) 2006 - 2017 Pentaho Corporation..  All rights reserved.
 */

package org.pentaho.reporting.engine.classic.extensions.datasources.olap4j;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.pentaho.reporting.engine.classic.core.ClassicEngineBoot;
import org.pentaho.reporting.engine.classic.core.DataFactory;
import org.pentaho.reporting.engine.classic.core.ParameterDataRow;
import org.pentaho.reporting.engine.classic.core.ReportDataFactoryException;
import org.pentaho.reporting.engine.classic.core.designtime.datafactory.DesignTimeDataFactoryContext;
import org.pentaho.reporting.engine.classic.core.testsupport.DebugJndiContextFactoryBuilder;
import org.pentaho.reporting.engine.classic.extensions.datasources.olap4j.connections.DriverConnectionProvider;

import javax.naming.spi.NamingManager;
import javax.swing.table.TableModel;

public class Prd5276T {
  private static final String query = Olap4JTestUtil.QUERY_UNION_OK;

  private static final String queryFlipped = Olap4JTestUtil.QUERY_UNION_FLIPPED;

  private static final String queryBroken = Olap4JTestUtil.QUERY_UNION_BROKEN;

  private static final String queryMultipleH =
    "SELECT [Product].Children ON COLUMNS, Hierarchize({[Time].[Years].Members, [Time].[Quarters].Members, [Time]"
      + ".[Months].Members}) ON ROWS FROM [SteelWheelsSales]";

  @Before
  public void setUp() throws Exception {
    ClassicEngineBoot.getInstance().start();
    if ( NamingManager.hasInitialContextFactoryBuilder() == false ) {
      NamingManager.setInitialContextFactoryBuilder( new DebugJndiContextFactoryBuilder() );
    }
  }

  /**
   * Validates that queries with empty results (no rows or no columns) are correctly handled by CachingDataFactory.
   * <p/>
   * http://jira.pentaho.com/browse/PRD-4628
   */
  @Test
  public void testQuery() throws ReportDataFactoryException {
    DataFactory dataFactory = createDataFactory( queryBroken );
    final TableModel tableModel = dataFactory.queryData( "default", new ParameterDataRow() );
    //    new DataPreviewDialog().showData(tableModel);

    Assert.assertEquals( "[Markets].[(All)]", tableModel.getColumnName( 2 ) );
    Assert.assertEquals( "[Product].[Line]", tableModel.getColumnName( 1 ) );
    Assert.assertNotNull( tableModel.getValueAt( 0, 3 ) );
    Assert.assertNotNull( tableModel.getValueAt( 1, 1 ) );
    Assert.assertNotNull( tableModel.getValueAt( 2, 2 ) );

    //    TableModelInfo.printTableModel(tableModel);
  }

  /**
   * Validates that queries with empty results (no rows or no columns) are correctly handled by CachingDataFactory.
   * <p/>
   * http://jira.pentaho.com/browse/PRD-4628
   */
  @Test
  public void testQueryOK() throws ReportDataFactoryException {
    DataFactory dataFactory = createDataFactory( query );
    final TableModel tableModel = dataFactory.queryData( "default", new ParameterDataRow() );
    //    new DataPreviewDialog().showData(tableModel);

    Assert.assertEquals( "[Markets].[(All)]", tableModel.getColumnName( 2 ) );
    Assert.assertEquals( "[Product].[Line]", tableModel.getColumnName( 1 ) );
    Assert.assertNotNull( tableModel.getValueAt( 1, 1 ) );
    Assert.assertNotNull( tableModel.getValueAt( 2, 2 ) );

    //    TableModelInfo.printTableModel(tableModel);
  }

  @Test
  public void testQueryMultipleH() throws ReportDataFactoryException {
    DataFactory dataFactory = createDataFactory( queryMultipleH );
    final TableModel tableModel = dataFactory.queryData( "default", new ParameterDataRow() );

    Assert.assertEquals( "[Time].[Quarters]", tableModel.getColumnName( 2 ) );
    Assert.assertEquals( "[Time].[Years]", tableModel.getColumnName( 1 ) );
    Assert.assertNotNull( tableModel.getValueAt( 1, 1 ) );
    Assert.assertNotNull( tableModel.getValueAt( 2, 2 ) );

    //    TableModelInfo.printTableModel(tableModel);
  }

  protected DataFactory createDataFactory( final String query ) throws ReportDataFactoryException {
    final DriverConnectionProvider provider = new DriverConnectionProvider();
    provider.setDriver( "mondrian.olap4j.MondrianOlap4jDriver" );
    provider.setProperty( "Catalog",
      "src/test/resources/org/pentaho/reporting/engine/classic/extensions/datasources/olap4j/steelwheels.mondrian.xml" );
    provider.setProperty( "JdbcUser", "sa" );
    provider.setProperty( "JdbcPassword", "" );
    provider.setProperty( "Jdbc", "jdbc:hsqldb:mem:SampleData" );
    provider.setProperty( "JdbcDrivers", "org.hsqldb.jdbcDriver" );
    provider.setUrl( "jdbc:mondrian:" );

    final BandedMDXDataFactory dataFactory = new BandedMDXDataFactory( provider );
    dataFactory.setQuery( "default", query, null, null );
    dataFactory.initialize( new DesignTimeDataFactoryContext() );
    return dataFactory;
  }

}