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

from kallithea.tests import *

from kallithea.model.repo import RepoModel

from kallithea.model.meta import Session

from kallithea.tests.fixture import Fixture

fixture = Fixture()

def _commit_div(sha, msg):

    return """<div id="C-%s" class="message">%s</div>""" % (sha, msg)

class TestCompareController(TestControllerPytest):

    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(_commit_div(cs1.raw_id, 'commit2'))

        response.mustcontain(_commit_div(cs2.raw_id, 'commit3'))

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

        response.mustcontain("""<a 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-small" 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(_commit_div(cs1.raw_id, 'commit2'))

        response.mustcontain(_commit_div(cs2.raw_id, 'commit3'))

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

        response.mustcontain("""<a 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-small" 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))

        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(_commit_div(cs1.raw_id, 'commit2'))

        response.mustcontain(_commit_div(cs2.raw_id, 'commit3'))

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

        response.mustcontain("""<a 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-small" 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))

        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(_commit_div(cs1.raw_id, 'commit2'))

        response.mustcontain(_commit_div(cs2.raw_id, 'commit3'))

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

        response.mustcontain("""<a 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-small" 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(_commit_div(cs2.raw_id, 'commit3'))

        response.mustcontain(_commit_div(cs3.raw_id, 'commit4'))

        response.mustcontain(_commit_div(cs4.raw_id, 'commit5'))

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

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

        response.mustcontain("""<a 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(_commit_div(cs3.raw_id, 'commit4'))

        response.mustcontain(_commit_div(cs4.raw_id, 'commit5'))

        response.mustcontain(_commit_div(cs5.raw_id, 'commit6'))

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

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

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

        ## files

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