~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_inv.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2009-06-03 15:02:09 UTC
  • mfrom: (4398.2.1 export-test-fix)
  • Revision ID: pqm@pqm.ubuntu.com-20090603150209-szap3popp2j8fpl3
(John Szakmeister) Fix error formatting for tar related KnowFailure
        on Mac

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007 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
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
 
from bzrlib import (
19
 
    chk_map,
20
 
    groupcompress,
21
 
    errors,
22
 
    inventory,
23
 
    osutils,
24
 
    repository,
25
 
    revision,
26
 
    tests,
27
 
    )
 
18
from bzrlib import errors, chk_map, inventory, osutils
28
19
from bzrlib.inventory import (CHKInventory, Inventory, ROOT_ID, InventoryFile,
29
20
    InventoryDirectory, InventoryEntry, TreeReference)
30
 
from bzrlib.tests import (
31
 
    TestCase,
32
 
    TestCaseWithTransport,
33
 
    condition_isinstance,
34
 
    multiply_tests,
35
 
    split_suite_by_condition,
36
 
    )
37
 
from bzrlib.tests.per_workingtree import workingtree_formats
38
 
 
39
 
 
40
 
def load_tests(standard_tests, module, loader):
41
 
    """Parameterise some inventory tests."""
42
 
    to_adapt, result = split_suite_by_condition(standard_tests,
43
 
        condition_isinstance(TestDeltaApplication))
44
 
    scenarios = [
45
 
        ('Inventory', {'apply_delta':apply_inventory_Inventory}),
46
 
        ]
47
 
    # Working tree basis delta application
48
 
    # Repository add_inv_by_delta.
49
 
    # Reduce form of the per_repository test logic - that logic needs to be
50
 
    # be able to get /just/ repositories whereas these tests are fine with
51
 
    # just creating trees.
52
 
    formats = set()
53
 
    for _, format in repository.format_registry.iteritems():
54
 
        scenarios.append((str(format.__name__), {
55
 
            'apply_delta':apply_inventory_Repository_add_inventory_by_delta,
56
 
            'format':format}))
57
 
    for format in workingtree_formats():
58
 
        scenarios.append(
59
 
            (str(format.__class__.__name__) + ".update_basis_by_delta", {
60
 
            'apply_delta':apply_inventory_WT_basis,
61
 
            'format':format}))
62
 
        scenarios.append(
63
 
            (str(format.__class__.__name__) + ".apply_inventory_delta", {
64
 
            'apply_delta':apply_inventory_WT,
65
 
            'format':format}))
66
 
    return multiply_tests(to_adapt, scenarios, result)
67
 
 
68
 
 
69
 
def create_texts_for_inv(repo, inv):
70
 
    for path, ie in inv.iter_entries():
71
 
        if ie.text_size:
72
 
            lines = ['a' * ie.text_size]
73
 
        else:
74
 
            lines = []
75
 
        repo.texts.add_lines((ie.file_id, ie.revision), [], lines)
76
 
    
77
 
def apply_inventory_Inventory(self, basis, delta):
78
 
    """Apply delta to basis and return the result.
79
 
    
80
 
    :param basis: An inventory to be used as the basis.
81
 
    :param delta: The inventory delta to apply:
82
 
    :return: An inventory resulting from the application.
83
 
    """
84
 
    basis.apply_delta(delta)
85
 
    return basis
86
 
 
87
 
 
88
 
def apply_inventory_WT(self, basis, delta):
89
 
    """Apply delta to basis and return the result.
90
 
 
91
 
    This sets the tree state to be basis, and then calls apply_inventory_delta.
92
 
    
93
 
    :param basis: An inventory to be used as the basis.
94
 
    :param delta: The inventory delta to apply:
95
 
    :return: An inventory resulting from the application.
96
 
    """
97
 
    control = self.make_bzrdir('tree', format=self.format._matchingbzrdir)
98
 
    control.create_repository()
99
 
    control.create_branch()
100
 
    tree = self.format.initialize(control)
101
 
    tree.lock_write()
102
 
    try:
103
 
        tree._write_inventory(basis)
104
 
    finally:
105
 
        tree.unlock()
106
 
    # Fresh object, reads disk again.
107
 
    tree = tree.bzrdir.open_workingtree()
108
 
    tree.lock_write()
109
 
    try:
110
 
        tree.apply_inventory_delta(delta)
111
 
    finally:
112
 
        tree.unlock()
113
 
    # reload tree - ensure we get what was written.
114
 
    tree = tree.bzrdir.open_workingtree()
115
 
    tree.lock_read()
116
 
    self.addCleanup(tree.unlock)
117
 
    # One could add 'tree._validate' here but that would cause 'early' failues 
118
 
    # as far as higher level code is concerned. Possibly adding an
119
 
    # expect_fail parameter to this function and if that is False then do a
120
 
    # validate call.
121
 
    return tree.inventory
122
 
 
123
 
 
124
 
