@Test public void test_something() { FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setColumnSensing(true); FlatXmlDataSet xmlDataSet = builder.build( this.getClass().getResourceAsStream("/dbunit/testset.xml")); // ...
public class ColumnDetectorXmlDataSetLoader extends FlatXmlDataSetLoader { @Override protected IDataSet createDataSet(Resource resource) throws Exception { FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setColumnSensing(true); InputStream inputStream = resource.getInputStream(); try { return builder.build(inputStream); } finally { inputStream.close(); } } }
private IDataSet loadXmlDataSet(final String xmlFile) throws DataSetException { final FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder(); flatXmlDataSetBuilder.setColumnSensing(true); addDtdIfDefined(flatXmlDataSetBuilder, xmlFile); return flatXmlDataSetBuilder.build(Thread.currentThread().getContextClassLoader().getResourceAsStream(xmlFile)); }
InputStream inputStream = ClassNameTest.class.getResourceAsStream(name); assertNotNull("file " + name + " not found in classpath", inputStream); FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setColumnSensing(true); return builder.build(inputStream);
private IDataSet loadXmlDataSet(final String xmlFile) throws DataSetException { final FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder(); flatXmlDataSetBuilder.setColumnSensing(true); addDtdIfDefined(flatXmlDataSetBuilder, xmlFile); return flatXmlDataSetBuilder.build(Thread.currentThread().getContextClassLoader().getResourceAsStream(xmlFile)); }
private IDataSet loadXmlDataSet(final String xmlFile) throws DataSetException { final FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder(); flatXmlDataSetBuilder.setColumnSensing(true); addDtdIfDefined(flatXmlDataSetBuilder, xmlFile); return flatXmlDataSetBuilder.build(Thread.currentThread().getContextClassLoader().getResourceAsStream(xmlFile)); }
private IDataSet loadXmlDataSet(final String xmlFile) throws DataSetException { final FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder(); flatXmlDataSetBuilder.setColumnSensing(true); addDtdIfDefined(flatXmlDataSetBuilder, xmlFile); return flatXmlDataSetBuilder.build(Thread.currentThread().getContextClassLoader().getResourceAsStream(xmlFile)); }
private IDataSet build(String file, String replaceStr, boolean cached) { try { if (file.endsWith(".xml")) { FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder(); flatXmlDataSetBuilder.setColumnSensing(cached); return flatXmlDataSetBuilder.build(readFileAndReplaceTableName(file, replaceStr)); } if (file.endsWith(".csv")) { URL resource = getClass().getClassLoader().getResource(file); if (resource == null) { throw new RuntimeException(String.format("file[%s] not found", new Object[]{file})); } return new QCsvDataSet(new File(resource.getPath())); } throw new IllegalStateException("DbUnit only supports CSV or Flat XML data sets for the moment"); } catch (Exception e) { throw new RuntimeException(e.getMessage(), e); } }
@Override protected IDataSet createDataSet(Resource resource) throws Exception { FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setColumnSensing(true); return buildDataSet(builder, resource); }
@Override protected IDataSet createDataSet(Resource resource) throws Exception { FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setColumnSensing(true); return buildDataSet(builder, resource); }
/** * Load a DataSet from XML file. * * @param filename data file xml. * @return an IDataSet instance of the entities loaded from xml. * @throws DataSetException if an error occurred while reading XML definition. */ protected IDataSet getDataSet(@NotNull @Size(min = 1) final String filename) throws DataSetException { // Check sanity Validate.notEmpty(filename, "filename"); final FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder(); flatXmlDataSetBuilder.setColumnSensing(true); final ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader(); final InputStream flatXmlFile = contextClassLoader.getResourceAsStream(filename); if (flatXmlFile == null) { throw new NullPointerException("Could not find a file from filename [" + filename + "]"); } // All done. return flatXmlDataSetBuilder.build(flatXmlFile); }
// since dbunit 2.4.7 FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setInputSource(new File("src/xml/flatXmlTableTest.xml")); builder.setColumnSensing(true); IDataSet dataSet = builder.build(); // or dbunit release <= 2.4.6: boolean enableColumnSensing = true; IDataSet dataSet = new FlatXmlDataSet( new File("src/xml/flatXmlTableTest.xml"), false, enableColumnSensing);
private DatabaseHelper dbh; private EntityManager em; private IDatabaseConnection connection; private IDataSet dataset; @BeforeClass private void setupDatabaseResource() throws Exception { // using JPA and a custom helper class em = dbh.getEntityManager(); connection = new DatabaseConnection(((SessionImpl) (em.getDelegate())).connection()); connection.getConfig().setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new HsqldbDataTypeFactory()); // full database export IDataSet fullDataSet = connection.createDataSet(); FlatXmlDataSet.write(fullDataSet, new FileOutputStream("target/generated-sources/test-dataset.xml")); FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder(); flatXmlDataSetBuilder.setColumnSensing(true); // keep the dataset in memory, for later restore points dataset = flatXmlDataSetBuilder.build(new FileInputStream("target/generated-sources/test-dataset.xml")); }
@Override protected IDataSet getDataSet() throws Exception { FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setColumnSensing(true); return builder.build(this.getClass().getClassLoader().getResourceAsStream(DATASET)); }
protected static IDataSet initDataSet() throws Exception { FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setColumnSensing(true); return builder.build(BaseDaoTestCase.class.getResourceAsStream("/fixtures.xml")); }
private ReplacementDataSet getDataSet(Description description) throws Exception { try (InputStream dataSetInputStream = description.getTestClass() .getResourceAsStream(xmlDataSet)) { ReplacementDataSet dataSet = new ReplacementDataSet( new FlatXmlDataSetBuilder().setColumnSensing(true).build(dataSetInputStream)); dataSet.addReplacementObject("[NULL]", null); return dataSet; } } }
/** * Imports DB data from the specified XML * @param xmlUrl URL to XML file location */ public ImportBuilder from(URL xmlUrl) { try { flatXmlDataSet = new FlatXmlDataSetBuilder() .setColumnSensing(true) .setCaseSensitiveTableNames(false) .build(xmlUrl); this.dataSetUrl = xmlUrl; return this; } catch (Exception ex) { throw new JuDbException("Couldn't import data from XML: " + xmlUrl, ex); } }
private IDataSet loadDataSet() throws IOException, DataSetException { InputStream in = target.getClass().getResourceAsStream(name); if (in == null) { throw new Error("Could not find resource '" + name + "'"); } return new LowerCaseDataSet(new FlatXmlDataSetBuilder() .setDtdMetadata(true) .setColumnSensing(true) .build(new InputStreamReader(in))); }
/** * {@inheritDoc} */ @Override protected IDataSet fromResource(Resource resource, DataSetFormatOptions options) throws DataSetException, IOException { FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder(); builder.setColumnSensing(options.isColumnSensing()); builder.setDtdMetadata(options.isDtdMetadata()); builder.setCaseSensitiveTableNames(options.isCaseSensitiveTableNames()); if (StringUtils.hasText(options.getDtdLocation())) { IDataSet metaDataSet = FLAT_DTD.loadUnique(null, options.getDtdLocation()); builder.setMetaDataSet(metaDataSet); } return builder.build(resource.getInputStream()); } },
public void loadDatset(URL resource) throws Throwable { InputStream in = Resources.asByteSource(resource).openStream(); Preconditions.checkNotNull("Cannot find xml file " + resource); LowerCaseDataSet dataSet = new LowerCaseDataSet(new FlatXmlDataSetBuilder() .setDtdMetadata(true) .setColumnSensing(true) .build(new InputStreamReader(in))); IDatabaseConnection dbUnitConnection = new MySqlConnection(connection, null); InsertIdentityOperation.INSERT.execute(dbUnitConnection, dataSet); }