View Javadoc
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  package org.dbunit.dataset.stream;
22  
23  import org.dbunit.dataset.ForwardOnlyTableTest;
24  import org.dbunit.dataset.ITable;
25  import org.dbunit.dataset.ITableIterator;
26  import org.dbunit.dataset.xml.FlatXmlDataSetTest;
27  import org.dbunit.dataset.xml.FlatXmlProducer;
28  import org.xml.sax.InputSource;
29  
30  import java.io.FileReader;
31  
32  /**
33   * @author Manuel Laflamme
34   * @since Apr 11, 2003
35   * @version $Revision$
36   */
37  public class StreamingTableTest extends ForwardOnlyTableTest
38  {
39      private static final String TEST_TABLE = "TEST_TABLE";
40  
41      public StreamingTableTest(String s)
42      {
43          super(s);
44      }
45  
46      protected ITable createTable() throws Exception
47      {
48          FileReader reader = new FileReader(FlatXmlDataSetTest.DATASET_FILE);
49  
50  //        IDataSetProducer source = new DataSetProducerAdapter(new FlatXmlDataSet(reader));
51          IDataSetProducer source = new FlatXmlProducer(new InputSource(reader));
52          ITableIterator iterator = new StreamingDataSet(source).iterator();
53          while(iterator.next())
54          {
55              ITable table = iterator.getTable();
56              String tableName = table.getTableMetaData().getTableName();
57              if (tableName.equals(TEST_TABLE))
58              {
59                  return table;
60              }
61          }
62  
63          throw new IllegalStateException();
64      }
65  }