def apply_inventory_WT_basis(self, basis, delta):
125
 
    """Apply delta to basis and return the result.
126
 
 
127
 
    This sets the parent and then calls update_basis_by_delta.
128
 
    It also puts the basis in the repository under both 'basis' and 'result' to
129
 
    allow safety checks made by the WT to succeed, and finally ensures that all
130
 
    items in the delta with a new path are present in the WT before calling
131
 
    update_basis_by_delta.
132
 
    
133
 
    :param basis: An inventory to be used as the basis.
134
 
    :param delta: The inventory delta to apply:
135
 
    :return: An inventory resulting from the application.
136
 
    """
137
 
    control = self.make_bzrdir('tree', format=self.format._matchingbzrdir)
138
 
    control.create_repository()
139
 
    control.create_branch()
140
 
    tree = self.format.initialize(control)
141
 
    tree.lock_write()
142
 
    try:
143
 
        repo = tree.branch.repository
144
 
        repo.start_write_group()
145
 
        try:
146
 
            rev = revision.Revision('basis', timestamp=0, timezone=None,
147
 
                message="", committer="foo@example.com")
148
 
            basis.revision_id = 'basis'
149
 
            create_texts_for_inv(tree.branch.repository, basis)
150
 
            repo.add_revision('basis', rev, basis)
151
 
            # Add a revision for the result, with the basis content - 
152
 
            # update_basis_by_delta doesn't check that the delta results in
153
 
            # result, and we want inconsistent deltas to get called on the
154
 
            # tree, or else the code isn't actually checked.
155
 
            rev = revision.Revision('result', timestamp=0, timezone=None,
156
 
                message="", committer="foo@example.com")
157
 
            basis.revision_id = 'result'
158
 
            repo.add_revision('result', rev, basis)
159
 
            repo.commit_write_group()
160
 
        except:
161
 
            repo.abort_write_group()
162
 
            raise
163
 
        # Set the basis state as the trees current state
164
 
        tree._write_inventory(basis)
165
 
        # This reads basis from the repo and puts it into the tree's local
166
 
        # cache, if it has one.
167
 
        tree.set_parent_ids(['basis'])
168
 
        paths = {}
169
 
        parents = set()
170
 
        for old, new, id, entry in delta:
171
 
            if None in (new, entry):
172
 
                continue
173
 
            paths[new] = (entry.file_id, entry.kind)
174
 
            parents.add(osutils.dirname(new))
175
 
        parents = osutils.minimum_path_selection(parents)
176
 
        parents.discard('')
177
 
        # Put place holders in the tree to permit adding the other entries.
178
 
        for pos, parent in enumerate(parents):
179
 
            if not tree.path2id(parent):
180
 
                # add a synthetic directory in the tree so we can can put the
181
 
                # tree0 entries in place for dirstate.
182
 
                tree.add([parent], ["id%d" % pos], ["directory"])
183
 
        if paths:
184
 
            # Many deltas may cause this mini-apply to fail, but we want to see what
185
 
            # the delta application code says, not the prep that we do to deal with 
186
 
            # limitations of dirstate's update_basis code.
187
 
            for path, (file_id, kind) in sorted(paths.items()):
188
 
                try:
189
 
                    tree.add([path], [file_id], [kind])
190
 
                except (KeyboardInterrupt, SystemExit):
191
 
                    raise
192
 
                except:
193
 
                    pass
194
 
    finally:
195
 
        tree.unlock()
196
 
    # Fresh lock, reads disk again.
197
 
    tree.lock_write()
198
 
    try:
199
 
        tree.update_basis_by_delta('result', delta)
200
 
    finally:
201
 
        tree.unlock()
202
 
    # reload tree - ensure we get what was written.
203
 
    tree = tree.bzrdir.open_workingtree()
204
 
    basis_tree = tree.basis_tree()
205
 
    basis_tree.lock_read()
206
 
    self.addCleanup(basis_tree.unlock)
207
 
    # Note, that if the tree does not have a local cache, the trick above of
208
 
    # setting the result as the basis, will come back to bite us. That said,
209
 
    # all the implementations in bzr do have a local cache.
210
 
    return basis_tree.inventory
211
 
 
212
 
 
213
 
def apply_inventory_Repository_add_inventory_by_delta(self, basis, delta):
214
 
    """Apply delta to basis and return the result.
215
 
    
216
 
    This inserts basis as a whole inventory and then uses
217
 
    add_inventory_by_delta to add delta.
218
 
 
219
 
    :param basis: An inventory to be used as the basis.
220
 
    :param delta: The inventory delta to apply:
221
 
    :return: An inventory resulting from the application.
222
 
    """
223
 
    format = self.format()
224
 
    control = self.make_bzrdir('tree', format=format._matchingbzrdir)
225
 
    repo = format.initialize(control)
226
 
    repo.lock_write()
227
 
    try:
228
 
        repo.start_write_group()
229
 
        try:
230
 
            rev = revision.Revision('basis', timestamp=0, timezone=None,
231
 
                message="", committer="foo@example.com")
232
 
            basis.revision_id = 'basis'
233
 
            create_texts_for_inv(repo, basis)
234
 
            repo.add_revision('basis', rev, basis)
235
 
            repo.commit_write_group()
236
 
        except:
237
 
            repo.abort_write_group()
238
 
            raise
