test_aiidalab.py

import pytest
import requests
import json
from packaging.version import parse


def test_notebook_service_available(notebook_service):
    response = requests.get(f"{notebook_service}/")
    assert response.status_code == 200


def test_pip_check(docker_exec, aiidalab_exec, nb_user):
    docker_exec("pip check")
    aiidalab_exec("pip check")
    docker_exec(f"mamba run -n aiida-homebase pip check")


def test_aiidalab_available(aiidalab_exec, nb_user, variant):
    if "lab" not in variant:
        pytest.skip()
    output = aiidalab_exec("aiidalab --version", user=nb_user).decode().strip().lower()
    assert "aiidalab" in output


def test_create_conda_environment(docker_exec, nb_user):
    output = docker_exec("conda create -y -n tmp", user=nb_user).decode().strip()
    assert "conda activate tmp" in output


def test_correct_python_version_installed(docker_exec, python_version):
    info = json.loads(docker_exec("mamba list --json --full-name python").decode())[0]
    assert info["name"] == "python"
    assert parse(info["version"]) == parse(python_version)

    info = json.loads(
        docker_exec("mamba list -n aiida-base --json --full-name python").decode()
    )[0]
    assert info["name"] == "python"
    assert parse(info["version"]) == parse(python_version)


def test_correct_aiida_version_installed(docker_exec, aiida_version):
    info = json.loads(
        docker_exec("mamba list -n aiida-base --json --full-name aiida-core").decode()
    )[0]
    assert info["name"] == "aiida-core"
    assert parse(info["version"]) == parse(aiida_version)


def test_correct_aiidalab_version_installed(docker_exec, aiidalab_version, variant):
    if "lab" not in variant:
        pytest.skip()
    info = json.loads(
        docker_exec("mamba list -n aiida-base --json --full-name aiidalab").decode()
    )[0]
    assert info["name"] == "aiidalab"
    assert parse(info["version"]) == parse(aiidalab_version)


def test_correct_aiidalab_home_version_installed(
    aiidalab_exec, aiidalab_home_version, variant
):
    if "lab" not in variant:
        pytest.skip()
    info = json.loads(
        aiidalab_exec("mamba list --json --full-name aiidalab-home").decode()
    )[0]
    assert info["name"] == "aiidalab-home"
    assert parse(info["version"]) == parse(aiidalab_home_version)


@pytest.mark.parametrize("package_manager", ["mamba", "pip"])
@pytest.mark.parametrize("incompatible_version", ["1.6.3"])
def test_prevent_installation_of_incompatible_aiida_version(
    aiidalab_exec, nb_user, aiida_version, package_manager, incompatible_version
):
    assert parse(aiida_version) != parse(incompatible_version)
    # Expected to succeed:
    aiidalab_exec(
        f"{package_manager} install aiida-core=={aiida_version}", user=nb_user
    )
    with pytest.raises(Exception):
        aiidalab_exec(
            f"{package_manager} install aiida-core={incompatible_version}", user=nb_user
        )


@pytest.mark.parametrize("package_manager", ["mamba", "pip"])
@pytest.mark.parametrize("incompatible_version", ["22.7.1"])
def test_prevent_installation_of_incompatible_aiidalab_version(
    aiidalab_exec,
    nb_user,
    package_manager,
    incompatible_version,
    variant,
):
    if "lab" not in variant:
        pytest.skip()
    with pytest.raises(Exception):
        aiidalab_exec(
            f"{package_manager} install aiidalab={incompatible_version}", user=nb_user
        )


def test_verdi_status(aiidalab_exec, nb_user):
    output = aiidalab_exec("verdi status", user=nb_user).decode().strip()
    assert "Connected to RabbitMQ" in output
    assert "Daemon is running" in output


@pytest.mark.integration
@pytest.mark.parametrize("package_name", ["aiidalab-widgets-base", "aiidalab-qe"])
@pytest.mark.skip(reason="Last AWB stable release doesn't support AiiDA-2.0 yet")
def test_install_apps_from_stable(aiidalab_exec, package_name, nb_user, variant):
    if "lab" not in variant:
        pytest.skip()
    output = (
        aiidalab_exec(f"aiidalab install --yes {package_name}", user=nb_user)
        .decode()
        .strip()
    )
    assert "ERROR" not in output
    assert "dependency conflict" not in output
    assert f"Installed '{package_name}' version" in output


@pytest.mark.integration
@pytest.mark.parametrize("package_name", ["aiidalab-widgets-base", "aiidalab-qe"])
def test_install_apps_from_master(aiidalab_exec, package_name, nb_user, variant):
    if "lab" not in variant:
        pytest.skip()
    output = (
        aiidalab_exec(
            f"aiidalab install --yes {package_name}@git+https://github.com/aiidalab/{package_name}.git",
            user=nb_user,
        )
        .decode()
        .strip()
    )
    assert "ERROR" not in output
    assert "dependency conflict" not in output
    assert "Successfully installed" in output
    # Disabling tests due to issues with restarting daemon through verdi
    # assert "Error:" not in output
    # assert f"Installed '{package_name}' version" in output


@pytest.mark.skip(reason="No longer adding this path")
def test_path_local_pip(docker_exec, nb_user):
    """test that the pip local bin path ~/.local/bin is added to PATH"""
    output = docker_exec("bash -c 'echo \"${PATH}\"'", user=nb_user).decode()
    assert f"/home/{nb_user}/.local/bin" in output