import os
import pytest
from buildstream import _yaml
from buildstream._exceptions import LoadError, LoadErrorReason
from tests.testutils.runcli import cli

# Project directory
DATA_DIR = os.path.join(
    os.path.dirname(os.path.realpath(__file__)),
    'assertion'
)


@pytest.mark.datafiles(DATA_DIR)
@pytest.mark.parametrize("target,opt_pony,opt_horsy,assertion", [
    # Test an unconditional (!) directly in the element
    ('raw-assertion.bst', 'False', 'False', 'Raw assertion boogey'),
    # Test an assertion in a conditional
    ('conditional-assertion.bst', 'True', 'False', "It's not pony time yet"),
    # Test that we get the first composited assertion
    ('ordered-assertion.bst', 'True', 'True', "It's not horsy time yet"),
])
def test_assertion_cli(cli, datafiles, target, opt_pony, opt_horsy, assertion):
    project = os.path.join(datafiles.dirname, datafiles.basename)
    result = cli.run(project=project, silent=True, args=[
        '--option', 'pony', opt_pony,
        '--option', 'horsy', opt_horsy,
        'show',
        '--deps', 'none',
        '--format', '%{vars}',
        target])

    assert result.exit_code != 0
    assert result.exception
    assert isinstance(result.exception, LoadError)
    assert result.exception.reason == LoadErrorReason.USER_ASSERTION

    # Assert that the assertion text provided by the user
    # is found in the exception text
    assert assertion in str(result.exception)