# Copyright (C) 2008 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program 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; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

"""Tests for branch implementations - test check() functionality."""

from io import BytesIO

from ... import tests, ui
from . import TestCaseWithBranch


class TestBranchCheck(TestCaseWithBranch):
    def test_check_detects_invalid_revhistory(self):
        # Different formats have different ways of handling invalid revision
        # histories, so the setup portion is customized
        tree = self.make_branch_and_tree("test")
        r1 = tree.commit("one")
        tree.commit("two")
        tree.commit("three")
        r4 = tree.commit("four")
        # create an alternate branch
        tree.set_parent_ids([r1])
        tree.branch.set_last_revision_info(1, r1)
        r2b = tree.commit("two-b")

        # now go back and merge the commit
        tree.set_parent_ids([r4, r2b])
        tree.branch.set_last_revision_info(4, r4)

        r5 = tree.commit("five")
        # Now, try to set an invalid history
        if getattr(tree.branch, "_set_revision_history", None) is not None:
            tree.branch._set_revision_history([r1, r2b, r5])
        else:
            tree.branch.set_last_revision_info(3, r5)

        tree.lock_read()
        self.addCleanup(tree.unlock)
        refs = self.make_refs(tree.branch)
        result = tree.branch.check(refs)
        ui.ui_factory = tests.TestUIFactory(stdout=BytesIO())
        result.report_results(True)
        self.assertContainsRe(
            b"revno does not match len", ui.ui_factory.stdout.getvalue()
        )

    def test_check_branch_report_results(self):
        """Checking a branch produces results which can be printed."""
        branch = self.make_branch(".")
        branch.lock_read()
        self.addCleanup(branch.unlock)
        result = branch.check(self.make_refs(branch))
        # reports results through logging
        result.report_results(verbose=True)
        result.report_results(verbose=False)

    def test__get_check_refs(self):
        tree = self.make_branch_and_tree(".")
        revid = tree.commit("foo")
        self.assertEqual(
            {("revision-existence", revid), ("lefthand-distance", revid)},
            set(tree.branch._get_check_refs()),
        )

    def make_refs(self, branch):
        needed_refs = branch._get_check_refs()
        refs = {}
        distances = set()
        existences = set()
        for ref in needed_refs:
            kind, value = ref
            if kind == "lefthand-distance":
                distances.add(value)
            elif kind == "revision-existence":
                existences.add(value)
            else:
                raise AssertionError("unknown ref kind for ref {}".format(ref))
        node_distances = branch.repository.get_graph().find_lefthand_distances(
            distances
        )
        for key, distance in node_distances.items():
            refs[("lefthand-distance", key)] = distance
            if key in existences and distance > 0:
                refs[("revision-existence", key)] = True
                existences.remove(key)
        parent_map = branch.repository.get_graph().get_parent_map(existences)
        for key in parent_map:
            refs[("revision-existence", key)] = True
            existences.remove(key)
        for key in existences:
            refs[("revision-existence", key)] = False
        return refs
