TestEmbeddedMySQL.java

package com.splout.db.engine;

/*
 * #%L
 * Splout SQL Server
 * %%
 * Copyright (C) 2012 - 2013 Datasalt Systems S.L.
 * %%
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */

import com.google.common.io.Files;
import com.mysql.management.util.QueryUtil;
import com.splout.db.common.PortUtils;
import com.splout.db.common.PortUtils.PortLock;
import com.splout.db.engine.EmbeddedMySQL.EmbeddedMySQLConfig;
import org.junit.Test;

import java.io.File;
import java.io.IOException;
import java.nio.charset.Charset;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import static org.junit.Assert.assertEquals;

public class TestEmbeddedMySQL {

  public static void insertData(Connection conn) throws IOException {
    String query = "CREATE TABLE `test_table` (`idpublisher` int(11) DEFAULT NULL, `idsite` int(11) DEFAULT NULL, `country_iso` varchar(4) DEFAULT NULL, `idzone` int(11) DEFAULT NULL, `hits` int(11) DEFAULT NULL, `cpm_value` double DEFAULT NULL, `ddate` varchar(16) DEFAULT NULL) ENGINE=InnoDB DEFAULT CHARSET=latin1";
    QueryUtil util = new QueryUtil(conn);
    util.execute(query);

    util.execute("BEGIN");

    for (String insert : Files.readLines(new File("src/test/resources/test.mysql"),
        Charset.defaultCharset())) {
      String[] fields = insert.split(",");
      String q = "INSERT INTO test_table VALUES (";
      for (int i = 0; i < fields.length; i++) {
        String val = fields[i];
        if (val.length() < 1) {
          val = "NULL";
        } else {
          if (i == 2 || i == 6) {
            val = "'" + val + "'";
          }
        }
        q += val;
        if (i != fields.length - 1) {
          q += ",";
        }
      }
      q += ");";
      util.execute(q);
    }

    util.execute("COMMIT");
  }

  @SuppressWarnings("rawtypes")
  @Test
  public void testPortLocking() throws InterruptedException {
    final int N_THREADS = 30;
    Thread[] pool = new Thread[N_THREADS];
    final ConcurrentHashMap map = new ConcurrentHashMap();
    for (int i = 0; i < N_THREADS; i++) {
      final int threadId = i;
      pool[i] = new Thread() {
        @SuppressWarnings("unchecked")
        public void run() {
          setName("thread_" + threadId);
          PortLock portLock = PortUtils.getNextAvailablePort(EmbeddedMySQLConfig.DEFAULT_PORT);
          map.put(getName(), portLock);
        }
      };
      pool[i].start();
    }

    long sleptSoFar = 0;
    do {
      Thread.sleep(500);
      sleptSoFar += 500;
      if (sleptSoFar > 5000) {
        throw new RuntimeException("Waited too much");
      }
    } while (map.keySet().size() < N_THREADS);

    Set<Integer> distinctPorts = new HashSet<Integer>();
    for (Object entry : map.entrySet()) {
      PortLock pLock = (PortLock) ((Map.Entry) entry).getValue();
      distinctPorts.add(pLock.getPort());
      pLock.release();
    }

    // Assert every thread locked on a different port
    assertEquals(distinctPorts.size(), map.keySet().size());
  }

  @Test
  public void test() throws ClassNotFoundException, SQLException, IOException, InterruptedException {
    EmbeddedMySQL mysql = new EmbeddedMySQL();
    mysql.start(true);
    Connection conn = null;
    try {
      MySQLManager manager = new MySQLManager(mysql);
      conn = manager.getJdbcManager().getConnectionFromPool();

      insertData(conn);

      List<?> l = new QueryUtil(conn).executeQuery("SELECT * FROM test_table LIMIT 10;");
      assertEquals(10, l.size());

    } finally {
      if (conn != null) {
        conn.close();
      }
      mysql.stop();
    }
  }
}