package in.gov.kerala.kwa.onlineservices.test;

import org.dbunit.database.DatabaseConfig;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.xml.FlatDtdDataSet;
import org.dbunit.ext.oracle.Oracle10DataTypeFactory;

import java.io.FileOutputStream;
import java.sql.Connection;
import java.sql.DriverManager;

public class CreateDTDDataSet {
    public static void main(String[] args) throws Exception {
        Class driverClass = Class.forName("oracle.jdbc.OracleDriver");
        Connection connection = DriverManager.getConnection("jdbc:oracle:thin:@//www.kaushikam.com:49161/XE",
                "testing", "redhat");
        IDatabaseConnection dbConnection = new DatabaseConnection(connection, "TESTING");
        DatabaseConfig config = dbConnection.getConfig();
        config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY,
                new Oracle10DataTypeFactory());
        config.setProperty(DatabaseConfig.FEATURE_QUALIFIED_TABLE_NAMES,
                Boolean.TRUE);
        FlatDtdDataSet.write(dbConnection.createDataSet(),
                new FileOutputStream("foreign_key.dtd"));
    }
}