Package org.dbunit.dataset

Examples of org.dbunit.dataset.CachedDataSet


   *
   * @param tableNames
   *            数据库中的表名
   */
  public void backupTables(String[] tableNames) throws Exception {
    cachedDataSet = new CachedDataSet(getDatabaseConnection().createDataSet(
        tableNames));
  }
View Full Code Here


      InputStreamReader inReader = new InputStreamReader(Dbunit.class
          .getResourceAsStream(path), "UTF-8");
      // FlatXmlDataSet fxset = new FlatXmlDataSet(inReader, true,
      // enableColumnSensing, false);
      // return fxset;
      return new CachedDataSet(new FlatXmlProducer(new InputSource(
          inReader), true, enableColumnSensing, false));

    } else {
      return new XmlDataSet(Dbunit.class.getResourceAsStream(path));
    }
View Full Code Here

  }

  private IDataSet getDatasetFromInputStream(InputStream in) {
    IDataSetProducer producer = new FlatXmlProducer(new InputSource(in), false);
    try {
      return new CachedDataSet(producer);
    } catch (DataSetException e) {
      throw new RuntimeException("Cannot get dataset.", e);
    }
  }
View Full Code Here

     */
    @SuppressWarnings("rawtypes")
    @Override
    protected IDataSet loadDataSet(final File source) throws DataSetException,
            IOException {
        final CachedDataSet dataSet = new CachedDataSet();
        dataSet.startDataSet();
        final CsvParser parser = new CsvParserImpl();
        final List readData = parser.parse(source);
        final List readColumns = ((List) readData.get(0));
        final Column[] columns = new Column[readColumns.size()];
        for (int i = 0; i < readColumns.size(); i++) {
            columns[i] = new Column((String) readColumns.get(i),
                    DataType.UNKNOWN);
        }
        final String tableName = source.getName().substring(0,
                source.getName().indexOf(".csv"));
        final ITableMetaData metaData = new DefaultTableMetaData(tableName,
                columns);
        dataSet.startTable(metaData);
        for (int rowIndex = 1; rowIndex < readData.size(); rowIndex++) {
            final List row = (List) readData.get(rowIndex);
            final Object[] values = row.toArray();
            for (int columnIndex = 0; columnIndex < values.length; columnIndex++) {
                if (values[columnIndex].equals(CsvDataSetWriter.NULL)) {
                    values[columnIndex] = null;
                }
            }
            dataSet.row(values);
        }
        dataSet.endTable();
        dataSet.endDataSet();
        return dataSet;
    }
View Full Code Here

            {
                producer = new FlatDtdProducer(getInputSource(src));
            }
            else if (format.equalsIgnoreCase(FORMAT_XLS))
            {
                return new CachedDataSet(new XlsDataSet(src));
            }
            else
            {
                throw new IllegalArgumentException("Type must be either 'flat'(default), 'xml', 'csv', 'xls' or 'dtd' but was: " + format);
            }

            if (forwardonly)
            {
                return new StreamingDataSet(producer);
            }
            return new CachedDataSet(producer);
        }
        catch (IOException e)
        {
            throw new DatabaseUnitException(e);
        }
View Full Code Here

   *
   * @param tableNames
   *            数据库中的表名
   */
  public void backupTables(String[] tableNames) throws Exception {
    cachedDataSet = new CachedDataSet(getDatabaseConnection().createDataSet(
        tableNames));
  }
View Full Code Here

            } else {
                tableName = source.getSourceFile().substring(slashPos + 1, dotPos);
            }
        }

        final CachedDataSet dataSet = new CachedDataSet();
        dataSet.startDataSet();
        final List readColumns = ((List)readData.get(0));
        final Column[] columns = new Column[readColumns.size()];
        for (int i = 0; i < readColumns.size(); i++) {
            columns[i] = new Column((String)readColumns.get(i), DataType.UNKNOWN);
        }

        final ITableMetaData metaData = new DefaultTableMetaData(tableName, columns);
        dataSet.startTable(metaData);
        for (int rowIndex = 1; rowIndex < readData.size(); rowIndex++) {
            final List row = (List)readData.get(rowIndex);
            final Object[] values = row.toArray();
            for (int columnIndex = 0; columnIndex < values.length; columnIndex++) {
                if (values[columnIndex].equals(CsvDataSetWriter.NULL)) {
                    values[columnIndex] = null;
                }
            }
            dataSet.row(values);
        }
        dataSet.endTable();
        dataSet.endDataSet();
        return dataSet;
    }
View Full Code Here

TOP

Related Classes of org.dbunit.dataset.CachedDataSet

Copyright © 2018 www.massapicom. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.