iotedgedev/tests/test_envvars.py

import os

import pytest

from iotedgedev.compat import PY2
from iotedgedev.envvars import EnvVars
from iotedgedev.output import Output

pytestmark = pytest.mark.unit


def test_valid_get_envvar():
    output = Output()
    envvars = EnvVars(output)
    deployment_template = envvars.get_envvar("DEPLOYMENT_CONFIG_TEMPLATE_FILE")
    assert deployment_template is not None


def test_invalid_get_envvar():
    output = Output()
    envvars = EnvVars(output)
    testerval = envvars.get_envvar("TESTER")
    assert not testerval


def test_valid_load():
    output = Output()
    envvars = EnvVars(output)
    envvars.load()
    assert envvars.DEPLOYMENT_CONFIG_TEMPLATE_FILE == "deployment.template.json"


def test_valid_verify_envvar_has_val():
    output = Output()
    envvars = EnvVars(output)
    envvars.load()
    result = envvars.verify_envvar_has_val("DEPLOYMENT_CONFIG_TEMPLATE_FILE", envvars.DEPLOYMENT_CONFIG_TEMPLATE_FILE)
    assert not result


def test_valid_get_envvar_key_if_val():
    output = Output()
    envvars = EnvVars(output)
    assert envvars.get_envvar_key_if_val("DEPLOYMENT_CONFIG_TEMPLATE_FILE")


def test_invalid_get_envvar_key_if_val():
    output = Output()
    envvars = EnvVars(output)
    assert not envvars.get_envvar_key_if_val("TESTER")


def test_set_envvar():
    output = Output()
    envvars = EnvVars(output)
    registry_server = envvars.get_envvar("DEPLOYMENT_CONFIG_TEMPLATE_FILE")
    envvars.set_envvar("DEPLOYMENT_CONFIG_TEMPLATE_FILE", "deployment.template_new.json")
    new_registry_server = envvars.get_envvar("DEPLOYMENT_CONFIG_TEMPLATE_FILE")
    assert new_registry_server == "deployment.template_new.json"
    envvars.set_envvar("DEPLOYMENT_CONFIG_TEMPLATE_FILE", registry_server)


def test_envvar_clean():
    output = Output()
    envvars = EnvVars(output)
    envvar_clean_name = u"IOTEDGEDEV_ENVVAR_CLEAN_TEST"
    os.environ[envvar_clean_name] = u"test unicode string"

    envvars.clean()

    if PY2:
        assert isinstance(os.environ[envvar_clean_name], str)


def test_in_command_list_true_1():
    output = Output()
    envvars = EnvVars(output)
    assert envvars.in_command_list("solution new test_solution", ["init", "e2e", "solution new", "new", "simulator stop"])


def test_in_command_list_true_2():
    output = Output()
    envvars = EnvVars(output)
    assert envvars.in_command_list("solution new", ["init", "e2e", "solution new", "new", "simulator stop"])


def test_in_command_list_false_1():
    output = Output()
    envvars = EnvVars(output)
    assert not envvars.in_command_list("solution add filtermodule", ["init", "e2e", "solution new", "new", "simulator stop"])


def test_in_command_list_false_2():
    output = Output()
    envvars = EnvVars(output)
    assert not envvars.in_command_list("solution addotherstuff filtermodule", ["init", "e2e", "solution add", "new", "simulator stop"])


def test_in_command_list_empty_1():
    output = Output()
    envvars = EnvVars(output)
    assert not envvars.in_command_list("", ["init", "e2e", "solution new", "new", "simulator stop"])


def test_in_command_list_empty_2():
    output = Output()
    envvars = EnvVars(output)
    assert not envvars.in_command_list("solution new test_solution", ["init", "e2e", "", "new", "simulator stop"])


def test_in_command_list_empty_3():
    output = Output()
    envvars = EnvVars(output)
    assert envvars.in_command_list("", ["init", "e2e", "", "new", "simulator stop"])


def test_is_terse_command_true():
    output = Output()
    envvars = EnvVars(output)
    assert envvars.is_terse_command("iothub setup --update-dotenv")


