/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/test_graph.py

Add a new repositoy method _generate_text_key_index for use by reconcile/check.

Show diffs side-by-side

added added

removed removed

Lines of Context:
145
145
        return self._real_parents_provider.get_parents(nodes)
146
146
 
147
147
 
148
 
class DictParentsProvider(object):
149
 
 
150
 
    def __init__(self, ancestry):
151
 
        self.ancestry = ancestry
152
 
 
153
 
    def __repr__(self):
154
 
        return 'DictParentsProvider(%r)' % self.ancestry
155
 
 
156
 
    def get_parents(self, revisions):
157
 
        return [self.ancestry.get(r, None) for r in revisions]
158
 
 
159
 
 
160
148
class TestGraph(TestCaseWithMemoryTransport):
161
149
 
162
150
    def make_graph(self, ancestors):
303
291
 
304
292
    def test_stacked_parents_provider(self):
305
293
 
306
 
        parents1 = DictParentsProvider({'rev2': ['rev3']})
307
 
        parents2 = DictParentsProvider({'rev1': ['rev4']})
 
294
        parents1 = _mod_graph.DictParentsProvider({'rev2': ['rev3']})
 
295
        parents2 = _mod_graph.DictParentsProvider({'rev1': ['rev4']})
308
296
        stacked = _mod_graph._StackedParentsProvider([parents1, parents2])
309
297
        self.assertEqual([['rev4',], ['rev3']],
310
298
                         stacked.get_parents(['rev1', 'rev2']))