diff --git a/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseBase.java b/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseBase.java index 88f0a32d..2931fccd 100644 --- a/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseBase.java +++ b/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseBase.java @@ -34,7 +34,7 @@ public static void setup() throws IOException { initialPing(); return; } - db = new ClickHouseContainer(ClickHouseTestHelpers.CLICKHOUSE_DOCKER_IMAGE); + db = new ClickHouseContainer(ClickHouseTestHelpers.CLICKHOUSE_DOCKER_IMAGE).withPassword("test_password").withEnv("CLICKHOUSE_DEFAULT_ACCESS_MANAGEMENT", "1"); db.start(); } diff --git a/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskSchemalessProxyTest.java b/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskSchemalessProxyTest.java index 07bb0433..24a46350 100644 --- a/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskSchemalessProxyTest.java +++ b/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskSchemalessProxyTest.java @@ -29,7 +29,11 @@ public class ClickHouseSinkTaskSchemalessProxyTest extends ClickHouseBase { public static void setup() throws IOException { Network network = Network.newNetwork(); // Note: we are using a different version of ClickHouse for the proxy - https://github.com/ClickHouse/ClickHouse/issues/58828 - db = new ClickHouseContainer(ClickHouseTestHelpers.CLICKHOUSE_FOR_PROXY_DOCKER_IMAGE).withNetwork(network).withNetworkAliases("clickhouse"); + db = new ClickHouseContainer(ClickHouseTestHelpers.CLICKHOUSE_FOR_PROXY_DOCKER_IMAGE) + .withNetwork(network) + .withNetworkAliases("clickhouse") + .withPassword("test_password") + .withEnv("CLICKHOUSE_DEFAULT_ACCESS_MANAGEMENT", "1"); db.start(); toxiproxy = new ToxiproxyContainer("ghcr.io/shopify/toxiproxy:2.7.0").withNetwork(network).withNetworkAliases("toxiproxy"); diff --git a/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskStringTest.java b/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskStringTest.java index 057e56d6..c1b3d499 100644 --- a/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskStringTest.java +++ b/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskStringTest.java @@ -23,7 +23,7 @@ public class ClickHouseSinkTaskStringTest extends ClickHouseBase { private int countRowsWithEmojis(ClickHouseHelperClient chc, String topic) { - String queryCount = "select count(*) from " + topic + " where str LIKE '%\uD83D\uDE00%'"; + String queryCount = "select count(*) from " + topic + " where str LIKE '%\uD83D\uDE00%' SETTINGS select_sequential_consistency = 1"; try { Records records = chc.getClient().queryRecords(queryCount).get(); String value = records.iterator().next().getString(1); @@ -35,7 +35,7 @@ private int countRowsWithEmojis(ClickHouseHelperClient chc, String topic) { } } private int countRows(ClickHouseHelperClient chc, String topic) { - String queryCount = String.format("select count(*) from `%s`", topic); + String queryCount = String.format("select count(*) from `%s` SETTINGS select_sequential_consistency = 1", topic); try { Records records = chc.getClient().queryRecords(queryCount).get(); String value = records.iterator().next().getString(1); diff --git a/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskWithSchemaProxyTest.java b/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskWithSchemaProxyTest.java index 72b79499..f6f5b1cc 100644 --- a/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskWithSchemaProxyTest.java +++ b/src/test/java/com/clickhouse/kafka/connect/sink/ClickHouseSinkTaskWithSchemaProxyTest.java @@ -39,7 +39,11 @@ public class ClickHouseSinkTaskWithSchemaProxyTest extends ClickHouseBase { public static void setup() throws IOException { Network network = Network.newNetwork(); // Note: we are using a different version of ClickHouse for the proxy - https://github.com/ClickHouse/ClickHouse/issues/58828 - db = new org.testcontainers.clickhouse.ClickHouseContainer(ClickHouseTestHelpers.CLICKHOUSE_FOR_PROXY_DOCKER_IMAGE).withNetwork(network).withNetworkAliases("clickhouse"); + db = new org.testcontainers.clickhouse.ClickHouseContainer(ClickHouseTestHelpers.CLICKHOUSE_FOR_PROXY_DOCKER_IMAGE) + .withNetwork(network) + .withNetworkAliases("clickhouse") + .withPassword("test_password") + .withEnv("CLICKHOUSE_DEFAULT_ACCESS_MANAGEMENT", "1"); db.start(); toxiproxy = new ToxiproxyContainer("ghcr.io/shopify/toxiproxy:2.7.0").withNetwork(network).withNetworkAliases("toxiproxy"); diff --git a/src/test/java/com/clickhouse/kafka/connect/sink/helper/ClickHouseTestHelpers.java b/src/test/java/com/clickhouse/kafka/connect/sink/helper/ClickHouseTestHelpers.java index 23eba246..6698c806 100644 --- a/src/test/java/com/clickhouse/kafka/connect/sink/helper/ClickHouseTestHelpers.java +++ b/src/test/java/com/clickhouse/kafka/connect/sink/helper/ClickHouseTestHelpers.java @@ -183,7 +183,7 @@ public static OperationMetrics optimizeTable(ClickHouseHelperClient chc, String } public static int countRows(ClickHouseHelperClient chc, String tableName) { - String queryCount = String.format("SELECT COUNT(*) FROM `%s`", tableName); + String queryCount = String.format("SELECT COUNT(*) FROM `%s` SETTINGS select_sequential_consistency = 1", tableName); try { optimizeTable(chc, tableName); @@ -212,7 +212,7 @@ public static int sumRows(ClickHouseHelperClient chc, String tableName, String c } public static int countRowsWithEmojis(ClickHouseHelperClient chc, String tableName) { - String queryCount = "SELECT COUNT(*) FROM `" + tableName + "` WHERE str LIKE '%\uD83D\uDE00%'"; + String queryCount = "SELECT COUNT(*) FROM `" + tableName + "` WHERE str LIKE '%\uD83D\uDE00%' SETTINGS select_sequential_consistency = 1"; try { Records records = chc.getClient().queryRecords(queryCount).get(CLOUD_TIMEOUT_VALUE, CLOUD_TIMEOUT_UNIT); String value = records.iterator().next().getString(1);