← 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

(jelmer) Use the absolute_import feature everywhere in bzrlib,
 and add a source test to make sure it's used everywhere. (Jelmer Vernooij)

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
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.
905
922
        # Unmapping the new node will check the existing nodes to see if they
906
923
        # would fit.
907
924
        # Clear the page cache so we ensure we have to read all the children
908
 
        chk_map._page_cache.clear()
 
925
        chk_map.clear_cache()
909
926
        chkmap.unmap(('aad',))
910
927
        self.assertIsInstance(chkmap._root_node._items['aaa'], LeafNode)
911
928
        self.assertIsInstance(chkmap._root_node._items['aab'], LeafNode)
945
962
        # Now clear the page cache, and only include 2 of the children in the
946
963
        # cache
947
964
        aab_key = chkmap._root_node._items['aab']
948
 
        aab_bytes = chk_map._page_cache[aab_key]
 
965
        aab_bytes = chk_map._get_cache()[aab_key]
949
966
        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
 
967
        aac_bytes = chk_map._get_cache()[aac_key]
 
968
        chk_map.clear_cache()
 
969
        chk_map._get_cache()[aab_key] = aab_bytes
 
970
        chk_map._get_cache()[aac_key] = aac_bytes
954
971
 
955
972
        # Unmapping the new node will check the nodes from the page cache
956
973
        # first, and not have to read in 'aaa'
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)))