# -*- coding: utf-8 -*-

from kallithea.tests.base import *

from kallithea.model.repo import RepoModel

from kallithea.model.meta import Session

from kallithea.tests.fixture import Fixture

fixture = Fixture()

class TestCompareController(TestController):

    def setup_method(self, method):

        self.r1_id = None

        self.r2_id = None

    def teardown_method(self, method):

        if self.r2_id:

            RepoModel().delete(self.r2_id)

        if self.r1_id:

            RepoModel().delete(self.r1_id)

        Session().commit()

        Session.remove()

    def test_compare_forks_on_branch_extra_commits_hg(self):

        self.log_user()

        repo1 = fixture.create_repo(u'one', repo_type='hg',

                                    repo_description='diff-test',

                                    cur_user=TEST_USER_ADMIN_LOGIN)

        self.r1_id = repo1.repo_id

        #commit something !

        cs0 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\n', message='commit1', vcs_type='hg',

                parent=None, newfile=True)

        #fork this repo

        repo2 = fixture.create_fork(u'one', u'one-fork')

        self.r2_id = repo2.repo_id

        #add two extra commit into fork

        cs1 = fixture.commit_change(repo2.repo_name, filename='file1',

                content='line1\nline2\n', message='commit2', vcs_type='hg',

                parent=cs0)

        cs2 = fixture.commit_change(repo2.repo_name, filename='file1',

                content='line1\nline2\nline3\n', message='commit3',

                vcs_type='hg', parent=cs1)

        rev1 = 'default'

        rev2 = 'default'

        response = self.app.get(url('compare_url',

                                    repo_name=repo1.repo_name,

                                    org_ref_type="branch",

                                    org_ref_name=rev2,

                                    other_repo=repo2.repo_name,

                                    other_ref_type="branch",

                                    other_ref_name=rev1,

                                    merge='1',))

        response.mustcontain('%s@%s' % (repo1.repo_name, rev2))

        response.mustcontain('%s@%s' % (repo2.repo_name, rev1))

        response.mustcontain("""Showing 2 commits""")

        response.mustcontain("""1 file changed with 2 insertions and 0 deletions""")

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r1:%s</a>""" % (repo2.repo_name, cs1.raw_id, cs1.short_id))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r2:%s</a>""" % (repo2.repo_name, cs2.raw_id, cs2.short_id))

        ## files

        response.mustcontain("""<a href="#C--826e8142e6ba">file1</a>""")

        #swap

        response.mustcontain("""<a class="btn btn-default btn-sm" href="/%s/compare/branch@%s...branch@%s?other_repo=%s&amp;merge=True"><i class="icon-arrows-cw"></i> Swap</a>""" % (repo2.repo_name, rev1, rev2, repo1.repo_name))

    def test_compare_forks_on_branch_extra_commits_git(self):

        self.log_user()

        repo1 = fixture.create_repo(u'one-git', repo_type='git',

                                    repo_description='diff-test',

                                    cur_user=TEST_USER_ADMIN_LOGIN)

        self.r1_id = repo1.repo_id

        #commit something !

        cs0 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\n', message='commit1', vcs_type='git',

                parent=None, newfile=True)

        #fork this repo

        repo2 = fixture.create_fork(u'one-git', u'one-git-fork')

        self.r2_id = repo2.repo_id

        #add two extra commit into fork

        cs1 = fixture.commit_change(repo2.repo_name, filename='file1',

                content='line1\nline2\n', message='commit2', vcs_type='git',

                parent=cs0)

        cs2 = fixture.commit_change(repo2.repo_name, filename='file1',

                content='line1\nline2\nline3\n', message='commit3',

                vcs_type='git', parent=cs1)

        rev1 = 'master'

        rev2 = 'master'

        response = self.app.get(url('compare_url',

                                    repo_name=repo1.repo_name,

                                    org_ref_type="branch",

                                    org_ref_name=rev2,

                                    other_repo=repo2.repo_name,

                                    other_ref_type="branch",

                                    other_ref_name=rev1,

                                    merge='1',))

        response.mustcontain('%s@%s' % (repo1.repo_name, rev2))

        response.mustcontain('%s@%s' % (repo2.repo_name, rev1))

        response.mustcontain("""Showing 2 commits""")

        response.mustcontain("""1 file changed with 2 insertions and 0 deletions""")

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r1:%s</a>""" % (repo2.repo_name, cs1.raw_id, cs1.short_id))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r2:%s</a>""" % (repo2.repo_name, cs2.raw_id, cs2.short_id))

        ## files

        response.mustcontain("""<a href="#C--826e8142e6ba">file1</a>""")

        #swap

        response.mustcontain("""<a class="btn btn-default btn-sm" href="/%s/compare/branch@%s...branch@%s?other_repo=%s&amp;merge=True"><i class="icon-arrows-cw"></i> Swap</a>""" % (repo2.repo_name, rev1, rev2, repo1.repo_name))

    def test_compare_forks_on_branch_extra_commits_origin_has_incoming_hg(self):

        self.log_user()

        repo1 = fixture.create_repo(u'one', repo_type='hg',

                                    repo_description='diff-test',

                                    cur_user=TEST_USER_ADMIN_LOGIN)

        self.r1_id = repo1.repo_id

        #commit something !

        cs0 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\n', message='commit1', vcs_type='hg',

                parent=None, newfile=True)

        #fork this repo

        repo2 = fixture.create_fork(u'one', u'one-fork')

        self.r2_id = repo2.repo_id

        #now commit something to origin repo

        cs1_prim = fixture.commit_change(repo1.repo_name, filename='file2',

                content='line1file2\n', message='commit2', vcs_type='hg',

                parent=cs0, newfile=True)

        #add two extra commit into fork

        cs1 = fixture.commit_change(repo2.repo_name, filename='file1',

                content='line1\nline2\n', message='commit2', vcs_type='hg',

                parent=cs0)

        cs2 = fixture.commit_change(repo2.repo_name, filename='file1',

                content='line1\nline2\nline3\n', message='commit3',

                vcs_type='hg', parent=cs1)

        rev1 = 'default'

        rev2 = 'default'

        response = self.app.get(url('compare_url',

                                    repo_name=repo1.repo_name,

                                    org_ref_type="branch",

                                    org_ref_name=rev2,

                                    other_repo=repo2.repo_name,

                                    other_ref_type="branch",

                                    other_ref_name=rev1,

                                    merge='1',))

        response.mustcontain('%s@%s' % (repo1.repo_name, rev2))

        response.mustcontain('%s@%s' % (repo2.repo_name, rev1))

        response.mustcontain("""Showing 2 commits""")

        response.mustcontain("""1 file changed with 2 insertions and 0 deletions""")

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r1:%s</a>""" % (repo2.repo_name, cs1.raw_id, cs1.short_id))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r2:%s</a>""" % (repo2.repo_name, cs2.raw_id, cs2.short_id))

        ## files

        response.mustcontain("""<a href="#C--826e8142e6ba">file1</a>""")

        #swap

        response.mustcontain("""<a class="btn btn-default btn-sm" href="/%s/compare/branch@%s...branch@%s?other_repo=%s&amp;merge=True"><i class="icon-arrows-cw"></i> Swap</a>""" % (repo2.repo_name, rev1, rev2, repo1.repo_name))

    def test_compare_forks_on_branch_extra_commits_origin_has_incoming_git(self):

        self.log_user()

        repo1 = fixture.create_repo(u'one-git', repo_type='git',

                                    repo_description='diff-test',

                                    cur_user=TEST_USER_ADMIN_LOGIN)

        self.r1_id = repo1.repo_id

        #commit something !

        cs0 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\n', message='commit1', vcs_type='git',

                parent=None, newfile=True)

        #fork this repo

        repo2 = fixture.create_fork(u'one-git', u'one-git-fork')

        self.r2_id = repo2.repo_id

        #now commit something to origin repo

        cs1_prim = fixture.commit_change(repo1.repo_name, filename='file2',

                content='line1file2\n', message='commit2', vcs_type='git',

                parent=cs0, newfile=True)

        #add two extra commit into fork

        cs1 = fixture.commit_change(repo2.repo_name, filename='file1',

                content='line1\nline2\n', message='commit2', vcs_type='git',

                parent=cs0)

        cs2 = fixture.commit_change(repo2.repo_name, filename='file1',

                content='line1\nline2\nline3\n', message='commit3',

                vcs_type='git', parent=cs1)

        rev1 = 'master'

        rev2 = 'master'

        response = self.app.get(url('compare_url',

                                    repo_name=repo1.repo_name,

                                    org_ref_type="branch",

                                    org_ref_name=rev2,

                                    other_repo=repo2.repo_name,

                                    other_ref_type="branch",

                                    other_ref_name=rev1,

                                    merge='1',))

        response.mustcontain('%s@%s' % (repo1.repo_name, rev2))

        response.mustcontain('%s@%s' % (repo2.repo_name, rev1))

        response.mustcontain("""Showing 2 commits""")

        response.mustcontain("""1 file changed with 2 insertions and 0 deletions""")

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r1:%s</a>""" % (repo2.repo_name, cs1.raw_id, cs1.short_id))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r2:%s</a>""" % (repo2.repo_name, cs2.raw_id, cs2.short_id))

        ## files

        response.mustcontain("""<a href="#C--826e8142e6ba">file1</a>""")

        #swap

        response.mustcontain("""<a class="btn btn-default btn-sm" href="/%s/compare/branch@%s...branch@%s?other_repo=%s&amp;merge=True"><i class="icon-arrows-cw"></i> Swap</a>""" % (repo2.repo_name, rev1, rev2, repo1.repo_name))

    def test_compare_cherry_pick_changesets_from_bottom(self):

