/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

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

  • Committer: Martin Pool
  • Date: 2009-12-14 06:06:59 UTC
  • mfrom: (4889 +trunk)
  • mto: This revision was merged to the branch mainline in revision 4891.
  • Revision ID: mbp@sourcefrog.net-20091214060659-1ucv8hpnky0cbnaj
merge trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/per_repository/test_check.py
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
18
"""Test operations that check the repository for corruption"""
19
19
 
 
20
import os
20
21
 
21
22
from bzrlib import (
 
23
    check,
 
24
    config as _mod_config,
22
25
    errors,
23
26
    inventory,
24
27
    revision as _mod_revision,
25
28
    )
26
29
from bzrlib.tests import TestNotApplicable
27
 
from bzrlib.tests.repository_implementations import TestCaseWithRepository
28
 
from bzrlib.tests.repository_implementations.helpers import (
 
30
from bzrlib.tests.per_repository import TestCaseWithRepository
 
31
from bzrlib.tests.per_repository.helpers import (
29
32
    TestCaseWithBrokenRevisionIndex,
30
33
    )
31
34
 
37
40
        tree = self.make_branch_and_tree('.')
38
41
        self.build_tree(['foo'])
39
42
        tree.smart_add(['.'])
40
 
        tree.commit('1')
 
43
        revid1 = tree.commit('1')
41
44
        self.build_tree(['bar'])
42
45
        tree.smart_add(['.'])
43
 
        tree.commit('2')
44
 
        # XXX: check requires a non-empty revision IDs list, but it ignores the
45
 
        # contents of it!
46
 
        check_object = tree.branch.repository.check(['ignored'])
47
 
        check_object.report_results(verbose=False)
 
46
        revid2 = tree.commit('2')
 
47
        check_object = tree.branch.repository.check([revid1, revid2])
 
48
        check_object.report_results(verbose=True)
48
49
        log = self._get_log(keep_log_file=True)
49
 
        self.assertContainsRe(
50
 
            log,
51
 
            "0 unreferenced text versions")
 
50
        self.assertContainsRe(log, "0 unreferenced text versions")
52
51
 
53
52
 
54
53
class TestFindInconsistentRevisionParents(TestCaseWithBrokenRevisionIndex):
101
100
            "revision-id has wrong parents in index: "
102
101
            r"\('incorrect-parent',\) should be \(\)")
103
102
 
 
103
 
 
104
class TestCallbacks(TestCaseWithRepository):
 
105
 
 
106
    def test_callback_tree_and_branch(self):
 
107
        # use a real tree to get actual refs that will work
 
108
        tree = self.make_branch_and_tree('foo')
 
109
        revid = tree.commit('foo')
 
110
        tree.lock_read()
 
111
        self.addCleanup(tree.unlock)
 
112
        needed_refs = {}
 
113
        for ref in tree._get_check_refs():
 
114
            needed_refs.setdefault(ref, []).append(tree)
 
115
        for ref in tree.branch._get_check_refs():
 
116
            needed_refs.setdefault(ref, []).append(tree.branch)
 
117
        self.tree_check = tree._check
 
118
        self.branch_check = tree.branch.check
 
119
        tree._check = self.tree_callback
 
120
        tree.branch.check = self.branch_callback
 
121
        self.callbacks = []
 
122
        tree.branch.repository.check([revid], callback_refs=needed_refs)
 
123
        self.assertNotEqual([], self.callbacks)
 
124
 
 
125
    def tree_callback(self, refs):
 
126
        self.callbacks.append(('tree', refs))
 
127
        return self.tree_check(refs)
 
128
 
 
129
    def branch_callback(self, refs):
 
130
        self.callbacks.append(('branch', refs))
 
131
        return self.branch_check(refs)
 
132
 
 
133
 
 
134
class TestCleanRepository(TestCaseWithRepository):
 
135
 
 
136
    def test_new_repo(self):
 
137
        repo = self.make_repository('foo')
 
138
        repo.lock_write()
 
139
        self.addCleanup(repo.unlock)
 
140
        config = _mod_config.Config()
 
141
        os.environ['BZR_EMAIL'] = 'foo@sample.com'
 
142
        builder = repo.get_commit_builder(None, [], config)
 
143
        list(builder.record_iter_changes(None, _mod_revision.NULL_REVISION, [
 
144
            ('TREE_ROOT', (None, ''), True, (False, True), (None, None),
 
145
            (None, ''), (None, 'directory'), (None, False))]))
 
146
        builder.finish_inventory()
 
147
        rev_id = builder.commit('first post')
 
148
        result = repo.check(None, check_repo=True)
 
149
        result.report_results(True)
 
150
        log = self._get_log(keep_log_file=True)
 
151
        self.assertFalse('Missing' in log, "Something was missing in %r" % log)