Add Clickhouse module

packages/modules/clickhouse/jest.config.ts

@@ -0,0 +1,11 @@
+import type { Config } from "jest";
+import * as path from "path";
+
+const config: Config = {
+  preset: "ts-jest",
+  moduleNameMapper: {
+    "^testcontainers$": path.resolve(__dirname, "../../testcontainers/src"),
+  },
+};
+
+export default config;

packages/modules/clickhouse/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@testcontainers/clickhouse",
+  "version": "10.2.1",
+  "license": "MIT",
+  "keywords": [
+    "clickhouse",
+    "testing",
+    "docker",
+    "testcontainers"
+  ],
+  "description": "Clickhouse module for Testcontainers",
+  "homepage": "https://github.com/testcontainers/testcontainers-node#readme",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/testcontainers/testcontainers-node"
+  },
+  "bugs": {
+    "url": "https://github.com/testcontainers/testcontainers-node/issues"
+  },
+  "main": "build/index.js",
+  "files": [
+    "build"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "prepack": "shx cp ../../../README.md . && shx cp ../../../LICENSE .",
+    "build": "tsc --project tsconfig.build.json"
+  },
+  "devDependencies": {
+    "@clickhouse/client": "^0.2.2"
+  },
+  "dependencies": {
+    "testcontainers": "^10.2.1"
+  }
+}

packages/modules/clickhouse/src/clickhouse-container.test.ts

@@ -0,0 +1,102 @@
+import { ClickhouseContainer, StartedClickhouseContainer } from "./clickhouse-container";
+import { ClickHouseClient, createClient } from "@clickhouse/client";
+import path from "path";
+
+describe("ClickhouseContainer", () => {
+  jest.setTimeout(240_000);
+
+  it("should work with defaults", async () => {
+    const container = await new ClickhouseContainer().start();
+    const client = createClickhouseContainerHttpClient(container);
+    await _test(client, container.getDatabase());
+    await client.close();
+    await container.stop();
+  });
+
+  it("should work with custom credentials", async () => {
+    const container = await new ClickhouseContainer()
+      .withUsername(`un${(Math.random()*1000000) | 0}`)
+      .withPassword(`pass${(Math.random()*1000000) | 0}`)
+      .start();
+    const client = createClickhouseContainerHttpClient(container);
+    await _test(client, container.getDatabase());
+    await client.close();
+    await container.stop();
+  });
+
+  it("should work with custom database", async () => {
+    const container = await new ClickhouseContainer()
+      .withDatabase(`db${(Math.random()*1000000) | 0}`)
+      .start();
+    const client = createClickhouseContainerHttpClient(container);
+    await _test(client, container.getDatabase());
+    await client.close();
+    await container.stop();
+  });
+
+  it("should work with custom image", async () => {
+    const container = await new ClickhouseContainer("23.8-alpine").start();
+    const client = createClickhouseContainerHttpClient(container);
+    await _test(client, container.getDatabase());
+    await client.close();
+    await container.stop();
+  });
+
+  it("should work with custom xml config", async () => {
+    const container = await new ClickhouseContainer()
+      .withPassword("")
+      .withXmlConfigFile(path.join("testdata", "config.xml"))
+      .start();
+    const client = createClickhouseContainerHttpClient(container);
+    await _test(client, container.getDatabase());
+    await client.close();
+    await container.stop();
+  });
+
+  it("should work with custom yaml config", async () => {
+    const container = await new ClickhouseContainer()
+      .withPassword("")
+      .withXmlConfigFile(path.join("testdata", "config.yaml"))
+      .start();
+    const client = createClickhouseContainerHttpClient(container);
+    await _test(client, container.getDatabase());
+    await client.close();
+    await container.stop();
+  });
+
+  function createClickhouseContainerHttpClient(container: StartedClickhouseContainer) {
+    return createClient({
+      host: `http://${container.getHost()}:${container.getHostPorts().http}`,
+      username: container.getUsername(),
+      password: container.getPassword(),
+    });
+  }
+
+  async function _test(client: ClickHouseClient, db: string) {
+    const tableName = 'array_json_each_row';
+    await client.command({
+      query: `DROP TABLE IF EXISTS ${db}.${tableName}`,
+    });
+    await client.command({
+      query: `
+        CREATE TABLE ${db}.${tableName}
+        (id UInt64, name String)
+        ENGINE MergeTree()
+        ORDER BY (id)
+      `,
+    });
+    await client.insert({
+      table: `${db}.${tableName}`,
+      values: [
+        { id: 42, name: 'foo' },
+        { id: 42, name: 'bar' },
+      ],
+      format: 'JSONEachRow',
+    });
+    const rows = await client.query({
+      query: `SELECT * FROM ${db}.${tableName}`,
+      format: 'JSONEachRow',
+    });
+    expect(await rows.json()).toEqual([{id:"42",name:"foo"},{id:"42",name:"bar"}]);   
+  }
+});

packages/modules/clickhouse/src/clickhouse-container.ts

