from IPython.utils.capture import capture_output

import pytest


def test_alias_lifecycle():
    name = "test_alias1"
    cmd = 'echo "Hello"'
    am = _ip.alias_manager
    am.clear_aliases()
    am.define_alias(name, cmd)
    assert am.is_alias(name)
    assert am.retrieve_alias(name) == cmd
    assert (name, cmd) in am.aliases

    # Test running the alias
    orig_system = _ip.system
    result = []
    _ip.system = result.append
    try:
        _ip.run_cell("%{}".format(name))
        result = [c.strip() for c in result]
        assert result == [cmd]
    finally:
        _ip.system = orig_system

    # Test removing the alias
    am.undefine_alias(name)
    assert not am.is_alias(name)
    with pytest.raises(ValueError):
        am.retrieve_alias(name)
    assert (name, cmd) not in am.aliases


def test_alias_args_error():
    """Error expanding with wrong number of arguments"""
    _ip.alias_manager.define_alias("parts", "echo first %s second %s")
    # capture stderr:
    with capture_output() as cap:
        _ip.run_cell("parts 1")

    assert cap.stderr.split(":")[0] == "UsageError"


def test_alias_args_commented():
    """Check that alias correctly ignores 'commented out' args"""
    _ip.run_line_magic("alias", "commentarg echo this is %%s a commented out arg")

    with capture_output() as cap:
        _ip.run_cell("commentarg")

    # strip() is for pytest compat; testing via iptest patch IPython shell
    # in testing.globalipapp and replace the system call which messed up the
    # \r\n
    assert cap.stdout.strip() == "this is %s a commented out arg"


def test_alias_args_commented_nargs():
    """Check that alias correctly counts args, excluding those commented out"""
    am = _ip.alias_manager
    alias_name = "comargcount"
    cmd = "echo this is %%s a commented out arg and this is not %s"

    am.define_alias(alias_name, cmd)
    assert am.is_alias(alias_name)

    thealias = am.get_alias(alias_name)
    assert thealias.nargs == 1
