depot/third_party/nixpkgs/nixos/tests/clickhouse.nix

import ./make-test-python.nix ({ pkgs, ... }: {
  name = "clickhouse";
  meta.maintainers = with pkgs.lib.maintainers; [ ma27 ];

  machine = {
    services.clickhouse.enable = true;
    virtualisation.memorySize = 4096;
  };

  testScript =
    let
      # work around quote/substitution complexity by Nix, Perl, bash and SQL.
      tableDDL = pkgs.writeText "ddl.sql" "CREATE TABLE `demo` (`value` FixedString(10)) engine = MergeTree PARTITION BY value ORDER BY tuple();";
      insertQuery = pkgs.writeText "insert.sql" "INSERT INTO `demo` (`value`) VALUES ('foo');";
      selectQuery = pkgs.writeText "select.sql" "SELECT * from `demo`";
    in
      ''
        machine.start()
        machine.wait_for_unit("clickhouse.service")
        machine.wait_for_open_port(9000)

        machine.succeed(
            "cat ${tableDDL} | clickhouse-client"
        )
        machine.succeed(
            "cat ${insertQuery} | clickhouse-client"
        )
        machine.succeed(
            "cat ${selectQuery} | clickhouse-client | grep foo"
        )
      '';
})