/*
    *
    * The DbUnit Database Testing Framework
    * Copyright (C)2002-2004, DbUnit.org
    *
    * This library is free software; you can redistribute it and/or
    * modify it under the terms of the GNU Lesser General Public
    * License as published by the Free Software Foundation; either
    * version 2.1 of the License, or (at your option) any later version.
   *
   * This library 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.
   *
   * You should have received a copy of the GNU Lesser General Public
   * License along with this library; if not, write to the Free Software
   * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
   *
   */
  
  package org.dbunit.dataset;
  
  import static org.junit.jupiter.api.Assertions.fail;
  
  import java.io.OutputStream;
  import java.nio.file.Files;
  import java.nio.file.Paths;
  
  import org.dbunit.AbstractDatabaseIT;
  import org.dbunit.DdlExecutor;
  import org.dbunit.database.QueryDataSet;
  import org.dbunit.dataset.datatype.DataType;
  import org.dbunit.dataset.xml.FlatXmlDataSet;
  import org.dbunit.testutil.TestUtils;
  import org.junit.jupiter.api.AfterEach;
  import org.junit.jupiter.api.BeforeEach;
  import org.junit.jupiter.api.Test;
  
  /**
   * Test Case for issue #1721870
   *
   * @author Sebastien Le Callonnec
   * @version $Revision$
   * @since Mar 11, 2008
   */
  class CompositeDataSetIterationIT extends AbstractDatabaseIT
  {
  
      private final String sqlFile = "hypersonic_simple_dataset.sql";
  
      @BeforeEach
      protected void setUpTables() throws Exception
      {
          DdlExecutor.dropTables(_connection.getConnection(), "A", "B", "C");
          DdlExecutor.executeDdlFile(TestUtils.getFile("sql/" + sqlFile),
                  _connection.getConnection(), false);
      }
  
      @AfterEach
      protected void tearDownTables() throws Exception
      {
          DdlExecutor.dropTables(_connection.getConnection(), "A", "B", "C");
          refreshConnection();
      }
  
      /**
       * Replaces {@code _connection} with a fresh connection after test tables
       * have been dropped, so that {@code AbstractDatabaseIT.tearDown()} uses an
       * up-to-date dataset that does not include the dropped tables.
       *
       * @throws Exception
       */
      private void refreshConnection() throws Exception
      {
          _connection.close();
          _connection = getDatabaseTester().getConnection();
          setUpDatabaseConfig(_connection.getConfig());
      }
  
      @Test
      void testCompositeIteration_withQueryAndDefaultDataSets_writesAllTablesToFile() throws Exception
      {
  
          // 1. QueryDataSet
          final QueryDataSet queryDataSet = new QueryDataSet(_connection);
          queryDataSet.addTable("B", "select * from B");
          queryDataSet.addTable("C", "select * from C");
  
          // 2. Hard-coded data set
          final DefaultDataSet plainDataSet = new DefaultDataSet();
  
          final Column id = new Column("id", DataType.DOUBLE);
          final Column name = new Column("name", DataType.VARCHAR);
  
          final Column[] cols = {id, name};
  
          final DefaultTable aTable = new DefaultTable("D", cols);
          final Object[] row1 = {Long.valueOf(1), "D1"};
         final Object[] row2 = {Long.valueOf(2), "D2"};
 
         aTable.addRow(row1);
         aTable.addRow(row2);
 
         plainDataSet.addTable(aTable);
 
         // 3. Composite
         final CompositeDataSet compositeDataSet =
                 new CompositeDataSet(queryDataSet, plainDataSet);
 
         // 4. Write
         try (OutputStream out = Files.newOutputStream(Paths.get("target/full.xml")))
         {
             FlatXmlDataSet.write(compositeDataSet, out);
         }
         catch (final Exception e)
         {
             fail(e.getMessage());
         }
     }
 }