@@ -0,0 +1,108 @@
+import { AbstractStartedContainer, GenericContainer, StartedTestContainer, Wait } from "testcontainers";
+
+const IMAGE_NAME = "clickhouse/clickhouse-server";
+const DEFAULT_IMAGE_VER = "23.3.8.21-alpine"
+
+const HTTP_PORT = 8123;
+const NATIVE_PORT = 9000;
+
+const CONFIG_PATH_XML = "/etc/clickhouse-server/config.d/config.xml";
+const CONFIG_PATH_YAML = "/etc/clickhouse-server/config.d/config.yaml";
+const CONFIG_FILE_MODE = parseInt("0644", 8)
+
+export class ClickhouseContainer extends GenericContainer {
+  private database = "test";
+  private username = "test";
+  private password = "test";
+  private hostConfigPathXml: string = "";
+  private hostConfigPathYaml: string = "";
+
+  constructor(imageVer = DEFAULT_IMAGE_VER) {
+    super(`${IMAGE_NAME}:${imageVer}`);
+  }
+
+  public withDatabase(database: string): this {
+    this.database = database;
+    return this;
+  }
+
+  public withUsername(username: string): this {
+    this.username = username;
+    return this;
+  }
+
+  public withPassword(password: string): this {
+    this.password = password;
+    return this;
+  }
+
+  public withXmlConfigFile(path: string): this {
+    this.hostConfigPathXml = path;
+    return this;
+  }
+
+  public withYamlConfigFile(path: string): this {
+    this.hostConfigPathYaml = path;
+    return this;
+  }
+
+  public override async start(): Promise<StartedClickhouseContainer> {
+    this.withExposedPorts(...(this.hasExposedPorts ? this.exposedPorts : [HTTP_PORT, NATIVE_PORT]))
+      .withEnvironment({
+        CLICKHOUSE_DB: this.database,
+        CLICKHOUSE_USER: this.username,
+        CLICKHOUSE_PASSWORD: this.password,
+      })
+      .withWaitStrategy(Wait.forHttp(`/ping`, HTTP_PORT)
+        .forStatusCode(200))
+      .withStartupTimeout(120_000);
+
+    if (this.hostConfigPathXml)
+      this.withCopyFilesToContainer([{source: this.hostConfigPathXml, target: CONFIG_PATH_XML, mode: CONFIG_FILE_MODE}]);
+    if (this.hostConfigPathYaml)
+      this.withCopyFilesToContainer([{source: this.hostConfigPathYaml, target: CONFIG_PATH_YAML, mode: CONFIG_FILE_MODE}]);
+
+    return new StartedClickhouseContainer(
+      await super.start(),
+      HTTP_PORT,
+      NATIVE_PORT,
+      this.database,
+      this.username,
+      this.password,
+    );
+  }
+}
+
+export class StartedClickhouseContainer extends AbstractStartedContainer {
+  private readonly httpHostPort: number;
+  private readonly nativeHostPort: number;
+
+  constructor(
+    startedTestContainer: StartedTestContainer,
+    httpPort: number,
+    nativePort: number,
+    private readonly database: string,
+    private readonly username: string,
+    private readonly password: string,
+  ) {
+    super(startedTestContainer);
+    this.httpHostPort = startedTestContainer.getMappedPort(httpPort);
+    this.nativeHostPort = startedTestContainer.getMappedPort(nativePort);
+  }
+
+  public getHostPorts(): {http: number; native: number} {
+    return {http: this.httpHostPort, native: this.nativeHostPort}
+  }
+
+  public getDatabase(): string {
+    return this.database;
+  }
+
+  public getUsername(): string {
+    return this.username;
+  }
+
+  public getPassword(): string {
+    return this.password;
+  }
+}

packages/modules/clickhouse/src/index.ts

@@ -0,0 +1 @@
+export { ClickhouseContainer, StartedClickhouseContainer } from "./clickhouse-container";

packages/modules/clickhouse/testdata/config.xml

@@ -0,0 +1,3 @@
+<clickhouse>
+  <allow_no_password>1</allow_no_password>
+</clickhouse>

packages/modules/clickhouse/testdata/config.yaml

@@ -0,0 +1 @@
+allow_no_password: true

packages/modules/clickhouse/tsconfig.build.json

@@ -0,0 +1,13 @@
+{
+  "extends": "./tsconfig.json",
+  "exclude": [
+    "build",
+    "jest.config.ts",
+    "src/**/*.test.ts"
+  ],
+  "references": [
+    {
+      "path": "../../testcontainers"
+    }
+  ]
+}

packages/modules/clickhouse/tsconfig.json

@@ -0,0 +1,21 @@
+{
+  "extends": "../../../tsconfig.base.json",
+  "compilerOptions": {
+    "rootDir": "src",
+    "outDir": "build",
+    "paths": {
+      "testcontainers": [
+        "../../testcontainers/src"
+      ]
+    }
+  },
+  "exclude": [
+    "build",
+    "jest.config.ts"
+  ],
+  "references": [
+    {
+      "path": "../../testcontainers"
+    }
+  ]
+}