1   /*
2    *
3    * The DbUnit Database Testing Framework
4    * Copyright (C)2002-2004, DbUnit.org
5    *
6    * This library is free software; you can redistribute it and/or
7    * modify it under the terms of the GNU Lesser General Public
8    * License as published by the Free Software Foundation; either
9    * version 2.1 of the License, or (at your option) any later version.
10   *
11   * This library is distributed in the hope that it will be useful,
12   * but WITHOUT ANY WARRANTY; without even the implied warranty of
13   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14   * Lesser General Public License for more details.
15   *
16   * You should have received a copy of the GNU Lesser General Public
17   * License along with this library; if not, write to the Free Software
18   * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19   *
20   */
21  
22  package org.dbunit.dataset;
23  
24  import org.dbunit.database.AmbiguousTableNameException;
25  import org.dbunit.dataset.xml.XmlDataSet;
26  import org.dbunit.testutil.TestUtils;
27  
28  import java.io.FileReader;
29  
30  /**
31   * @author Manuel Laflamme
32   * @version $Revision: 1162 $
33   * @since Feb 22, 2002
34   */
35  public class DefaultDataSetTest extends AbstractDataSetTest
36  {
37      public DefaultDataSetTest(String s)
38      {
39          super(s);
40      }
41  
42      protected IDataSet createDataSet() throws Exception
43      {
44          IDataSet dataSet = new XmlDataSet(
45                  TestUtils.getFileReader("xml/dataSetTest.xml"));
46          ITable[] tables = DataSetUtils.getTables(dataSet);
47  
48          return new DefaultDataSet(tables);
49      }
50  
51      protected IDataSet createDuplicateDataSet() throws Exception
52      {
53          return createDuplicateDataSet(false);
54      }
55      
56      protected IDataSet createMultipleCaseDuplicateDataSet() throws Exception 
57      {
58          return createDuplicateDataSet(true);
59      }
60  
61      private IDataSet createDuplicateDataSet(boolean multipleCase) throws AmbiguousTableNameException 
62      {
63          ITable[] tables = super.createDuplicateTables(multipleCase);
64          return new DefaultDataSet(tables);
65      }
66  
67      public void testAddTableThenReadBackAndDoItAgainDataSet() throws Exception
68      {
69      	String tableName1 = "TEST_TABLE";
70      	String tableName2 = "SECOND_TABLE";
71          DefaultDataSet dataSet = new DefaultDataSet();
72          
73          DefaultTable table1 = new DefaultTable(tableName1);
74          dataSet.addTable(table1);
75          assertEquals(table1, dataSet.getTable(tableName1));
76          
77          DefaultTable table2 = new DefaultTable(tableName2);
78          dataSet.addTable(table2);
79          assertEquals(table2, dataSet.getTable(tableName2));
80      }
81      
82  }
83  
84  
85  
86