nixos/tests/lxd/ui.nix

   1 import ../make-test-python.nix ({ pkgs, ... }: {
   2   name = "lxd-ui";
   3
   4   nodes.machine = { lib, ... }: {
   5     virtualisation = {
   6       lxd.enable = true;
   7       lxd.ui.enable = true;
   8     };
   9
  10     environment.systemPackages =
  11       let
  12         seleniumScript = pkgs.writers.writePython3Bin "selenium-script"
  13           {
  14             libraries = with pkgs.python3Packages; [ selenium ];
  15           } ''
  16           from selenium import webdriver
  17           from selenium.webdriver.common.by import By
  18           from selenium.webdriver.firefox.options import Options
  19           from selenium.webdriver.support.ui import WebDriverWait
  20
  21           options = Options()
  22           options.add_argument("--headless")
  23           service = webdriver.FirefoxService(executable_path="${lib.getExe pkgs.geckodriver}")  # noqa: E501
  24
  25           driver = webdriver.Firefox(options=options, service=service)
  26           driver.implicitly_wait(10)
  27           driver.get("https://localhost:8443/ui")
  28
  29           wait = WebDriverWait(driver, 60)
  30
  31           assert len(driver.find_elements(By.CLASS_NAME, "l-application")) > 0
  32           assert len(driver.find_elements(By.CLASS_NAME, "l-navigation__drawer")) > 0
  33
  34           driver.close()
  35         '';
  36       in
  37       with pkgs; [ curl firefox-unwrapped geckodriver seleniumScript ];
  38   };
  39
  40
  41   testScript = ''
  42     machine.wait_for_unit("sockets.target")
  43     machine.wait_for_unit("lxd.service")
  44     machine.wait_for_file("/var/lib/lxd/unix.socket")
  45
  46     # Wait for lxd to settle
  47     machine.succeed("lxd waitready")
  48
  49     # Configure LXC listen address
  50     machine.succeed("lxc config set core.https_address :8443")
  51     machine.succeed("systemctl restart lxd")
  52
  53     # Check that the LXD_UI environment variable is populated in the systemd unit
  54     machine.succeed("cat /etc/systemd/system/lxd.service | grep 'LXD_UI'")
  55
  56     # Ensure the endpoint returns an HTML page with 'LXD UI' in the title
  57     machine.succeed("curl -kLs https://localhost:8443/ui | grep '<title>LXD UI</title>'")
  58
  59     # Ensure the application is actually rendered by the Javascript
  60     machine.succeed("PYTHONUNBUFFERED=1 selenium-script")
  61   '';
  62 })