BatchDeleteQueryAllPrimitivesTest.java

/*
   Copyright (c) 2011, Oracle and/or its affiliates. All rights reserved.

   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; version 2 of the License.

   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, write to the Free Software
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301  USA
*/

package jdbctest;

import java.sql.PreparedStatement;
import java.sql.SQLException;

public class BatchDeleteQueryAllPrimitivesTest extends JDBCQueryTest {

    @Override
    public String tableName() {
        return "allprimitives";
    }

    @Override
    public void createInstances(int numberOfInstances) {
        for (int i = 0; i < numberOfInstances; ++i) {
            createAllPrimitiveInstance(i);
        }
    }

    public void testDeleteEqualByPrimaryKey() {
        deleteEqualQuery("id", "PRIMARY", 5, 1);
        deleteEqualQuery("id", "PRIMARY", 5, 0);
        try {
            connection.setAutoCommit(false);
            PreparedStatement preparedStatement = connection.prepareStatement("DELETE FROM allprimitives where id = ?");
            // delete 4 through 9 (excluding 5 which is already gone)
            for (int i = 4; i < 9; ++i) {
                preparedStatement.setInt(1, i);
                preparedStatement.addBatch();
            }
            int[] results = preparedStatement.executeBatch();
            connection.commit();
            for (int i = 0; i < 5; ++i) {
                int expected = (i == 1) ? 0:1;
                errorIfNotEqual("testDeleteEqualByPrimaryKey result " + i, expected, results[i]);
            }
        } catch (SQLException e) {
            error(e.getMessage());
        }
        equalQuery("id", "PRIMARY", 0, 0);
        equalQuery("id", "PRIMARY", 1, 1);
        equalQuery("id", "PRIMARY", 2, 2);
        equalQuery("id", "PRIMARY", 3, 3);
        equalQuery("id", "PRIMARY", 4);
        equalQuery("id", "PRIMARY", 5);
        equalQuery("id", "PRIMARY", 6);
        equalQuery("id", "PRIMARY", 7);
        equalQuery("id", "PRIMARY", 8);
        equalQuery("id", "PRIMARY", 9, 9);
        failOnError();
    }

}