239
 
    finally:
240
 
        repo.unlock()
241
 
    repo.lock_write()
242
 
    try:
243
 
        repo.start_write_group()
244
 
        try:
245
 
            inv_sha1 = repo.add_inventory_by_delta('basis', delta,
246
 
                'result', ['basis'])
247
 
        except:
248
 
            repo.abort_write_group()
249
 
            raise
250
 
        else:
251
 
            repo.commit_write_group()
252
 
    finally:
253
 
        repo.unlock()
254
 
    # Fresh lock, reads disk again.
255
 
    repo = repo.bzrdir.open_repository()
256
 
    repo.lock_read()
257
 
    self.addCleanup(repo.unlock)
258
 
    return repo.get_inventory('result')
259
 
 
260
 
 
261
 
class TestInventoryUpdates(TestCase):
262
 
 
263
 
    def test_creation_from_root_id(self):
264
 
        # iff a root id is passed to the constructor, a root directory is made
265
 
        inv = inventory.Inventory(root_id='tree-root')
266
 
        self.assertNotEqual(None, inv.root)
267
 
        self.assertEqual('tree-root', inv.root.file_id)
268
 
 
269
 
    def test_add_path_of_root(self):
270
 
        # if no root id is given at creation time, there is no root directory
271
 
        inv = inventory.Inventory(root_id=None)
272
 
        self.assertIs(None, inv.root)
273
 
        # add a root entry by adding its path
274
 
        ie = inv.add_path("", "directory", "my-root")
275
 
        ie.revision = 'test-rev'
276
 
        self.assertEqual("my-root", ie.file_id)
277
 
        self.assertIs(ie, inv.root)
278
 
 
279
 
    def test_add_path(self):
280
 
        inv = inventory.Inventory(root_id='tree_root')
281
 
        ie = inv.add_path('hello', 'file', 'hello-id')
282
 
        self.assertEqual('hello-id', ie.file_id)
283
 
        self.assertEqual('file', ie.kind)
284
 
 
285
 
    def test_copy(self):
286
 
        """Make sure copy() works and creates a deep copy."""
287
 
        inv = inventory.Inventory(root_id='some-tree-root')
288
 
        ie = inv.add_path('hello', 'file', 'hello-id')
289
 
        inv2 = inv.copy()
290
 
        inv.root.file_id = 'some-new-root'
291
 
        ie.name = 'file2'
292
 
        self.assertEqual('some-tree-root', inv2.root.file_id)
293
 
        self.assertEqual('hello', inv2['hello-id'].name)
294
 
 
295
 
    def test_copy_empty(self):
296
 
        """Make sure an empty inventory can be copied."""
297
 
        inv = inventory.Inventory(root_id=None)
298
 
        inv2 = inv.copy()
299
 
        self.assertIs(None, inv2.root)
300
 
 
301
 
    def test_copy_copies_root_revision(self):
302
 
        """Make sure the revision of the root gets copied."""
303
 
        inv = inventory.Inventory(root_id='someroot')
304
 
        inv.root.revision = 'therev'
305
 
        inv2 = inv.copy()
306
 
        self.assertEquals('someroot', inv2.root.file_id)
307
 
        self.assertEquals('therev', inv2.root.revision)
308
 
 
309
 
    def test_create_tree_reference(self):
310
 
        inv = inventory.Inventory('tree-root-123')
311
 
        inv.add(TreeReference('nested-id', 'nested', parent_id='tree-root-123',
312
 
                              revision='rev', reference_revision='rev2'))
313
 
 
314
 
    def test_error_encoding(self):
315
 
        inv = inventory.Inventory('tree-root')
316
 
        inv.add(InventoryFile('a-id', u'\u1234', 'tree-root'))
317
 
        e = self.assertRaises(errors.InconsistentDelta, inv.add,
318
 
            InventoryFile('b-id', u'\u1234', 'tree-root'))
319
 
        self.assertContainsRe(str(e), r'\\u1234')
320
 
 
321
 
    def test_add_recursive(self):
322
 
        parent = InventoryDirectory('src-id', 'src', 'tree-root')
323
 
        child = InventoryFile('hello-id', 'hello.c', 'src-id')
324
 
        parent.children[child.file_id] = child
325
 
        inv = inventory.Inventory('tree-root')
326
 
        inv.add(parent)
327
 
        self.assertEqual('src/hello.c', inv.id2path('hello-id'))
328
 
 
329
 
 
330
 
 
331
 
class TestDeltaApplication(TestCaseWithTransport):
332
 
 
333
 
    def get_empty_inventory(self, reference_inv=None):
334
 
        """Get an empty inventory.
335
 
 
336
 
        Note that tests should not depend on the revision of the root for
337
 
        setting up test conditions, as it has to be flexible to accomodate non
338
 
        rich root repositories.
339
 
 
340
 
        :param reference_inv: If not None, get the revision for the root from
341
 
            this inventory. This is useful for dealing with older repositories
342
 
            that routinely discarded the root entry data. If None, the root's
343
 
            revision is set to 'basis'.
344
 
        """
345
 
        inv = inventory.Inventory()
346
 
        if reference_inv is not None:
347
 
            inv.root.revision = reference_inv.root.revision
348
 
        else:
349
 
            inv.root.revision = 'basis'
350
 
        return inv
351
 
 
352
 
    def test_empty_delta(self):
353
 
        inv = self.get_empty_inventory()
354
 
        delta = []
355
 
        inv = self.apply_delta(self, inv, delta)
356
 
        inv2 = self.get_empty_inventory(inv)
357
 
        self.assertEqual([], inv2._make_delta(inv))
358
 
 
359
 
    def test_None_file_id(self):
360
 
        inv = self.get_empty_inventory()
361
 
        dir1 = inventory.InventoryDirectory(None, 'dir1', inv.root.file_id)
362
 
        dir1.revision = 'result'
363
 
        delta = [(None, u'dir1', None, dir1)]
364
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
365
 
            inv, delta)
366
 
 
367
 
    def test_unicode_file_id(self):
368
 
        inv = self.get_empty_inventory()
369
 
        dir1 = inventory.InventoryDirectory(u'dirid', 'dir1', inv.root.file_id)
370
 
        dir1.revision = 'result'
371
 
        delta = [(None, u'dir1', dir1.file_id, dir1)]
372
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
373
 
            inv, delta)
374
 
 
375
 
    def test_repeated_file_id(self):
376
 
        inv = self.get_empty_inventory()
377
 
        file1 = inventory.InventoryFile('id', 'path1', inv.root.file_id)
378
 
        file1.revision = 'result'
379
 
        file1.text_size = 0
380
 
        file1.text_sha1 = ""
381
 
        file2 = inventory.InventoryFile('id', 'path2', inv.root.file_id)
382
 
        file2.revision = 'result'
383
 
        file2.text_size = 0
384
 
        file2.text_sha1 = ""
385
 
        delta = [(None, u'path1', 'id', file1), (None, u'path2', 'id', file2)]
386
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
387
 
            inv, delta)
388
 
 
389
 
    def test_repeated_new_path(self):
390
 
        inv = self.get_empty_inventory()
391
 
        file1 = inventory.InventoryFile('id1', 'path', inv.root.file_id)
392
 
        file1.revision = 'result'
393
 
        file1.text_size = 0
394
 
        file1.text_sha1 = ""
395
 
        file2 = inventory.InventoryFile('id2', 'path', inv.root.file_id)
396
 
        file2.revision = 'result'
397
 
        file2.text_size = 0
398
 
        file2.text_sha1 = ""
399
 
        delta = [(None, u'path', 'id1', file1), (None, u'path', 'id2', file2)]
400
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
401
 
            inv, delta)
402
 
 
403
 
    def test_repeated_old_path(self):
404
 
        inv = self.get_empty_inventory()
405
 
        file1 = inventory.InventoryFile('id1', 'path', inv.root.file_id)
406
 
        file1.revision = 'result'
407
 
        file1.text_size = 0
408
 
        file1.text_sha1 = ""
409
 
        # We can't *create* a source inventory with the same path, but
410
 
        # a badly generated partial delta might claim the same source twice.
411
 
        # This would be buggy in two ways: the path is repeated in the delta,
412
 
        # And the path for one of the file ids doesn't match the source
413
 
        # location. Alternatively, we could have a repeated fileid, but that
414
 
        # is separately checked for.
415
 
        file2 = inventory.InventoryFile('id2', 'path2', inv.root.file_id)
416
 
        file2.revision = 'result'
417
 
        file2.text_size = 0
418
 
        file2.text_sha1 = ""
419
 
        inv.add(file1)
420
 
        inv.add(file2)
421
 
        delta = [(u'path', None, 'id1', None), (u'path', None, 'id2', None)]
422
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
423
 
            inv, delta)
424
 
 
425
 
    def test_mismatched_id_entry_id(self):
426
 
        inv = self.get_empty_inventory()
427
 
        file1 = inventory.InventoryFile('id1', 'path', inv.root.file_id)
428
 
        file1.revision = 'result'
429
 
        file1.text_size = 0
430
 
        file1.text_sha1 = ""
431
 
        delta = [(None, u'path', 'id', file1)]
432
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
433
 
            inv, delta)
434
 
 
435
 
    def test_mismatched_new_path_entry_None(self):
436
 
        inv = self.get_empty_inventory()
437
 
        delta = [(None, u'path', 'id', None)]
438
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
439
 
            inv, delta)
440
 
 
441
 
    def test_mismatched_new_path_None_entry(self):
442
 
        inv = self.get_empty_inventory()
443
 
        file1 = inventory.InventoryFile('id1', 'path', inv.root.file_id)
444
 
        file1.revision = 'result'
445
 
        file1.text_size = 0
446
 
        file1.text_sha1 = ""
447
 
        delta = [(u"path", None, 'id1', file1)]
448
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
449
 
            inv, delta)
450
 
 
451
 
    def test_parent_is_not_directory(self):
452
 
        inv = self.get_empty_inventory()
453
 
        file1 = inventory.InventoryFile('id1', 'path', inv.root.file_id)
454
 
        file1.revision = 'result'
