# -*- coding: UTF-8 -*-
#
# Copyright 2010-2019 The pygit2 contributors
#
# This file is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License, version 2,
# as published by the Free Software Foundation.
#
# In addition to the permissions in the GNU General Public License,
# the authors give you unlimited permission to link the compiled
# version of this file into combinations with other programs,
# and to distribute those combinations without any restriction
# coming from the use of this file.  (The General Public License
# restrictions do apply in other respects; for example, they cover
# modification of the file, and distribution when not linked into
# a combined executable.)
#
# This file is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; see the file COPYING.  If not, write to
# the Free Software Foundation, 51 Franklin Street, Fifth Floor,
# Boston, MA 02110-1301, USA.

"""Tests for branch methods."""

from __future__ import absolute_import
from __future__ import unicode_literals

import pytest

import pygit2
from . import utils

LAST_COMMIT = '2be5719152d4f82c7302b1c0932d8e5f0a4a0e98'
I18N_LAST_COMMIT = '5470a671a80ac3789f1a6a8cefbcf43ce7af0563'
ORIGIN_MASTER_COMMIT = '784855caf26449a1914d2cf62d12b9374d76ae78'
EXCLUSIVE_MASTER_COMMIT = '5ebeeebb320790caf276b9fc8b24546d63316533'
SHARED_COMMIT = '4ec4389a8068641da2d6578db0419484972284c8'


class BranchesObjectTestCase(utils.RepoTestCase):
    def test_lookup_branch_local(self):
        branch = self.repo.branches['master']
        assert branch.target.hex == LAST_COMMIT

        branch = self.repo.branches.local['i18n']
        assert branch.target.hex == I18N_LAST_COMMIT

        assert self.repo.branches.get('not-exists') is None

        with pytest.raises(KeyError): self.repo.branches['not-exists']

    def test_listall_branches(self):
        branches = sorted(self.repo.branches)
        assert branches == ['i18n', 'master']

    def test_create_branch(self):
        commit = self.repo[LAST_COMMIT]
        reference = self.repo.branches.create('version1', commit)
        assert 'version1' in self.repo.branches
        reference = self.repo.branches['version1']
        assert reference.target.hex == LAST_COMMIT

        # try to create existing reference
        with pytest.raises(ValueError):
            self.repo.branches.create('version1', commit)

        # try to create existing reference with force
        reference = self.repo.branches.create('version1', commit, True)
        assert reference.target.hex == LAST_COMMIT

    def test_delete(self):
        self.repo.branches.delete('i18n')

        assert self.repo.branches.get('i18n') is None

    def test_cant_delete_master(self):
        with pytest.raises(pygit2.GitError):
            self.repo.branches.delete('master')

    def test_branch_is_head_returns_true_if_branch_is_head(self):
        branch = self.repo.branches.get('master')
        assert branch.is_head()

    def test_branch_is_head_returns_false_if_branch_is_not_head(self):
        branch = self.repo.branches.get('i18n')
        assert not branch.is_head()

    def test_branch_rename_succeeds(self):
        new_branch = self.repo.branches['i18n'].rename('new-branch')
        assert new_branch.target.hex == I18N_LAST_COMMIT

        new_branch_2 = self.repo.branches.get('new-branch')
        assert new_branch_2.target.hex == I18N_LAST_COMMIT

    def test_branch_rename_fails_if_destination_already_exists(self):
        original_branch = self.repo.branches.get('i18n')
        with pytest.raises(ValueError): original_branch.rename('master')

    def test_branch_rename_not_fails_if_force_is_true(self):
        original_branch = self.repo.branches.get('master')
        new_branch = original_branch.rename('i18n', True)
        assert new_branch.target.hex == LAST_COMMIT

    def test_branch_rename_fails_with_invalid_names(self):
        original_branch = self.repo.branches.get('i18n')
        with pytest.raises(ValueError):
            original_branch.rename('abc@{123')

    def test_branch_name(self):
        branch = self.repo.branches.get('master')
        assert branch.branch_name == 'master'
        assert branch.name == 'refs/heads/master'

        branch = self.repo.branches.get('i18n')
        assert branch.branch_name == 'i18n'
        assert branch.name == 'refs/heads/i18n'

    def test_with_commit(self):
        branches = self.repo.branches.with_commit(EXCLUSIVE_MASTER_COMMIT)
        assert sorted(branches) == ['master']
        assert branches.get('i18n') is None
        assert branches['master'].branch_name == 'master'

        branches = self.repo.branches.with_commit(SHARED_COMMIT)
        assert sorted(branches) == ['i18n', 'master']

        branches = self.repo.branches.with_commit(LAST_COMMIT)
        assert sorted(branches) == ['master']

        branches = self.repo.branches.with_commit(self.repo[LAST_COMMIT])
        assert sorted(branches) == ['master']

        branches = self.repo.branches.remote.with_commit(LAST_COMMIT)
        assert sorted(branches) == []


