import angreal
import pytest
import os
import shutil

here = os.path.dirname(__file__)

def func():
    pass

def test_generate_context():
    toml = os.path.join(here,'..',"tests/common/test_assets/test_template/angreal.toml")
    x = angreal.generate_context(toml,False)
    assert isinstance(x,dict)
    assert x.get("key_1") == "value_1"
    assert x.get("key_2") == 1

def test_render_directory():
    toml = os.path.join(here,'..',"tests/common/test_assets/test_template/angreal.toml")
    ctx = angreal.generate_context(toml,False)
    src = os.path.join(here,'..',"tests/common/test_assets/test_template")
    x = angreal.render_directory(src=src, dst="",force=False, context=ctx )
    for f in x:
        assert os.path.exists(f)

    shutil.rmtree("folder_name")

def test_render_template():
    x = angreal.render_template("Hello {{ name }}!", dict(name='world'))

    assert x == "Hello world!"

def test_angreal_get_root():
    """ test that get root returns something useful"""
    angreal.get_root()

def test_angreal_required_good():
    """test required version equal to current version"""
    angreal.required_version(f"=={angreal.__version__}")

def test_angreal_required_bad():
    """test required version can't be greater than current version"""
    with pytest.raises(EnvironmentError):
        angreal.required_version(f">{angreal.__version__}")

def test_group_init():
    """Test group initialization"""

    group = angreal.Group(name="group", about="about")
    assert group.name == "group"
    assert group.about == "about"


def test_cmd_init():
    """Test command initialization"""

    group1 = angreal.Group(name="group1", about="about1")
    group2 = angreal.Group(name="group2", about="about2")

    cmd = angreal.Command(name="sub_command", func=func, group = [group1,group2])
    assert cmd.name == "sub_command"
    assert cmd.func == func
    assert cmd.about is None
    assert cmd.long_about is None
    assert cmd.group[0].name, cmd.group[1].name is ["group1","group2"]

    cmd = angreal.Command(
        name="sub_command", func=func, about="about", long_about="long_about"
    )
    assert cmd.name == "sub_command"
    assert cmd.func == func
    assert cmd.about == "about"
    assert cmd.long_about == "long_about"
    assert cmd.group is None

    with pytest.raises(TypeError):
        angreal.Command(name="sub_command")

    with pytest.raises(TypeError):
        angreal.Command(func=func)


def test_arg_init():
    """Test arg initialization"""

    arg = angreal.Arg(name="test_arg", command_name="sub_command")
    assert arg.name == "test_arg"
    assert arg.command_name == "sub_command"
    assert arg.takes_value is True
    assert arg.default_value is None
    assert arg.require_equals is None
    assert arg.multiple_values is None
    assert arg.max_values is None
    assert arg.min_values is None
    assert arg.python_type == "str"
    assert arg.short is None
    assert arg.long is None
    assert arg.long_help is None
    assert arg.help is None
    assert arg.required is None

    arg = angreal.Arg(
        name="test_arg_2",
        command_name="sub_command",
        takes_value=True,
        default_value="thang",
        require_equals=True,
        multiple_values=True,
        max_values=3,
        min_values=4,
        python_type="str",
        short="a",
        long="arg_2",
        long_help="This is a long help message",
        help="This is a help message",
        required=True,
    )

    assert arg.name == "test_arg_2"
    assert arg.command_name == "sub_command"
    assert arg.takes_value is True
    assert arg.default_value == "thang"
    assert arg.require_equals is True
    assert arg.multiple_values is True
    assert arg.max_values == 3
    assert arg.min_values == 4
    assert arg.python_type == "str"
    assert arg.short == "a"
    assert arg.long == "arg_2"
    assert arg.long_help == "This is a long help message"
    assert arg.help == "This is a help message"
    assert arg.required is True

    with pytest.raises(TypeError):
        angreal.Arg()

    with pytest.raises(TypeError):
        angreal.Arg(name="test")

    with pytest.raises(TypeError):
        angreal.Arg(command_name="test")


def test_group():

    group1 = angreal.command_group(name="group1")
    group2 = angreal.command_group(name="group2")

    @group1()
    @angreal.command(name="test")
    def test1():
        pass

    @group1()
    @group2()
    @angreal.command(name="test")
    def test2():
        pass