455
 
        file1.text_size = 0
456
 
        file1.text_sha1 = ""
457
 
        file2 = inventory.InventoryFile('id2', 'path2', 'id1')
458
 
        file2.revision = 'result'
459
 
        file2.text_size = 0
460
 
        file2.text_sha1 = ""
461
 
        inv.add(file1)
462
 
        delta = [(None, u'path/path2', 'id2', file2)]
463
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
464
 
            inv, delta)
465
 
 
466
 
    def test_parent_is_missing(self):
467
 
        inv = self.get_empty_inventory()
468
 
        file2 = inventory.InventoryFile('id2', 'path2', 'missingparent')
469
 
        file2.revision = 'result'
470
 
        file2.text_size = 0
471
 
        file2.text_sha1 = ""
472
 
        delta = [(None, u'path/path2', 'id2', file2)]
473
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
474
 
            inv, delta)
475
 
 
476
 
    def test_new_parent_path_has_wrong_id(self):
477
 
        inv = self.get_empty_inventory()
478
 
        parent1 = inventory.InventoryDirectory('p-1', 'dir', inv.root.file_id)
479
 
        parent1.revision = 'result'
480
 
        parent2 = inventory.InventoryDirectory('p-2', 'dir2', inv.root.file_id)
481
 
        parent2.revision = 'result'
482
 
        file1 = inventory.InventoryFile('id', 'path', 'p-2')
483
 
        file1.revision = 'result'
484
 
        file1.text_size = 0
485
 
        file1.text_sha1 = ""
486
 
        inv.add(parent1)
487
 
        inv.add(parent2)
488
 
        # This delta claims that file1 is at dir/path, but actually its at
489
 
        # dir2/path if you follow the inventory parent structure.
490
 
        delta = [(None, u'dir/path', 'id', file1)]
491
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
492
 
            inv, delta)
493
 
 
494
 
    def test_old_parent_path_is_wrong(self):
495
 
        inv = self.get_empty_inventory()
496
 
        parent1 = inventory.InventoryDirectory('p-1', 'dir', inv.root.file_id)
497
 
        parent1.revision = 'result'
498
 
        parent2 = inventory.InventoryDirectory('p-2', 'dir2', inv.root.file_id)
499
 
        parent2.revision = 'result'
500
 
        file1 = inventory.InventoryFile('id', 'path', 'p-2')
501
 
        file1.revision = 'result'
502
 
        file1.text_size = 0
503
 
        file1.text_sha1 = ""
504
 
        inv.add(parent1)
505
 
        inv.add(parent2)
506
 
        inv.add(file1)
507
 
        # This delta claims that file1 was at dir/path, but actually it was at
508
 
        # dir2/path if you follow the inventory parent structure.
509
 
        delta = [(u'dir/path', None, 'id', None)]
510
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
511
 
            inv, delta)
512
 
 
513
 
    def test_old_parent_path_is_for_other_id(self):
514
 
        inv = self.get_empty_inventory()
515
 
        parent1 = inventory.InventoryDirectory('p-1', 'dir', inv.root.file_id)
516
 
        parent1.revision = 'result'
517
 
        parent2 = inventory.InventoryDirectory('p-2', 'dir2', inv.root.file_id)
518
 
        parent2.revision = 'result'
519
 
        file1 = inventory.InventoryFile('id', 'path', 'p-2')
520
 
        file1.revision = 'result'
521
 
        file1.text_size = 0
522
 
        file1.text_sha1 = ""
523
 
        file2 = inventory.InventoryFile('id2', 'path', 'p-1')
524
 
        file2.revision = 'result'
525
 
        file2.text_size = 0
526
 
        file2.text_sha1 = ""
527
 
        inv.add(parent1)
528
 
        inv.add(parent2)
529
 
        inv.add(file1)
530
 
        inv.add(file2)
531
 
        # This delta claims that file1 was at dir/path, but actually it was at
532
 
        # dir2/path if you follow the inventory parent structure. At dir/path
533
 
        # is another entry we should not delete.
534
 
        delta = [(u'dir/path', None, 'id', None)]
535
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
536
 
            inv, delta)
537
 
 
538
 
    def test_add_existing_id_new_path(self):
539
 
        inv = self.get_empty_inventory()
540
 
        parent1 = inventory.InventoryDirectory('p-1', 'dir1', inv.root.file_id)
541
 
        parent1.revision = 'result'
542
 
        parent2 = inventory.InventoryDirectory('p-1', 'dir2', inv.root.file_id)
543
 
        parent2.revision = 'result'
544
 
        inv.add(parent1)
545
 
        delta = [(None, u'dir2', 'p-1', parent2)]
546
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
547
 
            inv, delta)
548
 
 
549
 
    def test_add_new_id_existing_path(self):
550
 
        inv = self.get_empty_inventory()
551
 
        parent1 = inventory.InventoryDirectory('p-1', 'dir1', inv.root.file_id)
552
 
        parent1.revision = 'result'
553
 
        parent2 = inventory.InventoryDirectory('p-2', 'dir1', inv.root.file_id)
554
 
        parent2.revision = 'result'
555
 
        inv.add(parent1)
