From 7a16e6212633ae74414052dad8ac21d03d972920 Mon Sep 17 00:00:00 2001
From: Jonathan Davies <jpds@protonmail.com>
Date: Sat, 25 Oct 2025 14:53:50 +0100
Subject: [PATCH] nixos/clickhouse: tests: Add basic Selenium test

---
 nixos/tests/clickhouse/default.nix |   6 ++
 nixos/tests/clickhouse/ui.nix      | 118 +++++++++++++++++++++++++++++
 2 files changed, 124 insertions(+)
 create mode 100644 nixos/tests/clickhouse/ui.nix

diff --git a/nixos/tests/clickhouse/default.nix b/nixos/tests/clickhouse/default.nix
index 88e01fcb1fdb..1299eb199f67 100644
--- a/nixos/tests/clickhouse/default.nix
+++ b/nixos/tests/clickhouse/default.nix
@@ -28,4 +28,10 @@
       inherit package;
     };
   };
+  ui = runTest {
+    imports = [ ./ui.nix ];
+    _module.args = {
+      inherit package;
+    };
+  };
 }
diff --git a/nixos/tests/clickhouse/ui.nix b/nixos/tests/clickhouse/ui.nix
new file mode 100644
index 000000000000..b49d8a8d8861
--- /dev/null
+++ b/nixos/tests/clickhouse/ui.nix
@@ -0,0 +1,118 @@
+{
+  lib,
+  pkgs,
+  package,
+  ...
+}:
+
+{
+  name = "clickhouse-ui";
+
+  nodes = {
+    browser =
+      {
+        config,
+        pkgs,
+        ...
+      }:
+      {
+        environment.systemPackages =
+          let
+            clickhouseSeleniumScript =
+              pkgs.writers.writePython3Bin "clickhouse-selenium-script"
+                {
+                  libraries = with pkgs.python3Packages; [ selenium ];
+                }
+                ''
+                  from selenium import webdriver
+                  from selenium.webdriver.common.by import By
+                  from selenium.webdriver.firefox.options import Options
+                  from selenium.webdriver.support.ui import WebDriverWait
+
+                  options = Options()
+                  options.add_argument("--headless")
+                  service = webdriver.FirefoxService(executable_path="${lib.getExe pkgs.geckodriver}")  # noqa: E501
+
+                  driver = webdriver.Firefox(options=options, service=service)
+                  driver.implicitly_wait(10)
+                  driver.get("http://clickhouse:8123/play")
+
+                  wait = WebDriverWait(driver, 60)
+
+                  assert len(driver.find_elements(
+                    By.ID, "query_div")) == 1
+
+                  server_info_element = driver.find_element(
+                    By.XPATH, "//span[@id='server_info']")
+                  assert "${
+                    lib.strings.replaceStrings [ "-stable" "-lts" ] [ "" "" ] package.version
+                  }" in server_info_element.text
+
+                  # Shouldn't show before query done
+                  assert len(driver.find_elements(
+                    By.CSS_SELECTOR, ".row-number")) == 0
+
+                  query_box = driver.find_element(
+                    By.XPATH, "//textarea[@id='query']")
+                  query_box.click()
+                  query_box.send_keys("SELECT 1")
+
+                  query_run_button = driver.find_element(
+                    By.XPATH, "//button[@id='run']").click()
+
+                  # Now verify results shown
+                  assert len(driver.find_elements(
+                    By.XPATH, "//div[@id='check-mark']")) == 1
+
+                  assert len(driver.find_elements(
+                    By.CSS_SELECTOR, ".row-number")) == 2
+
+                  driver.close()
+                '';
+          in
+          with pkgs;
+          [
+            curl
+            firefox-unwrapped
+            geckodriver
+            clickhouseSeleniumScript
+          ];
+      };
+
+    clickhouse =
+      { config, pkgs, ... }:
+      {
+        networking.firewall.allowedTCPPorts = [
+          8123
+          9000
+        ];
+
+        environment.etc = {
+          "clickhouse-server/config.d/listen.xml".text = ''
+            <clickhouse>
+              <listen_host>::</listen_host>
+            </clickhouse>
+          '';
+        };
+
+        services.clickhouse = {
+          enable = true;
+          inherit package;
+        };
+      };
+  };
+
+  testScript = ''
+    clickhouse.wait_for_unit("clickhouse")
+    clickhouse.wait_for_open_port(8123)
+    clickhouse.wait_for_open_port(9000)
+
+    browser.systemctl("start network-online.target")
+    browser.wait_for_unit("network-online.target")
+
+    browser.succeed("curl -kLs http://clickhouse:8123/play | grep 'ClickHouse Query'")
+
+    # Ensure the application is actually rendered by the Javascript
+    browser.succeed("PYTHONUNBUFFERED=1 clickhouse-selenium-script")
+  '';
+}