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: 815 $
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 }