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

from api import TestClient
from fixtures_flakes import FlakeForTest


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

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

    assert response.json() == {"machine": {"name": "test", "status": "unknown"}}

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

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


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

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

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

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

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

    # an invalid config missing the fileSystems
    invalid_config = dict(
        clan=dict(
            jitsi=dict(
                enable=True,
            ),
        ),
    )

    # verify an invalid config (fileSystems missing) fails
    response = api.put(
        f"/api/{test_flake.name}/machines/machine1/verify",
        json=invalid_config,
    )
    assert response.status_code == 200
    assert (
        "The ‘fileSystems’ option does not specify your root"
        in response.json()["error"]
    )

    # set come invalid config (fileSystems missing)
    response = api.put(
        f"/api/{test_flake.name}/machines/machine1/config",
        json=invalid_config,
    )
    assert response.status_code == 200

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

    # the part of the config that makes the evaluation pass
    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(
        f"/api/{test_flake.name}/machines/machine1/config",
        json=config2,
    )
    assert response.status_code == 200
    assert response.json() == {"config": config2}

    # get the config again
    response = api.get(f"/api/{test_flake.name}/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(
        f"/api/{test_flake.name}/machines/machine1/config",
        json=config2,
    )
    assert response.status_code == 200
    assert response.json() == {"config": config2}

    # verify the machine config evaluates
    response = api.get(f"/api/{test_flake.name}/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(
        f"/api/{test_flake.name}/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(
        f"/api/{test_flake.name}/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(
        f"/api/{test_flake.name}/machines/machine1/config",
        json=config_with_empty_imports,
    )
    assert response.status_code == 200
    assert response.json() == {"config": config_with_empty_imports}