# 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 StringIO import StringIO
from bzrlib import errors,tests,ui
from bzrlib.tests.per_branch 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')
r2 = tree.commit('two')
r3 = 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
try:
tree.branch.set_revision_history([r1, r2b, r5])
if tree.branch.last_revision_info() != (3, r5):
# RemoteBranch silently corrects an impossible revision
# history given to set_revision_history. It can be tricked
# with set_last_revision_info though.
tree.branch.set_last_revision_info(3, r5)
except errors.NotLefthandHistory:
# Branch5 allows set_revision_history to be wrong
# Branch6 raises NotLefthandHistory, but we can force bogus stuff
# with set_last_revision_info
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=StringIO())
result.report_results(True)
self.assertContainsRe('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(
set([('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 %s' % ref)
node_distances = branch.repository.get_graph().find_lefthand_distances(
distances)
for key, distance in node_distances.iteritems():
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
|