class BranchesObjectEmptyRepoTestCase(utils.EmptyRepoTestCase):
    def setUp(self):
        super(utils.EmptyRepoTestCase, self).setUp()

        remote = self.repo.remotes[0]
        remote.fetch()

    def test_lookup_branch_remote(self):
        branch = self.repo.branches.remote.get('origin/master')
        assert branch.target.hex == ORIGIN_MASTER_COMMIT

        assert self.repo.branches.remote.get('origin/not-exists') is None

    def test_listall_branches(self):
        branches = sorted(self.repo.branches.remote)
        assert branches == ['origin/master']

    def test_branch_remote_name(self):
        self.repo.remotes[0].fetch()
        branch = self.repo.branches.remote['origin/master']
        assert branch.remote_name == 'origin'

    def test_branch_upstream(self):
        self.repo.remotes[0].fetch()
        remote_master = self.repo.branches.remote['origin/master']
        master = self.repo.branches.create('master',
                                           self.repo[remote_master.target.hex])

        assert master.upstream is None
        master.upstream = remote_master
        assert master.upstream.branch_name == 'origin/master'

        def set_bad_upstream():
            master.upstream = 2.5

        with pytest.raises(TypeError): set_bad_upstream()

        master.upstream = None
        assert master.upstream is None

    def test_branch_upstream_name(self):
        self.repo.remotes[0].fetch()
        remote_master = self.repo.branches.remote['origin/master']
        master = self.repo.branches.create('master',
                                           self.repo[remote_master.target.hex])

        master.upstream = remote_master
        assert master.upstream_name == 'refs/remotes/origin/master'


