diff --git a/modules/databend/src/main/java/org/testcontainers/databend/DatabendContainer.java b/modules/databend/src/main/java/org/testcontainers/databend/DatabendContainer.java index 389c8c44963..7422c1aa1dc 100644 --- a/modules/databend/src/main/java/org/testcontainers/databend/DatabendContainer.java +++ b/modules/databend/src/main/java/org/testcontainers/databend/DatabendContainer.java @@ -37,6 +37,8 @@ public class DatabendContainer extends JdbcDatabaseContainer private String password = "databend"; + private boolean minioEnabled = true; + public DatabendContainer(String dockerImageName) { this(DockerImageName.parse(dockerImageName)); } @@ -53,6 +55,7 @@ public DatabendContainer(final DockerImageName dockerImageName) { protected void configure() { withEnv("QUERY_DEFAULT_USER", this.username); withEnv("QUERY_DEFAULT_PASSWORD", this.password); + withEnv("MINIO_ENABLED", this.minioEnabled); } @Override @@ -93,6 +96,10 @@ public String getDatabaseName() { return this.databaseName; } + public boolean isMinioEnabled() { + return this.minioEnabled; + } + @Override public String getTestQueryString() { return TEST_QUERY; @@ -109,4 +116,9 @@ public DatabendContainer withPassword(String password) { this.password = password; return this; } + + public DatabendContainer withMinioEnabled(boolean minioEnabled) { + this.minioEnabled = minioEnabled; + return this; + } } diff --git a/modules/databend/src/test/java/org/testcontainers/databend/DatabendContainerTest.java b/modules/databend/src/test/java/org/testcontainers/databend/DatabendContainerTest.java index 92f4842d119..ffcea967ce9 100644 --- a/modules/databend/src/test/java/org/testcontainers/databend/DatabendContainerTest.java +++ b/modules/databend/src/test/java/org/testcontainers/databend/DatabendContainerTest.java @@ -38,4 +38,25 @@ public void customCredentialsWithUrlParams() throws SQLException { assertThat(resultSetInt).isEqualTo(1); } } + + @Test + public void testInsertIntoWithMinioEnabled() throws SQLException { + try (DatabendContainer databend = new DatabendContainer("datafuselabs/databend:v1.2.615") + .withMinioEnabled(true) + .withUsername("test") + .withPassword("test") + .withUrlParam("ssl", "false") + ) { + databend.start(); + + performQuery(databend, "CREATE TABLE test_table (a int, b int);"); + performQuery(databend, "INSERT INTO test_table VALUES (1, 2);"); + + ResultSet resultSet = performQuery(databend, "SELECT * FROM test_table;"); + + resultSet.next(); + assertThat(resultSet.getInt(1)).isEqualTo(1); + assertThat(resultSet.getInt(2)).isEqualTo(2); + } + } }