Java Code Examples for org.dbunit.dataset.xml.FlatXmlDataSet#write()
The following examples show how to use
org.dbunit.dataset.xml.FlatXmlDataSet#write() .
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
| Source File: DataSetBuilderTest.java From database-rider with Apache License 2.0 | 6 votes |
|
@Test
public void shouldAddEmptyTable() throws IOException, DataSetException {
DataSetBuilder builder = new DataSetBuilder();
IDataSet dataSet = builder.
table("USER")
.build();
File datasetFile = Files.createTempFile("rider-empty-dataset", ".xml").toFile();
FileOutputStream fos = new FileOutputStream(datasetFile);
FlatXmlDataSet.write(dataSet, fos);
assertThat(contentOf(datasetFile)).
contains("<?xml version='1.0' encoding='UTF-8'?>"+NEW_LINE +
"<dataset>"+NEW_LINE +
" <USER/>"+NEW_LINE +
"</dataset>");
}
Example 2
| Source File: DataSetBuilderTest.java From database-rider with Apache License 2.0 | 6 votes |
|
@Test
public void shouldGeneratFlatXmlDataSetWithEmptyUserTableWithColumnValuesSyntax() throws DataSetException, IOException {
DataSetBuilder builder = new DataSetBuilder();
builder
.table("USER")
.table("TWEET")
.columns("ID","CONTENT","DATE")
.values("abcdef12345", "dbunit rules!", "[DAY,NOW]")
.table("FOLLOWER")
.columns("ID", "USER_ID","FOLLOWER_ID")
.values(1 , 1 , 2)
.build();
IDataSet dataSet = builder.build();
File datasetFile = Files.createTempFile("rider-dataset", ".xml").toFile();
FileOutputStream fos = new FileOutputStream(datasetFile);
FlatXmlDataSet.write(dataSet, fos);
assertThat(contentOf(datasetFile)).
contains("<?xml version='1.0' encoding='UTF-8'?>"+NEW_LINE +
"<dataset>"+NEW_LINE +
" <USER/>"+NEW_LINE +
" <TWEET ID=\"abcdef12345\" CONTENT=\"dbunit rules!\" DATE=\"[DAY,NOW]\"/>"+NEW_LINE +
" <FOLLOWER ID=\"1\" USER_ID=\"1\" FOLLOWER_ID=\"2\"/>"+NEW_LINE +
"</dataset>");
}
Example 3
| Source File: BasePaymentModuleDBTestCase.java From yes-cart with Apache License 2.0 | 6 votes |
|
protected void dumpDataBase(final String prefix, final String ... tables) {
if (!enabled) {
return; // no dumps
}
if (tables != null && tables.length > 0) {
final File dump = new File("target" + File.separator + prefix + "_dataset.xml");
try {
QueryDataSet queryDataSet = new QueryDataSet(dbTester.getConnection());
for (String tableName : tables) {
queryDataSet.addTable(tableName);
}
FlatXmlDataSet.write(queryDataSet, new FileOutputStream(dump));
System.out.println("DUMP: " + dump.getAbsoluteFile());
} catch (Exception exp) {
System.out.println("Unable to create dump file: " + dump.getAbsoluteFile());
exp.printStackTrace();
}
}
}
Example 4
| Source File: AbstractTestDAO.java From yes-cart with Apache License 2.0 | 6 votes |
|
protected void dumpDataBase(final String prefix, final String ... tables) {
if (!enabled) {
return; // no dumps
}
if (tables != null && tables.length > 0) {
final File dump = new File("target" + File.separator + prefix + "_dataset.xml");
try {
QueryDataSet queryDataSet = new QueryDataSet(dbTester.getConnection());
for (String tableName : tables) {
queryDataSet.addTable(tableName);
}
FlatXmlDataSet.write(queryDataSet, new FileOutputStream(dump));
System.out.println("DUMP: " + dump.getAbsoluteFile());
} catch (Exception exp) {
System.out.println("Unable to create dump file: " + dump.getAbsoluteFile());
exp.printStackTrace();
}
}
}
Example 5
| Source File: DbUnitUtil.java From Leo with Apache License 2.0 | 6 votes |
|
/**
* 把数据集写入到文件中
* @param dataSet
* @param destFilePath
*/
public static void writeToFileFromDataSet(IDataSet dataSet,String destFilePath) {
File destFile = new File(destFilePath);
try {
String destFileName = destFile.getName();
//转换为excel格式
if(destFileName.endsWith(".xls")){
XlsDataSet.write(dataSet, new FileOutputStream(destFile));
}else if(destFileName.endsWith(".xml")){
FlatXmlDataSet.write(dataSet, new FileOutputStream(destFile));
}else{
log.error("文件格式不是xls或者xml，不支持");
return;
}
log.info("写入数据到文件 : "+destFile.getAbsolutePath());
} catch (Exception e) {
log.error("写入数据到文件失败");
log.error(e.getMessage());
}
}
Example 6
| Source File: DataSetExecutorImpl.java From database-rider with Apache License 2.0 | 5 votes |
|
private void logDataSet(IDataSet resultingDataSet, Exception e) {
try {
File datasetFile = Files.createTempFile("dataset-log", ".xml").toFile();
log.info("Saving current dataset to " + datasetFile.getAbsolutePath());
try (FileOutputStream fos = new FileOutputStream(datasetFile)) {
FlatXmlDataSet.write(resultingDataSet, fos);
}
} catch (Exception e1) {
log.error("Could not log created dataset.", e);
}
}
Example 7
| Source File: DataSetBuilderTest.java From database-rider with Apache License 2.0 | 5 votes |
|
@Test
public void shouldGenerateFlatXmlDataSet() throws DataSetException, IOException {
DataSetBuilder builder = new DataSetBuilder();
ColumnSpec id = ColumnSpec.of("ID");
builder.table("USER")
.row()
.column("ID", 1)
.column("NAME", "@realpestano")
.table("USER")
.row().column(id, 2)
.column("NAME", "@dbunit")
.table("TWEET")
.row()
.column("ID", "abcdef12345")
.column("CONTENT", "dbunit rules!")
.column("DATE", "[DAY,NOW]")
.table("FOLLOWER")
.row()
.column(id, 1)
.column("USER_ID", 1)
.column("FOLLOWER_ID", 2)
.build();
IDataSet dataSet = builder.build();
File datasetFile = Files.createTempFile("rider-dataset", ".xml").toFile();
FileOutputStream fos = new FileOutputStream(datasetFile);
FlatXmlDataSet.write(dataSet, fos);
assertThat(contentOf(datasetFile)).
contains("<?xml version='1.0' encoding='UTF-8'?>"+NEW_LINE +
"<dataset>"+NEW_LINE +
" <USER ID=\"1\" NAME=\"@realpestano\"/>"+NEW_LINE +
" <USER ID=\"2\" NAME=\"@dbunit\"/>"+NEW_LINE +
" <TWEET ID=\"abcdef12345\" CONTENT=\"dbunit rules!\" DATE=\"[DAY,NOW]\"/>"+NEW_LINE +
" <FOLLOWER ID=\"1\" USER_ID=\"1\" FOLLOWER_ID=\"2\"/>"+NEW_LINE +
"</dataset>");
}
Example 8
| Source File: DataSetBuilderTest.java From database-rider with Apache License 2.0 | 5 votes |
|
@Test
public void shouldGeneratFlatXmlDataSetWithEmptyUserTable() throws DataSetException, IOException {
DataSetBuilder builder = new DataSetBuilder();
ColumnSpec id = ColumnSpec.of("ID");
builder.table("USER")
.table("TWEET")
.row()
.column("ID", "abcdef12345")
.column("CONTENT", "dbunit rules!")
.column("DATE", "[DAY,NOW]")
.table("FOLLOWER")
.row()
.column(id, 1)
.column("USER_ID", 1)
.column("FOLLOWER_ID", 2)
.build();
IDataSet dataSet = builder.build();
File datasetFile = Files.createTempFile("rider-dataset", ".xml").toFile();
FileOutputStream fos = new FileOutputStream(datasetFile);
FlatXmlDataSet.write(dataSet, fos);
assertThat(contentOf(datasetFile)).
contains("<?xml version='1.0' encoding='UTF-8'?>"+NEW_LINE +
"<dataset>"+NEW_LINE +
" <USER/>"+NEW_LINE +
" <TWEET ID=\"abcdef12345\" CONTENT=\"dbunit rules!\" DATE=\"[DAY,NOW]\"/>"+NEW_LINE +
" <FOLLOWER ID=\"1\" USER_ID=\"1\" FOLLOWER_ID=\"2\"/>"+NEW_LINE +
"</dataset>");
}
Example 9
| Source File: AbstractDbUnitTestCase.java From wetech-cms with MIT License | 5 votes |
|
private void writeBackupFile(IDataSet ds) throws IOException, DataSetException {
tempFile = File.createTempFile("back", "xml");
dtdFile = File.createTempFile("back","dtd");
//写dtd
FlatDtdDataSet.write(ds, new FileWriter(dtdFile));
//写数据表中的文件
FlatXmlDataSet.write(ds, new FileWriter(tempFile));
}
Example 10
| Source File: DatabaseCapable.java From jadira with Apache License 2.0 | 5 votes |
|
protected void writeExpectedFile(IDatabaseConnection dbunitConnection,
File outputFile, String tableName) throws IOException,
DataSetException {
QueryDataSet partialDataSet = new QueryDataSet(dbunitConnection);
partialDataSet.addTable(tableName);
FlatXmlDataSet.write(partialDataSet, new FileOutputStream(outputFile));
}
Example 11
| Source File: DBUnitDataExtractor.java From gazpachoquest with GNU General Public License v3.0 | 5 votes |
|
/**
* Performs the extraction. If no tables or queries are specified, data from
* entire database will be extracted.
* Otherwise, a partial extraction will be performed.
*
* @throws Exception
*/
public void extract() throws Exception {
Connection conn = null;
try {
conn = dataSource.getConnection();
logger.info("Beginning extraction from '" + conn.toString() + "'.");
IDatabaseConnection connection = new DatabaseConnection(conn, schema);
configConnection((DatabaseConnection) connection);
if (tableList != null || queryList != null) {
// partial database export
QueryDataSet partialDataSet = new QueryDataSet(connection);
addTables(partialDataSet);
addQueries(partialDataSet);
FlatXmlDataSet.write(partialDataSet, new FileOutputStream(dataSetName));
} else {
// full database export
IDataSet fullDataSet = connection.createDataSet();
FlatXmlDataSet.write(fullDataSet, new FileOutputStream(dataSetName));
// dependent tables database export: export table X and all
// tables that
// have a PK which is a FK on X, in the right order for
// insertion
String[] depTableNames = TablesDependencyHelper.getAllDependentTables(connection, "research");
IDataSet depDataset = connection.createDataSet(depTableNames);
FlatXmlWriter datasetWriter = new FlatXmlWriter(new FileOutputStream("target/dependents.xml"));
datasetWriter.write(depDataset);
}
} finally {
if (conn != null) {
conn.close();
}
}
logger.info("Completed extraction to '" + dataSetName + "'.");
}
Example 12
| Source File: AbstractDbUnitTestCase.java From wetech-cms with MIT License | 4 votes |
|
private void writeBackupFile(IDataSet ds) throws IOException, DataSetException {
tempFile = File.createTempFile("back","xml");
FlatXmlDataSet.write(ds, new FileWriter(tempFile));
}
Example 13
| Source File: AbstractDbUnitTestCase.java From SpringCloud with Apache License 2.0 | 4 votes |
|
private void writeBackupFile(IDataSet ds) throws IOException, DataSetException {
tempFile = File.createTempFile("back","xml");
FlatXmlDataSet.write(ds, new FileWriter(tempFile));
}
Example 14
| Source File: DBUnitTestExecutionListener.java From flyway-test-extensions with Apache License 2.0 | 4 votes |
|
/**
* Only annotation with saveIt will be executed
*/
public void afterTestMethod(final TestContext testContext) throws Exception {
// no we check for the DBResetForClass
final Method testMethod = testContext.getTestMethod();
final Annotation annotation = testMethod
.getAnnotation(DBUnitSupport.class);
if (annotation != null) {
final DBUnitSupport dbUnitAnnotaton = (DBUnitSupport) annotation;
final String saveIt = dbUnitAnnotaton.saveFileAfterRun();
final String[] tables = dbUnitAnnotaton.saveTableAfterRun();
if (saveIt != null && saveIt.trim().length() > 0) {
String executionInfo = "";
if (logger.isDebugEnabled()) {
executionInfo = ExecutionListenerHelper
.getExecutionInformation(testContext);
logger.debug("******** Start save information '"
+ executionInfo + "' info file '" + saveIt + "'.");
}
final DataSource ds = getSaveDataSource(testContext);
final IDatabaseConnection con = getConnection(ds, testContext);
// Issue 16 fix leaking database connection - will look better with Java 7 Closable
try {
final IDataSet dataSet = getDataSetToExport(tables, con);
final File fileToExport = getFileToExport(saveIt);
final FileWriter fileWriter = new FileWriter(fileToExport);
FlatXmlDataSet.write(dataSet, fileWriter);
} finally {
if (logger.isDebugEnabled()) {
logger.debug("******** Close database connection " + con);
}
con.close();
}
if (logger.isDebugEnabled()) {
logger.debug("******** Finished save information '"
+ executionInfo + "' info file '" + saveIt + "'.");
}
}
}
}
