← 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: Patch Queue Manager
  • Date: 2016-04-21 05:06:57 UTC
  • mfrom: (6603.4.1 bzr)
  • Revision ID: pqm@pqm.ubuntu.com-20160421050657-ygnzfybewvudf1j9
(richard-wilbur) Use initial_comment as commit_message for lp_propose.(Shawn
 Wang) (Shawn Wang)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_chk_map.py
1
 
# Copyright (C) 2008, 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2008-2011, 2016 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,
42
40
        self.assertTrue(len(common) <= len(key))
43
41
        self.assertStartsWith(prefix, common)
44
42
        self.assertStartsWith(key, common)
45
 
        self.assertEquals(expected_common, common)
 
43
        self.assertEqual(expected_common, common)
46
44
 
47
45
    def test_common_prefix(self):
48
46
        self.assertCommonPrefix('beg', 'beg', 'begin')
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
379
377
            else:
380
378
                # Leaf nodes must have identical contents
381
379
                self.assertEqual(node_one._items, node_two._items)
382
 
        self.assertEquals([], node_two_stack)
 
380
        self.assertEqual([], node_two_stack)
383
381
 
384
382
    def assertCanonicalForm(self, chkmap):
385
383
        """Assert that the chkmap is in 'canonical' form.
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)))