def test_is_terse_command_false():
    output = Output()
    envvars = EnvVars(output)
    assert not envvars.is_terse_command("solution add")


def test_is_terse_command_empty():
    output = Output()
    envvars = EnvVars(output)
    assert envvars.is_terse_command("")


def test_default_container_registry_server_key_exists():
    output = Output()
    envvars = EnvVars(output)
    envvars.load()
    assert "CONTAINER_REGISTRY_SERVER" in os.environ


def test_default_container_registry_server_value_exists():
    output = Output()
    envvars = EnvVars(output)
    server = envvars.get_envvar("CONTAINER_REGISTRY_SERVER")
    assert server is not None


def test_default_container_registry_username_value_exists_or_returns_empty_string():
    output = Output()
    envvars = EnvVars(output)
    username = envvars.get_envvar("CONTAINER_REGISTRY_USERNAME")
    assert username is not None


def test_default_container_registry_password_value_exists_or_returns_empty_string():
    output = Output()
    envvars = EnvVars(output)
    password = envvars.get_envvar("CONTAINER_REGISTRY_PASSWORD")
    assert password is not None


def test_container_registry_server_key_missing_sys_exit():
    output = Output()
    envvars = EnvVars(output)
    with pytest.raises(ValueError):
        envvars.get_envvar("CONTAINER_REGISTRY_SERVER_UNITTEST", required=True)


@pytest.fixture
def setup_test_env(request):
    output = Output()
    envvars = EnvVars(output)
    envvars.set_envvar("CONTAINER_REGISTRY_SERVER_UNITTEST", 'unittest.azurecr.io')
    envvars.set_envvar("CONTAINER_REGISTRY_USERNAME_UNITTEST", 'username')
    envvars.set_envvar("CONTAINER_REGISTRY_PASSWORD_UNITTEST", 'password')

    def clean():
        os.environ.pop("CONTAINER_REGISTRY_SERVER_UNITTEST")
        os.environ.pop("CONTAINER_REGISTRY_USERNAME_UNITTEST")
        os.environ.pop("CONTAINER_REGISTRY_PASSWORD_UNITTEST")
    request.addfinalizer(clean)

    return


def test_unique_container_registry_server_tokens():
    unique = set()
    length_container_registry_server = len('container_registry_server')
    is_unique = True
    output = Output()
    envvars = EnvVars(output)
    envvars.load()
    for key in os.environ:
        key = key.lower()
        if key.startswith('container_registry_server'):
            token = key[length_container_registry_server:]
            if token not in unique:
                unique.add(token)
            else:
                is_unique = False
    assert is_unique


def test_unique_container_registry_username_tokens():
    unique = set()
    length_container_registry_username = len('container_registry_username')
    is_unique = True
    output = Output()
    envvars = EnvVars(output)
    envvars.load()
    for key in os.environ:
        key = key.lower()
        if key.startswith('container_registry_username'):
            token = key[length_container_registry_username:]
            if token not in unique:
                unique.add(token)
            else:
                is_unique = False
    assert is_unique


def test_unique_container_registry_password_tokens():
    unique = set()
    length_container_registry_password = len('container_registry_password')
    is_unique = True
    output = Output()
    envvars = EnvVars(output)
    envvars.load()
    for key in os.environ:
        key = key.lower()
        if key.startswith('container_registry_password'):
            token = key[length_container_registry_password:]
            if token not in unique:
                unique.add(token)
            else:
                is_unique = False
    assert is_unique


def test_additional_container_registry_map_has_val(setup_test_env):
    output = Output()
    envvars = EnvVars(output)
    envvars.load()
    assert len(envvars.CONTAINER_REGISTRY_MAP) == 2
    assert 'UNITTEST' in envvars.CONTAINER_REGISTRY_MAP.keys()
    assert envvars.CONTAINER_REGISTRY_MAP['UNITTEST'].server == 'unittest.azurecr.io'
    assert envvars.CONTAINER_REGISTRY_MAP['UNITTEST'].username == 'username'
    assert envvars.CONTAINER_REGISTRY_MAP['UNITTEST'].password == 'password'