clan-core/pkgs/clan-cli/tests/test_machines_api.py

from pathlib import Path

from api import TestClient


def test_machines(api: TestClient, test_flake: Path) -> None:
    response = api.get("/api/machines")
    assert response.status_code == 200
    assert response.json() == {"machines": []}

    response = api.post("/api/machines", json={"name": "test"})
    assert response.status_code == 201
    assert response.json() == {"machine": {"name": "test", "status": "unknown"}}

    response = api.get("/api/machines/test")
    assert response.status_code == 200
    assert response.json() == {"machine": {"name": "test", "status": "unknown"}}

    response = api.get("/api/machines")
    assert response.status_code == 200
    assert response.json() == {"machines": [{"name": "test", "status": "unknown"}]}


def test_configure_machine(api: TestClient, test_flake: Path) -> None:
    # ensure error 404 if machine does not exist when accessing the config
    response = api.get("/api/machines/machine1/config")
    assert response.status_code == 404

    # ensure error 404 if machine does not exist when writing to the config
    response = api.put("/api/machines/machine1/config", json={})
    assert response.status_code == 404

    # create the machine
    response = api.post("/api/machines", json={"name": "machine1"})
    assert response.status_code == 201

    # ensure an empty config is returned by default for a new machine
    response = api.get("/api/machines/machine1/config")
    assert response.status_code == 200
    assert response.json() == {"config": {}}

    # get jsonschema for machine
    response = api.get("/api/machines/machine1/schema")
    assert response.status_code == 200
    json_response = response.json()
    assert "schema" in json_response and "properties" in json_response["schema"]

    # set come invalid config (fileSystems missing)
    config = dict(
        clan=dict(
            jitsi=dict(
                enable=True,
            ),
        ),
    )
    response = api.put(
        "/api/machines/machine1/config",
        json=config,
    )
    assert response.status_code == 400
    assert (
        "The ‘fileSystems’ option does not specify your root"
        in response.json()["detail"]
    )

    # ensure config is still empty after the invalid attempt
    response = api.get("/api/machines/machine1/config")
    assert response.status_code == 200
    assert response.json() == {"config": {}}

    fs_config = dict(
        fileSystems={
            "/": dict(
                device="/dev/fake_disk",
                fsType="ext4",
            ),
        },
        boot=dict(
            loader=dict(
                grub=dict(
                    devices=["/dev/fake_disk"],
                ),
            ),
        ),
    )

    # set some valid config
    config2 = dict(
        clan=dict(
            jitsi=dict(
                enable=True,
            ),
        ),
        **fs_config,
    )
    response = api.put(
        "/api/machines/machine1/config",
        json=config2,
    )
    assert response.status_code == 200
    assert response.json() == {"config": config2}

    # ensure that the config has actually been updated
    response = api.get("/api/machines/machine1/config")
    assert response.status_code == 200
    assert response.json() == {"config": config2}

    # ensure PUT on the config is idempotent by passing the config again
    # For example, this should not result in the boot.loader.grub.devices being
    #   set twice (eg. merged)
    response = api.put(
        "/api/machines/machine1/config",
        json=config2,
    )
    assert response.status_code == 200
    assert response.json() == {"config": config2}

    # verify the machine config evaluates
    response = api.get("/api/machines/machine1/verify")
    assert response.status_code == 200
    assert response.json() == {"success": True, "error": None}

    # get the schema with an extra module imported
    response = api.put(
        "/api/machines/machine1/schema",
        json={"clanImports": ["fake-module"]},
    )
    # expect the result schema to contain the fake-module.fake-flag option
    assert response.status_code == 200
    assert (
        response.json()["schema"]["properties"]["fake-module"]["properties"][
            "fake-flag"
        ]["type"]
        == "boolean"
    )

    # new config importing an extra clanModule (clanModules.fake-module)
    config_with_imports: dict = {
        "clanImports": ["fake-module"],
        "clan": {
            "fake-module": {
                "fake-flag": True,
            },
        },
        **fs_config,
    }

    # set the fake-module.fake-flag option to true
    response = api.put(
        "/api/machines/machine1/config",
        json=config_with_imports,
    )
    assert response.status_code == 200
    assert response.json() == {
        "config": {
            "clanImports": ["fake-module"],
            "clan": {
                "fake-module": {
                    "fake-flag": True,
                },
            },
            **fs_config,
        }
    }

    # remove the import from the config
    config_with_empty_imports = dict(
        clanImports=[],
        **fs_config,
    )
    response = api.put(
        "/api/machines/machine1/config",
        json=config_with_empty_imports,
    )
    assert response.status_code == 200
    assert response.json() == {"config": config_with_empty_imports}