← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_chk_map.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_chk_map.py
1
 
# Copyright (C) 2008, 2009 Canonical Ltd
 
1
# Copyright (C) 2008, 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
467
467
        # updated key.
468
468
        self.assertEqual(new_root, chkmap._root_node._key)
469
469
 
 
470
    def test_apply_delete_to_internal_node(self):
 
471
        # applying a delta should be convert an internal root node to a leaf
 
472
        # node if the delta shrinks the map enough.
 
473
        store = self.get_chk_bytes()
 
474
        chkmap = CHKMap(store, None)
 
475
        # Add three items: 2 small enough to fit in one node, and one huge to
 
476
        # force multiple nodes.
 
477
        chkmap._root_node.set_maximum_size(100)
 
478
        chkmap.map(('small',), 'value')
 
479
        chkmap.map(('little',), 'value')
 
480
        chkmap.map(('very-big',), 'x' * 100)
 
481
        # (Check that we have constructed the scenario we want to test)
 
482
        self.assertIsInstance(chkmap._root_node, InternalNode)
 
483
        # Delete the huge item so that the map fits in one node again.
 
484
        delta = [(('very-big',), None, None)]
 
485
        chkmap.apply_delta(delta)
 
486
        self.assertCanonicalForm(chkmap)
 
487
        self.assertIsInstance(chkmap._root_node, LeafNode)
 
488
 
470
489
    def test_apply_new_keys_must_be_new(self):
471
490
        # applying a delta (None, "a", "b") to a map with 'a' in it generates
472
491
        # an error.
905
924
        # Unmapping the new node will check the existing nodes to see if they
906
925
        # would fit.
907
926
        # Clear the page cache so we ensure we have to read all the children
908
 
        chk_map._page_cache.clear()
 
927
        chk_map.clear_cache()
909
928
        chkmap.unmap(('aad',))
910
929
        self.assertIsInstance(chkmap._root_node._items['aaa'], LeafNode)
911
930
        self.assertIsInstance(chkmap._root_node._items['aab'], LeafNode)
945
964
        # Now clear the page cache, and only include 2 of the children in the
946
965
        # cache
947
966
        aab_key = chkmap._root_node._items['aab']
948
 
        aab_bytes = chk_map._page_cache[aab_key]
 
967
        aab_bytes = chk_map._get_cache()[aab_key]
949
968
        aac_key = chkmap._root_node._items['aac']
950
 
        aac_bytes = chk_map._page_cache[aac_key]
951
 
        chk_map._page_cache.clear()
952
 
        chk_map._page_cache[aab_key] = aab_bytes
953
 
        chk_map._page_cache[aac_key] = aac_bytes
 
969
        aac_bytes = chk_map._get_cache()[aac_key]
 
970
        chk_map.clear_cache()
 
971
        chk_map._get_cache()[aab_key] = aab_bytes
 
972
        chk_map._get_cache()[aac_key] = aac_bytes
954
973
 
955
974
        # Unmapping the new node will check the nodes from the page cache
956
975
        # first, and not have to read in 'aaa'