556
 
        delta = [(None, u'dir1', 'p-2', parent2)]
557
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
558
 
            inv, delta)
559
 
 
560
 
    def test_remove_dir_leaving_dangling_child(self):
561
 
        inv = self.get_empty_inventory()
562
 
        dir1 = inventory.InventoryDirectory('p-1', 'dir1', inv.root.file_id)
563
 
        dir1.revision = 'result'
564
 
        dir2 = inventory.InventoryDirectory('p-2', 'child1', 'p-1')
565
 
        dir2.revision = 'result'
566
 
        dir3 = inventory.InventoryDirectory('p-3', 'child2', 'p-1')
567
 
        dir3.revision = 'result'
568
 
        inv.add(dir1)
569
 
        inv.add(dir2)
570
 
        inv.add(dir3)
571
 
        delta = [(u'dir1', None, 'p-1', None),
572
 
            (u'dir1/child2', None, 'p-3', None)]
573
 
        self.assertRaises(errors.InconsistentDelta, self.apply_delta, self,
574
 
            inv, delta)
575
 
 
576
 
 
577
 
class TestInventory(TestCase):
578
 
 
579
 
    def test_is_root(self):
580
 
        """Ensure our root-checking code is accurate."""
581
 
        inv = inventory.Inventory('TREE_ROOT')
582
 
        self.assertTrue(inv.is_root('TREE_ROOT'))
583
 
        self.assertFalse(inv.is_root('booga'))
584
 
        inv.root.file_id = 'booga'
585
 
        self.assertFalse(inv.is_root('TREE_ROOT'))
586
 
        self.assertTrue(inv.is_root('booga'))
587
 
        # works properly even if no root is set
588
 
        inv.root = None
589
 
        self.assertFalse(inv.is_root('TREE_ROOT'))
590
 
        self.assertFalse(inv.is_root('booga'))
591
 
 
592
 
    def test_entries_for_empty_inventory(self):
593
 
        """Test that entries() will not fail for an empty inventory"""
594
 
        inv = Inventory(root_id=None)
595
 
        self.assertEqual([], inv.entries())
 
21
from bzrlib.tests import TestCase, TestCaseWithTransport
596
22
 
597
23
 
598
24
class TestInventoryEntry(TestCase):
611
37
 
612
38
    def test_dir_detect_changes(self):
613
39
        left = inventory.InventoryDirectory('123', 'hello.c', ROOT_ID)
 
40
        left.text_sha1 = 123
 
41
        left.executable = True
 
42
        left.symlink_target='foo'
614
43
        right = inventory.InventoryDirectory('123', 'hello.c', ROOT_ID)
 
44
        right.text_sha1 = 321
 
45
        right.symlink_target='bar'
615
46
        self.assertEqual((False, False), left.detect_changes(right))
616
47
        self.assertEqual((False, False), right.detect_changes(left))
617
48
 
631
62
 
632
63
    def test_symlink_detect_changes(self):
633
64
        left = inventory.InventoryLink('123', 'hello.c', ROOT_ID)
 
65
        left.text_sha1 = 123
 
66
        left.executable = True
634
67
        left.symlink_target='foo'
635
68
        right = inventory.InventoryLink('123', 'hello.c', ROOT_ID)
 
69
        right.text_sha1 = 321
636
70
        right.symlink_target='foo'
637
71
        self.assertEqual((False, False), left.detect_changes(right))
638
72
        self.assertEqual((False, False), right.detect_changes(left))
734
168
        self.assertEqual(expected_change, change)
735
169
 
736
170
 
737
 
class TestCHKInventory(tests.TestCaseWithMemoryTransport):
 
171
class TestCHKInventory(TestCaseWithTransport):
738
172
 
739
173
    def get_chk_bytes(self):
740
 
        factory = groupcompress.make_pack_factory(True, True, 1)
741
 
        trans = self.get_transport('')
742
 
        return factory(trans)
 
174
        # The easiest way to get a CHK store is a development6 repository and
 
175
        # then work with the chk_bytes attribute directly.
 
176
        repo = self.make_repository(".", format="development6-rich-root")
 
177
        repo.lock_write()
 
178
        self.addCleanup(repo.unlock)
 
179
        repo.start_write_group()
 
180
        self.addCleanup(repo.abort_write_group)
 
181
        return repo.chk_bytes
743
182
 
744
183
    def read_bytes(self, chk_bytes, key):
745
184
        stream = chk_bytes.get_record_stream([key], 'unordered', True)
858
297
        inv.root.revision = "rootrev"
859
298
        chk_bytes = self.get_chk_bytes()
860
299
        chk_inv = CHKInventory.from_inventory(chk_bytes, inv, 120)
861
 
        chk_inv.id_to_entry._ensure_root()
862
300
        self.assertEqual(120, chk_inv.id_to_entry._root_node.maximum_size)
863
 
        self.assertEqual(1, chk_inv.id_to_entry._root_node._key_width)
864
 
        p_id_basename = chk_inv.parent_id_basename_to_file_id
865
 
        p_id_basename._ensure_root()
866
 
        self.assertEqual(120, p_id_basename._root_node.maximum_size)