#        repo1:

#            cs0:

#            cs1:

#        repo1-fork- in which we will cherry pick bottom changesets

#            cs0:

#            cs1:

#            cs2: x

#            cs3: x

#            cs4: x

#            cs5:

        #make repo1, and cs1+cs2

        self.log_user()

        repo1 = fixture.create_repo(u'repo1', repo_type='hg',

                                    repo_description='diff-test',

                                    cur_user=TEST_USER_ADMIN_LOGIN)

        self.r1_id = repo1.repo_id

        #commit something !

        cs0 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\n', message='commit1', vcs_type='hg',

                parent=None, newfile=True)

        cs1 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\n', message='commit2', vcs_type='hg',

                parent=cs0)

        #fork this repo

        repo2 = fixture.create_fork(u'repo1', u'repo1-fork')

        self.r2_id = repo2.repo_id

        #now make cs3-6

        cs2 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\nline3\n', message='commit3',

                vcs_type='hg', parent=cs1)

        cs3 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\nline3\nline4\n', message='commit4',

                vcs_type='hg', parent=cs2)

        cs4 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\nline3\nline4\nline5\n',

                message='commit5', vcs_type='hg', parent=cs3)

        cs5 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\nline3\nline4\nline5\nline6\n',

                message='commit6', vcs_type='hg', parent=cs4)

        response = self.app.get(url('compare_url',

                                    repo_name=repo2.repo_name,

                                    org_ref_type="rev",

                                    org_ref_name=cs1.short_id,  # parent of cs2, in repo2

                                    other_repo=repo1.repo_name,

                                    other_ref_type="rev",

                                    other_ref_name=cs4.short_id,

                                    merge='True',

                                    ))

        response.mustcontain('%s@%s' % (repo2.repo_name, cs1.short_id))

        response.mustcontain('%s@%s' % (repo1.repo_name, cs4.short_id))

        response.mustcontain("""Showing 3 commits""")

        response.mustcontain("""1 file changed with 3 insertions and 0 deletions""")

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r2:%s</a>""" % (repo1.repo_name, cs2.raw_id, cs2.short_id))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r3:%s</a>""" % (repo1.repo_name, cs3.raw_id, cs3.short_id))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r4:%s</a>""" % (repo1.repo_name, cs4.raw_id, cs4.short_id))

        ## files

        response.mustcontain("""#C--826e8142e6ba">file1</a>""")

    def test_compare_cherry_pick_changesets_from_top(self):

