Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions modules/clickhouse/build.gradle
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
description = "Testcontainers :: JDBC :: ClickHouse"

dependencies {
compile project(':testcontainers')
compile project(':jdbc')
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,72 @@
package org.testcontainers.containers;

import org.testcontainers.containers.wait.HttpWaitStrategy;

import java.time.Duration;

public class ClickHouseContainer extends JdbcDatabaseContainer {
public static final String NAME = "clickhouse";
public static final String IMAGE = "yandex/clickhouse-server";
public static final String DEFAULT_TAG = "18.10.3";

public static final Integer HTTP_PORT = 8123;
public static final Integer NATIVE_PORT = 9000;

private static final String DRIVER_CLASS_NAME = "ru.yandex.clickhouse.ClickHouseDriver";
private static final String JDBC_URL_PREFIX = "jdbc:" + NAME + "://";
private static final String TEST_QUERY = "SELECT 1";

private String databaseName = "default";
private String username = "default";
private String password = "";

public ClickHouseContainer() {
super(IMAGE + ":" + DEFAULT_TAG);
}

public ClickHouseContainer(String dockerImageName) {
super(dockerImageName);
}

@Override
protected void configure() {
withExposedPorts(HTTP_PORT, NATIVE_PORT);
waitingFor(
new HttpWaitStrategy()
.forStatusCode(200)
.forResponsePredicate(responseBody -> "Ok.".equals(responseBody))
.withStartupTimeout(Duration.ofMinutes(1))
);
}

@Override
protected Integer getLivenessCheckPort() {
return getMappedPort(HTTP_PORT);
}

@Override
public String getDriverClassName() {
return DRIVER_CLASS_NAME;
}

@Override
public String getJdbcUrl() {
return JDBC_URL_PREFIX + getContainerIpAddress() + ":" + getMappedPort(HTTP_PORT) + "/" + databaseName;
}

@Override
public String getUsername() {
return username;
}

@Override
public String getPassword() {
return password;
}

@Override
public String getTestQueryString() {
return TEST_QUERY;
}

}
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
package org.testcontainers.containers;

public class ClickHouseProvider extends JdbcDatabaseContainerProvider {
@Override
public boolean supports(String databaseType) {
return databaseType.equals(ClickHouseContainer.NAME);
}

@Override
public JdbcDatabaseContainer newInstance(String tag) {
return new ClickHouseContainer(ClickHouseContainer.IMAGE + ":" + tag);
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
org.testcontainers.containers.ClickHouseProvider
2 changes: 2 additions & 0 deletions modules/jdbc-test/build.gradle
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,15 @@ dependencies {
compile project(':mariadb')
compile project(':oracle-xe')
compile project(':mssqlserver')
compile project(':clickhouse')

testCompile 'com.google.guava:guava:18.0'
testCompile 'org.postgresql:postgresql:42.0.0'
testCompile 'mysql:mysql-connector-java:5.1.45'
testCompile 'org.mariadb.jdbc:mariadb-java-client:1.4.6'
testCompile 'com.oracle:ojdbc6:12.1.0.1-atlassian-hosted'
testCompile 'com.microsoft.sqlserver:mssql-jdbc:6.1.0.jre8'
testCompile 'ru.yandex.clickhouse:clickhouse-jdbc:0.1.41'

testCompile 'com.zaxxer:HikariCP-java6:2.3.8'
testCompile 'org.apache.tomcat:tomcat-jdbc:8.5.4'
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ private enum Options {
CharacterSet,
CustomIniFile,
JDBCParams,
PmdKnownBroken
}

@Parameter
Expand All @@ -56,7 +57,9 @@ public static Iterable<Object[]> data() {
{"jdbc:tc:mariadb:10.2.14://hostname/databasename?TC_INITSCRIPT=somepath/init_unicode_mysql.sql&useUnicode=yes&characterEncoding=utf8", EnumSet.of(Options.CharacterSet)},
{"jdbc:tc:mariadb:10.2.14://hostname/databasename?TC_INITSCRIPT=somepath/init_mariadb.sql", EnumSet.of(Options.ScriptedSchema)},
{"jdbc:tc:mariadb:10.2.14://hostname/databasename?TC_INITFUNCTION=org.testcontainers.jdbc.JDBCDriverTest::sampleInitFunction", EnumSet.of(Options.ScriptedSchema)},
{"jdbc:tc:mariadb:10.2.14://hostname/databasename?TC_MY_CNF=somepath/mariadb_conf_override", EnumSet.of(Options.CustomIniFile)}});
{"jdbc:tc:mariadb:10.2.14://hostname/databasename?TC_MY_CNF=somepath/mariadb_conf_override", EnumSet.of(Options.CustomIniFile)},
{"jdbc:tc:clickhouse://hostname/databasename", EnumSet.of(Options.PmdKnownBroken)},
});
}

public static void sampleInitFunction(Connection connection) throws SQLException {
Expand Down Expand Up @@ -102,7 +105,7 @@ public void test() throws SQLException {

private void performSimpleTest(String jdbcUrl) throws SQLException {
try (HikariDataSource dataSource = getDataSource(jdbcUrl, 1)) {
boolean result = new QueryRunner(dataSource).query("SELECT 1", rs -> {
boolean result = new QueryRunner(dataSource, options.contains(Options.PmdKnownBroken)).query("SELECT 1", rs -> {
rs.next();
int resultSetInt = rs.getInt(1);
assertEquals("A basic SELECT query succeeds", 1, resultSetInt);
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
package org.testcontainers.junit;

import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.junit.Rule;
import org.junit.Test;
import org.testcontainers.containers.ClickHouseContainer;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import static org.rnorth.visibleassertions.VisibleAssertions.assertEquals;

public class SimpleClickhouseTest {

@Rule
public ClickHouseContainer clickhouse = new ClickHouseContainer();

@Test
public void testSimple() throws SQLException {
HikariConfig hikariConfig = new HikariConfig();
hikariConfig.setJdbcUrl(clickhouse.getJdbcUrl());
hikariConfig.setUsername(clickhouse.getUsername());
hikariConfig.setPassword(clickhouse.getPassword());

HikariDataSource ds = new HikariDataSource(hikariConfig);
Statement statement = ds.getConnection().createStatement();
statement.execute("SELECT 1");
ResultSet resultSet = statement.getResultSet();

resultSet.next();
int resultSetInt = resultSet.getInt(1);
assertEquals("A basic SELECT query succeeds", 1, resultSetInt);
}
}