~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/per_repository/test_check.py

  • Committer: Vincent Ladeuil
  • Date: 2011-05-26 20:30:53 UTC
  • mfrom: (5920 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5924.
  • Revision ID: v.ladeuil+lp@free.fr-20110526203053-hbjn6yuzwg03wnuv
MergeĀ fromĀ trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
40
40
        self.assertContainsRe(self.get_log(), "0 unreferenced text versions")
41
41
 
42
42
 
43
 
class TestCallbacks(TestCaseWithRepository):
44
 
 
45
 
    def test_callback_tree_and_branch(self):
46
 
        # use a real tree to get actual refs that will work
47
 
        tree = self.make_branch_and_tree('foo')
48
 
        revid = tree.commit('foo')
49
 
        tree.lock_read()
50
 
        self.addCleanup(tree.unlock)
51
 
        needed_refs = {}
52
 
        for ref in tree._get_check_refs():
53
 
            needed_refs.setdefault(ref, []).append(tree)
54
 
        for ref in tree.branch._get_check_refs():
55
 
            needed_refs.setdefault(ref, []).append(tree.branch)
56
 
        self.tree_check = tree._check
57
 
        self.branch_check = tree.branch.check
58
 
        tree._check = self.tree_callback
59
 
        tree.branch.check = self.branch_callback
60
 
        self.callbacks = []
61
 
        tree.branch.repository.check([revid], callback_refs=needed_refs)
62
 
        self.assertNotEqual([], self.callbacks)
63
 
 
64
 
    def tree_callback(self, refs):
65
 
        self.callbacks.append(('tree', refs))
66
 
        return self.tree_check(refs)
67
 
 
68
 
    def branch_callback(self, refs):
69
 
        self.callbacks.append(('branch', refs))
70
 
        return self.branch_check(refs)
71
 
 
72
 
 
73
43
class TestCleanRepository(TestCaseWithRepository):
74
44
 
75
45
    def test_new_repo(self):