import pytest
import os
from calculate.templates.template_processor import TemplateAction, DIR
from calculate.templates.template_engine import ParametersContainer


template_action = TemplateAction()

TEST_DIRECTORY_PATH = os.path.join(os.getcwd(), 'tests/templates/testfiles')


@pytest.mark.template_action
class TestTemplateAction:
    def test_chown_directory(self):
        pass

    def test_chmod_directory(self):
        pass

    def test_create_directory(self):
        path_to_create = os.path.join(TEST_DIRECTORY_PATH,
                                      'dir_tests/new_dir/new_subdir')
        template_action.template_parameters = TemplateParameters(
                                               {'chown':
                                                '{0}:{1}'.format(os.getuid(),
                                                                 os.getgid())},
                                               DIR)
        template_action._create_directory(target_path=path_to_create)
        assert os.path.exists(path_to_create)

    def test_link_directory(self):
        path_to_source = os.path.join(TEST_DIRECTORY_PATH,
                                      'dir_tests/new_dir/new_subdir')
        path_to_link = os.path.join(TEST_DIRECTORY_PATH,
                                    'dir_tests/new_dir/new_subdir_link')
        template_action.source = path_to_source
        template_action._link_directory(path_to_link)

        template_action.source = False
        assert os.path.exists(path_to_link) and os.path.islink(path_to_link)

    def test_remove_directory(self):
        directory_to_remove = os.path.join(TEST_DIRECTORY_PATH,
                                           'dir_tests/dir')
        link_to_remove = os.path.join(TEST_DIRECTORY_PATH,
                                      'dir_tests/new_dir/new_subdir_link')

        template_action._remove_directory(link_to_remove)
        assert not os.path.exists(link_to_remove)

        template_action._remove_directory(directory_to_remove)
        assert not os.path.exists(directory_to_remove)