Package org.dbunit.dataset

Examples of org.dbunit.dataset.IDataSet


                Set<ColumnValues> pks = getPksFromQuery(connection, tableName, query);
                pksToLoad.put(config.getClassForEntity(entity), pks);
            }

            IDataSet data = null;

            if (pksToLoad.isEmpty()) {
                data = connection.createDataSet();
            } else {
                EntityRelationshipFilter filter = new EntityRelationshipFilter(connection, pksToLoad, new ConfigurableDependencyInclusionResolver(config));
View Full Code Here


    private static Set<ColumnValues> getPksFromQuery(IDatabaseConnection connection, String table, String query)
        throws DataSetException, SQLException {

        Set<ColumnValues> ret = new HashSet<ColumnValues>();

        IDataSet data = connection.createDataSet(new String[] { table });
        Column[] tablePks = data.getTableMetaData(table).getPrimaryKeys();
        String pkName = tablePks[0].getColumnName();

        if (query == null) {
            query = "SELECT " + pkName + " FROM " + table;
        }
View Full Code Here

            throws Exception {
        // load the data set
        ClassLoader cl = Thread.currentThread().getContextClassLoader();
        InputStream stream = cl.getResource(getDataSetXml()).openStream();

        IDataSet data;

        if (getUseFlatXmlDataSet()) {
            data = new FlatXmlDataSet(stream);
        } else {
            data = new XmlDataSet(stream);
        }

        // order the data set
        DatabaseSequenceFilter filter = new DatabaseSequenceFilter(connection,
                data.getTableNames());
        return new FilteredDataSet(filter, data);
    }
View Full Code Here

            if (dbProdName.toLowerCase().contains("hsql")) {
                DatabaseConfig config = dbunitConn.getConfig();
                config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY,
                        new HsqlDataTypeFactory());
            }
            IDataSet dataSet = loadDataSet(dbunitConn);
            databaseTester = new DefaultDatabaseTester(dbunitConn);
            databaseTester.setDataSet(dataSet);
            databaseTester.setSetUpOperation(DatabaseOperation.CLEAN_INSERT);
            databaseTester.onSetup();
        } catch (Exception ex) {
View Full Code Here

    private IDataSet loadDataSet( IDatabaseConnection connection ) throws Exception {
        // load the data set
        ClassLoader cl = Thread.currentThread().getContextClassLoader();
        InputStream stream = cl.getResource( getDataSetXml() ).openStream();

        IDataSet data;

        if (getUseFlatXmlDataSet()) {
            data = new FlatXmlDataSet( stream );
        } else {
            data = new XmlDataSet( stream );
        }

        // order the data set
        DatabaseSequenceFilter filter = new DatabaseSequenceFilter( connection, data.getTableNames() );
        return new FilteredDataSet( filter, data );
    }
View Full Code Here

    EntityManager em = null;
    try {
      em = entityManagerFactory.createEntityManager();
      Connection conn = getConnection(em);
      IDatabaseConnection dbunitConn = new DatabaseConnection(conn);
      IDataSet dataSet = loadDataSet(dbunitConn);
      databaseTester = new DefaultDatabaseTester(dbunitConn);
      databaseTester.setDataSet(dataSet);
      databaseTester.setSetUpOperation(DatabaseOperation.CLEAN_INSERT);
      databaseTester.onSetup();
    } catch (Exception ex) {
View Full Code Here

                    PersistenceAdapter adapter = dbService.getAdapter();
                    connection = adapter.getConnection();
                }
                IDatabaseConnection dbConnection
                        = new DatabaseConnection(connection);
                IDataSet fullDataSet = dbConnection.createDataSet();
                FlatXmlDataSet.write(fullDataSet,
                                     new FileOutputStream(outputPath));
            } catch (Exception exception) {
                exception.printStackTrace();
                System.exit(1);
View Full Code Here

        importer.apply();

        // verify data between the two master databases
        IDatabaseConnection masterConn = getConnection(master);
        IDatabaseConnection migratedConn = getConnection(migrated);
        IDataSet expectedDataSet = masterConn.createDataSet();
        IDataSet actualDataSet = migratedConn.createDataSet();

        // NOTE: don't care about the contents of the 'seeds' and 'system_data'
        // tables as neither are explicitly migrated, and the contents of
        // each may differ
        checkQuery("destinations", DESTINATIONS_QUERY, masterConn,
View Full Code Here

    // assertDataset(DATA_SET_FILE_INSERTED_KEYGEN);
    // }

    private void assertDataset(String datasetlocation) throws Exception {
        String tablename1 = "book";
        IDataSet actualDataSet = new DefaultDatabaseTester(
                new DatabaseConnection(_jdo.getConnectionFactory()
                        .createConnection())).getConnection().createDataSet(
                new String[] { tablename1 });
        IDataSet expectedDataSet = new XmlDataSet(new FileInputStream(
                getClass().getResource(datasetlocation).getFile()));
        Assertion.assertEquals(expectedDataSet, actualDataSet);

        ITable actualTable = actualDataSet.getTable(tablename1);
        ITable expectedTable = expectedDataSet.getTable(tablename1);
        Assertion.assertEquals(expectedTable, actualTable);

    }
View Full Code Here

     */
    private void assertDataset(String datasetlocation) throws Exception{
        String tablename1 = "employee";
        String tablename2 = "address";
       
        IDataSet actualDataSet = new DefaultDatabaseTester(
                new DatabaseConnection(_jdo.getConnectionFactory()
                        .createConnection())).getConnection().createDataSet(
                new String[] { tablename1, tablename2 });
        IDataSet expectedDataSet = new XmlDataSet(new FileInputStream(
                getClass().getResource(datasetlocation).getFile()));
       
        Assertion.assertEquals(expectedDataSet, actualDataSet);
   
        ITable actualTable = actualDataSet.getTable(tablename1);
        ITable expectedTable = expectedDataSet.getTable(tablename1);
        Assertion.assertEquals(expectedTable, actualTable);
   
        actualTable = actualDataSet.getTable(tablename2);
        expectedTable = expectedDataSet.getTable(tablename2);
        Assertion.assertEquals(expectedTable, actualTable);
    }
View Full Code Here

TOP

Related Classes of org.dbunit.dataset.IDataSet

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.