"""Unit tests: set_yadm_dirs"""
import pytest


@pytest.mark.parametrize(
    'condition', [
        'basic',
        'override',
        'override_data',
        'xdg_config_home',
        'xdg_data_home'
        ],
    )
def test_set_yadm_dirs(runner, yadm, condition):
    """Test set_yadm_dirs"""
    setup = ''
    if condition == 'override':
        setup = 'YADM_DIR=/override'
    elif condition == 'override_data':
        setup = 'YADM_DATA=/override'
    elif condition == 'xdg_config_home':
        setup = 'XDG_CONFIG_HOME=/xdg'
    elif condition == 'xdg_data_home':
        setup = 'XDG_DATA_HOME=/xdg'
    script = f"""
        HOME=/testhome
        YADM_TEST=1 source {yadm}
        XDG_CONFIG_HOME=
        XDG_DATA_HOME=
        {setup}
        set_yadm_dirs
        echo "YADM_DIR=$YADM_DIR"
        echo "YADM_DATA=$YADM_DATA"
    """
    run = runner(command=['bash'], inp=script)
    assert run.success
    assert run.err == ''
    if condition == 'basic':
        assert 'YADM_DIR=/testhome/.config/yadm' in run.out
        assert 'YADM_DATA=/testhome/.local/share/yadm' in run.out
    elif condition == 'override':
        assert 'YADM_DIR=/override' in run.out
    elif condition == 'override_data':
        assert 'YADM_DATA=/override' in run.out
    elif condition == 'xdg_config_home':
        assert 'YADM_DIR=/xdg/yadm' in run.out
    elif condition == 'xdg_data_home':
        assert 'YADM_DATA=/xdg/yadm' in run.out