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