querydsl/querydsl-sql/src/test/java/com/mysema/query/sql/DerbyTest.java

/*
 * Copyright (c) 2009 Mysema Ltd.
 * All rights reserved.
 * 
 */
package com.mysema.query.sql;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.sql.Statement;

import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.runner.RunWith;

import com.mysema.query.FilteringTestRunner;
import com.mysema.query.Label;

@RunWith(FilteringTestRunner.class)
@Label("derby")
public class DerbyTest extends SqlQueryTest {

    @BeforeClass
    public static void setUp() throws Exception {
        String sql;
        Connection c = getDerbyConnection();
        Statement stmt = c.createStatement();

        connHolder.set(c);
        stmtHolder.set(stmt);

        // survey
        // stmt.execute("drop table survey if exists");
        safeExecute(stmt, "drop table survey");
        stmt.execute("create table survey (id int,name varchar(30))");
        stmt.execute("insert into survey values (1, 'Hello World')");

        // test
        // stmt.execute("drop table test if exists");
        safeExecute(stmt, "drop table test");
        stmt.execute("create table test(name varchar(255))");
        sql = "insert into test values(?)";
        PreparedStatement pstmt = c.prepareStatement(sql);
        try{
            for (int i = 0; i < 10000; i++) {
                pstmt.setString(1, "name" + i);
                pstmt.addBatch();
            }
            pstmt.executeBatch();    
        }finally{
            pstmt.close();
        }        

        // employee
        // stmt.execute("drop table employee if exists");
        safeExecute(stmt, "drop table employee2");
        stmt.execute("create table employee2(id int, "
                + "firstname VARCHAR(50), " + "lastname VARCHAR(50), "
                + "salary decimal(10, 2), " + "superior_id int, "
                + "CONSTRAINT PK_employee PRIMARY KEY (id), "
                + "CONSTRAINT FK_superior FOREIGN KEY (superior_id) "
                + "REFERENCES employee2(ID))");
        addEmployee(1, "Mike", "Smith", 160000, -1);
        addEmployee(2, "Mary", "Smith", 140000, -1);

        // Employee under Mike
        addEmployee(10, "Joe", "Divis", 50000, 1);
        addEmployee(11, "Peter", "Mason", 45000, 1);
        addEmployee(12, "Steve", "Johnson", 40000, 1);
        addEmployee(13, "Jim", "Hood", 35000, 1);

        // Employee under Mike
        addEmployee(20, "Jennifer", "Divis", 60000, 2);
        addEmployee(21, "Helen", "Mason", 50000, 2);
        addEmployee(22, "Daisy", "Johnson", 40000, 2);
        addEmployee(23, "Barbara", "Hood", 30000, 2);

        // date_test and time_test
        // stmt.execute("drop table time_test if exists");
        // stmt.execute("drop table date_test if exists");
        safeExecute(stmt, "drop table time_test");
        safeExecute(stmt, "drop table date_test");
        stmt.execute("create table time_test(time_test time)");
        stmt.execute("create table date_test(date_test date)");
    }

    private static void safeExecute(Statement stmt, String sql) {
        try {
            stmt.execute(sql);
        } catch (SQLException e) {
            // do nothing
        }
    }

    @Before
    public void setUpForTest() {
        dialect = SQLTemplatesFactory.forDerby().newLineToSingleSpace();
    }

    private static void addEmployee(int id, String firstName, String lastName,
            double salary, int superiorId) throws Exception {
        stmtHolder.get().execute(
                "insert into employee2 values(" + id + ", '" + firstName
                        + "', '" + lastName + "', " + salary + ", "
                        + (superiorId <= 0 ? "null" : ("" + superiorId)) + ")");
    }

    private static Connection getDerbyConnection() throws Exception {
        Class.forName("org.apache.derby.jdbc.EmbeddedDriver");
        String url = "jdbc:derby:demoDB;create=true";
        return DriverManager.getConnection(url, "", "");
    }

}