1
# Copyright (C) 2009 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
"""Tests for checking of trees."""
22
from bzrlib.tests.workingtree_implementations import TestCaseWithWorkingTree
25
class TestCheck(TestCaseWithWorkingTree):
27
def test__get_check_refs_new(self):
28
tree = self.make_branch_and_tree('tree')
29
self.assertEqual(set([('trees', 'null:')]),
30
set(tree._get_check_refs()))
32
def test__get_check_refs_basis(self):
33
# with a basis, all current bzr trees cache it and so need the
34
# inventory to cross-check.
35
tree = self.make_branch_and_tree('tree')
36
revid = tree.commit('first post')
37
self.assertEqual(set([('trees', revid)]),
38
set(tree._get_check_refs()))
40
def test__check_with_refs(self):
41
# _check can be called with a dict of the things required.
42
tree = self.make_branch_and_tree('tree')
44
self.addCleanup(tree.unlock)
45
revid = tree.commit('first post')
46
needed_refs = tree._get_check_refs()
47
repo = tree.branch.repository
48
for ref in needed_refs:
52
refs[ref] = repo.revision_tree(revid)
54
self.fail('unknown ref kind')