867
 
        self.assertEqual(2, p_id_basename._root_node._key_width)
868
301
 
869
302
    def test___iter__(self):
870
303
        inv = Inventory()
998
431
        inv.add_path("", "directory", "myrootid", None)
999
432
        inv.revision_id = "expectedid"
1000
433
        reference_inv = CHKInventory.from_inventory(chk_bytes, inv)
1001
 
        delta = [("", None, base_inv.root.file_id, None),
1002
 
            (None, "",  "myrootid", inv.root)]
 
434
        delta = [(None, "",  "myrootid", inv.root)]
1003
435
        new_inv = base_inv.create_by_apply_delta(delta, "expectedid")
1004
436
        self.assertEquals(reference_inv.root, new_inv.root)
1005
437
 
1022
454
        # new_inv should be the same as reference_inv.
1023
455
        self.assertEqual(reference_inv.revision_id, new_inv.revision_id)
1024
456
        self.assertEqual(reference_inv.root_id, new_inv.root_id)
1025
 
        reference_inv.id_to_entry._ensure_root()
1026
 
        new_inv.id_to_entry._ensure_root()
1027
457
        self.assertEqual(reference_inv.id_to_entry._root_node._key,
1028
458
            new_inv.id_to_entry._root_node._key)
1029
459
 
1043
473
        reference_inv = CHKInventory.from_inventory(chk_bytes, inv)
1044
474
        delta = [(None, "A",  "A-id", a_entry)]
1045
475
        new_inv = base_inv.create_by_apply_delta(delta, "expectedid")
1046
 
        reference_inv.id_to_entry._ensure_root()
1047
 
        reference_inv.parent_id_basename_to_file_id._ensure_root()
1048
 
        new_inv.id_to_entry._ensure_root()
1049
 
        new_inv.parent_id_basename_to_file_id._ensure_root()
1050
476
        # new_inv should be the same as reference_inv.
1051
477
        self.assertEqual(reference_inv.revision_id, new_inv.revision_id)
1052
478
        self.assertEqual(reference_inv.root_id, new_inv.root_id)
1210
636
        self.assertIsInstance(ie2.name, unicode)
1211
637
        self.assertEqual(('tree\xce\xa9name', 'tree-root-id', 'tree-rev-id'),
1212
638
                         inv._bytes_to_utf8name_key(bytes))
1213
 
 
1214
 
 
1215
 
class TestCHKInventoryExpand(tests.TestCaseWithMemoryTransport):
1216
 
 
1217
 
    def get_chk_bytes(self):
1218
 
        factory = groupcompress.make_pack_factory(True, True, 1)
1219
 
        trans = self.get_transport('')
1220
 
        return factory(trans)
1221
 
 
1222
 
    def make_dir(self, inv, name, parent_id):
1223
 
        inv.add(inv.make_entry('directory', name, parent_id, name + '-id'))
1224
 
 
1225
 
    def make_file(self, inv, name, parent_id, content='content\n'):
1226
 
        ie = inv.make_entry('file', name, parent_id, name + '-id')
1227
 
        ie.text_sha1 = osutils.sha_string(content)
1228
 
        ie.text_size = len(content)
1229
 
        inv.add(ie)
1230
 
 
1231
 
    def make_simple_inventory(self):
1232
 
        inv = Inventory('TREE_ROOT')
1233
 
        inv.revision_id = "revid"
1234
 
        inv.root.revision = "rootrev"
1235
 
        # /                 TREE_ROOT
1236
 
        # dir1/             dir1-id
1237
 
        #   sub-file1       sub-file1-id
1238
 
        #   sub-file2       sub-file2-id
1239
 
        #   sub-dir1/       sub-dir1-id
1240
 
        #     subsub-file1  subsub-file1-id
1241
 
        # dir2/             dir2-id
1242
 
        #   sub2-file1      sub2-file1-id
1243
 
        # top               top-id
1244
 
        self.make_dir(inv, 'dir1', 'TREE_ROOT')
1245
 
        self.make_dir(inv, 'dir2', 'TREE_ROOT')
1246
 
        self.make_dir(inv, 'sub-dir1', 'dir1-id')
1247
 
        self.make_file(inv, 'top', 'TREE_ROOT')
1248
 
        self.make_file(inv, 'sub-file1', 'dir1-id')
1249
 
        self.make_file(inv, 'sub-file2', 'dir1-id')
1250
 
        self.make_file(inv, 'subsub-file1', 'sub-dir1-id')
1251
 
        self.make_file(inv, 'sub2-file1', 'dir2-id')
1252
 
        chk_bytes = self.get_chk_bytes()
1253
 
        #  use a small maximum_size to force internal paging structures
1254
 
        chk_inv = CHKInventory.from_inventory(chk_bytes, inv,
1255
 
                        maximum_size=100,
1256
 
                        search_key_name='hash-255-way')
1257
 
        bytes = ''.join(chk_inv.to_lines())
1258
 
        return CHKInventory.deserialise(chk_bytes, bytes, ("revid",))
1259
 
 
1260
 
    def assert_Getitems(self, expected_fileids, inv, file_ids):
