Skip to content

Expose MySQL and PostgreSQL protocol ports in ClickHouseContainer #12001

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
junho0831 wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Expose MySQL and PostgreSQL protocol ports in ClickHouseContainer #12001

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
2 changes: 2 additions & 0 deletions modules/clickhouse/build.gradle
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,8 @@ dependencies {
testRuntimeOnly(group: 'com.clickhouse', name: 'clickhouse-jdbc', version: '0.9.8', classifier: 'all')

testImplementation 'com.clickhouse:client-v2:0.9.8'
testImplementation 'org.mariadb.jdbc:mariadb-java-client:3.5.9'
testImplementation 'org.postgresql:postgresql:42.7.12'
testImplementation testFixtures(project(':testcontainers-r2dbc'))
testRuntimeOnly(group: 'com.clickhouse', name: 'clickhouse-r2dbc', version: '0.9.8', classifier: 'http')
}
48 changes: 47 additions & 1 deletion modules/clickhouse/src/main/java/org/testcontainers/clickhouse/ClickHouseContainer.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,8 @@
* <ul>
* <li>Database: 8123</li>
* <li>Console: 9000</li>
* <li>MySQL: 9004</li>
* <li>PostgreSQL: 9005</li>
* </ul>
*/
public class ClickHouseContainer extends JdbcDatabaseContainer<ClickHouseContainer> {
Expand All @@ -29,6 +31,10 @@ public class ClickHouseContainer extends JdbcDatabaseContainer<ClickHouseContain

static final Integer NATIVE_PORT = 9000;

static final Integer MYSQL_PORT = 9004;

static final Integer POSTGRESQL_PORT = 9005;

private static final String LEGACY_V1_DRIVER_CLASS_NAME = "com.clickhouse.jdbc.ClickHouseDriver";

private static final String DRIVER_CLASS_NAME = "com.clickhouse.jdbc.Driver";
Expand All @@ -55,7 +61,7 @@ public ClickHouseContainer(final DockerImageName dockerImageName) {
super(dockerImageName);
dockerImageName.assertCompatibleWith(CLICKHOUSE_IMAGE_NAME);

addExposedPorts(HTTP_PORT, NATIVE_PORT);
addExposedPorts(HTTP_PORT, NATIVE_PORT, MYSQL_PORT, POSTGRESQL_PORT);
waitingFor(
Wait
.forHttp("/")
Expand Down Expand Up @@ -105,6 +111,46 @@ public String getHttpUrl() {
return "http://" + getHost() + ":" + getMappedPort(HTTP_PORT);
}

public String getMysqlJdbcUrl() {
return (
"jdbc:mysql://" +
getHost() +
":" +
getMappedPort(MYSQL_PORT) +
"/" +
this.databaseName +
constructUrlParameters("?", "&")
);
}

public Integer getHttpPort() {
return getMappedPort(HTTP_PORT);
}

public Integer getNativePort() {
return getMappedPort(NATIVE_PORT);
}

public Integer getMysqlPort() {
return getMappedPort(MYSQL_PORT);
}

public Integer getPostgresqlPort() {
return getMappedPort(POSTGRESQL_PORT);
}

public String getPostgresqlJdbcUrl() {
return (
"jdbc:postgresql://" +
getHost() +
":" +
getMappedPort(POSTGRESQL_PORT) +
"/" +
this.databaseName +
constructUrlParameters("?", "&")
);
}

@Override
public String getUsername() {
return username;
Expand Down
6 changes: 5 additions & 1 deletion modules/clickhouse/src/main/java/org/testcontainers/containers/ClickHouseContainer.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,10 @@ public class ClickHouseContainer extends JdbcDatabaseContainer<ClickHouseContain

public static final Integer NATIVE_PORT = 9000;

public static final Integer MYSQL_PORT = 9004;

public static final Integer POSTGRESQL_PORT = 9005;

private static final String LEGACY_DRIVER_CLASS_NAME = "ru.yandex.clickhouse.ClickHouseDriver";

private static final String DRIVER_CLASS_NAME = "com.clickhouse.jdbc.ClickHouseDriver";
Expand Down Expand Up @@ -64,7 +68,7 @@ public ClickHouseContainer(final DockerImageName dockerImageName) {
dockerImageName.assertCompatibleWith(DEFAULT_IMAGE_NAME, CLICKHOUSE_IMAGE_NAME);
supportsNewDriver = isNewDriverSupported(dockerImageName);

addExposedPorts(HTTP_PORT, NATIVE_PORT);
addExposedPorts(HTTP_PORT, NATIVE_PORT, MYSQL_PORT, POSTGRESQL_PORT);
this.waitStrategy =
new HttpWaitStrategy()
.forStatusCode(200)
Expand Down
65 changes: 65 additions & 0 deletions modules/clickhouse/src/test/java/org/testcontainers/clickhouse/ClickHouseContainerTest.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,11 @@
import org.testcontainers.ClickhouseTestImages;
import org.testcontainers.db.AbstractContainerDatabaseTest;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;
Expand Down Expand Up @@ -89,4 +92,66 @@ void testGetHttpMethodWithHttpClient() {
client.close();
}
}

@Test
void testMysqlProtocol() throws SQLException {
try (ClickHouseContainer clickhouse = new ClickHouseContainer(ClickhouseTestImages.CLICKHOUSE_24_12_IMAGE)) {
clickhouse.start();

assertThat(clickhouse.getMysqlPort()).isNotNull();
assertThat(clickhouse.getHttpPort()).isNotNull();
assertThat(clickhouse.getNativePort()).isNotNull();

String mysqlUrl = clickhouse.getMysqlJdbcUrl();
assertThat(mysqlUrl).startsWith("jdbc:mysql://");

String mariadbUrl =
"jdbc:mariadb://" +
clickhouse.getHost() +
":" +
clickhouse.getMysqlPort() +
"/" +
clickhouse.getDatabaseName();

try (
Connection connection = DriverManager.getConnection(
mariadbUrl,
clickhouse.getUsername(),
clickhouse.getPassword()
);
Statement statement = connection.createStatement();
ResultSet resultSet = statement.executeQuery("SELECT 1")
) {
resultSet.next();
int resultSetInt = resultSet.getInt(1);
assertThat(resultSetInt).isEqualTo(1);
}
}
}

@Test
void testPostgresqlProtocol() throws SQLException {
try (ClickHouseContainer clickhouse = new ClickHouseContainer(ClickhouseTestImages.CLICKHOUSE_24_12_IMAGE)) {
clickhouse.start();

assertThat(clickhouse.getPostgresqlPort()).isNotNull();

String pgUrl = clickhouse.getPostgresqlJdbcUrl() + "?sslmode=disable&preferQueryMode=simple";
assertThat(pgUrl).startsWith("jdbc:postgresql://");

try (
Connection connection = DriverManager.getConnection(
pgUrl,
clickhouse.getUsername(),
clickhouse.getPassword()
);
Statement statement = connection.createStatement();
ResultSet resultSet = statement.executeQuery("SELECT 1")
) {
resultSet.next();
int resultSetInt = resultSet.getInt(1);
assertThat(resultSetInt).isEqualTo(1);
}
}
}
}