import os
import shutil
from unittest import TestCase

from neatlynx.git_wrapper import GitWrapperI


class TestDvcPathTest(TestCase):
    def setUp(self):
        self.test_dir = os.path.realpath('/tmp/ntx_unit_test/dvc_path')
        shutil.rmtree(self.test_dir, ignore_errors=True)
        os.makedirs(os.path.join(self.test_dir, 'data'))
        os.makedirs(os.path.join(self.test_dir, 'code', 'lib'))
        os.makedirs(os.path.join(self.test_dir, 'd1', 'd2', 'dir3', 'd4', 'dir5'))

        self.git = GitWrapperI(self.test_dir)

    def _validate_dvc_path(self, path, dvc_file_name, relative_file_name):
        self.assertEqual(path.dvc, dvc_file_name)
        self.assertEqual(path.filename, os.path.basename(dvc_file_name))
        self.assertEqual(path.relative, relative_file_name)
        self.assertEquals(path.abs[0], os.path.sep)
        self.assertTrue(path.abs.endswith(dvc_file_name))

    def basic_test(self):
        print('----------------------------------------------')
        os.chdir(self.test_dir)

        file = os.path.join('data', 'file.txt')
        path = self.git.build_dvc_path(file)
        self._validate_dvc_path(path, file, file)
        pass

    def from_dir_test(self):
        os.chdir(os.path.join(self.test_dir, 'code'))

        file_dvc_path = os.path.join('data', 'file1.txt')
        file_relative_path = os.path.join('..', file_dvc_path)

        path = self.git.build_dvc_path(file_relative_path)
        self._validate_dvc_path(path, file_dvc_path, file_relative_path)
        print('+++++++++++++++++++++++++++++')
        pass

    def from_deep_dirs_test(self):
        deep_dir = os.path.join('d1', 'd2', 'dir3')
        os.chdir(os.path.join(self.test_dir, deep_dir))

        file_dvc = os.path.join('code', 'lib', 'context_switcher_structs.asm')
        file_relative = os.path.join('..', '..', '..', file_dvc)

        path = self.git.build_dvc_path(file_relative)
        self._validate_dvc_path(path, file_dvc, file_relative)
        pass

    def go_deeper_test(self):
        deep_dir = os.path.join('d1', 'd2', 'dir3')
        os.chdir(os.path.join(self.test_dir, deep_dir))

        file_relative_path = os.path.join(deep_dir, 'd4', 'dir5', 'rawdata.tsv')
        file_dvc_path = os.path.join(deep_dir, file_relative_path)

        path = self.git.build_dvc_path(file_relative_path)
        self._validate_dvc_path(path, file_dvc_path, file_relative_path)
        pass