← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2009-08-06 02:23:37 UTC
  • mfrom: (4332.3.36 check)
  • Revision ID: pqm@pqm.ubuntu.com-20090806022337-7c2oni07fsjq6gun
(robertc) Partial overhaul of check to do less duplicate work.
        (Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/per_repository/test_check.py
36
36
        tree = self.make_branch_and_tree('.')
37
37
        self.build_tree(['foo'])
38
38
        tree.smart_add(['.'])
39
 
        tree.commit('1')
 
39
        revid1 = tree.commit('1')
40
40
        self.build_tree(['bar'])
41
41
        tree.smart_add(['.'])
42
 
        tree.commit('2')
43
 
        # XXX: check requires a non-empty revision IDs list, but it ignores the
44
 
        # contents of it!
45
 
        check_object = tree.branch.repository.check(['ignored'])
 
42
        revid2 = tree.commit('2')
 
43
        check_object = tree.branch.repository.check([revid1, revid2])
46
44
        check_object.report_results(verbose=True)
47
45
        log = self._get_log(keep_log_file=True)
48
 
        self.assertContainsRe(
49
 
            log,
50
 
            "0 unreferenced text versions")
 
46
        self.assertContainsRe(log, "0 unreferenced text versions")
51
47
 
52
48
 
53
49
class TestFindInconsistentRevisionParents(TestCaseWithBrokenRevisionIndex):
100
96
            "revision-id has wrong parents in index: "
101
97
            r"\('incorrect-parent',\) should be \(\)")
102
98
 
 
99
 
 
100
class TestCallbacks(TestCaseWithRepository):
 
101
 
 
102
    def test_callback_tree_and_branch(self):
 
103
        # use a real tree to get actual refs that will work
 
104
        tree = self.make_branch_and_tree('foo')
 
105
        revid = tree.commit('foo')
 
106
        tree.lock_read()
 
107
        self.addCleanup(tree.unlock)
 
108
        needed_refs = {}
 
109
        for ref in tree._get_check_refs():
 
110
            needed_refs.setdefault(ref, []).append(tree)
 
111
        for ref in tree.branch._get_check_refs():
 
112
            needed_refs.setdefault(ref, []).append(tree.branch)
 
113
        self.tree_check = tree._check
 
114
        self.branch_check = tree.branch.check
 
115
        tree._check = self.tree_callback
 
116
        tree.branch.check = self.branch_callback
 
117
        self.callbacks = []
 
118
        tree.branch.repository.check([revid], callback_refs=needed_refs)
 
119
        self.assertNotEqual([], self.callbacks)
 
120
 
 
121
    def tree_callback(self, refs):
 
122
        self.callbacks.append(('tree', refs))
 
123
        return self.tree_check(refs)
 
124
 
 
125
    def branch_callback(self, refs):
 
126
        self.callbacks.append(('branch', refs))
 
127
        return self.branch_check(refs)