← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_index.py

Implement add_node/add_nodes to the GraphIndexPrefixAdapter.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_index.py
756
756
 
757
757
class TestGraphIndexPrefixAdapter(TestCaseWithMemoryTransport):
758
758
 
759
 
    def make_index(self, ref_lists=1, key_elements=2, nodes=[]):
 
759
    def make_index(self, ref_lists=1, key_elements=2, nodes=[], add_callback=False):
760
760
        result = InMemoryGraphIndex(ref_lists, key_elements=key_elements)
761
761
        result.add_nodes(nodes)
762
 
        adapter = GraphIndexPrefixAdapter(result, ('prefix', ), key_elements - 1)
 
762
        if add_callback:
 
763
            add_nodes_callback=result.add_nodes
 
764
        else:
 
765
            add_nodes_callback=None
 
766
        adapter = GraphIndexPrefixAdapter(result, ('prefix', ), key_elements - 1,
 
767
            add_nodes_callback=add_nodes_callback)
763
768
        return result, adapter
764
769
 
 
770
    def test_add_node(self):
 
771
        index, adapter = self.make_index(add_callback=True)
 
772
        adapter.add_node(('key',), 'value', ((('ref',),),))
 
773
        self.assertEqual(set([(('prefix', 'key'), 'value', ((('prefix', 'ref'),),))]),
 
774
            set(index.iter_all_entries()))
 
775
 
 
776
    def test_add_nodes(self):
 
777
        index, adapter = self.make_index(add_callback=True)
 
778
        adapter.add_nodes((
 
779
            (('key',), 'value', ((('ref',),),)),
 
780
            (('key2',), 'value2', ((),)),
 
781
            ))
 
782
        self.assertEqual(set([
 
783
            (('prefix', 'key2'), 'value2', ((),)),
 
784
            (('prefix', 'key'), 'value', ((('prefix', 'ref'),),))
 
785
            ]),
 
786
            set(index.iter_all_entries()))
 
787
 
765
788
    def test_construct(self):
766
789
        index = InMemoryGraphIndex()
767
790
        adapter = GraphIndexPrefixAdapter(index, ('prefix', ), 1)