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