import org.junit.Before;

import org.junit.Test;

import java.sql.Connection;

import java.sql.Driver;

import java.sql.PreparedStatement;

import java.sql.SQLException;

import java.sql.Statement;

import java.util.Properties;

public class MaxPacketAllowedTest {

    private Properties properties;

    @Before

    public void setUp() throws Exception {

        properties = new Properties();

        properties.put("user", "root");

        properties.put("password", "");

    }

    @Test

    public void testMariaDbConnector() throws Exception {

        executeBatch(new org.mariadb.jdbc.Driver());

    }

    @Test

    public void testMySqlDbConnector() throws Exception {

        executeBatch(new com.mysql.jdbc.Driver());

    }

    private void executeBatch(Driver driver) throws SQLException {

        Connection connection = driver.connect(

                "jdbc:mysql://localhost:3306/max?rewriteBatchedStatements=true",

                properties);

        Statement statement = connection.createStatement();

        statement.execute("DROP TABLE IF EXISTS test");

        statement.execute("CREATE TABLE test (id INT, text VARCHAR(50))");

        PreparedStatement preparedStatement = connection.prepareStatement("INSERT INTO test VALUES(?, ?)");

        for (int i = 0; i < 100000; i++) {

            preparedStatement.setInt(1, i);

            preparedStatement.setString(2, "test text");

            preparedStatement.addBatch();

        }

        preparedStatement.executeBatch();

    }

}