nixos/tests/cockpit.nix

   1 import ./make-test-python.nix (
   2   { pkgs, lib, ... }:
   3
   4   let
   5     user = "alice"; # from ./common/user-account.nix
   6     password = "foobar"; # from ./common/user-account.nix
   7   in {
   8     name = "cockpit";
   9     meta = {
  10       maintainers = with lib.maintainers; [ lucasew ];
  11     };
  12     nodes = {
  13       server = { config, ... }: {
  14         imports = [ ./common/user-account.nix ];
  15         security.polkit.enable = true;
  16         users.users.${user} = {
  17           extraGroups = [ "wheel" ];
  18         };
  19         services.cockpit = {
  20           enable = true;
  21           openFirewall = true;
  22           settings = {
  23             WebService = {
  24               Origins = "https://server:9090";
  25             };
  26           };
  27         };
  28       };
  29       client = { config, ... }: {
  30         imports = [ ./common/user-account.nix ];
  31         environment.systemPackages = let
  32           seleniumScript = pkgs.writers.writePython3Bin "selenium-script" {
  33             libraries = with pkgs.python3Packages; [ selenium ];
  34             } ''
  35             from selenium import webdriver
  36             from selenium.webdriver.common.by import By
  37             from selenium.webdriver.firefox.options import Options
  38             from selenium.webdriver.support.ui import WebDriverWait
  39             from selenium.webdriver.support import expected_conditions as EC
  40             from time import sleep
  41
  42
  43             def log(msg):
  44                 from sys import stderr
  45                 print(f"[*] {msg}", file=stderr)
  46
  47
  48             log("Initializing")
  49
  50             options = Options()
  51             options.add_argument("--headless")
  52
  53             service = webdriver.FirefoxService(executable_path="${lib.getExe pkgs.geckodriver}")  # noqa: E501
  54             driver = webdriver.Firefox(options=options, service=service)
  55
  56             driver.implicitly_wait(10)
  57
  58             log("Opening homepage")
  59             driver.get("https://server:9090")
  60
  61             wait = WebDriverWait(driver, 60)
  62
  63
  64             def wait_elem(by, query):
  65                 wait.until(EC.presence_of_element_located((by, query)))
  66
  67
  68             def wait_title_contains(title):
  69                 wait.until(EC.title_contains(title))
  70
  71
  72             def find_element(by, query):
  73                 return driver.find_element(by, query)
  74
  75
  76             def set_value(elem, value):
  77                 script = 'arguments[0].value = arguments[1]'
  78                 return driver.execute_script(script, elem, value)
  79
  80
  81             log("Waiting for the homepage to load")
  82
  83             # cockpit sets initial title as hostname
  84             wait_title_contains("server")
  85             wait_elem(By.CSS_SELECTOR, 'input#login-user-input')
  86
  87             log("Homepage loaded!")
  88
  89             log("Filling out username")
  90             login_input = find_element(By.CSS_SELECTOR, 'input#login-user-input')
  91             set_value(login_input, "${user}")
  92
  93             log("Filling out password")
  94             password_input = find_element(By.CSS_SELECTOR, 'input#login-password-input')
  95             set_value(password_input, "${password}")
  96
  97             log("Submitting credentials for login")
  98             driver.find_element(By.CSS_SELECTOR, 'button#login-button').click()
  99
 100             # driver.implicitly_wait(1)
 101             # driver.get("https://server:9090/system")
 102
 103             log("Waiting dashboard to load")
 104             wait_title_contains("${user}@server")
 105
 106             log("Waiting for the frontend to initialize")
 107             sleep(1)
 108
 109             log("Looking for that banner that tells about limited access")
 110             container_iframe = find_element(By.CSS_SELECTOR, 'iframe.container-frame')
 111             driver.switch_to.frame(container_iframe)
 112
 113             assert "Web console is running in limited access mode" in driver.page_source
 114
 115             driver.close()
 116           '';
 117         in with pkgs; [ firefox-unwrapped geckodriver seleniumScript ];
 118       };
 119     };
 120
 121     testScript = ''
 122       start_all()
 123
 124       server.wait_for_open_port(9090)
 125       server.wait_for_unit("network.target")
 126       server.wait_for_unit("multi-user.target")
 127       server.systemctl("start", "polkit")
 128
 129       client.wait_for_unit("multi-user.target")
 130
 131       client.succeed("curl -k https://server:9090 -o /dev/stderr")
 132       print(client.succeed("whoami"))
 133       client.succeed('PYTHONUNBUFFERED=1 selenium-script')
 134     '';
 135   }
 136 )