Package org.dbunit.database

Examples of org.dbunit.database.QueryDataSet

122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
  {
    logger.debug("createQueryDataSet(tables={}, connection={})", tables, connection);
   
    List queryDataSets = new ArrayList();
   
        QueryDataSet queryDataSet = new QueryDataSet(connection);
       
        for (Iterator it = tables.iterator(); it.hasNext();)
        {
            Object item = it.next();
           
            if(item instanceof QuerySet) {
        if(queryDataSet.getTableNames().length > 0)
                queryDataSets.add(queryDataSet);
       
        QueryDataSet newQueryDataSet = (((QuerySet)item).getQueryDataSet(connection));
        queryDataSets.add(newQueryDataSet);
        queryDataSet = new QueryDataSet(connection);
            }
            else if (item instanceof Query)
            {
                Query queryItem = (Query)item;
                queryDataSet.addTable(queryItem.getName(), queryItem.getSql());
View Full Code Here
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
* @return
*/
public static IDataSet getDataSetFromDataBase(IDatabaseConnection databaseConnection,String querySqlsSplitBySemicolon){
  //分割多个sql语句
  String[] querySqls = querySqlsSplitBySemicolon.split(";");
  QueryDataSet queryDataSet=null;
  try {
    queryDataSet = new QueryDataSet(databaseConnection);
    //处理多个语句情况
    for (int i = 0; i < querySqls.length; i++) {
      queryDataSet.addTable(getTableNameFromQuerySql(querySqls[i]), querySqls[i]);
    }
  } catch (Exception e) {
    log.error("查询数据失败!");
    log.error(e.getMessage());
  }
View Full Code Here
758
759
760
761
762
763
764
765
766
767
768
769
    public void assertEqualsTable(final String tableName, final String orderColumnName) {
      this.dbDataUtil.execute(new DbUnitWork() {
        @Override
        public void execute(IDatabaseConnection conn) {
          try {
            QueryDataSet tableDataSet = new QueryDataSet(conn);
            tableDataSet.addTable(tableName, String.format("select * from %s order by %s", tableName, orderColumnName));
           
            Assertion.assertEquals(flatXmlDataSet, tableDataSet);
          } catch (Exception ex) {
            throw new JuDbException("Couldn't assert DB data", ex);
          }
View Full Code Here
50
51
52
53
54
55
56
57
58
59
60
   * @param conn IDatabaseConnection instance
   * @return IDataSet instance
   */
  public IDataSet createDataSet(IDatabaseConnection conn) {
    if (this.items.size() > 0) {
      QueryDataSet dataSet = new QueryDataSet(conn, false);
      for (ExportItem item : this.items) {
        item.addToQueryDataSet(dataSet, casedTableNames);
      }
      return dataSet;
    } else {
View Full Code Here
38
39
40
41
42
43
44
45
46
47
48
49
50
51
        Connection jdbcConnection =
                DriverManager.getConnection(
                        "jdbc:mysql://localhost:3306/zanata", "root", "");
        IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);

        QueryDataSet dataSet = new QueryDataSet(connection);

        for (Map.Entry<String, String> entry : tableNameToQueryMap) {
            dataSet.addTable(entry.getKey(), entry.getValue());
        }

        FlatDtdDataSet.write(dataSet, new FileOutputStream(new File(TEMP_DIR,
                dataSetName + ".dbunit.dtd")));
        File dbunitFile = new File(TEMP_DIR, dataSetName + ".dbunit.xml");
View Full Code Here
171
172
173
174
175
176
177
178
179
   * @throws FileNotFoundException
   * @throws SQLException
   * @throws IOException
   */
  public void exportFromDBToFile(String tableName,String sqlQuery,File flatXMLFile) throws DataSetException, FileNotFoundException, SQLException, IOException {
    QueryDataSet partialDataSet = new QueryDataSet(getDatabaseConnection());
    partialDataSet.addTable(tableName,sqlQuery);
    FlatXmlDataSet.write(partialDataSet, new FileOutputStream(flatXMLFile));
  }
View Full Code Here
203
204
205
206
207
208
209
210
211
212
213
   * @throws DataSetException
   * @throws FileNotFoundException
   * @throws IOException
   */ 
  public void exportFromDBToFile(String[] tableNames,File flatXMLFile) throws SQLException, DataSetException, FileNotFoundException, IOException {
    QueryDataSet partialDataSet = new QueryDataSet(getDatabaseConnection());
    for(int i = 0; i < tableNames.length; i++ ) {
      partialDataSet.addTable(tableNames[i]);
    }
    FlatXmlDataSet.write(partialDataSet, new FileOutputStream(flatXMLFile));
  }
View Full Code Here
TOP

Related Classes of org.dbunit.database.QueryDataSet

  • ch.inftec.ju.testing.db.DbDataUtil$AssertBuilder
  • ch.inftec.ju.testing.db.ExportItems
  • cn.baiweigang.qtaf.toolkit.dbunit.DbUnitUtil
  • cn.org.rapid_framework.test.dbunit.DBUnitFlatXmlHelper
  • com.common.test.DBUnitFlatXmlHelper
  • mydatagenerator.core.database.operations.DatabaseExporter
  • org.dbunit.ant.AbstractStep
  • org.dbunit.ant.QuerySet
  • org.exolab.jms.tools.migration.ExportImportTest
  • org.zanata.DBUnitDataExtractor
    • 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.