1261
 
        self.assertEqual(sorted(expected_fileids),
1262
 
                         sorted([ie.file_id for ie in inv._getitems(file_ids)]))
1263
 
 
1264
 
    def assertExpand(self, all_ids, inv, file_ids):
1265
 
        (val_all_ids,
1266
 
         val_children) = inv._expand_fileids_to_parents_and_children(file_ids)
1267
 
        self.assertEqual(set(all_ids), val_all_ids)
1268
 
        entries = inv._getitems(val_all_ids)
1269
 
        expected_children = {}
1270
 
        for entry in entries:
1271
 
            s = expected_children.setdefault(entry.parent_id, [])
1272
 
            s.append(entry.file_id)
1273
 
        val_children = dict((k, sorted(v)) for k, v
1274
 
                            in val_children.iteritems())
1275
 
        expected_children = dict((k, sorted(v)) for k, v
1276
 
                            in expected_children.iteritems())
1277
 
        self.assertEqual(expected_children, val_children)
1278
 
 
1279
 
    def test_make_simple_inventory(self):
1280
 
        inv = self.make_simple_inventory()
1281
 
        layout = []
1282
 
        for path, entry in inv.iter_entries_by_dir():
1283
 
            layout.append((path, entry.file_id))
1284
 
        self.assertEqual([
1285
 
            ('', 'TREE_ROOT'),
1286
 
            ('dir1', 'dir1-id'),
1287
 
            ('dir2', 'dir2-id'),
1288
 
            ('top', 'top-id'),
1289
 
            ('dir1/sub-dir1', 'sub-dir1-id'),
1290
 
            ('dir1/sub-file1', 'sub-file1-id'),
1291
 
            ('dir1/sub-file2', 'sub-file2-id'),
1292
 
            ('dir1/sub-dir1/subsub-file1', 'subsub-file1-id'),
1293
 
            ('dir2/sub2-file1', 'sub2-file1-id'),
1294
 
            ], layout)
1295
 
 
1296
 
    def test__getitems(self):
1297
 
        inv = self.make_simple_inventory()
1298
 
        # Reading from disk
1299
 
        self.assert_Getitems(['dir1-id'], inv, ['dir1-id'])
1300
 
        self.assertTrue('dir1-id' in inv._fileid_to_entry_cache)
1301
 
        self.assertFalse('sub-file2-id' in inv._fileid_to_entry_cache)
1302
 
        # From cache
1303
 
        self.assert_Getitems(['dir1-id'], inv, ['dir1-id'])
1304
 
        # Mixed
1305
 
        self.assert_Getitems(['dir1-id', 'sub-file2-id'], inv,
1306
 
                             ['dir1-id', 'sub-file2-id'])
1307
 
        self.assertTrue('dir1-id' in inv._fileid_to_entry_cache)
1308
 
        self.assertTrue('sub-file2-id' in inv._fileid_to_entry_cache)
1309
 
 
1310
 
    def test_single_file(self):
1311
 
        inv = self.make_simple_inventory()
1312
 
        self.assertExpand(['TREE_ROOT', 'top-id'], inv, ['top-id'])
1313
 
 
1314
 
    def test_get_all_parents(self):
1315
 
        inv = self.make_simple_inventory()
1316
 
        self.assertExpand(['TREE_ROOT', 'dir1-id', 'sub-dir1-id',
1317
 
                           'subsub-file1-id',
1318
 
                          ], inv, ['subsub-file1-id'])
1319
 
 
1320
 
    def test_get_children(self):
1321
 
        inv = self.make_simple_inventory()
1322
 
        self.assertExpand(['TREE_ROOT', 'dir1-id', 'sub-dir1-id',
1323
 
                           'sub-file1-id', 'sub-file2-id', 'subsub-file1-id',
1324
 
                          ], inv, ['dir1-id'])
1325
 
 
1326
 
    def test_from_root(self):
1327
 
        inv = self.make_simple_inventory()
1328
 
        self.assertExpand(['TREE_ROOT', 'dir1-id', 'dir2-id', 'sub-dir1-id',
1329
 
                           'sub-file1-id', 'sub-file2-id', 'sub2-file1-id',
1330
 
                           'subsub-file1-id', 'top-id'], inv, ['TREE_ROOT'])
1331
 
 
1332
 
    def test_top_level_file(self):
1333
 
        inv = self.make_simple_inventory()
1334
 
        self.assertExpand(['TREE_ROOT', 'top-id'], inv, ['top-id'])
1335
 
 
1336
 
    def test_subsub_file(self):
1337
 
        inv = self.make_simple_inventory()
1338
 
        self.assertExpand(['TREE_ROOT', 'dir1-id', 'sub-dir1-id',
1339
 
                           'subsub-file1-id'], inv, ['subsub-file1-id'])
1340
 
 
1341
 
    def test_sub_and_root(self):
1342
 
        inv = self.make_simple_inventory()
1343
 
        self.assertExpand(['TREE_ROOT', 'dir1-id', 'sub-dir1-id', 'top-id',
1344
 
                           'subsub-file1-id'], inv, ['top-id', 'subsub-file1-id'])