From 7e7c2c7b85b87e8e7af9be6f42dd5c10d0031516 Mon Sep 17 00:00:00 2001
From: Yuya Ebihara <yuya.ebihara@starburstdata.com>
Date: Tue, 11 Jan 2022 19:06:36 +0900
Subject: [PATCH] Set keep_alive_timeout as 10 seconds in ClickHouse testing
 server

For avoiding below failure during copy of TPCH tables.
 ClickHouse exception, code: 1002, host: localhost, port: <port>;
 localhost:<port> failed to respond
---
 .../io/trino/plugin/clickhouse/TestingClickHouseServer.java  | 2 ++
 plugin/trino-clickhouse/src/test/resources/custom.xml        | 5 +++++
 2 files changed, 7 insertions(+)
 create mode 100644 plugin/trino-clickhouse/src/test/resources/custom.xml

diff --git a/plugin/trino-clickhouse/src/test/java/io/trino/plugin/clickhouse/TestingClickHouseServer.java b/plugin/trino-clickhouse/src/test/java/io/trino/plugin/clickhouse/TestingClickHouseServer.java
index 36144a1c0983..7bf507169145 100644
--- a/plugin/trino-clickhouse/src/test/java/io/trino/plugin/clickhouse/TestingClickHouseServer.java
+++ b/plugin/trino-clickhouse/src/test/java/io/trino/plugin/clickhouse/TestingClickHouseServer.java
@@ -22,6 +22,7 @@
 
 import static java.lang.String.format;
 import static org.testcontainers.containers.ClickHouseContainer.HTTP_PORT;
+import static org.testcontainers.utility.MountableFile.forClasspathResource;
 
 public class TestingClickHouseServer
         implements Closeable
@@ -33,6 +34,7 @@ public TestingClickHouseServer()
     {
         // Use 2nd stable version
         dockerContainer = (ClickHouseContainer) new ClickHouseContainer(CLICKHOUSE_IMAGE)
+                .withCopyFileToContainer(forClasspathResource("custom.xml"), "/etc/clickhouse-server/config.d/custom.xml")
                 .withStartupAttempts(10);
 
         dockerContainer.start();
diff --git a/plugin/trino-clickhouse/src/test/resources/custom.xml b/plugin/trino-clickhouse/src/test/resources/custom.xml
new file mode 100644
index 000000000000..f4fa0e8b59d5
--- /dev/null
+++ b/plugin/trino-clickhouse/src/test/resources/custom.xml
@@ -0,0 +1,5 @@
+<?xml version="1.0"?>
+<yandex>
+    <!-- To avoid "failed to response" error message during copy of TPCH tables -->
+    <keep_alive_timeout>10</keep_alive_timeout>
+</yandex>