class BranchesTestCase(utils.RepoTestCase):
    def test_lookup_branch_local(self):
        branch = self.repo.lookup_branch('master')
        assert branch.target.hex == LAST_COMMIT

        branch = self.repo.lookup_branch('i18n', pygit2.GIT_BRANCH_LOCAL)
        assert branch.target.hex == I18N_LAST_COMMIT

        assert self.repo.lookup_branch('not-exists') is None

    def test_listall_branches(self):
        branches = sorted(self.repo.listall_branches())
        assert branches == ['i18n', 'master']

    def test_create_branch(self):
        commit = self.repo[LAST_COMMIT]
        reference = self.repo.create_branch('version1', commit)
        refs = self.repo.listall_branches()
        assert 'version1' in refs
        reference = self.repo.lookup_branch('version1')
        assert reference.target.hex == LAST_COMMIT

        # try to create existing reference
        with pytest.raises(ValueError):
            self.repo.create_branch('version1', commit)

        # try to create existing reference with force
        reference = self.repo.create_branch('version1', commit, True)
        assert reference.target.hex == LAST_COMMIT

    def test_delete(self):
        branch = self.repo.lookup_branch('i18n')
        branch.delete()

        assert self.repo.lookup_branch('i18n') is None

    def test_cant_delete_master(self):
        branch = self.repo.lookup_branch('master')

        with pytest.raises(pygit2.GitError): branch.delete()

    def test_branch_is_head_returns_true_if_branch_is_head(self):
        branch = self.repo.lookup_branch('master')
        assert branch.is_head()

    def test_branch_is_head_returns_false_if_branch_is_not_head(self):
        branch = self.repo.lookup_branch('i18n')
        assert not branch.is_head()

    def test_branch_is_checked_out_returns_true_if_branch_is_checked_out(self):
        branch = self.repo.lookup_branch('master')
        assert branch.is_checked_out()

    def test_branch_is_checked_out_returns_false_if_branch_is_not_checked_out(self):
        branch = self.repo.lookup_branch('i18n')
        assert not branch.is_checked_out()

    def test_branch_rename_succeeds(self):
        original_branch = self.repo.lookup_branch('i18n')
        new_branch = original_branch.rename('new-branch')
        assert new_branch.target.hex == I18N_LAST_COMMIT

        new_branch_2 = self.repo.lookup_branch('new-branch')
        assert new_branch_2.target.hex == I18N_LAST_COMMIT

    def test_branch_rename_fails_if_destination_already_exists(self):
        original_branch = self.repo.lookup_branch('i18n')
        with pytest.raises(ValueError): original_branch.rename('master')

    def test_branch_rename_not_fails_if_force_is_true(self):
        original_branch = self.repo.lookup_branch('master')
        new_branch = original_branch.rename('i18n', True)
        assert new_branch.target.hex == LAST_COMMIT

    def test_branch_rename_fails_with_invalid_names(self):
        original_branch = self.repo.lookup_branch('i18n')
        with pytest.raises(ValueError):
            original_branch.rename('abc@{123')

    def test_branch_name(self):
        branch = self.repo.lookup_branch('master')
        assert branch.branch_name == 'master'
        assert branch.name == 'refs/heads/master'

        branch = self.repo.lookup_branch('i18n')
        assert branch.branch_name == 'i18n'
        assert branch.name == 'refs/heads/i18n'


class BranchesEmptyRepoTestCase(utils.EmptyRepoTestCase):
    def setUp(self):
        super(utils.EmptyRepoTestCase, self).setUp()

        remote = self.repo.remotes[0]
        remote.fetch()

    def test_lookup_branch_remote(self):
        branch = self.repo.lookup_branch('origin/master',
                                         pygit2.GIT_BRANCH_REMOTE)
        assert branch.target.hex == ORIGIN_MASTER_COMMIT

        assert self.repo.lookup_branch('origin/not-exists', pygit2.GIT_BRANCH_REMOTE) is None

    def test_listall_branches(self):
        branches = sorted(self.repo.listall_branches(pygit2.GIT_BRANCH_REMOTE))
        assert branches == ['origin/master']

    def test_branch_remote_name(self):
        self.repo.remotes[0].fetch()
        branch = self.repo.lookup_branch('origin/master',
                                         pygit2.GIT_BRANCH_REMOTE)
        assert branch.remote_name == 'origin'

    def test_branch_upstream(self):
        self.repo.remotes[0].fetch()
        remote_master = self.repo.lookup_branch('origin/master',
                                                pygit2.GIT_BRANCH_REMOTE)
        master = self.repo.create_branch('master',
                                         self.repo[remote_master.target.hex])

        assert master.upstream is None
        master.upstream = remote_master
        assert master.upstream.branch_name == 'origin/master'

        def set_bad_upstream():
            master.upstream = 2.5

        with pytest.raises(TypeError): set_bad_upstream()

        master.upstream = None
        assert master.upstream is None

    def test_branch_upstream_name(self):
        self.repo.remotes[0].fetch()
        remote_master = self.repo.lookup_branch('origin/master',
                                                pygit2.GIT_BRANCH_REMOTE)
        master = self.repo.create_branch('master',
                                         self.repo[remote_master.target.hex])

        master.upstream = remote_master
        assert master.upstream_name == 'refs/remotes/origin/master'