import os

import pytest

from kallithea.lib.utils2 import safe_str

from kallithea.lib.vcs.backends.hg import MercurialRepository, MercurialChangeset

from kallithea.lib.vcs.exceptions import RepositoryError, VCSError, NodeDoesNotExistError

from kallithea.lib.vcs.nodes import NodeKind, NodeState

from kallithea.tests.vcs.conf import TEST_HG_REPO, TEST_HG_REPO_CLONE, \

    TEST_HG_REPO_PULL, TEST_TMP_PATH

from kallithea.lib.vcs.utils.compat import unittest

class MercurialRepositoryTest(unittest.TestCase):

    def __check_for_existing_repo(self):

        if os.path.exists(TEST_HG_REPO_CLONE):

            pytest.fail('Cannot test mercurial clone repo as location %s already '

                      'exists. You should manually remove it first.'

                      % TEST_HG_REPO_CLONE)

    def setUp(self):

        self.repo = MercurialRepository(safe_str(TEST_HG_REPO))

    def test_wrong_repo_path(self):

        wrong_repo_path = os.path.join(TEST_TMP_PATH, 'errorrepo')

        self.assertRaises(RepositoryError, MercurialRepository, wrong_repo_path)

    def test_unicode_path_repo(self):

        self.assertRaises(VCSError,lambda:MercurialRepository(u'iShouldFail'))

    def test_repo_clone(self):

        self.__check_for_existing_repo()

        repo = MercurialRepository(safe_str(TEST_HG_REPO))

        repo_clone = MercurialRepository(TEST_HG_REPO_CLONE,

            src_url=TEST_HG_REPO, update_after_clone=True)

        self.assertEqual(len(repo.revisions), len(repo_clone.revisions))

        # Checking hashes of changesets should be enough

        for changeset in repo.get_changesets():

            raw_id = changeset.raw_id

            self.assertEqual(raw_id, repo_clone.get_changeset(raw_id).raw_id)

    def test_repo_clone_with_update(self):

        repo = MercurialRepository(safe_str(TEST_HG_REPO))

        repo_clone = MercurialRepository(TEST_HG_REPO_CLONE + '_w_update',

            src_url=TEST_HG_REPO, update_after_clone=True)

        self.assertEqual(len(repo.revisions), len(repo_clone.revisions))

        #check if current workdir was updated

        self.assertEqual(os.path.isfile(os.path.join(TEST_HG_REPO_CLONE \

                                                    + '_w_update',

                                                    'MANIFEST.in')), True,)

    def test_repo_clone_without_update(self):

        repo = MercurialRepository(safe_str(TEST_HG_REPO))

        repo_clone = MercurialRepository(TEST_HG_REPO_CLONE + '_wo_update',

            src_url=TEST_HG_REPO, update_after_clone=False)

        self.assertEqual(len(repo.revisions), len(repo_clone.revisions))

        self.assertEqual(os.path.isfile(os.path.join(TEST_HG_REPO_CLONE \

                                                    + '_wo_update',

                                                    'MANIFEST.in')), False,)

    def test_pull(self):

        if os.path.exists(TEST_HG_REPO_PULL):

            pytest.fail('Cannot test mercurial pull command as location %s '

                      'already exists. You should manually remove it first'

                      % TEST_HG_REPO_PULL)

        repo_new = MercurialRepository(TEST_HG_REPO_PULL, create=True)

        self.assertTrue(len(self.repo.revisions) > len(repo_new.revisions))

        repo_new.pull(self.repo.path)

        repo_new = MercurialRepository(TEST_HG_REPO_PULL)

        self.assertTrue(len(self.repo.revisions) == len(repo_new.revisions))

    def test_revisions(self):

        # there are 21 revisions at bitbucket now

        # so we can assume they would be available from now on

        subset = set(['b986218ba1c9b0d6a259fac9b050b1724ed8e545',

                 '3d8f361e72ab303da48d799ff1ac40d5ac37c67e',

                 '6cba7170863a2411822803fa77a0a264f1310b35',

                 '56349e29c2af3ac913b28bde9a2c6154436e615b',

                 '2dda4e345facb0ccff1a191052dd1606dba6781d',

                 '6fff84722075f1607a30f436523403845f84cd9e',

                 '7d4bc8ec6be56c0f10425afb40b6fc315a4c25e7',

                 '3803844fdbd3b711175fc3da9bdacfcd6d29a6fb',

                 'dc5d2c0661b61928834a785d3e64a3f80d3aad9c',

                 'be90031137367893f1c406e0a8683010fd115b79',

                 'db8e58be770518cbb2b1cdfa69146e47cd481481',

                 '84478366594b424af694a6c784cb991a16b87c21',

                 '17f8e105dddb9f339600389c6dc7175d395a535c',

                 '20a662e756499bde3095ffc9bc0643d1def2d0eb',

                 '2e319b85e70a707bba0beff866d9f9de032aa4f9',

                 '786facd2c61deb9cf91e9534735124fb8fc11842',

                 '94593d2128d38210a2fcd1aabff6dda0d6d9edf8',

                 'aa6a0de05b7612707db567078e130a6cd114a9a7',

                 'eada5a770da98ab0dd7325e29d00e0714f228d09'

                ])

        self.assertTrue(subset.issubset(set(self.repo.revisions)))

        # check if we have the proper order of revisions

        org = ['b986218ba1c9b0d6a259fac9b050b1724ed8e545',

                '3d8f361e72ab303da48d799ff1ac40d5ac37c67e',

                '6cba7170863a2411822803fa77a0a264f1310b35',

                '56349e29c2af3ac913b28bde9a2c6154436e615b',

                '2dda4e345facb0ccff1a191052dd1606dba6781d',

                '6fff84722075f1607a30f436523403845f84cd9e',

                '7d4bc8ec6be56c0f10425afb40b6fc315a4c25e7',

                '3803844fdbd3b711175fc3da9bdacfcd6d29a6fb',