~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_chk_map.py

  • Committer: Jelmer Vernooij
  • Date: 2011-09-26 15:21:01 UTC
  • mfrom: (6165.2.3 avoid-revision-history)
  • mto: This revision was merged to the branch mainline in revision 6216.
  • Revision ID: jelmer@samba.org-20110926152101-afcxw1hikybyivfd
merge avoid-revision-history.

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
"""Tests for maps built on a CHK versionedfiles facility."""
18
18
 
19
 
from itertools import izip
20
 
 
21
19
from bzrlib import (
22
20
    chk_map,
23
21
    errors,
226
224
            "      ('ddd',) 'initial ddd content'\n",
227
225
            c_map._dump_tree())
228
226
 
229
 
    def test_one_deep_map_16(self):
 
227
    def test_root_only_aaa_ddd_16(self):
230
228
        c_map = self.make_root_only_aaa_ddd_map(
231
229
                search_key_func=chk_map._search_key_16)
232
230
        # We use 'aaa' and 'ddd' because they happen to map to 'F' when using
467
465
        # updated key.
468
466
        self.assertEqual(new_root, chkmap._root_node._key)
469
467
 
 
468
    def test_apply_delete_to_internal_node(self):
 
469
        # applying a delta should be convert an internal root node to a leaf
 
470
        # node if the delta shrinks the map enough.
 
471
        store = self.get_chk_bytes()
 
472
        chkmap = CHKMap(store, None)
 
473
        # Add three items: 2 small enough to fit in one node, and one huge to
 
474
        # force multiple nodes.
 
475
        chkmap._root_node.set_maximum_size(100)
 
476
        chkmap.map(('small',), 'value')
 
477
        chkmap.map(('little',), 'value')
 
478
        chkmap.map(('very-big',), 'x' * 100)
 
479
        # (Check that we have constructed the scenario we want to test)
 
480
        self.assertIsInstance(chkmap._root_node, InternalNode)
 
481
        # Delete the huge item so that the map fits in one node again.
 
482
        delta = [(('very-big',), None, None)]
 
483
        chkmap.apply_delta(delta)
 
484
        self.assertCanonicalForm(chkmap)
 
485
        self.assertIsInstance(chkmap._root_node, LeafNode)
 
486
 
470
487
    def test_apply_new_keys_must_be_new(self):
471
488
        # applying a delta (None, "a", "b") to a map with 'a' in it generates
472
489
        # an error.
1091
1108
        basis_get = basis._store.get_record_stream
1092
1109
        def get_record_stream(keys, order, fulltext):
1093
1110
            if ('sha1:1adf7c0d1b9140ab5f33bb64c6275fa78b1580b7',) in keys:
1094
 
                self.fail("'aaa' pointer was followed %r" % keys)
 
1111
                raise AssertionError("'aaa' pointer was followed %r" % keys)
1095
1112
            return basis_get(keys, order, fulltext)
1096
1113
        basis._store.get_record_stream = get_record_stream
1097
1114
        result = sorted(list(target.iter_changes(basis)))