#        repo1:

#            cs0:

#            cs1:

#        repo1-fork- in which we will cherry pick bottom changesets

#            cs0:

#            cs1:

#            cs2:

#            cs3: x

#            cs4: x

#            cs5: x

#

        #make repo1, and cs1+cs2

        self.log_user()

        repo1 = fixture.create_repo(u'repo1', repo_type='hg',

                                    repo_description='diff-test',

                                    cur_user=TEST_USER_ADMIN_LOGIN)

        self.r1_id = repo1.repo_id

        #commit something !

        cs0 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\n', message='commit1', vcs_type='hg',

                parent=None, newfile=True)

        cs1 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\n', message='commit2', vcs_type='hg',

                parent=cs0)

        #fork this repo

        repo2 = fixture.create_fork(u'repo1', u'repo1-fork')

        self.r2_id = repo2.repo_id

        #now make cs3-6

        cs2 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\nline3\n', message='commit3',

                vcs_type='hg', parent=cs1)

        cs3 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\nline3\nline4\n', message='commit4',

                vcs_type='hg', parent=cs2)

        cs4 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\nline3\nline4\nline5\n',

                message='commit5', vcs_type='hg', parent=cs3)

        cs5 = fixture.commit_change(repo1.repo_name, filename='file1',

                content='line1\nline2\nline3\nline4\nline5\nline6\n',

                message='commit6', vcs_type='hg', parent=cs4)

        response = self.app.get(url('compare_url',

                                    repo_name=repo1.repo_name,

                                    org_ref_type="rev",

                                    org_ref_name=cs2.short_id, # parent of cs3, not in repo2

                                    other_ref_type="rev",

                                    other_ref_name=cs5.short_id,

                                    merge='1',))

        response.mustcontain('%s@%s' % (repo1.repo_name, cs2.short_id))

        response.mustcontain('%s@%s' % (repo1.repo_name, cs5.short_id))

        response.mustcontain("""Showing 3 commits""")

        response.mustcontain("""1 file changed with 3 insertions and 0 deletions""")

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r3:%s</a>""" % (repo1.repo_name, cs3.raw_id, cs3.short_id))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r4:%s</a>""" % (repo1.repo_name, cs4.raw_id, cs4.short_id))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/%s">r5:%s</a>""" % (repo1.repo_name, cs5.raw_id, cs5.short_id))

        ## files

        response.mustcontain("""#C--826e8142e6ba">file1</a>""")

    def test_compare_cherry_pick_changeset_mixed_branches(self):

        #TODO: write this

        assert 1

    def test_compare_remote_branches_hg(self):

        self.log_user()

        repo2 = fixture.create_fork(HG_REPO, HG_FORK)

        self.r2_id = repo2.repo_id

        rev1 = '56349e29c2af'

        rev2 = '7d4bc8ec6be5'

        response = self.app.get(url('compare_url',

                                    repo_name=HG_REPO,

                                    org_ref_type="rev",

                                    org_ref_name=rev1,

                                    other_ref_type="rev",

                                    other_ref_name=rev2,

                                    other_repo=HG_FORK,

                                    merge='1',))

        response.mustcontain('%s@%s' % (HG_REPO, rev1))

        response.mustcontain('%s@%s' % (HG_FORK, rev2))

        ## outgoing changesets between those revisions

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/2dda4e345facb0ccff1a191052dd1606dba6781d">r4:2dda4e345fac</a>""" % (HG_FORK))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/6fff84722075f1607a30f436523403845f84cd9e">r5:6fff84722075</a>""" % (HG_FORK))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/7d4bc8ec6be56c0f10425afb40b6fc315a4c25e7">r6:%s</a>""" % (HG_FORK, rev2))

        ## files

        response.mustcontain("""<a href="#C--9c390eb52cd6">vcs/backends/hg.py</a>""")

        response.mustcontain("""<a href="#C--41b41c1f2796">vcs/backends/__init__.py</a>""")

        response.mustcontain("""<a href="#C--2f574d260608">vcs/backends/base.py</a>""")

    def test_compare_remote_branches_git(self):

        self.log_user()

        repo2 = fixture.create_fork(GIT_REPO, GIT_FORK)

        self.r2_id = repo2.repo_id

        rev1 = '102607b09cdd60e2793929c4f90478be29f85a17'

        rev2 = 'd7e0d30fbcae12c90680eb095a4f5f02505ce501'

        response = self.app.get(url('compare_url',

                                    repo_name=GIT_REPO,

                                    org_ref_type="rev",

                                    org_ref_name=rev1,

                                    other_ref_type="rev",

                                    other_ref_name=rev2,

                                    other_repo=GIT_FORK,

                                    merge='1',))

        response.mustcontain('%s@%s' % (GIT_REPO, rev1))

        response.mustcontain('%s@%s' % (GIT_FORK, rev2))

        ## outgoing changesets between those revisions

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/49d3fd156b6f7db46313fac355dca1a0b94a0017">r4:49d3fd156b6f</a>""" % (GIT_FORK))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/2d1028c054665b962fa3d307adfc923ddd528038">r5:2d1028c05466</a>""" % (GIT_FORK))

        response.mustcontain("""<a class="changeset_hash" href="/%s/changeset/d7e0d30fbcae12c90680eb095a4f5f02505ce501">r6:%s</a>""" % (GIT_FORK, rev2[:12]))

        ## files

        response.mustcontain("""<a href="#C--9c390eb52cd6">vcs/backends/hg.py</a>""")

        response.mustcontain("""<a href="#C--41b41c1f2796">vcs/backends/__init__.py</a>""")

        response.mustcontain("""<a href="#C--2f574d260608">vcs/backends/base.py</a>""")

    def test_org_repo_new_commits_after_forking_simple_diff_hg(self):

        self.log_user()

        repo1 = fixture.create_repo(u'one', repo_type='hg',

                                    repo_description='diff-test',

                                    cur_user=TEST_USER_ADMIN_LOGIN)

        self.r1_id = repo1.repo_id

        r1_name = repo1.repo_name

        cs0 = fixture.commit_change(repo=r1_name, filename='file1',

                content='line1', message='commit1', vcs_type='hg', newfile=True)

        Session().commit()

        assert repo1.scm_instance.revisions == [cs0.raw_id]

        #fork the repo1

        repo2 = fixture.create_repo(u'one-fork', repo_type='hg',

                                    repo_description='diff-test',

                                    cur_user=TEST_USER_ADMIN_LOGIN,

                                    clone_uri=repo1.repo_full_path,

                                    fork_of='one')

        Session().commit()

        assert repo2.scm_instance.revisions == [cs0.raw_id]

        self.r2_id = repo2.repo_id

        r2_name = repo2.repo_name