ProarcDatabaseTest.java

/*
 * Copyright (C) 2013 Jan Pokorsky
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package cz.cas.lib.proarc.common.dao.empiredb;

import java.sql.Connection;
import org.apache.empire.db.DBCmdType;
import org.apache.empire.db.DBDatabase;
import org.apache.empire.db.DBDatabaseDriver;
import org.apache.empire.db.DBRelation;
import org.apache.empire.db.DBSQLScript;
import org.apache.empire.db.DBTable;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.IDataSet;
import org.dbunit.operation.DatabaseOperation;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.*;

/**
 *
 * @author Jan Pokorsky
 */
public class ProarcDatabaseTest {

    private DbUnitSupport support;
    private ProarcDatabase schema;
    private EmpireConfiguration emireCfg;
    private DBDatabaseDriver driver;

    public ProarcDatabaseTest() {
    }

    @BeforeClass
    public static void setUpClass() {
    }

    @AfterClass
    public static void tearDownClass() {
    }

    @Before
    public void setUp() {
        support = new DbUnitSupport();
        emireCfg = support.getEmireCfg();
        schema = emireCfg.getSchema();
        driver = emireCfg.getDriver();
    }

    @After
    public void tearDown() {
    }

    @Test
    public void testCreateSchemaScript() throws Exception {
        Connection conn = emireCfg.getConnection();
        try {
            schema.open(driver, conn);
            DBSQLScript script = new DBSQLScript();
            schema.getCreateDDLScript(schema.getDriver(), script);
            System.out.println("### create script:\n" + script);
        } finally {
            conn.close();
        }
    }

    @Test
    public void testDropSchemaScript() throws Exception {
        Connection conn = emireCfg.getConnection();
        try {
            schema.open(driver, conn);
            DBSQLScript script = new DBSQLScript();
            dropConstraints(schema, script);
            dropTables(schema, script);
            System.out.println("### drop script:\n" + script);
        } finally {
            conn.close();
        }
    }
    
    @Test
    public void testInit() throws Exception {
        // clear DB
        dropSchema(schema);
        schema.init(emireCfg);
    }

    @Test
    public void testUpgrade() throws Exception {
        ProarcDatabaseV1 v1 = new ProarcDatabaseV1();
        ProarcDatabaseV2 v2 = new ProarcDatabaseV2();
        ProarcDatabaseV3 v3 = new ProarcDatabaseV3();
        final IDatabaseConnection con = support.getConnection();
        try {
            // clear DB
            dropSchema(schema);
            dropSchema(v3);
            dropSchema(v2);
            dropSchema(v1);
            v1.init(emireCfg);
            assertEquals(1, ProarcDatabase.schemaExists(schema, con.getConnection()));

            IDataSet db = support.loadFlatXmlDataStream(getClass(), "proarc_v1.xml", true);
            try {
                DatabaseOperation.INSERT.execute(con, db);
                con.getConnection().commit();
            } finally {
                support.clearDtdSchema();
            }
            schema.init(emireCfg);
            assertEquals(ProarcDatabase.VERSION, ProarcDatabase.schemaExists(schema, con.getConnection()));
        } finally {
            con.close();
            dropSchema(schema);
            dropSchema(v1);
        }
    }

    @Test
    public void testSchemaExists() throws Exception {
        schema.init(emireCfg);
        Connection c = emireCfg.getConnection();
        try {
            boolean result = ProarcDatabase.schemaExists(schema, c) > 0;
            assertTrue(result);
        } finally {
            c.close();
        }
    }

    @Test
    public void testSchemaNotExists() throws Exception {
        dropSchema(schema);
        Connection c = emireCfg.getConnection();
        try {
            boolean result = ProarcDatabase.schemaExists(schema, c) > 0;
            assertFalse(result);
        } finally {
            c.close();
        }
    }

    private void dropSchema(DBDatabase schema) throws Exception {
        Connection conn = emireCfg.getConnection();
        try {
            schema.open(driver, conn);
            DBSQLScript script = new DBSQLScript();
            dropConstraints(schema, script);
            dropTables(schema, script);
//            System.out.println("### drop script:\n" + script);
            conn.setAutoCommit(true);
            try {
                script.run(driver, conn, true);
            } finally {
                conn.setAutoCommit(false);
            }
        } finally {
            conn.close();
        }
    }

    private void dropTables(DBDatabase schema, DBSQLScript script) {
        for (DBTable table : schema.getTables()) {
            driver.getDDLScript(DBCmdType.DROP, table, script);
        }
    }

    private void dropConstraints(DBDatabase schema, DBSQLScript script) {
        for (DBRelation relation : schema.getRelations()) {
            EmpireUtils.dropRelation(relation, driver, script);
        }
    }

}