import pytest
from calculate.templates.template_engine import TemplateEngine, Variables,\
                                                ConditionFailed, DIR, FILE
from calculate.templates.template_processor import TemplateAction


APPENDS_SET = TemplateAction().available_appends


@pytest.mark.template_engine
class TestTemplateEngine():
    def test_if_an_input_template_contains_calculate_tag_with_some_parameters__the_template_engine_object_will_collect_its_parameters(self):
        input_template = '''{% calculate name = 'filename', path = '/etc/path', force %}'''
        template_engine = TemplateEngine(appends_set=APPENDS_SET)
        template_engine.process_template_from_string(input_template, DIR)

        output_parameters = template_engine.parameters
        assert (output_parameters.name == 'filename' and
                output_parameters.path == '/etc/path' and
                output_parameters.force)

    def test_if_an_input_template_binded_with_datavars_module__variables_available_in_a_template(self):
        input_template = '''{% calculate name = vars.var_1, path = vars.var_2, autoupdate %}'''
        datavars_module = Variables({'vars':
                                     Variables({'var_1': 'filename',
                                                'var_2': '/etc/path'})})
        template_engine = TemplateEngine(appends_set=APPENDS_SET,
                                         datavars_module=datavars_module)
        template_engine.process_template_from_string(input_template, FILE)

        output_parameters = template_engine.parameters
        assert (output_parameters.name == 'filename' and
                output_parameters.path == '/etc/path' and
                output_parameters.autoupdate)

    def test_if_an_input_template_contains_env_parameter_in_which_module_name_is_assigned__the_variables_from_this_module_can_be_used_in_template_without_determining_of_their_module(self):
        input_template = '''{% calculate name = vars.var_1, env = 'other_vars', path = var_3 %}'''
        datavars_module = Variables({'vars':
                                     Variables({'var_1': 'filename',
                                                'var_2': '/etc/path'}),
                                     'other_vars':
                                     Variables({'var_3': '/etc/other_path'})})
        template_engine = TemplateEngine(appends_set=APPENDS_SET,
                                         datavars_module=datavars_module)
        template_engine.process_template_from_string(input_template, DIR)

        output_parameters = template_engine.parameters
        assert (output_parameters.name == 'filename' and
                output_parameters.path == '/etc/other_path' and
                output_parameters.env == {'other_vars'})

    def test_if_an_input_template_contains_condition_and_it_is_True__the_template_engine_object_will_be_initialized_without_any_exceptions(self):
        input_template = '''{% calculate vars.var_1 < vars.var_2 or (not (var_3 == 'required status') and vars.var_4), env = 'vars' %}'''
        datavars_module = Variables({'vars':
                                     Variables({'var_1': 12,
                                                'var_2': 1.2,
                                                'var_3': 'unrequired status',
                                                'var_4': True})})
        template_engine = TemplateEngine(appends_set=APPENDS_SET,
                                         datavars_module=datavars_module)

        try:
            template_engine.process_template_from_string(input_template, FILE)
        except ConditionFailed:
            pytest.fail('Unexpected ConditionFailed exception.')

    def test_if_an_input_template_contains_several_conditions_and_it_is_False__the_template_engine_raises_ConditionFailed_exception(self):
        input_template = '''{% calculate name = vars.var_1 %}
                            {% calculate env = 'other_vars'%}
                            {% calculate var_4 < var_5 %}
                            {% calculate path = var_3, var_6 == 'value' %}'''

        datavars_module = Variables({'vars':
                                     Variables({'var_1': 'filename',
                                                'var_2': '/etc/path'}),
                                     'other_vars':
                                     Variables({'var_3': '/etc/other_path',
                                                'var_4': 12, 'var_5': 1.2,
                                                'var_6': 'value'})})
        template_engine = TemplateEngine(appends_set=APPENDS_SET,
                                         datavars_module=datavars_module)

        with pytest.raises(ConditionFailed):
            template_engine.process_template_from_string(input_template, DIR)

    def test_if_an_input_template_contains_several_calculate_tags__the_template_engine_will_parse_them_all_and_will_contain_all_parameters_and_result_of_all_conditions(self):
        input_template = '''{% calculate name = vars.var_1 %}
                            {% calculate env = 'other_vars, vars' %}
                            {% calculate var_4 > var_5 %}
                            {% calculate path = var_3, var_6 == 'value' %}'''
        datavars_module = Variables({'vars':
                                     Variables({'var_1': 'filename',
                                                'var_2': '/etc/path'}),
                                     'other_vars':
                                     Variables({'var_3': '/etc/other_path',
                                                'var_4': 12, 'var_5': 1.2,
                                                'var_6': 'value'})})
        template_engine = TemplateEngine(appends_set=APPENDS_SET,
                                         datavars_module=datavars_module)
        template_engine.process_template_from_string(input_template, FILE)

        output_parameters = template_engine.parameters
        assert (output_parameters.name == 'filename' and
                output_parameters.path == '/etc/other_path' and
                output_parameters.env == {'other_vars', 'vars'})

    def test_if_an_input_template_contains_variables_in_its_text__the_rendered_text_will_contain_values_of_this_variables(self):
        input_template = '''{% calculate name = 'filename', force -%}
        parameter_1 = {{ vars_2.var_3 }}
        {% calculate env = 'vars_1' -%}
        parameter_2 = {{ vars_2.var_4 }}
        parameter_3 = {{ var_1 }}'''
        output_text = '''parameter_1 = value_1
        parameter_2 = value_2
        parameter_3 = 12'''
        datavars_module = Variables({'vars_1':
                                     Variables({'var_1': 12,
                                                'var_2': 1.2}),
                                     'vars_2':
                                     Variables({'var_3': 'value_1',
                                                'var_4': 'value_2'})})
        template_engine = TemplateEngine(appends_set=APPENDS_SET,
                                         datavars_module=datavars_module)
        template_engine.process_template_from_string(input_template, DIR)

        text = template_engine.template_text
        assert text == output_text

    def test_if_value_of_variable_is_set_using_save_tag__the_new_value_of_variable_can_be_used_in_template(self):
        input_template = '''{% calculate name = 'filename', force -%}
        {% save vars_1.var_3 = 13 -%}
        parameter = {{ vars_1.var_3 - 1 + vars_1.var_2 * 2 }}
        {% save.session vars_1.var_4 = True -%}
        bool parameter = {{ vars_1.var_4 }}'''
        output_text = '''parameter = 14.4
        bool parameter = True'''
        datavars_module = Variables({'vars_1':
                                     Variables({'var_1': 12,
                                                'var_2': 1.2})})
        template_engine = TemplateEngine(appends_set=APPENDS_SET,
                                         datavars_module=datavars_module)
        template_engine.process_template_from_string(input_template, FILE)

        text = template_engine.template_text
        assert text == output_text

    def test_if_an_input_template_contains_pkg_function_with_existing_package_as_its_argument__it_works_correctly_and_pkg_function_returns_version_value(self):
        input_template = '''{% calculate name = 'filename', force -%}
        {% if pkg('dev-lang/python') > 2.7 -%}
        pkg() works correctly.
        {%- else -%}
        pkg() does not work correctly.
        {%- endif -%}'''
        output_text = 'pkg() works correctly.'
        template_engine = TemplateEngine(appends_set=APPENDS_SET)
        template_engine.process_template_from_string(input_template, FILE)

        text = template_engine.template_text
        assert text == output_text

    def test_if_an_input_template_contains_pkg_function_with_package_that_does_not_exist_in_its_argument__it_works_correctly_and_pkg_function_returns_empty_version_value(self):
        input_template = '''{% calculate name = 'filename', force -%}
        {% if pkg('category/name') == Version() -%}
        pkg() works correctly.
        {%- else -%}
        pkg() does not work correctly.
        {%- endif -%}'''
        output_text = 'pkg() works correctly.'
        template_engine = TemplateEngine(appends_set=APPENDS_SET)
        template_engine.process_template_from_string(input_template, FILE)

        text = template_engine.template_text
        assert text == output_text

    def test_if_an_input_template_contains_pkg_function_without_any_arguments_but_with_package_parameter_in_calculate_tag__the_pkg_function_returns_version_value_of_the_package_from_package_parameter(self):
        input_template = '''{% calculate name = 'filename', package='dev-lang/python', force -%}
        {% if pkg() > 2.7 -%}
        pkg() works correctly.
        {%- else -%}
        pkg() does not work correctly.
        {%- endif -%}'''
        output_text = 'pkg() works correctly.'
        template_engine = TemplateEngine(appends_set=APPENDS_SET)
        template_engine.process_template_from_string(input_template, FILE)

        text = template_engine.template_text
        assert text == output_text

    def test_if_an_input_template_contains_pkg_function_without_any_arguments_and_without_package_parameter_in_calculate_tag__the_pkg_function_returns_empty_version_value(self):
        input_template = '''{% calculate name = 'filename', force -%}
        {% if pkg() == Version() -%}
        pkg() works correctly.
        {%- else -%}
        pkg() does not work correctly.
        {%- endif -%}'''
        output_text = 'pkg() works correctly.'
        template_engine = TemplateEngine(appends_set=APPENDS_SET)
        template_engine.process_template_from_string(input_template, FILE)

        text = template_engine.template_text
        assert text == output_text

    def test_if_an_input_template_calculate_tag_contains_pkg_function_with_an_existing_package_in_its_argument__the_pkg_function_returns_version_value_of_the_package_from_package_parameter_without_any_exceptions(self):
        input_template = '''{% calculate name = 'filename', package='dev-lang/python', force -%}
        {% calculate pkg() > 2.7 -%}'''

        try:
            template_engine = TemplateEngine(appends_set=APPENDS_SET)
            template_engine.process_template_from_string(input_template, FILE)
        except ConditionFailed:
            pytest.fail('Unexpected ConditionFailed exception.')